Dart factory + equitable generation (Close #24) (#40)

* Dart factory + equitable generation (Close #24)

Author: kubk

src/Dto/DtoType.php

@@ -14,17 +14,6 @@ public function __construct(
     ) {
     }
 
-    public function isNumericEnum(): bool
-    {
-        foreach ($this->properties as $property) {
-            if (!($property instanceof DtoEnumProperty && $property->isNumeric())) {
-                return false;
-            }
-        }
-
-        return true;
-    }
-
     public function getName(): string
     {
         return $this->name;

src/Language/Dart/DartClassFactoryGenerator.php

@@ -0,0 +1,77 @@
+<?php
+
+namespace Riverwaysoft\DtoConverter\Language\Dart;
+
+use Riverwaysoft\DtoConverter\Dto\DtoEnumProperty;
+use Riverwaysoft\DtoConverter\Dto\DtoList;
+use Riverwaysoft\DtoConverter\Dto\DtoType;
+use Riverwaysoft\DtoConverter\Dto\PhpType\PhpListType;
+use Riverwaysoft\DtoConverter\Dto\PhpType\PhpTypeInterface;
+use Riverwaysoft\DtoConverter\Dto\PhpType\PhpUnionType;
+use Riverwaysoft\DtoConverter\Dto\PhpType\PhpUnknownType;
+use Webmozart\Assert\Assert;
+
+class DartClassFactoryGenerator
+{
+    public function __construct(private string|null $includePattern = null)
+    {
+
+    }
+
+    public function generateClassFactory(DtoType $dto, DtoList $dtoList): string
+    {
+        if (empty($dtoList->getList())) {
+            return '';
+        }
+
+        if ($this->includePattern && !preg_match(pattern: $this->includePattern, subject: $dto->getName())) {
+           return '';
+        }
+
+        $factoryProperties = '';
+
+        foreach ($dto->getProperties() as $property) {
+            Assert::false($property instanceof DtoEnumProperty, "Dart factories only work in a class context, not enum");
+            $propertyValue = $this->resolveFactoryProperty($property->getName(), $property->getType(), $dtoList);
+            $factoryProperties .= sprintf("      %s: %s,\n", $property->getName(), $propertyValue);
+        }
+
+        return sprintf("\n  factory %s.fromJson(Map<String, dynamic> json) {
+    return %s(\n%s    );
+  }\n",
+            $dto->getName(),
+            $dto->getName(),
+            $factoryProperties,
+        );
+    }
+
+    private function resolveFactoryProperty(string $propertyName, PhpTypeInterface $type, DtoList $dtoList): string
+    {
+        if ($type instanceof PhpUnionType && $type->isNullable()) {
+            return sprintf("json['{$propertyName}'] != null ? %s : null", $this->resolveFactoryProperty($propertyName, $type->getFirstNotNullType(), $dtoList));
+        }
+
+        if ($type instanceof PhpListType) {
+            $collectionType = $type->getType();
+            if (!($collectionType instanceof PhpUnknownType)) {
+                throw new \Exception('Only class instance can be converted to collection');
+            }
+            $class = $collectionType->getName();
+            return sprintf("List<%s>.from(json['%s'].map((e) => %s.fromJson(e)))", $class, $propertyName, $class);
+        }
+
+        if ($type instanceof PhpUnknownType) {
+            if ($type->getName() === 'DateTime' || $type->getName() === 'DateTimeImmutable') {
+                return sprintf("DateTime.parse(json['%s'])", $propertyName);
+            }
+
+            if ($dtoList->getDtoByType($type->getName())?->getExpressionType()->isAnyEnum()) {
+                return sprintf("%s.values[json['%s']]", $type->getName(), $propertyName);
+            }
+
+            return sprintf("%s.fromJson(json['%s'])", $type->getName(), $propertyName);
+        }
+
+        return sprintf("json['%s']", $propertyName);
+    }
+}

src/Language/Dart/DartEquitableGenerator.php

@@ -0,0 +1,42 @@
+<?php
+
+namespace Riverwaysoft\DtoConverter\Language\Dart;
+
+use Riverwaysoft\DtoConverter\Dto\DtoClassProperty;
+use Riverwaysoft\DtoConverter\Dto\DtoType;
+
+class DartEquitableGenerator
+{
+    public function __construct(private string|null $includePattern = null)
+    {
+
+    }
+
+    public function generateEquitableHeader(DtoType $dto): string
+    {
+        if ($this->doesntMatch($dto)) {
+            return '';
+        }
+
+        return ' extends Equatable';
+    }
+
+    private function doesntMatch(DtoType $dto): bool
+    {
+        return $this->includePattern && !preg_match(pattern: $this->includePattern, subject: $dto->getName());
+    }
+
+    public function generateEquitableId(DtoType $dto): string
+    {
+        if ($this->doesntMatch($dto)) {
+            return '';
+        }
+
+        $properties = $dto->getProperties();
+        $propertiesNames = array_map(fn(DtoClassProperty $property) => $property->getName(), $properties);
+        $propertiesString = implode(separator: ', ', array: $propertiesNames);
+
+        return sprintf("\n  @override
+  List<dynamic> get props => [%s];\n", $propertiesString);
+    }
+}

src/Language/Dart/DartGenerator.php

@@ -27,7 +27,9 @@ public function __construct(
         private OutputWriterInterface $outputWriter,
         /** @var UnknownTypeResolverInterface[] */
         private array $unknownTypeResolvers = [],
-        private ?OutputFilesProcessor $outputFilesProcessor = null
+        private ?OutputFilesProcessor $outputFilesProcessor = null,
+        private ?DartClassFactoryGenerator $classFactoryGenerator = null,
+        private ?DartEquitableGenerator $equitableGenerator = null,
     ) {
         $this->outputFilesProcessor = $this->outputFilesProcessor ?? new OutputFilesProcessor();
     }
@@ -44,15 +46,20 @@ public function generate(DtoList $dtoList): array
         return $this->outputFilesProcessor->process($this->outputWriter->getTypes());
     }
 
+
+
     private function convertToDartType(DtoType $dto, DtoList $dtoList): string
     {
         if ($dto->getExpressionType()->equals(ExpressionType::class())) {
             return sprintf(
-                "class %s {%s\n\n  %s({%s\n  }) {}\n}",
+                "class %s%s {%s\n\n  %s({%s\n  }) {}\n%s%s}",
                 $dto->getName(),
+                $this->equitableGenerator ? $this->equitableGenerator->generateEquitableHeader($dto) : '',
                 $this->convertToDartProperties($dto, $dtoList),
                 $dto->getName(),
                 $this->generateConstructor($dto->getProperties()),
+                $this->classFactoryGenerator ? $this->classFactoryGenerator->generateClassFactory($dto, $dtoList) : '',
+                $this->equitableGenerator ? $this->equitableGenerator->generateEquitableId($dto) : '',
             );
         }
 

src/Language/TypeScript/TypeScriptGenerator.php

@@ -30,8 +30,8 @@ public function __construct(
         private OutputWriterInterface $outputWriter,
         /** @var UnknownTypeResolverInterface[] $unknownTypeResolvers */
         private array $unknownTypeResolvers = [],
+        private ?OutputFilesProcessor $outputFilesProcessor = null,
         ?TypeScriptGeneratorOptions $options = null,
-        private ?OutputFilesProcessor $outputFilesProcessor = null
     ) {
         $this->options = $options ?? new TypeScriptGeneratorOptions(useTypesInsteadOfEnums: false);
         $this->outputFilesProcessor = $this->outputFilesProcessor ?? new OutputFilesProcessor();