Add properties

yiisoft · Aug 31, 2023 · b08de00 · b08de00
1 parent 9b0b4e4
commit b08de00
Show file tree

Hide file tree

Showing 2 changed files with 45 additions and 6 deletions.
diff --git a/src/Generator/ActiveRecord/Generator.php b/src/Generator/ActiveRecord/Generator.php
@@ -5,15 +5,28 @@
 namespace Yiisoft\Yii\Gii\Generator\ActiveRecord;
 
 use InvalidArgumentException;
+use Yiisoft\Aliases\Aliases;
+use Yiisoft\Db\Connection\ConnectionInterface;
+use Yiisoft\Validator\ValidatorInterface;
 use Yiisoft\Yii\Gii\Component\CodeFile\CodeFile;
 use Yiisoft\Yii\Gii\Generator\AbstractGenerator;
 use Yiisoft\Yii\Gii\GeneratorCommandInterface;
+use Yiisoft\Yii\Gii\ParametersProvider;
 
 /**
  * This generator will generate a controller and one or a few action view files.
  */
 final class Generator extends AbstractGenerator
 {
+    public function __construct(
+        Aliases $aliases,
+        ValidatorInterface $validator,
+        ParametersProvider $parametersProvider,
+        private ConnectionInterface $connection,
+    ) {
+        parent::__construct($aliases, $validator, $parametersProvider);
+    }
+
     public static function getId(): string
     {
         return 'active-record';
@@ -46,10 +59,24 @@ public function doGenerate(GeneratorCommandInterface $command): array
 
         $rootPath = $this->aliases->get('@root');
 
+        $properties = [];
+        if ($schema = $this->connection->getTableSchema($command->getTableName(), true)){
+            foreach ($schema->getColumns() as $columnSchema) {
+                $properties[] = [
+                    'name' => $columnSchema->getName(),
+                    'type' => match ($columnSchema->getPhpType()) {
+                        'integer' => 'int',
+                        default => 'string',
+                    },
+                    'isAllowNull' => $columnSchema->isAllowNull(),
+                    'defaultValue' => $columnSchema->getDefaultValue(),
+                ];
+            }
+        }
         $path = $this->getControllerFile($command);
         $codeFile = (new CodeFile(
             $path,
-            $this->render($command, 'model.php')
+            $this->render($command, 'model.php', ['properties' => $properties])
         ))->withBasePath($rootPath);
         $files[$codeFile->getId()] = $codeFile;
 

diff --git a/src/Generator/ActiveRecord/default/model.php b/src/Generator/ActiveRecord/default/model.php
@@ -1,13 +1,14 @@
 <?php
-declare(strict_types=1);
-/**
- * This is the template for generating a controller class file.
- */
 
-/* @var $command Yiisoft\Yii\Gii\Generator\ActiveRecord\Command */
+declare(strict_types=1);
 
 use Yiisoft\Strings\StringHelper;
 
+/**
+ * @var Yiisoft\Yii\Gii\Generator\ActiveRecord\Command $command
+ * @var array<string, string> $properties
+ */
+
 echo "<?php\n";
 ?>
 
@@ -19,6 +20,17 @@
 
 final class <?= $command->getModelName(); ?> extends <?= StringHelper::baseName($command->getBaseClass()) . PHP_EOL ?>
 {
+<?php foreach ($properties as $property): ?>
+    private <?=sprintf(
+        '%s%s $%s',
+        $property['isAllowNull'] ? '?' : '',
+        $property['type'],
+        $property['name'],
+    )?>;
+<?php endforeach; ?>
+<?php if (!empty($properties)) {
+    echo PHP_EOL;
+} ?>
     public function tableName(): string
     {
         return '<?= $command->getTableName() ?>';