Cleanup

Author: GromNaN

generator/src/FluentStageFactoryGenerator.php

@@ -19,9 +19,11 @@
 use MongoDB\Builder\Stage;
 use MongoDB\Builder\Type\AccumulatorInterface;
 use MongoDB\Builder\Type\ExpressionInterface;
+use MongoDB\Builder\Type\FieldQueryInterface;
 use MongoDB\Builder\Type\Optional;
 use MongoDB\Builder\Type\QueryInterface;
 use MongoDB\Builder\Type\Sort;
+use MongoDB\Builder\Type\StageInterface;
 use MongoDB\CodeGenerator\Definition\GeneratorDefinition;
 use MongoDB\Model\BSONArray;
 use Nette\PhpGenerator\ClassType;
@@ -61,6 +63,8 @@ private function createFluentFactoryClass(GeneratorDefinition $definition): PhpN
         $namespace = new PhpNamespace($definition->namespace);
         $class = $namespace->addClass('FluentFactory');
 
+        $namespace->addUse(StageInterface::class);
+        $namespace->addUse(FieldQueryInterface::class);
         $namespace->addUse(Pipeline::class);
         $namespace->addUse(Decimal128::class);
         $namespace->addUse(Document::class);
@@ -84,6 +88,7 @@ private function createFluentFactoryClass(GeneratorDefinition $definition): PhpN
         $namespace->addUse(self::FACTORY_CLASS);
         $class->addProperty('pipeline')
             ->setType('array')
+            ->setComment('@var list<StageInterface>')
             ->setValue([]);
         $class->addMethod('getPipeline')
             ->setReturnType(Pipeline::class)

src/Builder/Stage/FluentFactory.php

@@ -2,8 +2,10 @@
 
 declare(strict_types=1);
 
-namespace MongoDB\Builder;
+namespace MongoDB\Tests\Builder;
 
+use MongoDB\Builder\Pipeline;
+use MongoDB\Builder\Query;
 use MongoDB\Builder\Stage\FluentFactory;
 use MongoDB\Builder\Type\Sort;
 use PHPUnit\Framework\TestCase;