custom/plugins/MoorlProductAccessories/src/Core/Content/Accessory/AccessoryCollection.php line 78

Open in your IDE?
  1. <?php declare(strict_types=1);
  3. namespace MoorlProductAccessories\Core\Content\Accessory;
  5. use MoorlProductAccessories\Core\Content\Category\CategoryCollection;
  6. use MoorlProductAccessories\Core\Content\Category\CategoryEntity;
  7. use Shopware\Core\Framework\DataAbstractionLayer\EntityCollection;
  8. use Shopware\Core\Framework\Uuid\Uuid;
  10. /**
  11.  * @method void            add(AccessoryEntity $entity)
  12.  * @method void            set(string $key, AccessoryEntity $entity)
  13.  * @method AccessoryEntity[]    getIterator()
  14.  * @method AccessoryEntity[]    getElements()
  15.  * @method AccessoryEntity|null get(string $key)
  16.  * @method AccessoryEntity|null first()
  17.  * @method AccessoryEntity|null last()
  18.  */
  19. class AccessoryCollection extends EntityCollection
  20. {
  21.     protected function getExpectedClass(): string
  22.     {
  23.         return AccessoryEntity::class;
  24.     }
  26.     public function filterByCategoryId(?string $id = null): self
  27.     {
  28.         return $this->filter(function (AccessoryEntity $entity) use ($id) {
  29.             return ($entity->getCategoryId() === $id && $entity->getAccessory()->getActive() === true);
  30.         });
  31.     }
  33.     public function hasDuplicate(AccessoryEntity $struct): bool
  34.     {
  35.         foreach ($this->getElements() as $entity) {
  36.             if ($entity->getId() === $struct->getId()) {
  37.                 continue;
  38.             }
  39.             if ($entity->getAccessoryId() === $struct->getAccessoryId() && $entity->getCategoryId() === $struct->getCategoryId()) {
  40.                 return true;
  41.             }
  42.         }
  43.         return false;
  44.     }
  46.     public function removeDuplicates(): void
  47.     {
  48.         foreach ($this->getElements() as $entity) {
  49.             if ($entity->getSwCategoryId() && $this->hasDuplicate($entity)) {
  50.                 $this->remove($entity->getId());
  51.             }
  52.         }
  53.     }
  55.     public function sortByName(): void
  56.     {
  57.         $this->sort(function (AccessoryEntity $a, AccessoryEntity $b) {
  58.             return strnatcasecmp($a->getAccessory()->getTranslated()['name'], $b->getAccessory()->getTranslated()['name']);
  59.         });
  60.     }
  62.     public function getByAccessoryId(string $id): ?AccessoryEntity
  63.     {
  64.         return $this->filter(function (AccessoryEntity $entity) use ($id) {
  65.             return ($entity->getAccessoryId() === $id);
  66.         })->first();
  67.     }
  69.     public function getIdByAccessoryId(string $id): ?string
  70.     {
  71.         $entity = $this->getByAccessoryId($id);
  72.         if (!$entity) {
  73.             return null;
  74.         }
  75.         return $entity->getId();
  76.     }
  78.     public function getCategories(): CategoryCollection
  79.     {
  80.         $this->removeDuplicates();
  81.         $this->sortByName();
  83.         $categories = new CategoryCollection();
  85.         foreach ($this->getElements() as $entity) {
  86.             if ($entity->getCategoryId() && !$categories->has($entity->getCategoryId())) {
  87.                 $categories->add($entity->getCategory());
  88.             }
  89.         }
  91.         foreach ($categories as $category) {
  92.             $category->setAccessories($this->filterByCategoryId($category->getId()));
  93.         }
  95.         if ($this->filterByCategoryId()->count() > 0) {
  96.             $category = new CategoryEntity();
  97.             $category->setSortOrder(999); // is Last
  98.             $category->setAccessories($this->filterByCategoryId());
  99.             $category->setType('multi');
  100.             $category->setId(Uuid::randomHex());
  101.             $categories->add($category);
  102.         }
  104.         $categories->sortByOrder();
  106.         return $categories;
  107.     }
  108. }