Fixes #3668

craftcms · Sep 13, 2024 · 4fe4ebe · 4fe4ebe
1 parent 40f75e1
commit 4fe4ebe
Show file tree

Hide file tree

Showing 3 changed files with 33 additions and 12 deletions.
diff --git a/src/collections/UpdateInventoryLevelCollection.php b/src/collections/UpdateInventoryLevelCollection.php
@@ -40,4 +40,14 @@ public static function make($items = [])
         $collection = parent::make($items);
         return $collection;
     }
+
+    /**
+     * @return array
+     */
+    public function getPurchasables(): array
+    {
+        return $this->map(function(UpdateInventoryLevel $updateInventoryLevel) {
+            return $updateInventoryLevel->inventoryItem->getPurchasable();
+        })->all();
+    }
 }
diff --git a/src/services/Inventory.php b/src/services/Inventory.php
@@ -306,8 +306,9 @@ public function executeUpdateInventoryLevels(UpdateInventoryLevelCollection $upd
 
             $transaction->commit();
 
-            // TODO: Update stock value on purchasable stores
-            // Craft::$app->getElements()->invalidateCachesForElement($this);
+            foreach ($updateInventoryLevels->getPurchasables() as $purchasable) {
+                Plugin::getInstance()->getPurchasables()->updateStoreStockCache($purchasable, true);
+            }
 
             return true;
         } catch (\Exception $e) {
@@ -734,7 +735,7 @@ public function orderCompleteHandler(Order $order)
 
         foreach ($selectedInventoryLevelForItem as $inventoryLevel) {
             $purchasable = $inventoryLevel->getPurchasable();
-            Plugin::getInstance()->getPurchasables()->updateStoreStockCache($purchasable);
+            Plugin::getInstance()->getPurchasables()->updateStoreStockCache($purchasable, true);
         }
     }
 }
diff --git a/src/services/Purchasables.php b/src/services/Purchasables.php
@@ -155,20 +155,30 @@ public function isPurchasableShippable(PurchasableInterface $purchasable, Order
      * Updated the cached stock value for the purchasable in a store.
      *
      * @param Purchasable $purchasable
+     * @param bool $allSites Update across all sites (stores).
      * @return void
      * @throws \yii\base\InvalidConfigException
      * @throws \yii\db\Exception
      */
-    public function updateStoreStockCache(Purchasable $purchasable): void
+    public function updateStoreStockCache(Purchasable $purchasable, bool $allSites = false): void
     {
-        $stock = Plugin::getInstance()->getInventory()->getInventoryLevelsForPurchasable($purchasable)->sum('availableTotal');
-
-        Craft::$app->getDb()->createCommand()
-            ->update(
-                table:Table::PURCHASABLES_STORES,
-                columns: ['stock' => $stock],
-                condition: ['id' => $purchasable->id, 'storeId' => $purchasable->getStore()->id])
-            ->execute();
+        if ($allSites) {
+            $purchasables = $purchasable::find()->siteid('*')->id($purchasable->id)->status(null)->all();
+        } else {
+            $purchasables = [$purchasable];
+        }
+
+        /** @var Purchasable $purchasable */
+        foreach ($purchasables as $purchasable) {
+            $stock = Plugin::getInstance()->getInventory()->getInventoryLevelsForPurchasable($purchasable)->sum('availableTotal');
+
+            Craft::$app->getDb()->createCommand()
+                ->update(
+                    table: Table::PURCHASABLES_STORES,
+                    columns: ['stock' => $stock],
+                    condition: ['purchasableId' => $purchasable->id, 'storeId' => $purchasable->getStore()->id])
+                ->execute();
+        }
     }
 
     /**