Merge pull request #75 from ajthinking/add-insert-stmt-closure-with-b…

…uilder Allow passing a closure to insertStmt
ajthinking · May 25, 2022 · 62abebd · 62abebd
2 parents 4d84592 + 490a6b6
commit 62abebd
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 0 deletions.
diff --git a/src/Support/AST/ASTQueryBuilder.php b/src/Support/AST/ASTQueryBuilder.php
@@ -20,6 +20,7 @@
 use Exception;
 use Illuminate\Support\Arr;
 use Illuminate\Support\Collection;
+use PhpParser\BuilderFactory;
 use PhpParser\ConstExprEvaluator;
 
 class ASTQueryBuilder
@@ -351,6 +352,8 @@ public function insertStmts($newNodes): self
 
     public function insertStmt($newNode): self
     {
+		if($newNode instanceof Closure) $newNode = $newNode(new BuilderFactory);
+
         $this->currentNodes()->each(function ($node) use ($newNode) {
             $target = $node->result;
 

diff --git a/tests/Unit/Support/AST/ASTQueryBuilderTest.php b/tests/Unit/Support/AST/ASTQueryBuilderTest.php
@@ -203,4 +203,16 @@
 		->render();
 
 	assertEquals($original, $after);
+});
+
+it('can insert stmt with a closure', function() {
+	PHPFile::make()->class(\App\Dummy::class)
+		->astQuery()
+		->class()
+		->insertStmt(function(BuilderFactory $builder) {
+			return $builder->property('someProperty')->getNode();
+		})
+		->commit()
+		->end()
+		->preview();
 });