add code argument in TheliaCollectionItem loop

thelia-modules · Feb 12, 2024 · 40b5d4c · 40b5d4c
1 parent 964c1c0
commit 40b5d4c
Showing 1 changed file with 11 additions and 3 deletions.
diff --git a/Loop/TheliaCollectionItem.php b/Loop/TheliaCollectionItem.php
@@ -18,6 +18,7 @@
  * @method string getItemType()
  * @method int    getItemId()
  * @method int    getTheliaCollectionId()
+ * @method string getCode()
  */
 class TheliaCollectionItem extends BaseLoop implements PropelSearchLoopInterface
 {
@@ -27,7 +28,8 @@ protected function getArgDefinitions()
             Argument::createIntTypeArgument('id'),
             Argument::createIntTypeArgument('item_id'),
             Argument::createAlphaNumStringTypeArgument('item_type'),
-            Argument::createIntTypeArgument('thelia_collection_id')
+            Argument::createIntTypeArgument('thelia_collection_id'),
+            Argument::createAlphaNumStringTypeArgument('code')
         );
     }
 
@@ -43,6 +45,13 @@ public function buildModelCriteria()
             $query->filterByTheliaCollectionId($theliaCollectionId);
         }
 
+        if (null !== $code = $this->getCode()) {
+            $query
+                ->useTheliaCollectionQuery()
+                ->filterByCode($code)
+                ->endUse();
+        }
+
         if (null !== $itemId = $this->getItemId()) {
             $query->filterByItemId($itemId);
         }
@@ -66,8 +75,7 @@ public function parseResults(LoopResult $loopResult)
                 ->set('THELIA_COLLECTION_ID', $entry->getTheliaCollectionId())
                 ->set('POSITION', $entry->getPosition())
                 ->set('ITEM_TYPE', $entry->getItemType())
-                ->set('ITEM_ID', $entry->getItemId())
-            ;
+                ->set('ITEM_ID', $entry->getItemId());
 
             $this->addOutputFields($row, $entry);