Added prune variable support

LaravelFreelancerNL · May 15, 2022 · 051b6e1 · 051b6e1
1 parent 3fe8feb
commit 051b6e1
Show file tree

Hide file tree

Showing 3 changed files with 53 additions and 3 deletions.
diff --git a/src/AQL/HasGraphClauses.php b/src/AQL/HasGraphClauses.php
@@ -143,14 +143,15 @@ public function prune(
         object|array|string|int|float|bool|null $leftOperand,
         string|QueryBuilder|Expression $comparisonOperator = null,
         object|array|string|int|float|bool|null $rightOperand = null,
-        string|QueryBuilder|Expression $logicalOperator = null
+        string|QueryBuilder|Expression $logicalOperator = null,
+        string $pruneVariable = null
     ): self {
         $predicates = $leftOperand;
         if (! is_array($predicates)) {
             $predicates = [[$leftOperand, $comparisonOperator, $rightOperand, $logicalOperator]];
         }
 
-        $this->addCommand(new PruneClause($predicates));
+        $this->addCommand(new PruneClause($predicates, $pruneVariable));
 
         return $this;
     }

diff --git a/src/Clauses/PruneClause.php b/src/Clauses/PruneClause.php
@@ -4,16 +4,41 @@
 
 namespace LaravelFreelancerNL\FluentAQL\Clauses;
 
+use LaravelFreelancerNL\FluentAQL\Expressions\PredicateExpression;
 use LaravelFreelancerNL\FluentAQL\QueryBuilder;
 
 class PruneClause extends FilterClause
 {
+    protected ?string $pruneVariable = null;
+
+    /**
+     * Filter statement.
+     *
+     * @param array<mixed> $predicates
+     */
+    public function __construct(
+        array|PredicateExpression $predicates,
+        string $pruneVariable = null
+    ) {
+        parent::__construct($predicates);
+
+        $this->pruneVariable = $pruneVariable;
+    }
+
+
     public function compile(QueryBuilder $queryBuilder): string
     {
+        $aql = 'PRUNE ';
+
+        if (isset($this->pruneVariable)) {
+            $pruneVariable = $queryBuilder->normalizeArgument($this->pruneVariable, 'Variable');
+            $aql .= $pruneVariable->compile($queryBuilder) . ' = ';
+        }
+
         $this->predicates = $queryBuilder->normalizePredicates($this->predicates);
 
         $compiledPredicates = $queryBuilder->compilePredicates($this->predicates);
 
-        return 'PRUNE ' . rtrim($compiledPredicates);
+        return  $aql . rtrim($compiledPredicates);
     }
 }
diff --git a/tests/Unit/AQL/GraphClausesTest.php b/tests/Unit/AQL/GraphClausesTest.php
@@ -158,4 +158,28 @@ public function testPruneClauseWithMultiplePredicates()
              ->get();
         self::assertEquals('FOR u IN Users PRUNE u.active == true AND u.age > 18', $result->query);
     }
+
+    /**
+     * @covers \LaravelFreelancerNL\FluentAQL\Clauses\PruneClause
+     */
+    public function testPruneClauseWithVariable()
+    {
+        $result = (new QueryBuilder())
+            ->for('u', 'users')
+            ->prune('u.active', '==', 'true', null, 'pruneCondition')
+            ->get();
+        self::assertEquals('FOR u IN users PRUNE pruneCondition = u.active == true', $result->query);
+
+        $result = (new QueryBuilder())
+            ->for('u', 'Users')
+            ->prune('u.active', '==', 'true', 'OR', 'pruneCondition')
+            ->get();
+        self::assertEquals('FOR u IN Users PRUNE pruneCondition = u.active == true', $result->query);
+
+        $result = (new QueryBuilder())
+            ->for('u', 'Users')
+            ->prune(['u.active', '==', 'true'], pruneVariable: 'pruneCondition')
+            ->get();
+        self::assertEquals('FOR u IN Users PRUNE pruneCondition = u.active == true', $result->query);
+    }
 }