Merge pull request #11 from justbetter/feature/check-customers

Author: VincentBean

composer.json

@@ -19,6 +19,7 @@
     "require": {
         "ext-json": "*",
         "magento/module-catalog": "*",
+        "magento/module-customer": "*",
         "magento/module-eav": "*"
     },
     "autoload": {

phpstan.neon

@@ -6,6 +6,7 @@ parameters:
         - src
     level: 1
     ignoreErrors:
+        - '#Magento\\Customer\\Model\\ResourceModel\\Customer\\CollectionFactory#'
         - '#JustBetter\\CustomerPricing\\Model\\ResourceModel\\CustomerPricing\\CollectionFactory#'
         - '#JustBetter\\CustomerPricing\\Model\\ResourceModel\\CustomerPricingFactory#'
         - '#JustBetter\\CustomerPricing\\Model\\CustomerPricingFactory#'

src/Model/CustomerPricingManagement.php

@@ -4,26 +4,38 @@
 
 use JustBetter\CustomerPricing\Api\CustomerPricingManagementInterface;
 use JustBetter\CustomerPricing\Model\CustomerPricingFactory as ModelFactory;
-use JustBetter\CustomerPricing\Model\ResourceModel\CustomerPricingFactory as ResourceModelFactory;
 use JustBetter\CustomerPricing\Model\ResourceModel\CustomerPricing\Collection;
+use JustBetter\CustomerPricing\Model\ResourceModel\CustomerPricingFactory as ResourceModelFactory;
 use Magento\Catalog\Api\ProductRepositoryInterface;
 use Magento\Catalog\Model\Product;
+use Magento\Customer\Model\ResourceModel\Customer\CollectionFactory as CustomerCollectionFactory;
 
 class CustomerPricingManagement implements CustomerPricingManagementInterface
 {
     public function __construct(
         protected ProductRepositoryInterface $productRepository,
         protected ModelFactory $customerFactory,
         protected ResourceModelFactory $customerPricingFactory,
-        protected Collection $customerPricingCollection
-    ) {
-    }
+        protected Collection $customerPricingCollection,
+        protected CustomerCollectionFactory $customerCollectionFactory
+    ) {}
 
     /**
-     * @inheritDoc
+     * {@inheritDoc}
      */
     public function saveCustomerPrices(string $sku, mixed $customerPrices): bool
     {
+        $customerIds = array_unique(array_column($customerPrices, 'customer_id'));
+
+        $customerCollection = $this->customerCollectionFactory->create()
+            ->addFieldToFilter('entity_id', ['in' => $customerIds])
+            ->addFieldToSelect('entity_id');
+
+        $validCustomerIds = array_map(
+            static fn ($customer) => (int) $customer->getId(),
+            $customerCollection->getItems()
+        );
+
         /** @var Product $product */
         $product = $this->productRepository->get($sku);
 
@@ -43,6 +55,10 @@ public function saveCustomerPrices(string $sku, mixed $customerPrices): bool
             $quantity = $customerPriceData['quantity'];
             $price = $customerPriceData['price'];
 
+            if (! in_array($customerId, $validCustomerIds)) {
+                continue;
+            }
+
             $updated = false;
 
             /** @var CustomerPricing $currentCustomerPrice */