<?php
/**
 * @author Amasty Team
 * @copyright Copyright (c) Amasty (https://www.amasty.com)
 * @package Cookie Consent (GDPR) for Magento 2
 */

namespace Amasty\GdprCookie\Controller\Adminhtml\Cookie;

use Amasty\GdprCookie\Api\Data\CookieInterface;
use Amasty\GdprCookie\Controller\Adminhtml\AbstractCookie;
use Amasty\GdprCookie\Model\CookieFactory;
use Amasty\GdprCookie\Model\OptionSource\Cookie\Groups;
use Amasty\GdprCookie\Model\Repository\CookieRepository;
use Magento\Backend\App\Action\Context;
use Magento\Framework\App\Request\DataPersistorInterface;
use Magento\Framework\Exception\LocalizedException;
use Psr\Log\LoggerInterface;

class Save extends AbstractCookie
{
    /**
     * @var CookieRepository
     */
    private $cookieRepository;

    /**
     * @var DataPersistorInterface
     */
    private $dataPersistor;

    /**
     * @var CookieFactory
     */
    private $cookieFactory;

    /**
     * @var LoggerInterface
     */
    private $logger;

    public function __construct(
        Context $context,
        CookieRepository $cookieRepository,
        DataPersistorInterface $dataPersistor,
        CookieFactory $cookieFactory,
        LoggerInterface $logger
    ) {
        parent::__construct($context);
        $this->cookieRepository = $cookieRepository;
        $this->dataPersistor = $dataPersistor;
        $this->cookieFactory = $cookieFactory;
        $this->logger = $logger;
    }

    /**
     * Save action
     */
    public function execute()
    {
        $formData = $this->getRequest()->getPostValue('cookie');
        $storeId = (int)$this->getRequest()->getParam('store');

        try {
            $data = $formData;
            $model = isset($formData['id'])
                ? $this->cookieRepository->getById($formData['id'])
                : $this->cookieFactory->create();

            if ($data[CookieInterface::GROUP_ID] === (string)Groups::NONE_GROUP_ID) {
                $data[CookieInterface::GROUP_ID] = null;
            }

            if ($storeId) {
                $this->modifyUseDefaultsData($data);
            }

            $model->setData($data);
            $this->cookieRepository->save($model, $storeId);
            $this->messageManager->addSuccessMessage(__('You saved the item.'));

            if ($this->getRequest()->getParam('back')) {
                return $this->resultRedirectFactory->create()->setPath(
                    '*/*/edit',
                    ['id' => $model->getId(), 'store' => $storeId]
                );
            }
        } catch (LocalizedException $e) {
            $this->messageManager->addErrorMessage($e->getMessage());

            return $this->redirectIfError($formData);
        } catch (\Exception $e) {
            $this->messageManager->addErrorMessage(__('An error has occurred.'));
            $this->logger->critical($e);

            return $this->redirectIfError($formData);
        }

        return $this->resultRedirectFactory->create()->setPath('*/*');
    }

    private function modifyUseDefaultsData(array &$data)
    {
        $useDefaultData = $this->getRequest()->getPostValue('use_default');

        if (isset($useDefaultData) && is_array($useDefaultData)) {
            foreach ($useDefaultData as $field => $isUseDefault) {
                if (!$isUseDefault) {
                    continue;
                }
                switch ($field) {
                    case 'is_enabled':
                        $data[$field] = 1;
                        break;
                    default:
                        $data[$field] = null;
                        break;
                }
            }
        }
    }

    /**
     * @param array $formData
     * @return \Magento\Framework\Controller\Result\Redirect
     */
    private function redirectIfError($formData)
    {
        $this->dataPersistor->set('formData', $formData);

        $resultRedirect = $this->resultRedirectFactory->create();
        if ($id = (int)$this->getRequest()->getParam('id')) {
            $resultRedirect->setPath('*/*/edit', ['id' => $id]);
        } else {
            $resultRedirect->setPath('*/*/new', ['id' => $id]);
        }

        return $resultRedirect;
    }
}