<?php

namespace Leadlion\Autogedal\Plugin\Ajax;

use Amasty\Shopby\Model\Ajax\AjaxResponseBuilder;
use Amasty\Shopby\Model\Ajax\Counter\CounterDataProvider;
use Amasty\Shopby\Model\Ajax\RequestResponseUtils;
use Magento\Framework\App\ActionFlag;
use Magento\Framework\App\ActionInterface;
use Magento\Framework\App\ObjectManager;
use Magento\Framework\App\RequestInterface;
use Magento\Framework\App\ResponseInterface;
use Magento\Framework\Controller\Result\Raw;
use Magento\Framework\View\Result\Page;
use Magento\Framework\View\LayoutInterface;

class CategoryViewAjax extends \Amasty\Shopby\Plugin\Ajax\CategoryViewAjax
{
    /**
     * @var LayoutInterface
      */
    private $layout;

    /**
     * @var ActionFlag
     */
    private $actionFlag;

    /**
     * @var RequestResponseUtils
     */
    private $utils;

    /**
     * @var AjaxResponseBuilder
     */
    private $ajaxResponseBuilder;

    /**
     * @var CounterDataProvider
     */
    private $counterDataProvider;

    /**
     * @var RequestInterface
     */
    private RequestInterface $request;

    /**
     * @var ResponseInterface
     */
    private ResponseInterface $response;

    public function __construct(
        ActionFlag $actionFlag,
        RequestResponseUtils $utils,
        AjaxResponseBuilder $ajaxResponseBuilder,
        CounterDataProvider $counterDataProvider,
        RequestInterface $request,
        LayoutInterface $layout,
        ?ResponseInterface $response = null
    ) {
        parent::__construct(
            $actionFlag,
            $utils,
            $ajaxResponseBuilder,
            $counterDataProvider,
            $request,
            $response
        );
        $this->layout = $layout;
        $this->actionFlag = $actionFlag;
        $this->utils = $utils;
        $this->ajaxResponseBuilder = $ajaxResponseBuilder;
        $this->counterDataProvider = $counterDataProvider;
        $this->request = $request;
        $this->response = $response ?? ObjectManager::getInstance()->get(ResponseInterface::class);
    }

    public function beforeExecute(ActionInterface $controller): void
    {
        if ($this->utils->isAjaxNavigation($this->request)) {
            $this->actionFlag->set('', 'no-renderLayout', true);
        }
    }

    /**
     * @param Action $controller
     * @param Page $page
     *
     * @return Raw|Page
     */
    public function afterExecute(ActionInterface $controller, $page)
    {
        if (!$this->utils->isAjaxNavigation($this->request)) {
            return $page;
        }

        $responseData = $this->utils->isCounterRequest($this->request)
            ? $this->counterDataProvider->execute()
            : $this->ajaxResponseBuilder->build();

        $responseData['category_subtitle'] = $this->getBlockHtml($this->layout, 'page.title.counter');
        $responseData['description'] = $this->layout->getBlock('category_desc_main_column')
                ? $this->layout->renderElement('category_desc_main_column')
                : '';
        $this->response->clearHeader('Location');
        $response = $this->utils->prepareResponse($responseData);

        return $response;
    }

    /**
     * @param $layout
     * @param $blockName
     * @return string
     */
    private function getBlockHtml($layout, $blockName)
    {
        return $layout->getBlock($blockName) ? $layout->getBlock($blockName)->toHtml() : '';
    }
}