move entity changes to trait (#8)

* fix #2 - move entity changes to trait * ignore /.idea
JoppeDC · Oct 13, 2019 · 2d82a98 · 2d82a98
1 parent f460bd1
commit 2d82a98
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 10 deletions.
diff --git a/.gitignore b/.gitignore
@@ -10,3 +10,5 @@
 /behat.yml
 /phpspec.yml
 docker-compose.yml
+
+/.idea
diff --git a/src/Entity/Product.php → src/Traits/SeoTrait.php b/src/Entity/Product.php → src/Traits/SeoTrait.php
@@ -2,31 +2,35 @@
 
 declare(strict_types=1);
 
-namespace JoppeDc\SyliusBetterSeoPlugin\Entity;
+namespace JoppeDc\SyliusBetterSeoPlugin\Traits;
 
-use Sylius\Component\Core\Model\Product as SyliusProduct;
+use JoppeDc\SyliusBetterSeoPlugin\Entity\ProductSeo;
 
-class Product extends SyliusProduct
+trait SeoTrait
 {
     /**
      * @var ProductSeo
      */
     protected $seo;
 
-    public function __construct()
-    {
-        parent::__construct();
-
-        $this->seo = new ProductSeo();
-    }
-
+    /**
+     * @return ProductSeo|null
+     */
     public function getSeo(): ?ProductSeo
     {
         return $this->seo;
     }
 
+    /**
+     * @param ProductSeo|null $seo
+     */
     public function setSeo(?ProductSeo $seo): void
     {
         $this->seo = $seo;
     }
+
+    protected function initSeo(): void
+    {
+        $this->seo = new ProductSeo();
+    }
 }