added code generator back to repository

2026-08-19 04:23:59 +00:00 · 2020-09-14 16:36:40 +08:00
parent 0cab8e6456
commit 06b024ce9f
84 changed files with 5955 additions and 1 deletions
@@ -0,0 +1,92 @@
+<?php
+
+namespace App\Classes\CodeGenerator\Generators;
+
+use Illuminate\Support\Facades\File;
+use App\Classes\CodeGenerator\Common\CommandData;
+use App\Classes\CodeGenerator\Utils\FileUtil;
+use Illuminate\Support\Str;
+
+class MigrationGenerator extends BaseGenerator
+{
+    /** @var CommandData */
+    private $commandData;
+
+    /** @var string */
+    private $path;
+
+    public function __construct($commandData)
+    {
+        parent::__construct();
+
+        $this->commandData = $commandData;
+        $this->path = database_path('migrations/');
+    }
+
+    public function generate()
+    {
+        $templateData = $this->generatorHelpers->get_template('Migration.migration');
+
+        $templateData = $this->generatorHelpers->fill_template($this->commandData->dynamicVars, $templateData);
+
+        $templateData = str_replace('$FIELDS$', $this->generateFields(), $templateData);
+
+        $tableName = $this->commandData->dynamicVars['$TABLE_NAME$'];
+
+        $fileName = date('Y_m_d_His').'_'.'create_'.Str::snake(Str::plural($tableName)).'_table.php';
+
+        FileUtil::createFile($this->path, $fileName, $templateData);
+
+        $this->commandData->commandComment("\nMigration created: ");
+        $this->commandData->commandInfo($fileName);
+    }
+
+    private function generateFields()
+    {
+        $fields = [];
+        $foreignKeys = [];
+        $createdAtField = null;
+        $updatedAtField = null;
+
+        $fields[] = '$table->id();';
+
+        foreach ($this->commandData->fields as $field) {
+            $fields[] = $field->migrationText;
+            if (!empty($field->foreignKeyText)) {
+                $foreignKeys[] = $field->foreignKeyText;
+            }
+        }
+
+        $fields[] = '$table->timestamps();';
+
+        if ($this->commandData->getOption('softDelete')) {
+            $fields[] = '$table->softDeletes();';
+        }
+
+        return implode($this->generatorHelpers->generator_nl_tab(1, 3), array_merge($fields, $foreignKeys));
+    }
+
+    public function rollback()
+    {
+        $fileName = 'create_'.$this->commandData->config->tableName.'_table.php';
+
+        $allFiles = File::allFiles($this->path);
+
+        $files = [];
+
+        foreach ($allFiles as $file) {
+            $files[] = $file->getFilename();
+        }
+
+        $files = array_reverse($files);
+
+        foreach ($files as $file) {
+            if (Str::contains($file, $fileName)) {
+                if ($this->rollbackFile($this->path, $file)) {
+                    $this->commandData->commandComment('Migration file deleted: '.$file);
+                }
+                break;
+            }
+        }
+    }
+}