EmicoEcommerce/Magento2Tweakwise

<?php

namespace Tweakwise\Magento2Tweakwise\Model\Client\Request;

use Tweakwise\Magento2Tweakwise\Model\Config;
use Magento\Catalog\Api\Data\CategoryInterface;
use Magento\Store\Api\Data\StoreInterface;
use Magento\Store\Model\Store;

trait SearchRequestTrait
{
    /**
     * @var Config
     */
    protected $config;

    /**
     * @param string $parameter
     * @param string|null $value
     * @return mixed
     */
    abstract public function setParameter(string $parameter, string $value = null);

    /**
     * @param string $parameter
     * @return mixed|null
     */
    abstract public function getParameter(string $parameter);

    /**
     * @return StoreInterface|null
     */
    abstract protected function getStore();

    /**
     * @param CategoryInterface|int $category
     */
    abstract protected function addCategoryFilter($category);

    /**
     * @param string $query
     */
    public function setSearch(string $query): void
    {
        $this->setParameter('tn_q', $query);
        $this->setDefaultCategory();
        $this->setSearchLanguage();
    }

    /**
     * @return void
     */
    protected function setDefaultCategory(): void
    {
        if ($this->getParameter('tn_cid') === null) {
            $rootCategoryId = $this->getStoreRootCategoryId() ?: 2;
            $this->addCategoryFilter($rootCategoryId);
        }
    }

    /**
     * @return int|null
     */
    protected function getStoreRootCategoryId(): ?int
    {
        $store = $this->getStore();
        if ($store instanceof Store) {
            /** @noinspection PhpUnhandledExceptionInspection */
            return (int) $store->getRootCategoryId();
        }

        return null;
    }

    /**
     * @return void
     */
    protected function setSearchLanguage(): void
    {
        $language = $this->config->getSearchLanguage();
        if (!$language) {
            return;
        }

        $this->setParameter('tn_lang', $language);
    }
}