<?php
/************************************************************************
 * This file is part of Real Estate extension for EspoCRM.
 *
 * EspoCRM - Open Source CRM application.
 * Copyright (C) 2014-2022 Yurii Kuznietsov, Taras Machyshyn, Oleksii Avramenko
 * Website: https://www.espocrm.com
 *
 * Real Estate extension is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * Real Estate extension is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with EspoCRM. If not, see http://www.gnu.org/licenses/.
 *
 * The interactive user interfaces in modified source and object code versions
 * of this program must display Appropriate Legal Notices, as required under
 * Section 5 of the GNU General Public License version 3.
 *
 * In accordance with Section 7(b) of the GNU General Public License version 3,
 * these Appropriate Legal Notices must retain the display of the "EspoCRM" word.
 ************************************************************************/

namespace Espo\Modules\RealEstate\Repositories;

use Espo\ORM\Entity;

class Opportunity extends \Espo\Modules\Crm\Repositories\Opportunity
{
    public function beforeSave(Entity $entity, array $options = [])
    {
        parent::beforeSave($entity, $options);

        if (
            $entity->has('closeDate') &&
            !$entity->get('closeDate') &&
            $entity->get('stage') == 'Closed Won' &&
            (
                $entity->isAttributeChanged('stage') || $entity->isNew()
            )
        ) {
            $entity->set('closeDate', date('Y-m-d'));
        }

        if ($entity->get('requestId') && $entity->get('propertyId')) {
            $request = $this->entityManager->getEntity('RealEstateRequest', $entity->get('requestId'));
            $property = $this->entityManager->getEntity('RealEstateProperty', $entity->get('propertyId'));

            if ($request && $property) {
                $name = $property->get('name') . ' - ' . $request->get('name');
                $entity->set('name', $name);
            }
        }

        if (!$entity->get('name') && $entity->get('propertyId') && $entity->get('contactId')) {
               $property = $this->entityManager->getEntity('RealEstateProperty', $entity->get('propertyId'));
               $contact =  $this->entityManager->getEntity('Contact', $entity->get('contactId'));
               $name = $property->get('name') . ' - ' . $contact->get('name');
               $entity->set('name', $name);
        }
        if (!$entity->get('name')) {
               $entity->set('name', 'WARNING: Detail Missing');
        }
    }

    public function afterSave(Entity $entity, array $options = [])
    {
        parent::afterSave($entity, $options);

        if (
            $entity->get('stage') == 'Closed Won' &&
            (
                $entity->isAttributeChanged('stage') || $entity->isNew()
            )
        ) {
            if ($entity->get('requestId') && $entity->get('propertyId')) {
                $request = $this->entityManager->getEntity('RealEstateRequest', $entity->get('requestId'));

                $request->set('status', 'Completed');

                $this->entityManager->saveEntity($request);

                $property = $this->entityManager->getEntity('RealEstateProperty', $entity->get('propertyId'));
                $property->set('status', 'Completed');
                $this->entityManager->saveEntity($property);

                $opportunityList = $this
                    ->where([
                        'requestId' => $entity->get('requestId'),
                        'propertyId' => $entity->get('propertyId'),
                    ])
                    ->find();

                foreach ($opportunityList as $opportunity) {
                    if ($entity->getId() == $opportunity->getId()) {
                        continue;
                    }

                    $opportunity->set('stage', 'Closed Lost');
                    $opportunity->set('closeDate', date('Y-m-d'));

                    $this->save($opportunity);
                }
            }
        }

        if ($entity->isNew() && $entity->get('status') !== 'Closed Lost') {
            if ($entity->get('requestId') && $entity->get('propertyId')) {
                $note = $this->entityManager->getEntity('Note');

                $note->set([
                    'type' => 'CreateRelated',
                    'parentId' => $entity->get('propertyId'),
                    'parentType' => 'RealEstateProperty',
                    'data' => [
                        'action' => 'created',
                    ],
                    'relatedId' => $entity->getId(),
                    'relatedType' => 'Opportunity'
                ]);

                $this->entityManager->saveEntity($note);

                $note = $this->entityManager->getEntity('Note');

                $note->set([
                    'type' => 'CreateRelated',
                    'parentId' => $entity->get('requestId'),
                    'parentType' => 'RealEstateRequest',
                    'data' => [
                        'action' => 'created',
                    ],
                    'relatedId' => $entity->getId(),
                    'relatedType' => 'Opportunity'
                ]);

                $this->entityManager->saveEntity($note);
            }
        }

        if ($entity->isNew() || $entity->isAttributeChanged('stage')) {
            if ($entity->get('requestId') && $entity->get('propertyId')) {
                $property = $this->entityManager->getEntity('RealEstateProperty', $entity->get('propertyId'));

                if ($property) {
                    if ($entity->get('stage') !== 'Closed Lost') {
                        $this->entityManager
                            ->getRDBRepository('RealEstateProperty')
                            ->getRelation($property, 'requests')
                            ->unrelateById($entity->get('requestId'));
                    }
                    else {
                        $this->entityManager
                            ->getRDBRepository('RealEstateProperty')
                            ->getRelation($property, 'requests')
                            ->relateById($entity->get('requestId'), ['interestDegree' => 0]);
                    }
                }
            }
        }
    }
}