<?php

namespace AmeliaBooking\Application\Commands\Bookable\Category;

use AmeliaBooking\Application\Commands\CommandHandler;
use AmeliaBooking\Application\Commands\CommandResult;
use AmeliaBooking\Domain\Collection\Collection;
use AmeliaBooking\Domain\Common\Exceptions\InvalidArgumentException;
use AmeliaBooking\Domain\Entity\Bookable\Service\Category;
use AmeliaBooking\Domain\Entity\Bookable\Service\Extra;
use AmeliaBooking\Domain\Entity\Bookable\Service\Service;
use AmeliaBooking\Domain\Entity\Entities;
use AmeliaBooking\Domain\Factory\Bookable\Service\CategoryFactory;
use AmeliaBooking\Application\Common\Exceptions\AccessDeniedException;
use AmeliaBooking\Domain\ValueObjects\Number\Integer\Id;
use AmeliaBooking\Infrastructure\Common\Exceptions\QueryExecutionException;
use AmeliaBooking\Infrastructure\Repository\Bookable\Service\CategoryRepository;
use AmeliaBooking\Infrastructure\Repository\Bookable\Service\ExtraRepository;
use AmeliaBooking\Infrastructure\Repository\Bookable\Service\ServiceRepository;

/**
 * Class AddCategoryCommandHandler
 *
 * @package AmeliaBooking\Application\Commands\Bookable\Category
 */
class AddCategoryCommandHandler extends CommandHandler
{
    protected $mandatoryFields = [
        'name'
    ];

    /**
     * @param AddCategoryCommand $command
     *
     * @return CommandResult
     * @throws \Slim\Exception\ContainerValueNotFoundException
     * @throws AccessDeniedException
     * @throws QueryExecutionException
     * @throws InvalidArgumentException
     */
    public function handle(AddCategoryCommand $command)
    {
        if (!$command->getPermissionService()->currentUserCanWrite(Entities::SERVICES)) {
            throw new AccessDeniedException('You are not allowed to add category.');
        }

        $result = new CommandResult();

        $this->checkMandatoryFields($command);

        $categoryArray = $command->getFields();

        $categoryArray = apply_filters('amelia_before_category_added_filter', $categoryArray);

        do_action('amelia_before_category_added', $categoryArray);

        /** @var Category $category */
        $category = CategoryFactory::create($categoryArray);

        if (!($category instanceof Category)) {
            $result->setResult(CommandResult::RESULT_ERROR);
            $result->setMessage('Could not create category.');

            return $result;
        }

        /** @var CategoryRepository $categoryRepository */
        $categoryRepository = $this->container->get('domain.bookable.category.repository');
        /** @var ServiceRepository $serviceRepository */
        $serviceRepository = $this->container->get('domain.bookable.service.repository');
        /** @var ExtraRepository $extraRepository */
        $extraRepository = $this->container->get('domain.bookable.extra.repository');

        $categoryRepository->beginTransaction();

        $categoryId = $categoryRepository->add($category);

        $category->setId(new Id($categoryId));

        if ($category->getServiceList() && $category->getServiceList()->length()) {
            foreach ($category->getServiceList()->getItems() as $service) {
                /** @var Service $service */
                $service->setCategoryId(new Id($categoryId));

                $serviceId = $serviceRepository->add($service);

                /** @var Extra $extra */
                foreach ($service->getExtras()->getItems() as $extra) {
                    $extra->setServiceId(new Id($serviceId));

                    $extraId = $extraRepository->add($extra);

                    $extra->setId(new Id($extraId));
                }

                $service->setId(new Id($serviceId));
            }
        }

        $categoryRepository->commit();

        do_action('amelia_after_category_added', $categoryArray);

        $result->setResult(CommandResult::RESULT_SUCCESS);
        $result->setMessage('Successfully added new category.');
        $result->setData(
            [
                Entities::CATEGORY => $category->toArray()
            ]
        );

        return $result;
    }
}