Add Distinct query modifier and CountDistinct operand (#209)

* create Distinct query modifier

* create CountDistinct operand

* remove not used classes

src/Operand/CountDistinct.php

@@ -0,0 +1,35 @@
+<?php
+
+namespace Happyr\DoctrineSpecification\Operand;
+
+use Doctrine\ORM\QueryBuilder;
+
+class CountDistinct implements Operand
+{
+    /**
+     * @var Operand|string
+     */
+    private $field;
+
+    /**
+     * @param Operand|string $field
+     */
+    public function __construct($field)
+    {
+        $this->field = $field;
+    }
+
+    /**
+     * @param QueryBuilder $qb
+     * @param string       $dqlAlias
+     *
+     * @return string
+     */
+    public function transform(QueryBuilder $qb, $dqlAlias)
+    {
+        $field = ArgumentToOperandConverter::toField($this->field);
+        $field = $field->transform($qb, $dqlAlias);
+
+        return sprintf('COUNT(DISTINCT %s)', $field);
+    }
+}

src/Query/Distinct.php

@@ -0,0 +1,17 @@
+<?php
+
+namespace Happyr\DoctrineSpecification\Query;
+
+use Doctrine\ORM\QueryBuilder;
+
+class Distinct implements QueryModifier
+{
+    /**
+     * @param QueryBuilder $qb
+     * @param string       $dqlAlias
+     */
+    public function modify(QueryBuilder $qb, $dqlAlias)
+    {
+        $qb->distinct();
+    }
+}

src/Spec.php

@@ -21,6 +21,7 @@
 use Happyr\DoctrineSpecification\Operand\BitOr;
 use Happyr\DoctrineSpecification\Operand\BitRightShift;
 use Happyr\DoctrineSpecification\Operand\BitXor;
+use Happyr\DoctrineSpecification\Operand\CountDistinct;
 use Happyr\DoctrineSpecification\Operand\Division;
 use Happyr\DoctrineSpecification\Operand\Field;
 use Happyr\DoctrineSpecification\Operand\LikePattern;
@@ -32,6 +33,7 @@
 use Happyr\DoctrineSpecification\Operand\Value;
 use Happyr\DoctrineSpecification\Operand\Values;
 use Happyr\DoctrineSpecification\Query\AddSelect;
+use Happyr\DoctrineSpecification\Query\Distinct;
 use Happyr\DoctrineSpecification\Query\GroupBy;
 use Happyr\DoctrineSpecification\Query\IndexBy;
 use Happyr\DoctrineSpecification\Query\InnerJoin;
@@ -245,6 +247,14 @@ public static function groupBy($field, $dqlAlias = null)
         return new GroupBy($field, $dqlAlias);
     }
 
+    /**
+     * @return Distinct
+     */
+    public static function distinct()
+    {
+        return new Distinct();
+    }
+
     /*
      * Selection
      */
@@ -581,6 +591,16 @@ public static function likePattern($value, $format = LikePattern::CONTAINS)
         return new LikePattern($value, $format);
     }
 
+    /**
+     * @param Operand|string $field
+     *
+     * @return CountDistinct
+     */
+    public static function countDistinct($field)
+    {
+        return new CountDistinct($field);
+    }
+
     /*
      * Arithmetic operands
      */

tests/Operand/CountDistinctSpec.php

@@ -0,0 +1,36 @@
+<?php
+
+namespace tests\Happyr\DoctrineSpecification\Operand;
+
+use Doctrine\ORM\QueryBuilder;
+use Happyr\DoctrineSpecification\Operand\CountDistinct;
+use Happyr\DoctrineSpecification\Operand\Operand;
+use PhpSpec\ObjectBehavior;
+
+/**
+ * @mixin CountDistinct
+ */
+class CountDistinctSpec extends ObjectBehavior
+{
+    private $field = 'foo';
+
+    public function let()
+    {
+        $this->beConstructedWith($this->field);
+    }
+
+    public function it_is_a_count_distinct()
+    {
+        $this->shouldBeAnInstanceOf(CountDistinct::class);
+    }
+
+    public function it_is_a_operand()
+    {
+        $this->shouldBeAnInstanceOf(Operand::class);
+    }
+
+    public function it_is_transformable(QueryBuilder $qb)
+    {
+        $this->transform($qb, 'a')->shouldReturn('COUNT(DISTINCT a.foo)');
+    }
+}

tests/Query/DistinctSpec.php

@@ -0,0 +1,30 @@
+<?php
+
+namespace tests\Happyr\DoctrineSpecification\Query;
+
+use Doctrine\ORM\QueryBuilder;
+use Happyr\DoctrineSpecification\Query\Distinct;
+use Happyr\DoctrineSpecification\Query\QueryModifier;
+use PhpSpec\ObjectBehavior;
+
+/**
+ * @mixin Distinct
+ */
+class DistinctSpec extends ObjectBehavior
+{
+    public function it_is_a_distinct()
+    {
+        $this->shouldBeAnInstanceOf(Distinct::class);
+    }
+
+    public function it_is_a_query_modifier()
+    {
+        $this->shouldHaveType(QueryModifier::class);
+    }
+
+    public function it_add_having(QueryBuilder $qb)
+    {
+        $qb->distinct()->shouldBeCalled();
+        $this->modify($qb, 'a');
+    }
+}

tests/SpecSpec.php

@@ -8,12 +8,14 @@
 use Happyr\DoctrineSpecification\Operand\Alias;
 use Happyr\DoctrineSpecification\Operand\BitAnd;
 use Happyr\DoctrineSpecification\Operand\BitOr;
+use Happyr\DoctrineSpecification\Operand\CountDistinct;
 use Happyr\DoctrineSpecification\Operand\Division;
 use Happyr\DoctrineSpecification\Operand\Modulo;
 use Happyr\DoctrineSpecification\Operand\Multiplication;
 use Happyr\DoctrineSpecification\Operand\PlatformFunction;
 use Happyr\DoctrineSpecification\Operand\Subtraction;
 use Happyr\DoctrineSpecification\Query\AddSelect;
+use Happyr\DoctrineSpecification\Query\Distinct;
 use Happyr\DoctrineSpecification\Query\Select;
 use Happyr\DoctrineSpecification\Query\Selection\SelectAs;
 use Happyr\DoctrineSpecification\Query\Selection\SelectEntity;
@@ -31,6 +33,16 @@ public function it_creates_an_x_specification()
         $this->andX()->shouldReturnAnInstanceOf(LogicX::class);
     }
 
+    public function it_creates_distinct()
+    {
+        $this->distinct()->shouldReturnAnInstanceOf(Distinct::class);
+    }
+
+    public function it_creates_count_distinct()
+    {
+        $this->countDistinct('foo')->shouldReturnAnInstanceOf(CountDistinct::class);
+    }
+
     public function it_creates_add_operand()
     {
         $this->add('foo', 'bar')->shouldReturnAnInstanceOf(Addition::class);