implemented uml configuration

Author: reecube

README.md

@@ -151,7 +151,7 @@ uay_entities:
 - [x] Generate app repositories
 - [x] Generate app enums
 - [x] Configure generated names
-- [ ] Better UML configuration
+- [x] Better UML configuration
 - [ ] Recursive namespace configurations
 - [ ] Config validation
 - [ ] Generate base repositories

src/DataModelGenerator/DataModelGenerator.php

@@ -263,7 +263,7 @@ public function read(): void
 
     protected function generatePlantTextUml(): void
     {
-        $plantTextGenerator = new PlantTextGenerator($this->entities);
+        $plantTextGenerator = new PlantTextGenerator($this->inputModel, $this->entities);
 
         $plantTextGenerator->write($this->pathEntities . DIRECTORY_SEPARATOR . static::FILE_PLANTTEXT);
 

src/DataModelGenerator/InputModel.php

@@ -13,11 +13,12 @@ class InputModel
 
     protected const CONFIG_KEY_NAMESPACE = 'namespace';
     protected const CONFIG_KEY_CLASS_POSTFIX = 'classPostfix';
+    protected const CONFIG_KEY_UML = 'uml';
 
     /**
      * @var array
      */
-    protected $configuration = [
+    protected static $defaultConfiguration = [
         self::CONFIG_KEY_NAMESPACE => [
             'app' => 'App',
             'base' => 'Generated',
@@ -29,8 +30,16 @@ class InputModel
             'entity' => 'Generated',
             'repository' => 'Repository',
         ],
+        self::CONFIG_KEY_UML => [
+            'valid' => false,
+        ],
     ];
 
+    /**
+     * @var array
+     */
+    protected $configuration;
+
     /**
      * @var array
      */
@@ -48,7 +57,7 @@ public function __construct(array $config)
             $this->{$property} = ArrayUtil::getValueByPath($config, $requiredPath);
         }
 
-        $this->configuration = ArrayUtil::fillRecursive($this->configuration, $config);
+        $this->configuration = ArrayUtil::fillRecursive(static::$defaultConfiguration, $config);
     }
 
     protected function getConfigStringOrThrow(string $group, string $name): string
@@ -78,6 +87,15 @@ public function getClassPostfix(string $name): string
         return $this->getConfigStringOrThrow(static::CONFIG_KEY_CLASS_POSTFIX, $name);
     }
 
+    /**
+     * @param string $name
+     * @return string
+     */
+    public function getUML(string $name): string
+    {
+        return $this->getConfigStringOrThrow(static::CONFIG_KEY_UML, $name);
+    }
+
     public function getEntitiesData(): array
     {
         return $this->entitiesData;

src/DataModelGenerator/PlantTextGenerator.php

@@ -7,6 +7,11 @@
 
 class PlantTextGenerator
 {
+    /**
+     * @var InputModel
+     */
+    protected $inputModel;
+
     /**
      * @var string
      */
@@ -16,13 +21,15 @@ protected function parseEntity(Entity $entity): array
     {
         $result = [];
 
+        $shouldProduceValidUML = $this->inputModel->getUML('valid');
+
         $type = $entity->getType() === Entity::TYPE_ENUM ? 'enum' : 'class';
 
         $result[] = "{$type} {$entity->getName()} {";
 
         $plantTypeField = 'field';
         $plantTypeMethod = 'method';
-        $plantTypeExternal = $plantTypeMethod;
+        $plantTypeExternal = $shouldProduceValidUML ? $plantTypeField : $plantTypeMethod;
 
         if ($entity->getType() === Entity::TYPE_ENTITY) {
             $result[] = "  {{$plantTypeField}}+integer id";
@@ -116,10 +123,13 @@ protected function generatePlantRelation(array $members): string
     }
 
     /**
+     * @param InputModel $inputModel
      * @param Entity[] $entities
      */
-    public function __construct(array $entities)
+    public function __construct(InputModel $inputModel, array $entities)
     {
+        $this->inputModel = $inputModel;
+
         $plantText = [];
 
         $plantText[] = '@startuml';