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

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;
  19. use Sylius\Component\Resource\Model\TimestampableTrait;
  21. class Wishlist implements WishlistInterface
  22. {
  23.     use TimestampableTrait;
  25.     protected ?int $id = null;
  27.     protected ?string $name;
  29.     /** @var Collection|WishlistProductInterface[] */
  30.     protected $wishlistProducts;
  32.     protected ?ShopUserInterface $shopUser = null;
  34.     /** @var WishlistTokenInterface|null */
  35.     protected $token;
  37.     protected ?ChannelInterface $channel;
  39.     public function __construct()
  40.     {
  41.         $this->wishlistProducts = new ArrayCollection();
  42.         $this->token = new WishlistToken();
  43.         $this->id = null;
  45.         $this->createdAt = new \DateTime();
  46.     }
  48.     public function getId(): ?int
  49.     {
  50.         return $this->id;
  51.     }
  53.     public function getName(): ?string
  54.     {
  55.         return $this->name;
  56.     }
  58.     public function setName(?string $name): void
  59.     {
  60.         $this->name = $name;
  61.     }
  63.     public function getProducts(): Collection
  64.     {
  65.         $products = [];
  67.         foreach ($this->wishlistProducts as $wishlistProduct) {
  68.             $products[] = $wishlistProduct->getProduct();
  69.         }
  71.         return new ArrayCollection($products);
  72.     }
  74.     /**
  75.      * @return Collection<int,ProductVariantInterface|null>
  76.      */
  77.     public function getProductVariants(): Collection
  78.     {
  79.         $variants = [];
  81.         foreach ($this->wishlistProducts as $wishlistProduct) {
  82.             $variants[] = $wishlistProduct->getVariant();
  83.         }
  85.         return new ArrayCollection($variants);
  86.     }
  88.     public function hasProductVariant(ProductVariantInterface $productVariant): bool
  89.     {
  90.         foreach ($this->wishlistProducts as $wishlistProduct) {
  91.             if ($productVariant === $wishlistProduct->getVariant()) {
  92.                 return true;
  93.             }
  94.         }
  96.         return false;
  97.     }
  99.     public function getWishlistProducts(): Collection
  100.     {
  101.         return $this->wishlistProducts;
  102.     }
  104.     public function hasProduct(ProductInterface $product): bool
  105.     {
  106.         foreach ($this->wishlistProducts as $wishlistProduct) {
  107.             if ($product === $wishlistProduct->getProduct()) {
  108.                 return true;
  109.             }
  110.         }
  112.         return false;
  113.     }
  115.     public function setWishlistProducts(Collection $wishlistProducts): void
  116.     {
  117.         $this->wishlistProducts = $wishlistProducts;
  118.     }
  120.     public function hasWishlistProduct(WishlistProductInterface $wishlistProduct): bool
  121.     {
  122.         return $this->wishlistProducts->contains($wishlistProduct);
  123.     }
  125.     public function addWishlistProduct(WishlistProductInterface $wishlistProduct): void
  126.     {
  127.         /** @var ProductVariantInterface $variant */
  128.         $variant = $wishlistProduct->getVariant();
  129.         if (!$this->hasProductVariant($variant)) {
  130.             $wishlistProduct->setWishlist($this);
  131.             $this->wishlistProducts->add($wishlistProduct);
  132.         }
  133.     }
  135.     public function getShopUser(): ?ShopUserInterface
  136.     {
  137.         return $this->shopUser;
  138.     }
  140.     public function setShopUser(ShopUserInterface $shopUser): void
  141.     {
  142.         $this->shopUser = $shopUser;
  143.     }
  145.     public function getToken(): string
  146.     {
  147.         return (string) $this->token;
  148.     }
  150.     public function setToken(string $token): void
  151.     {
  152.         $this->token = new WishlistToken($token);
  153.     }
  155.     public function removeProduct(WishlistProductInterface $product): self
  156.     {
  157.         if ($this->hasWishlistProduct($product)) {
  158.             $this->wishlistProducts->removeElement($product);
  159.         }
  161.         return $this;
  162.     }
  164.     public function removeProductVariant(ProductVariantInterface $variant): self
  165.     {
  166.         foreach ($this->wishlistProducts as $wishlistProduct) {
  167.             if ($wishlistProduct->getVariant() === $variant) {
  168.                 $this->wishlistProducts->removeElement($wishlistProduct);
  169.             }
  170.         }
  172.         return $this;
  173.     }
  175.     public function clear(): void
  176.     {
  177.         $this->wishlistProducts = new ArrayCollection();
  178.     }
  180.     public function getChannel(): ?ChannelInterface
  181.     {
  182.         return $this->channel;
  183.     }
  185.     public function setChannel(?ChannelInterface $channel): void
  186.     {
  187.         $this->channel = $channel;
  188.     }
  189. }