RegistryLocator.php

Go to the documentation of this file.

<?php
namespace Magento\Catalog\Model\Locator;

use Magento\Catalog\Api\Data\ProductInterface;
use Magento\Framework\Exception\NotFoundException;
use Magento\Framework\Registry;
use Magento\Store\Api\Data\StoreInterface;

class RegistryLocator implements LocatorInterface
{
    private $registry;

    private $product;

    private $store;

    public function __construct(Registry $registry)
    {
        $this->registry = $registry;
    }

    public function getProduct()
    {
        if (null !== $this->product) {
            return $this->product;
        }

        if ($product = $this->registry->registry('current_product')) {
            return $this->product = $product;
        }

        throw new NotFoundException(__("The product wasn't registered."));
    }

    public function getStore()
    {
        if (null !== $this->store) {
            return $this->store;
        }

        if ($store = $this->registry->registry('current_store')) {
            return $this->store = $store;
        }

        throw new NotFoundException(__("The store wasn't registered. Verify the store and try again."));
    }

    public function getWebsiteIds()
    {
        return $this->getProduct()->getWebsiteIds();
    }

    public function getBaseCurrencyCode()
    {
        return $this->getStore()->getBaseCurrencyCode();
    }
}