vendor/bitbag/wishlist-plugin/src/Entity/Wishlist.php line 20

Open in your IDE?
  1. <?php
  3. /*
  4.  * This file was created by developers working at BitBag
  5.  * Do you need more information about us and what we do? Visit our https://bitbag.io website!
  6.  * We are hiring developers from all over the world. Join us and start your new, exciting adventure and become part of us: https://bitbag.io/career
  7. */
  9. declare(strict_types=1);
  11. namespace BitBag\SyliusWishlistPlugin\Entity;
  13. use Doctrine\Common\Collections\ArrayCollection;
  14. use Doctrine\Common\Collections\Collection;
  15. use Sylius\Component\Channel\Model\ChannelInterface;
  16. use Sylius\Component\Core\Model\ProductInterface;
  17. use Sylius\Component\Core\Model\ProductVariantInterface;
  18. use Sylius\Component\Core\Model\ShopUserInterface;
  20. class Wishlist implements WishlistInterface
  21. {
  22.     protected ?int $id = null;
  24.     protected ?string $name;
  26.     /** @var Collection|WishlistProductInterface[] */
  27.     protected $wishlistProducts;
  29.     protected ?ShopUserInterface $shopUser = null;
  31.     /** @var WishlistTokenInterface|null */
  32.     protected $token;
  34.     protected ?ChannelInterface $channel;
  36.     public function __construct()
  37.     {
  38.         $this->wishlistProducts = new ArrayCollection();
  39.         $this->token = new WishlistToken();
  40.         $this->id = null;
  41.     }
  43.     public function getId(): ?int
  44.     {
  45.         return $this->id;
  46.     }
  48.     public function getName(): ?string
  49.     {
  50.         return $this->name;
  51.     }
  53.     public function setName(?string $name): void
  54.     {
  55.         $this->name = $name;
  56.     }
  58.     public function getProducts(): Collection
  59.     {
  60.         $products = [];
  62.         foreach ($this->wishlistProducts as $wishlistProduct) {
  63.             $products[] = $wishlistProduct->getProduct();
  64.         }
  66.         return new ArrayCollection($products);
  67.     }
  69.     /**
  70.      * @return Collection<int,ProductVariantInterface|null>
  71.      */
  72.     public function getProductVariants(): Collection
  73.     {
  74.         $variants = [];
  76.         foreach ($this->wishlistProducts as $wishlistProduct) {
  77.             $variants[] = $wishlistProduct->getVariant();
  78.         }
  80.         return new ArrayCollection($variants);
  81.     }
  83.     public function hasProductVariant(ProductVariantInterface $productVariant): bool
  84.     {
  85.         foreach ($this->wishlistProducts as $wishlistProduct) {
  86.             if ($productVariant === $wishlistProduct->getVariant()) {
  87.                 return true;
  88.             }
  89.         }
  91.         return false;
  92.     }
  94.     public function getWishlistProducts(): Collection
  95.     {
  96.         return $this->wishlistProducts;
  97.     }
  99.     public function hasProduct(ProductInterface $product): bool
  100.     {
  101.         foreach ($this->wishlistProducts as $wishlistProduct) {
  102.             if ($product === $wishlistProduct->getProduct()) {
  103.                 return true;
  104.             }
  105.         }
  107.         return false;
  108.     }
  110.     public function setWishlistProducts(Collection $wishlistProducts): void
  111.     {
  112.         $this->wishlistProducts = $wishlistProducts;
  113.     }
  115.     public function hasWishlistProduct(WishlistProductInterface $wishlistProduct): bool
  116.     {
  117.         return $this->wishlistProducts->contains($wishlistProduct);
  118.     }
  120.     public function addWishlistProduct(WishlistProductInterface $wishlistProduct): void
  121.     {
  122.         /** @var ProductVariantInterface $variant */
  123.         $variant = $wishlistProduct->getVariant();
  124.         if (!$this->hasProductVariant($variant)) {
  125.             $wishlistProduct->setWishlist($this);
  126.             $this->wishlistProducts->add($wishlistProduct);
  127.         }
  128.     }
  130.     public function getShopUser(): ?ShopUserInterface
  131.     {
  132.         return $this->shopUser;
  133.     }
  135.     public function setShopUser(ShopUserInterface $shopUser): void
  136.     {
  137.         $this->shopUser = $shopUser;
  138.     }
  140.     public function getToken(): string
  141.     {
  142.         return (string) $this->token;
  143.     }
  145.     public function setToken(string $token): void
  146.     {
  147.         $this->token = new WishlistToken($token);
  148.     }
  150.     public function removeProduct(WishlistProductInterface $product): self
  151.     {
  152.         if ($this->hasWishlistProduct($product)) {
  153.             $this->wishlistProducts->removeElement($product);
  154.         }
  156.         return $this;
  157.     }
  159.     public function removeProductVariant(ProductVariantInterface $variant): self
  160.     {
  161.         foreach ($this->wishlistProducts as $wishlistProduct) {
  162.             if ($wishlistProduct->getVariant() === $variant) {
  163.                 $this->wishlistProducts->removeElement($wishlistProduct);
  164.             }
  165.         }
  167.         return $this;
  168.     }
  170.     public function clear(): void
  171.     {
  172.         $this->wishlistProducts = new ArrayCollection();
  173.     }
  175.     public function getChannel(): ?ChannelInterface
  176.     {
  177.         return $this->channel;
  178.     }
  180.     public function setChannel(?ChannelInterface $channel): void
  181.     {
  182.         $this->channel = $channel;
  183.     }
  184. }