Add set method in both CurrentCategory and CurrentProduct

weprovide · Jan 24, 2023 · 4abee45 · 4abee45
1 parent 99e6ba2
commit 4abee45
Show file tree

Hide file tree

Showing 3 changed files with 47 additions and 11 deletions.
diff --git a/Service/CurrentCategory.php b/Service/CurrentCategory.php
@@ -4,11 +4,14 @@
 
 namespace WeProvide\Core\Service;
 
-use Magento\Catalog\Model\Category;
+use Magento\Catalog\Api\Data\CategoryInterface;
 use Magento\Framework\Registry;
 
 class CurrentCategory
 {
+    /** @var string */
+    protected const CURRENT_CATEGORY = 'current_category';
+
     /** @var Registry */
     protected $registry;
 
@@ -22,10 +25,25 @@ public function __construct(
     }
 
     /**
-     * @return Category|null
+     * Gets the current category
+     *
+     * @return CategoryInterface|null
      */
-    public function getCurrentCategory(): ?Category
+    public function getCurrentCategory(): ?CategoryInterface
     {
-        return $this->registry->registry('current_category');
+        return $this->registry->registry(static::CURRENT_CATEGORY);
+    }
+
+    /**
+     * Sets the current category
+     *
+     * @param CategoryInterface|null $category
+     * @return CurrentCategory
+     */
+    public function setCurrentCategory(?CategoryInterface $category): CurrentCategory
+    {
+        $this->registry->register(static::CURRENT_CATEGORY, $category);
+
+        return $this;
     }
-}
+}
diff --git a/Service/CurrentProduct.php b/Service/CurrentProduct.php
@@ -4,11 +4,14 @@
 
 namespace WeProvide\Core\Service;
 
-use Magento\Catalog\Model\Product;
+use Magento\Catalog\Api\Data\ProductInterface;
 use Magento\Framework\Registry;
 
 class CurrentProduct
 {
+    /** @var string */
+    protected const CURRENT_PRODUCT = 'current_product';
+
     /** @var Registry */
     protected $registry;
 
@@ -22,10 +25,25 @@ public function __construct(
     }
 
     /**
-     * @return Product|null
+     * Gets the current product
+     *
+     * @return ProductInterface|null
      */
-    public function getCurrentProduct(): ?Product
+    public function getCurrentProduct(): ?ProductInterface
     {
-        return $this->registry->registry('current_product');
+        return $this->registry->registry(static::CURRENT_PRODUCT);
+    }
+
+    /**
+     * Sets the current product
+     *
+     * @param ProductInterface|null $product
+     * @return CurrentProduct
+     */
+    public function setCurrentProduct(?ProductInterface $product): CurrentProduct
+    {
+        $this->registry->register(static::CURRENT_PRODUCT, $product);
+
+        return $this;
     }
-}
+}
diff --git a/composer.json b/composer.json
@@ -2,7 +2,7 @@
     "name": "weprovide/magento2-core",
     "description": "",
     "type": "magento2-module",
-    "version": "1.3.0",
+    "version": "1.3.1",
     "license": "MIT",
     "authors": [
         {