generate setter for simple arrays

Author: PawelSuwinski

phpstan.neon

@@ -61,6 +61,7 @@ parameters:
                  author: false|string,
                  fieldVisibility: 'private'|'protected'|'public'|null,
                  accessorMethods: boolean,
+                 useSimpleArraySetter: boolean,
                  fluentMutatorMethods: boolean,
                  rangeMapping: array<string, string>,
                  allTypes: boolean,

src/Model/Class_.php

@@ -218,6 +218,7 @@ public function toNetteFile(array $config, InflectorInterface $inflector, ?PhpFi
     {
         $useDoctrineCollections = $config['doctrine']['useCollection'];
         $useAccessors = $config['accessorMethods'];
+        $useSimpleArraySetter = $config['useSimpleArraySetter'];
         $useFluentMutators = $config['fluentMutatorMethods'];
         $fileHeader = $config['header'] ?? null;
         $fieldVisibility = $config['fieldVisibility'];
@@ -333,7 +334,7 @@ public function toNetteFile(array $config, InflectorInterface $inflector, ?PhpFi
             foreach ($sortedProperties as $property) {
                 foreach ($property->generateNetteMethods(static function ($string) use ($inflector) {
                     return $inflector->singularize($string)[0];
-                }, $namespace, $useDoctrineCollections, $useFluentMutators) as $method) {
+                }, $namespace, $useDoctrineCollections, $useFluentMutators, $useSimpleArraySetter) as $method) {
                     $methods[] = $method;
                 }
             }

src/Model/Property.php

@@ -81,6 +81,11 @@ public function isArray(): bool
         return $this->type instanceof ArrayType;
     }
 
+    public function isSimpleArray(): bool
+    {
+        return 'array' === $this->typeHint;
+    }
+
     public function addAnnotation(string $annotation): self
     {
         if ('' === $annotation || !\in_array($annotation, $this->annotations, true)) {
@@ -154,7 +159,7 @@ public function toNetteProperty(PhpNamespace $namespace, ?string $visibility = n
             $property->setType($this->resolveName($namespace, $this->typeHint));
         }
 
-        if (!$this->isArray() || $this->isTypeHintedAsCollection()) {
+        if (!$this->isArray() || $this->isSimpleArray() || $this->isTypeHintedAsCollection()) {
             $property->setNullable($this->isNullable);
         }
 
@@ -195,9 +200,10 @@ public function generateNetteMethods(
         PhpNamespace $namespace,
         bool $useDoctrineCollections = true,
         bool $useFluentMutators = false,
+        bool $useSimpleArraySetter = false,
     ): array {
         return array_merge(
-            $this->generateMutators($singularize, $namespace, $useDoctrineCollections, $useFluentMutators),
+            $this->generateMutators($singularize, $namespace, $useDoctrineCollections, $useFluentMutators, $useSimpleArraySetter),
             $this->isReadable ? [$this->generateGetter($namespace)] : []
         );
     }
@@ -214,7 +220,7 @@ private function generateGetter(PhpNamespace $namespace): Method
         }
         if ($this->typeHint) {
             $getter->setReturnType($this->resolveName($namespace, $this->typeHint));
-            if ($this->isNullable && !$this->isArray()) {
+            if ($this->isNullable && (!$this->isArray() || $this->isSimpleArray())) {
                 $getter->setReturnNullable();
             }
         }
@@ -231,13 +237,14 @@ private function generateMutators(
         PhpNamespace $namespace,
         bool $useDoctrineCollections = true,
         bool $useFluentMutators = false,
+        bool $useSimpleArraySetter = false,
     ): array {
         if (!$this->isWritable) {
             return [];
         }
 
         $mutators = [];
-        if ($this->isArray()) {
+        if ($this->isArray() && (!$this->isSimpleArray() || !$useSimpleArraySetter)) {
             $singularProperty = $singularize($this->name());
 
             $adder = (new Method('add'.ucfirst($singularProperty)))->setVisibility(ClassType::VISIBILITY_PUBLIC);

src/SchemaGeneratorConfiguration.php

@@ -167,6 +167,7 @@ public function getConfigTreeBuilder(): TreeBuilder
                 ->scalarNode('author')->defaultFalse()->info('The value of the phpDoc\'s @author annotation')->example('Kévin Dunglas <[email protected]>')->end()
                 ->enumNode('fieldVisibility')->values(['private', 'protected', 'public'])->defaultValue('private')->cannotBeEmpty()->info('Visibility of entities fields')->end()
                 ->booleanNode('accessorMethods')->defaultTrue()->info('Set this flag to false to not generate getter, setter, adder and remover methods')->end()
+                ->booleanNode('useSimpleArraySetter')->defaultFalse()->info('Set this flag to true to generate setter method for simple arrays instead of adder and remover methods')->end()
                 ->booleanNode('fluentMutatorMethods')->defaultFalse()->info('Set this flag to true to generate fluent setter, adder and remover methods')->end()
                 ->arrayNode('rangeMapping')
                     ->useAttributeAsKey('name')

tests/Command/DumpConfigurationTest.php

@@ -154,6 +154,9 @@ interface:            App\Model # Example: App\Model
     # Set this flag to false to not generate getter, setter, adder and remover methods
     accessorMethods:      true
 
+    # Set this flag to true to generate setter method for simple arrays instead of adder and remover methods
+    useSimpleArraySetter: false
+
     # Set this flag to true to generate fluent setter, adder and remover methods
     fluentMutatorMethods: false
     rangeMapping:

tests/Command/GenerateCommandTest.php

@@ -538,6 +538,43 @@ class Page extends Object_
         self::assertFalse($this->fs->exists("$outputDir/App/Entity/Travel.php"));
     }
 
+    public function testGeneratedSimpleArray(): void
+    {
+        $outputDir = __DIR__.'/../../build/simple-array';
+        $config = __DIR__.'/../config/simple-array.yaml';
+
+        $this->fs->mkdir($outputDir);
+
+        $commandTester = new CommandTester(new GenerateCommand());
+        $this->assertEquals(0, $commandTester->execute(['output' => $outputDir, 'config' => $config]));
+        $source = file_get_contents("$outputDir/App/Entity/Project.php");
+
+        $this->assertStringContainsString(<<<'PHP'
+    /**
+     * @see _:shareWith
+     */
+    #[ORM\Column(type: 'simple_array', nullable: true)]
+    #[Assert\Unique]
+    private ?array $shareWith = [];
+PHP
+            , $source);
+
+        $this->assertStringContainsString(<<<'PHP'
+    public function setShareWith(?array $shareWith): void
+    {
+        $this->shareWith = $shareWith;
+    }
+
+    public function getShareWith(): ?array
+    {
+        return $this->shareWith;
+    }
+PHP
+            , $source);
+        $this->assertStringNotContainsString('function addShareWith', $source);
+        $this->assertStringNotContainsString('function removeShareWith', $source);
+    }
+
     public function testGenerationWithoutConfigFileQuestion(): void
     {
         // No config file is given.

tests/TypesGeneratorTest.php

@@ -105,7 +105,7 @@ public function testGenerate(): void
         $article = file_get_contents("$this->outputDir/App/Entity/Article.php");
         $this->assertStringContainsString('abstract class Article extends CreativeWork', $article);
         $this->assertStringContainsString('private ?string $articleBody = null;', $article);
-        $this->assertStringContainsString('private array $articleSection = [];', $article);
+        $this->assertStringContainsString('private ?array $articleSection = [];', $article);
         $this->assertStringContainsString('public function setArticleBody(?string $articleBody): void', $article);
         $this->assertStringContainsString('public function getArticleBody(): ?string', $article);
         $this->assertStringContainsString('public function addArticleSection(string $articleSection): void', $article);

tests/config/simple-array.yaml

@@ -0,0 +1,12 @@
+useSimpleArraySetter: true
+types:
+  Project:
+    properties:
+      name:
+        range: "https://schema.org/Text"
+      shareWith:
+        range: "https://schema.org/email"
+        cardinality: "(0..*)"
+        attributes:
+          Assert\Unique: ~
+          ORM\Column: { type: "simple_array" }