Merge pull request #151 from spryker-sdk/feature/sdk-2855-add-missed-…

…plugin-classes SDK 2855 add missed plugin classes
spryker-sdk · Oct 27, 2023 · 431e07e · 431e07e
2 parents b16d7d2 + 3e188ed
commit 431e07e
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/src/Builder/Visitor/AddPluginToPluginListVisitor.php b/src/Builder/Visitor/AddPluginToPluginListVisitor.php
@@ -211,7 +211,10 @@ protected function addNewPluginWithConditionIntoList(ClassMethod $node): Node
             $newStmts[] = $this->createNewConditionStatement($returnStmt->expr);
         }
         if ($returnStmt->expr instanceof FuncCall || $returnStmt->expr instanceof Array_) {
-            $newStmts[] = $this->createAssignEmptyArray(static::PLUGINS_VARIBLE);
+            $newStmts[] = $this->createAssignArray(
+                static::PLUGINS_VARIBLE,
+                $returnStmt->expr instanceof Array_ ? $returnStmt->expr : null,
+            );
             $newStmts[] = $this->createNewConditionStatement((new BuilderFactory())->var(static::PLUGINS_VARIBLE));
             $returnStmt->expr = (new BuilderFactory())->var(static::PLUGINS_VARIBLE);
         }
@@ -223,15 +226,16 @@ protected function addNewPluginWithConditionIntoList(ClassMethod $node): Node
 
     /**
      * @param string $varName
+     * @param \PhpParser\Node\Expr\Array_|null $initialArray
      *
      * @return \PhpParser\Node\Stmt\Expression
      */
-    protected function createAssignEmptyArray(string $varName): Expression
+    protected function createAssignArray(string $varName, ?Array_ $initialArray = null): Expression
     {
         return new Expression(
             new Assign(
                 (new BuilderFactory())->var($varName),
-                new Array_(),
+                $initialArray ?? new Array_(),
             ),
         );
     }