<?php

namespace Autogedal\Extend\Plugin\Catalog\Product;

use Amasty\Sorting\Model\Method\ApplyGlobalSorting;
use Amasty\Sorting\Model\MethodProvider;
use Amasty\Sorting\Model\ResourceModel\Method\Image as ImageMethodResource;
use Amasty\Sorting\Model\ResourceModel\Method\Instock as InstockMethodResource;
use Amasty\Sorting\Model\SortingAdapterFactory;
use Magento\Framework\DB\Select;
use Magento\Framework\Search\Adapter\Mysql\TemporaryStorage;
use Amasty\Sorting\Model\Elasticsearch\IsElasticSort;

/**
 * Plugin Collection
 * plugin name: Amasty_Sorting::SortingMethodsProcessor
 * type: \Magento\Catalog\Model\ResourceModel\Product\Collection
 */
class Collection
{
    public const PROCESS_FLAG = 'amsort_process';

    /**
     * @var MethodProvider
     */
    private $methodProvider;

    /**
     * @var SortingAdapterFactory
     */
    private $adapterFactory;

    /**
     * @var ImageMethodResource
     */
    private $imageMethod;

    /**
     * @var InstockMethodResource
     */
    private $stockMethod;

    /**
     * @var array
     */
    private $skipAttributes = [];

    /**
     * @var ApplyGlobalSorting
     */
    private $applyGlobalSorting;

    /**
     * @var IsElasticSort
     */
    private $isElasticSort;
    
    /**
     * @var bool
     */
    private $isFulltextCollection;

    public function __construct(
        MethodProvider $methodProvider,
        ImageMethodResource $imageMethod,
        InstockMethodResource $stockMethod,
        SortingAdapterFactory $adapterFactory,
        ApplyGlobalSorting $applyGlobalSorting,
        IsElasticSort $isElasticSort,
        bool $isFulltextCollection = false
    ) {
        $this->methodProvider = $methodProvider;
        $this->adapterFactory = $adapterFactory;
        $this->imageMethod = $imageMethod;
        $this->stockMethod = $stockMethod;
        $this->applyGlobalSorting = $applyGlobalSorting;
        $this->isElasticSort = $isElasticSort;
        $this->isFulltextCollection = $isFulltextCollection;
    }

    /**
     * @param \Magento\Catalog\Model\ResourceModel\Product\Collection $subject
     * @param string                                                  $attribute
     * @param string                                                  $dir  'ASC'|'DESC'
     *
     * @return array
     */
    public function beforeSetOrder($subject, $attribute, $dir = Select::SQL_DESC)
    {
        $subject->setFlag(self::PROCESS_FLAG, true);
        $this->applyHighPriorityOrders($subject, $dir);
        $flagName = $this->getFlagName($attribute);
        if ($subject->getFlag($flagName)) {
            if ($this->isElasticSort()) {
                $this->skipAttributes[] = $flagName;
            } else {
                // attribute already used in sorting; disable double sorting by renaming attribute into not existing
                $attribute .= '_ignore';
            }
        } else {
            $method = $this->methodProvider->getMethodByCode($attribute);
            if ($method) {
                $method->applyCustomAttribute($subject, $dir);
                if (!$this->isElasticSort()) {
                    $method->apply($subject, $dir);
                }
                $attribute = $method->getAlias();
            }
            if (!$this->isElasticSort()) {
                if ($attribute == 'relevance' && !$subject->getFlag($this->getFlagName('am_relevance'))) {
                    $this->addRelevanceSorting($subject, $dir);
                    $attribute = 'am_relevance';
                }
                if ($attribute == 'price') {
                    $subject->addOrder($attribute, $dir);
                    $attribute .= '_ignore';
                }
            }
            $subject->setFlag($flagName, true);
        }

        $subject->setFlag(self::PROCESS_FLAG, false);

        return [$attribute, $dir];
    }

    /**
     * @param \Magento\Catalog\Model\ResourceModel\Product\Collection $subject
     * @param callable $proceed
     * @param string $attribute
     * @param string $dir
     * @return mixed
     */
    public function aroundSetOrder($subject, callable $proceed, $attribute, $dir = Select::SQL_DESC)
    {
        $flagName = $this->getFlagName($attribute);
        if (!in_array($flagName, $this->skipAttributes)) {
            $proceed($attribute, $dir);
        }

        return $subject;
    }

    private function getFlagName($attribute)
    {
        if ($attribute == 'price_asc' || $attribute == 'price_desc') {
            $attribute = 'price';
        }
        if (is_string($attribute)) {
            return 'sorted_by_' . $attribute;
        }

        return 'amasty_sorting';
    }

    /**
     * @param \Magento\Catalog\Model\ResourceModel\Product\Collection $collection
     * @param string $dir
     */
    private function applyHighPriorityOrders($collection, $dir)
    {
        if (!$collection->getFlag($this->getFlagName('high'))) {
            $collection->setFlag($this->getFlagName('high'), true);
            if ($this->isElasticSort()) {
                if ($this->imageMethod->isMethodActive($collection)) {
                    $collection->setOrder('non_images_last', Select::SQL_DESC);
                }
                $collection->setOrder('product_priority_high', Select::SQL_DESC);
                $collection->setOrder('low_priority_subcategories', Select::SQL_DESC);
                if ($this->stockMethod->isMethodActive($collection)) {
                    $collection->setOrder('out_of_stock_last', Select::SQL_DESC);
                }
            } else {
                $this->stockMethod->apply($collection, $dir);
                $this->imageMethod->apply($collection, $dir);
            }
            $this->applyGlobalSorting->execute($collection);
        }
    }

    /**
     * @param \Magento\Catalog\Model\ResourceModel\Product\Collection $collection
     */
    private function addRelevanceSorting($collection)
    {
        if (!$this->isFulltextCollection) {
            return;
        }

        $collection->getSelect()->columns(['am_relevance' => new \Zend_Db_Expr(
            'search_result.'. TemporaryStorage::FIELD_SCORE
        )]);
        $collection->addExpressionAttributeToSelect('am_relevance', 'am_relevance', []);

        // remove last item from columns because e.am_relevance from addExpressionAttributeToSelect not exist
        $columns = $collection->getSelect()->getPart(Select::COLUMNS);
        array_pop($columns);
        $collection->getSelect()->setPart(Select::COLUMNS, $columns);
        $collection->setFlag($this->getFlagName('am_relevance'), true);
    }

    /**
     * @param $subject
     * @param $attribute
     * @param string $dir
     * @return array
     */
    public function beforeAddOrder($subject, $attribute, $dir = Select::SQL_DESC)
    {
        if (!$subject->getFlag(self::PROCESS_FLAG)) {
            $result =  $this->beforeSetOrder($subject, $attribute, $dir);
        }

        return $result ?? [$attribute, $dir];
    }

    /**
     * @param $subject
     * @param callable $proceed
     * @param $attribute
     * @param string $dir
     * @return mixed
     */
    public function aroundAddOrder($subject, callable $proceed, $attribute, $dir = Select::SQL_DESC)
    {
        return $this->aroundSetOrder($subject, $proceed, $attribute, $dir);
    }

    private function isElasticSort(): bool
    {
        return $this->isFulltextCollection && $this->isElasticSort->execute();
    }
}