basic support for generating validation rules

cebe · Jun 26, 2019 · 665b773 · 665b773
1 parent 9bff78d
commit 665b773
Show file tree

Hide file tree

Showing 9 changed files with 98 additions and 1 deletion.
diff --git a/src/generator/ApiGenerator.php b/src/generator/ApiGenerator.php
@@ -625,15 +625,23 @@ protected function generateModels()
                     $type = $type[0];
                 }
 
-                $attributes[] = [
+                $attributes[$name] = [
                     'name' => $name,
                     'type' => $type,
                     'dbType' => $dbType,
                     'dbName' => $dbName,
+                    'required' => false,
+                    'readOnly' => $resolvedProperty->readOnly ?? false,
                     'description' => $resolvedProperty->description,
                     'faker' => $this->guessModelFaker($name, $type, $resolvedProperty),
                 ];
             }
+            foreach ($schema->required as $property) {
+                if (!isset($attributes[$property])) {
+                    continue;
+                }
+                $attributes[$property]['required'] = true;
+            }
 
             $models[$schemaName] = [
                 'name' => $schemaName,

diff --git a/src/generator/default/model.php b/src/generator/default/model.php
@@ -19,6 +19,56 @@ public static function tableName()
         return <?= var_export($tableName) ?>;
     }
 
+    public function rules()
+    {
+        return [
+<?php
+    $safeAttributes = [];
+    $requiredAttributes = [];
+    $integerAttributes = [];
+    $stringAttributes = [];
+
+    foreach ($attributes as $attribute) {
+        if ($attribute['readOnly']) {
+            continue;
+        }
+        if ($attribute['required']) {
+            $requiredAttributes[$attribute['name']] = $attribute['name'];
+        }
+        switch ($attribute['type']) {
+            case 'integer':
+                $integerAttributes[$attribute['name']] = $attribute['name'];
+                break;
+            case 'string':
+                $stringAttributes[$attribute['name']] = $attribute['name'];
+                break;
+            default:
+            case 'array':
+                $safeAttributes[$attribute['name']] = $attribute['name'];
+                break;
+        }
+    }
+    if (!empty($stringAttributes)) {
+        echo "            [['" . implode("', '", $stringAttributes) . "'], 'trim'],\n";
+    }
+    if (!empty($requiredAttributes)) {
+        echo "            [['" . implode("', '", $requiredAttributes) . "'], 'required'],\n";
+    }
+    if (!empty($stringAttributes)) {
+        echo "            [['" . implode("', '", $stringAttributes) . "'], 'string'],\n";
+    }
+    if (!empty($integerAttributes)) {
+        echo "            [['" . implode("', '", $integerAttributes) . "'], 'integer'],\n";
+    }
+    if (!empty($safeAttributes)) {
+        echo "            // TODO define more concreate validation rules!\n";
+        echo "            [['" . implode("','", $safeAttributes) . "'], 'safe'],\n";
+    }
+
+?>
+        ];
+    }
+
 <?php foreach ($relations as $relationName => $relation): ?>
     public function get<?= ucfirst($relationName) ?>()
     {

diff --git a/tests/specs/petstore.yaml b/tests/specs/petstore.yaml
@@ -90,6 +90,7 @@ components:
         id:
           type: integer
           format: int64
+          readOnly: True
         name:
           type: string
         tag:

diff --git a/tests/specs/petstore/models/Pet.php b/tests/specs/petstore/models/Pet.php
@@ -16,4 +16,13 @@ public static function tableName()
         return '{{%pets}}';
     }
 
+    public function rules()
+    {
+        return [
+            [['name', 'tag'], 'trim'],
+            [['name'], 'required'],
+            [['name', 'tag'], 'string'],
+        ];
+    }
+
 }
diff --git a/tests/specs/petstore_arrayref.yaml b/tests/specs/petstore_arrayref.yaml
@@ -86,6 +86,7 @@ components:
         id:
           type: integer
           format: int64
+          readOnly: True
         name:
           type: string
         tag:

diff --git a/tests/specs/petstore_arrayref/models/Pet.php b/tests/specs/petstore_arrayref/models/Pet.php
@@ -16,4 +16,13 @@ public static function tableName()
         return '{{%pets}}';
     }
 
+    public function rules()
+    {
+        return [
+            [['name', 'tag'], 'trim'],
+            [['name'], 'required'],
+            [['name', 'tag'], 'string'],
+        ];
+    }
+
 }
diff --git a/tests/specs/petstore_namespace/mymodels/Pet.php b/tests/specs/petstore_namespace/mymodels/Pet.php
@@ -16,4 +16,13 @@ public static function tableName()
         return '{{%pets}}';
     }
 
+    public function rules()
+    {
+        return [
+            [['name', 'tag'], 'trim'],
+            [['name'], 'required'],
+            [['name', 'tag'], 'string'],
+        ];
+    }
+
 }
diff --git a/tests/specs/petstore_wrapped.yaml b/tests/specs/petstore_wrapped.yaml
@@ -98,6 +98,7 @@ components:
         id:
           type: integer
           format: int64
+          readOnly: True
         name:
           type: string
         tag:

diff --git a/tests/specs/petstore_wrapped/models/Pet.php b/tests/specs/petstore_wrapped/models/Pet.php
@@ -16,4 +16,13 @@ public static function tableName()
         return '{{%pets}}';
     }
 
+    public function rules()
+    {
+        return [
+            [['name', 'tag'], 'trim'],
+            [['name'], 'required'],
+            [['name', 'tag'], 'string'],
+        ];
+    }
+
 }