custom/plugins/NetiNextStoreLocator/src/Decorator/EntityRepositoryDecorator.php line 130

Open in your IDE?
  1. <?php declare(strict_types=1);
  3. namespace NetInventors\NetiNextStoreLocator\Decorator;
  5. use NetInventors\NetiNextStoreLocator\Subscriber\ImportExportSubscriber;
  6. use Shopware\Core\Framework\Context;
  7. use Shopware\Core\Framework\DataAbstractionLayer\EntityDefinition;
  8. use Shopware\Core\Framework\DataAbstractionLayer\EntityRepositoryInterface;
  9. use Shopware\Core\Framework\DataAbstractionLayer\Event\EntityWrittenContainerEvent;
  10. use Shopware\Core\Framework\DataAbstractionLayer\Search\AggregationResult\AggregationResultCollection;
  11. use Shopware\Core\Framework\DataAbstractionLayer\Search\Criteria;
  12. use Shopware\Core\Framework\DataAbstractionLayer\Search\EntitySearchResult;
  13. use Shopware\Core\Framework\DataAbstractionLayer\Search\Filter\EqualsFilter;
  14. use Shopware\Core\Framework\DataAbstractionLayer\Search\IdSearchResult;
  15. use Shopware\Core\Framework\DataAbstractionLayer\Write\CloneBehavior;
  16. use Shopware\Core\Framework\Uuid\Uuid;
  18. class EntityRepositoryDecorator implements EntityRepositoryInterface
  19. {
  20.     private const COUNTRY_ISOS = [
  21.         '2'  => 'DE',
  22.         '3'  => 'AE',
  23.         '4'  => 'AU',
  24.         '5'  => 'BE',
  25.         '7'  => 'DK',
  26.         '8'  => 'FI',
  27.         '9'  => 'FR',
  28.         '10' => 'GR',
  29.         '11' => 'GB',
  30.         '12' => 'IE',
  31.         '13' => 'IS',
  32.         '14' => 'IT',
  33.         '15' => 'JP',
  34.         '16' => 'CA',
  35.         '18' => 'LU',
  36.         '20' => 'NA',
  37.         '21' => 'NL',
  38.         '22' => 'NO',
  39.         '23' => 'AT',
  40.         '24' => 'PT',
  41.         '25' => 'SE',
  42.         '26' => 'CH',
  43.         '27' => 'ES',
  44.         '28' => 'US',
  45.         '29' => 'LI',
  46.         '30' => 'PL',
  47.         '31' => 'HU',
  48.         '32' => 'TR',
  49.         '33' => 'CZ',
  50.         '34' => 'SK',
  51.         '35' => 'RO',
  52.         '36' => 'BR',
  53.         '37' => 'IL',
  54.     ];
  56.     /**
  57.      * @var EntityRepositoryInterface
  58.      */
  59.     private $repository;
  61.     /**
  62.      * @var EntityRepositoryInterface
  63.      */
  64.     private $salesChannelRepository;
  66.     /**
  67.      * @var EntityRepositoryInterface
  68.      */
  69.     private $countryRepository;
  71.     /**
  72.      * @var EntityRepositoryInterface
  73.      */
  74.     private $countryStateRepository;
  76.     /** @var array<string, string> */
  77.     private array $cachedCountryIds       = [];
  79.     /** @var array<string, string> */
  80.     private array $cachedIsoCountryIds    = [];
  82.     /** @var array<string, string> */
  83.     private array $cachedSalesChannelIds  = [];
  85.     /** @var string[] */
  86.     private array $defaultSalesChannelIds = [];
  88.     /**
  89.      * EntityRepositoryDecorator constructor.
  90.      *
  91.      * @param EntityRepositoryInterface $repository
  92.      * @param EntityRepositoryInterface $salesChannelRepository
  93.      * @param EntityRepositoryInterface $countryRepository
  94.      * @param EntityRepositoryInterface $countryStateRepository
  95.      */
  96.     public function __construct(
  97.         EntityRepositoryInterface $repository,
  98.         EntityRepositoryInterface $salesChannelRepository,
  99.         EntityRepositoryInterface $countryRepository,
  100.         EntityRepositoryInterface $countryStateRepository
  101.     ) {
  102.         $this->repository             = $repository;
  103.         $this->salesChannelRepository = $salesChannelRepository;
  104.         $this->countryRepository      = $countryRepository;
  105.         $this->countryStateRepository = $countryStateRepository;
  106.     }
  108.     public function getDefinition(): EntityDefinition
  109.     {
  110.         return $this->repository->getDefinition();
  111.     }
  113.     public function aggregate(Criteria $criteria, Context $context): AggregationResultCollection
  114.     {
  115.         return $this->repository->aggregate($criteria, $context);
  116.     }
  118.     public function searchIds(Criteria $criteria, Context $context): IdSearchResult
  119.     {
  120.         return $this->repository->searchIds($criteria, $context);
  121.     }
  123.     public function clone(string $id, Context $context, ?string $newId = null, ?CloneBehavior $behavior = null): EntityWrittenContainerEvent
  124.     {
  125.         return $this->repository->clone($id, $context, $newId, $behavior);
  126.     }
  128.     public function search(Criteria $criteria, Context $context): EntitySearchResult
  129.     {
  130.         return $this->repository->search($criteria, $context);
  131.     }
  133.     public function update(array $data, Context $context): EntityWrittenContainerEvent
  134.     {
  135.         return $this->repository->update($data, $context);
  136.     }
  138.     public function upsert(array $data, Context $context): EntityWrittenContainerEvent
  139.     {
  140.         if ($context->hasState(ImportExportSubscriber::STATE_IMPORT)) {
  141.             $data = $this->prepareDataForImport($data, $context);
  142.         }
  144.         return $this->repository->upsert($data, $context);
  145.     }
  147.     public function create(array $data, Context $context): EntityWrittenContainerEvent
  148.     {
  149.         return $this->repository->create($data, $context);
  150.     }
  152.     public function delete(array $data, Context $context): EntityWrittenContainerEvent
  153.     {
  154.         return $this->repository->delete($data, $context);
  155.     }
  157.     public function createVersion(string $id, Context $context, ?string $name = null, ?string $versionId = null): string
  158.     {
  159.         return $this->repository->createVersion($id, $context, $name, $versionId);
  160.     }
  162.     public function merge(string $versionId, Context $context): void
  163.     {
  164.         $this->repository->merge($versionId, $context);
  165.     }
  167.     private function prepareDataForImport(array $data, Context $context): array
  168.     {
  169.         $data[0]['countryId'] = $this->getCountryId($data[0]['countryId'], $context);
  170.         $data[0]['id']        = $this->createStoreId($data[0], $context);
  172.         /**
  173.          * Due to the condition above data must be an array
  174.          *
  175.          * @psalm-suppress MixedArgument
  176.          */
  177.         $data[0]['showAlways'] = $this->setShowAlways($data[0]);
  179.         if (isset($data[0]['salesChannels'])) {
  180.             /**
  181.              * @var array $salesChannels
  182.              * @psalm-suppress MixedArrayAccess - It has to be available because it is checked by isset
  183.              */
  184.             $salesChannels = $data[0]['salesChannels'];
  186.             /** @psalm-suppress MixedArrayAssignment */
  187.             $data[0]['salesChannels'] = $this->assignSalesChannel($salesChannels, $context);
  188.         } else {
  189.             $salesChannelIds = $this->getSalesChannelsIds($context);
  190.             unset($data[0]['salesChannels']);
  191.             foreach ($salesChannelIds as $channelId) {
  192.                 $data[0]['salesChannels'][] = [ 'id' => $channelId ];
  193.             }
  194.         }
  196.         if (isset($data[0]['countryId'])) {
  197.             $data[0]['countryId'] = $this->assignCountryIdByIso($data[0], $context);
  199.             if (isset($data[0]['country'])) {
  200.                 unset($data[0]['country']);
  201.             }
  203.             if (isset($data[0]['countryState'])) {
  204.                 $data[0]['countryStateId'] = $this->assignCountryStateByShortCode($data[0], $context);
  205.                 unset($data[0]['countryState']);
  206.             }
  207.         }
  209.         return $data;
  210.     }
  212.     private function getSalesChannelsIds(Context $context): array
  213.     {
  214.         if (empty($this->defaultSalesChannelIds)) {
  215.             /** @var string[] $ids */
  216.             $ids = $this->salesChannelRepository->searchIds(new Criteria(), $context)->getIds();
  217.             $this->defaultSalesChannelIds = $ids;
  218.         }
  220.         return $this->defaultSalesChannelIds;
  222.     }
  224.     /**
  225.      * @param string  $id
  226.      * @param Context $context
  227.      *
  228.      * @return string
  229.      */
  230.     private function getCountryId(string $id, Context $context): ?string
  231.     {
  232.         if ($this->isMatchingCountryId($id, $context)) {
  233.             return $id;
  234.         }
  236.         return $this->getCountryByIso(self::COUNTRY_ISOS[$id] ?? 'DE', $context);
  237.     }
  239.     /**
  240.      * @param array   $data
  241.      * @param Context $context
  242.      *
  243.      * @return string
  244.      */
  245.     private function createStoreId(array $data, Context $context): string
  246.     {
  247.         unset($data['salesChannels'], $data['id']);
  249.         $criteria = new Criteria();
  250.         foreach ($data as $key => $value) {
  251.             if ($key !== 'country' && !empty($value) && !is_array($value)) {
  252.                 $criteria->addFilter(new EqualsFilter($key, $value));
  253.             }
  254.         }
  256.         $result = $this->repository->searchIds($criteria, $context)->firstId();
  258.         if (null === $result) {
  259.             $result = Uuid::randomHex();
  260.         }
  262.         return $result;
  263.     }
  265.     /**
  266.      * @param array $data
  267.      *
  268.      * @return string
  269.      */
  270.     private function setShowAlways(array $data): string
  271.     {
  272.         $newDisplay = 'no';
  274.         if (!isset($data['showAlways'])) {
  275.             return $newDisplay;
  276.         }
  278.         $showAlways = (string) $data['showAlways'];
  280.         switch ($showAlways) {
  281.             case '0':
  282.                 break;
  283.             case '1':
  284.                 $newDisplay = 'top';
  285.                 break;
  286.             case '2':
  287.                 $newDisplay = 'bottom';
  288.                 break;
  289.         }
  291.         return $newDisplay;
  292.     }
  294.     /**
  295.      * @param string  $salesChannelId
  296.      * @param Context $context
  297.      *
  298.      * @return bool
  299.      */
  300.     private function isMatchingSalesChannel(string $salesChannelId, Context $context): bool
  301.     {
  302.         if (isset($this->cachedSalesChannelIds[$salesChannelId])) {
  303.             return true;
  304.         }
  306.         $criteria = new Criteria([ $salesChannelId ]);
  307.         $result   = $this->salesChannelRepository->searchIds($criteria, $context)->firstId();
  309.         if (is_string($result)) {
  310.             $this->cachedSalesChannelIds[$salesChannelId] = $result;
  311.         }
  313.         return isset($this->cachedSalesChannelIds[$salesChannelId]);
  314.     }
  316.     /**
  317.      * @param array   $salesChannels
  318.      * @param Context $context
  319.      *
  320.      * @return array
  321.      */
  322.     private function assignSalesChannel(array $salesChannels, Context $context): array
  323.     {
  324.         foreach ($salesChannels as $key => $salesChannel) {
  325.             if (!$this->isMatchingSalesChannel($salesChannel['id'], $context)) {
  326.                 unset($salesChannels[$key]);
  327.             }
  328.         }
  330.         if (empty($salesChannels)) {
  331.             $defaultSalesChannelIds = $this->getSalesChannelsIds($context);
  332.             foreach ($defaultSalesChannelIds as $defaultSalesChannelId) {
  333.                 $salesChannels[] = [ 'id' => $defaultSalesChannelId ];
  334.             }
  335.         }
  337.         return $salesChannels;
  338.     }
  340.     /**
  341.      * @param array   $data
  342.      * @param Context $context
  343.      *
  344.      * @return string
  345.      */
  346.     private function assignCountryIdByIso(array $data, Context $context): ?string
  347.     {
  348.         if (isset($data['countryId'], $data['country']['iso']) && !$this->isMatchingCountryId($data['countryId'], $context)) {
  349.             $data['countryId'] = $this->getCountryByIso($data['country']['iso'], $context);
  350.         }
  352.         if (!isset($data['countryId'])) {
  353.             $data['countryId'] = $this->getCountryId('deutsch', $context);
  354.         }
  356.         return $data['countryId'];
  357.     }
  359.     /**
  360.      * @param array   $data
  361.      * @param Context $context
  362.      *
  363.      * @return string
  364.      */
  365.     private function assignCountryStateByShortCode(array $data, Context $context): string
  366.     {
  367.         if (isset($data['countryStateId'], $data['countryState']['shortCode']) && !$this->isMatchingCountryStateId($data['countryStateId'], $context)) {
  368.             $data['countryStateId'] = $this->getCountryStateByShortCode($data['countryState']['shortCode'], $context);
  369.         }
  371.         return $data['countryStateId'];
  372.     }
  374.     /**
  375.      * @param string  $countryId
  376.      * @param Context $context
  377.      *
  378.      * @return bool
  379.      */
  380.     private function isMatchingCountryId(string $countryId, Context $context): bool
  381.     {
  382.         if (isset($this->cachedCountryIds[$countryId])) {
  383.             return true;
  384.         }
  386.         $criteria = new Criteria([ $countryId ]);
  387.         $result   = $this->countryRepository->searchIds($criteria, $context)->firstId();
  389.         if (is_string($result)) {
  390.             $this->cachedCountryIds[$result] = $result;
  391.         }
  393.         return isset($this->cachedCountryIds[$countryId]);
  394.     }
  396.     /**
  397.      * @param string  $iso
  398.      * @param Context $context
  399.      *
  400.      * @return string|null
  401.      */
  402.     private function getCountryByIso(string $iso, Context $context): ?string
  403.     {
  404.         if (isset($this->cachedIsoCountryIds[$iso])) {
  405.             return $this->cachedIsoCountryIds[$iso];
  406.         }
  408.         $criteria = new Criteria();
  409.         $criteria->addFilter(new EqualsFilter('iso', $iso));
  411.         $countryId = $this->countryRepository->searchIds($criteria, $context)->firstId();
  413.         if (is_string($countryId)) {
  414.             $this->cachedIsoCountryIds[$iso] = $countryId;
  415.         }
  417.         return $countryId;
  418.     }
  420.     /**
  421.      * @param string  $countryStateId
  422.      * @param Context $context
  423.      *
  424.      * @return bool
  425.      */
  426.     private function isMatchingCountryStateId(string $countryStateId, Context $context): bool
  427.     {
  428.         $criteria = new Criteria([ $countryStateId]);
  429.         $result   = $this->countryStateRepository->searchIds($criteria, $context)->getIds();
  431.         return !empty($result);
  432.     }
  434.     /**
  435.      * @param string  $shortCode
  436.      * @param Context $context
  437.      *
  438.      * @return string|null
  439.      */
  440.     private function getCountryStateByShortCode(string $shortCode, Context $context): ?string
  441.     {
  442.         $criteria = new Criteria();
  443.         $criteria->addFilter(new EqualsFilter('shortCode', $shortCode));
  445.         return $this->countryStateRepository->searchIds($criteria, $context)->firstId();
  446.     }
  447. }