feat: option to generate code with strict types

Author: MrMeshok

src/Configuration.php

@@ -36,6 +36,10 @@ public function __construct(
          * Does the mapper should throw an exception if the target is read-only.
          */
         public bool $allowReadOnlyTargetToPopulate = false,
+        /**
+         * Add declare(strict_types=1) to generated code.
+         */
+        public bool $strictTypes = false,
     ) {
     }
 }

src/Generator/MapperGenerator.php

@@ -20,6 +20,9 @@
 use PhpParser\Node\Stmt;
 use Symfony\Component\ExpressionLanguage\ExpressionLanguage;
 
+use function AutoMapper\PhpParser\create_declare_item;
+use function AutoMapper\PhpParser\create_scalar_int;
+
 /**
  * Generates code for a mapping class.
  *
@@ -32,6 +35,7 @@
     private MapperConstructorGenerator $mapperConstructorGenerator;
     private InjectMapperMethodStatementsGenerator $injectMapperMethodStatementsGenerator;
     private MapMethodStatementsGenerator $mapMethodStatementsGenerator;
+    private bool $declareStrictTypes;
     private bool $disableGeneratedMapper;
 
     public function __construct(
@@ -52,28 +56,38 @@ public function __construct(
 
         $this->injectMapperMethodStatementsGenerator = new InjectMapperMethodStatementsGenerator();
 
+        $this->declareStrictTypes = $configuration->strictTypes;
         $this->disableGeneratedMapper = !$configuration->autoRegister;
     }
 
     /**
      * Generate Class AST given metadata for a mapper.
      *
+     * @return Stmt[]
+     *
      * @throws CompileException
      * @throws InvalidMappingException
      */
-    public function generate(GeneratorMetadata $metadata): Stmt\Class_
+    public function generate(GeneratorMetadata $metadata): array
     {
         if ($this->disableGeneratedMapper) {
             throw new InvalidMappingException('No mapper found for source ' . $metadata->mapperMetadata->source . ' and target ' . $metadata->mapperMetadata->target);
         }
 
-        return (new Builder\Class_($metadata->mapperMetadata->className))
+        $statements = [];
+        if ($this->declareStrictTypes) {
+            // @phpstan-ignore argument.type
+            $statements[] = new Stmt\Declare_([create_declare_item('strict_types', create_scalar_int(1))]);
+        }
+        $statements[] = (new Builder\Class_($metadata->mapperMetadata->className))
             ->makeFinal()
             ->extend(GeneratedMapper::class)
             ->addStmt($this->constructorMethod($metadata))
             ->addStmt($this->mapMethod($metadata))
             ->addStmt($this->registerMappersMethod($metadata))
             ->getNode();
+
+        return $statements;
     }
 
     /**

src/Loader/EvalLoader.php

@@ -31,9 +31,9 @@ public function __construct(
 
     public function loadClass(MapperMetadata $mapperMetadata): void
     {
-        $class = $this->generator->generate($this->metadataFactory->getGeneratorMetadata($mapperMetadata->source, $mapperMetadata->target));
-
-        eval($this->printer->prettyPrint([$class]));
+        eval($this->printer->prettyPrint($this->generator->generate(
+            $this->metadataFactory->getGeneratorMetadata($mapperMetadata->source, $mapperMetadata->target)
+        )));
     }
 
     public function buildMappers(MetadataRegistry $registry): bool

src/Loader/FileLoader.php

@@ -78,8 +78,9 @@ public function createGeneratedMapper(MapperMetadata $mapperMetadata): string
         $className = $mapperMetadata->className;
         $classPath = $this->directory . \DIRECTORY_SEPARATOR . $className . '.php';
 
-        $generatorMetadata = $this->metadataFactory->getGeneratorMetadata($mapperMetadata->source, $mapperMetadata->target);
-        $classCode = $this->printer->prettyPrint([$this->generator->generate($generatorMetadata)]);
+        $classCode = $this->printer->prettyPrint($this->generator->generate(
+            $this->metadataFactory->getGeneratorMetadata($mapperMetadata->source, $mapperMetadata->target)
+        ));
 
         $this->write($classPath, "<?php\n\n" . $classCode . "\n");
 

src/php-parser.php

@@ -43,3 +43,19 @@ function create_expr_array_item(Expr $value, Expr $key = null, bool $byRef = fal
 
     return new $class($value, $key, $byRef, $attributes, $unpack);
 }
+
+/**
+ * Constructs a declare key=>value pair node.
+ *
+ * @param string|Node\Identifier $key        Key
+ * @param Expr                   $value      Value
+ * @param array<string, mixed>   $attributes Additional attributes
+ *
+ * @internal
+ */
+function create_declare_item($key, Expr $value, array $attributes = []): Node\DeclareItem|Node\Stmt\DeclareDeclare
+{
+    $class = class_exists(Node\DeclareItem::class) ? Node\DeclareItem::class : Node\Stmt\DeclareDeclare::class;
+
+    return new $class($key, $value, $attributes);
+}

tests/AutoMapperTest.php

@@ -785,6 +785,15 @@ public function testNoAutoRegister(): void
         $automapper->getMapper(Fixtures\User::class, Fixtures\UserDTO::class);
     }
 
+    public function testStrictTypes(): void
+    {
+        $this->expectException(\TypeError::class);
+
+        $automapper = AutoMapper::create(new Configuration(strictTypes: true, classPrefix: 'StrictTypes_'));
+        $data = ['foo' => 1.1];
+        $automapper->map($data, Fixtures\IntDTO::class);
+    }
+
     public function testWithMixedArray(): void
     {
         $user = new Fixtures\User(1, 'yolo', '13');

tests/Fixtures/IntDTO.php

@@ -0,0 +1,13 @@
+<?php
+
+declare(strict_types=1);
+
+namespace AutoMapper\Tests\Fixtures;
+
+readonly class IntDTO
+{
+    public function __construct(
+        public int $foo,
+    ) {
+    }
+}