ltsecurityinc安防产品在线商城代码

app/code/Lts/Inventory/Model/SourceCustomerRepository.php

@@ -0,0 +1,190 @@
+<?php
+/**
+ * Copyright ©  All rights reserved.
+ * See COPYING.txt for license details.
+ */
+declare(strict_types=1);
+
+namespace Lts\Inventory\Model;
+
+use Lts\Inventory\Api\Data\SourceCustomerInterface;
+use Lts\Inventory\Api\Data\SourceCustomerInterfaceFactory;
+use Lts\Inventory\Api\Data\SourceCustomerSearchResultsInterfaceFactory;
+use Lts\Inventory\Api\SourceCustomerRepositoryInterface;
+use Lts\Inventory\Model\ResourceModel\SourceCustomer as ResourceInventorySourceCustomer;
+use Lts\Inventory\Model\ResourceModel\SourceCustomer\CollectionFactory as SourceCustomerCollectionFactory;
+use Magento\Framework\Api\SearchCriteria\CollectionProcessorInterface;
+use Magento\Framework\Exception\CouldNotDeleteException;
+use Magento\Framework\Exception\CouldNotSaveException;
+use Magento\Framework\Exception\NoSuchEntityException;
+use Magento\InventoryApi\Api\Data\SourceInterface;
+use Magento\InventoryApi\Api\SourceRepositoryInterface;
+
+class SourceCustomerRepository implements SourceCustomerRepositoryInterface
+{
+
+    /**
+     * @var SourceCustomerInterfaceFactory
+     */
+    protected $sourceCustomerFactory;
+
+    /**
+     * @var CollectionProcessorInterface
+     */
+    protected $collectionProcessor;
+
+    /**
+     * @var SourceCustomerCollectionFactory
+     */
+    protected $sourceCustomerCollectionFactory;
+
+    /**
+     * @var ResourceInventorySourceCustomer
+     */
+    protected $resource;
+
+    /**
+     * @var SourceCustomer
+     */
+    protected $searchResultsFactory;
+    private \Magento\Framework\Api\SearchCriteriaBuilderFactory $searchCriteriaBuilderFactory;
+    private SourceRepositoryInterface $sourceRepository;
+
+
+    /**
+     * @param ResourceInventorySourceCustomer $resource
+     * @param SourceCustomerInterfaceFactory $sourceCustomerFactory
+     * @param sourceCustomerCollectionFactory $sourceCustomerCollectionFactory
+     * @param SourceCustomerSearchResultsInterfaceFactory $searchResultsFactory
+     * @param CollectionProcessorInterface $collectionProcessor
+     */
+    public function __construct(
+        ResourceInventorySourceCustomer $resource,
+        SourceCustomerInterfaceFactory $sourceCustomerFactory,
+        SourceCustomerCollectionFactory $sourceCustomerCollectionFactory,
+        SourceCustomerSearchResultsInterfaceFactory $searchResultsFactory,
+        CollectionProcessorInterface $collectionProcessor,
+        \Magento\Framework\Api\SearchCriteriaBuilderFactory $searchCriteriaBuilderFactory,
+        SourceRepositoryInterface $sourceRepository,
+    ) {
+        $this->resource = $resource;
+        $this->sourceCustomerFactory = $sourceCustomerFactory;
+        $this->sourceCustomerCollectionFactory = $sourceCustomerCollectionFactory;
+        $this->searchResultsFactory = $searchResultsFactory;
+        $this->collectionProcessor = $collectionProcessor;
+        $this->searchCriteriaBuilderFactory = $searchCriteriaBuilderFactory;
+        $this->sourceRepository = $sourceRepository;
+    }
+
+    /**
+     * @inheritDoc
+     */
+    public function save(
+        SourceCustomerInterface $sourceCustomer
+    ) {
+        try {
+            $this->sourceRepository->get($sourceCustomer->getSourceCode());
+            if (!$sourceCustomer->getEntityId() && $sourceCustomer->getCustomerId()) {
+                $collection = $this->sourceCustomerCollectionFactory->create()
+                    ->setItemObjectClass(SourceCustomer::class)
+                    ->addFilter(SourceCustomerInterface::CUSTOMER_ID, $sourceCustomer->getCustomerId());
+
+                $temp = $collection->getFirstItem();
+                if ($temp->getEntityId() && !$sourceCustomer->getEntityId()) {
+                    $sourceCustomer->setEntityId($temp->getEntityId());
+                }
+            }
+            $this->resource->save($sourceCustomer);
+        } catch (\Exception $exception) {
+            throw new CouldNotSaveException(__(
+                'Could not save the inventorySourceCustomer: %1',
+                $exception->getMessage()
+            ));
+        }
+        return $sourceCustomer;
+    }
+
+    public function getFromCustomerId(int $customerId): ?SourceCustomerInterface
+    {
+        $inventorySourceCustomer = $this->sourceCustomerFactory->create();
+        $this->resource->load($inventorySourceCustomer, $customerId, 'customer_id');
+        if (!$inventorySourceCustomer->getId()) {
+            throw new NoSuchEntityException(__('SourceCustomer with customer id "%1" does not exist.', $customerId));
+        }
+        return $inventorySourceCustomer;
+    }
+    /**
+     * @inheritDoc
+     */
+    public function get($id)
+    {
+        $inventorySourceCustomer = $this->sourceCustomerFactory->create();
+        $this->resource->load($inventorySourceCustomer, $id);
+        if (!$inventorySourceCustomer->getId()) {
+            throw new NoSuchEntityException(__('SourceCustomer with id "%1" does not exist.', $id));
+        }
+        return $inventorySourceCustomer;
+    }
+
+    /**
+     * @inheritDoc
+     */
+    public function getList(
+        string $customerId
+    ) {
+
+        $criteria = $this->searchCriteriaBuilderFactory->create()
+            ->addFilter(SourceInterface::ENABLED, true)
+            ->addFilter('is_pickup_location_active', 1)
+            ->create();
+
+        $searchResults = $this->sourceRepository->getList($criteria);
+        
+        $items = [];
+        $index = 0;
+        /* @var $model SourceInterface */
+        $sourceCustomer = $this->sourceCustomerFactory->create();
+        $this->resource->load($sourceCustomer, $customerId, 'customer_id');
+
+        foreach ($searchResults->getItems() as $model) {
+            if ($sourceCustomer->getSourceCode() == $model->getSourceCode()) {
+                $items[0] = $sourceCustomer->setSource($model);
+            } else {
+                $index += 1;
+                $items[$index] = $this->sourceCustomerFactory->create()->setSource($model);
+            }
+        }
+        ksort($items);
+        $searchResults->setItems(array_values($items));
+        $searchResults->setTotalCount($searchResults->getTotalCount());
+        return $searchResults;
+    }
+
+    /**
+     * @inheritDoc
+     */
+    public function delete(
+        SourceCustomerInterface $inventorySourceCustomer
+    ) {
+        try {
+            $inventorySourceCustomerModel = $this->sourceCustomerFactory->create();
+            $this->resource->load($inventorySourceCustomerModel, $inventorySourceCustomer->getInventorysourcecustomerId());
+            $this->resource->delete($inventorySourceCustomerModel);
+        } catch (\Exception $exception) {
+            throw new CouldNotDeleteException(__(
+                'Could not delete the SourceCustomer: %1',
+                $exception->getMessage()
+            ));
+        }
+        return true;
+    }
+
+    /**
+     * @inheritDoc
+     */
+    public function deleteById(int $entityId)
+    {
+        return $this->delete($this->get($entityId));
+    }
+}
+