0byt3m1n1-V2

Path: /home/dev2.destoffenstraat.com/app/code/Neklo/Core/Service/ [ Home ]
File: CategoryRegistry.php
<?php
/*
NOTICE OF LICENSE

This source file is subject to the NekloEULA that is bundled with this package in the file ICENSE.txt.

It is also available through the world-wide-web at this URL: http://store.neklo.com/LICENSE.txt

Copyright (c)  Neklo (http://store.neklo.com/)
*/


declare(strict_types=1);

namespace Neklo\Core\Service;

use Magento\Catalog\Api\Data\CategoryInterface;
use Magento\Framework\Exception\LocalizedException;
use Magento\Framework\Registry;

class CategoryRegistry
{
    /**
     * Registry key list.
     */
    const REGISTRY_KEY_CURRENT_CATEGORY = 'current_category';
    const REGISTRY_KEY_CATEGORY = 'category';

    /**
     * @var Registry
     */
    private $registry;

    /**
     * @var string[]
     */
    private $registryKeyMap;

    /**
     * @param Registry $registry
     * @param array $registryKeyMap
     */
    public function __construct(
        Registry $registry,
        array $registryKeyMap = [
            self::REGISTRY_KEY_CURRENT_CATEGORY,
            self::REGISTRY_KEY_CATEGORY,
        ]
    ) {
        $this->registry = $registry;
        $this->registryKeyMap = $registryKeyMap;
    }

    /**
     * @param CategoryInterface $category
     */
    public function setCurrentCategory(CategoryInterface $category)
    {
        $this->unsetCurrentCategory();
        foreach ($this->registryKeyMap as $registryKey) {
            if (!$this->registry->registry($registryKey) instanceof CategoryInterface) {
                $this->registry->register($registryKey, $category);
            }
        }
    }

    /**
     * @return CategoryInterface
     * @throws LocalizedException
     */
    public function getCurrentCategory(): CategoryInterface
    {
        foreach ($this->registryKeyMap as $registryKey) {
            if ($this->registry->registry($registryKey) instanceof CategoryInterface) {
                return $this->registry->registry($registryKey);
            }
        }

        throw new LocalizedException(__("The current category isn't initialized"));
    }

    /**
     * @return void
     */
    public function unsetCurrentCategory()
    {
        foreach ($this->registryKeyMap as $registryKey) {
            $this->registry->unregister($registryKey);
        }
    }
}