From e1d0e06f330abb6e373cf010177112b1c2f9abf4 Mon Sep 17 00:00:00 2001
From: Yurun <admin@yurunsoft.com>
Date: Thu, 21 Mar 2024 08:25:47 +0800
Subject: [PATCH] =?UTF-8?q?=E5=A2=9E=E5=8A=A0=E5=B9=B6=E5=AE=8C=E5=96=84?=
 =?UTF-8?q?=E6=A8=A1=E5=9E=8B=E7=94=9F=E6=88=90=E8=A1=A8=E4=BA=8B=E4=BB=B6?=
 =?UTF-8?q?=20(#684)?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 doc/dev/generate/table.md                     | 33 +++++++++++++++----
 .../Cli/Table/Event/GenerateModelEvents.php   | 22 -------------
 .../Table/Event/Param/AfterGenerateTable.php  | 12 +++++++
 .../Table/Event/Param/AfterGenerateTables.php | 11 +++++++
 .../Table/Event/Param/BeforeGenerateTable.php | 12 +++++++
 .../Event/Param/BeforeGenerateTables.php      | 11 +++++++
 .../Cli/Table/Event/Param/TGenerateTable.php  | 32 ++++++++++++++++++
 .../model/src/Cli/Table/TableGenerate.php     | 17 +++++++---
 8 files changed, 118 insertions(+), 32 deletions(-)
 delete mode 100644 src/Components/model/src/Cli/Table/Event/GenerateModelEvents.php
 create mode 100644 src/Components/model/src/Cli/Table/Event/Param/AfterGenerateTable.php
 create mode 100644 src/Components/model/src/Cli/Table/Event/Param/AfterGenerateTables.php
 create mode 100644 src/Components/model/src/Cli/Table/Event/Param/BeforeGenerateTable.php
 create mode 100644 src/Components/model/src/Cli/Table/Event/Param/BeforeGenerateTables.php
 create mode 100644 src/Components/model/src/Cli/Table/Event/Param/TGenerateTable.php

diff --git a/doc/dev/generate/table.md b/doc/dev/generate/table.md
index 14869e7aa..29774f9a4 100644
--- a/doc/dev/generate/table.md
+++ b/doc/dev/generate/table.md
@@ -28,14 +28,35 @@ vendor/bin/imi-xxx generate/table
 
 ## 事件监听
 
-### imi.generate_model.before
+在生成表时，会触发以下事件：
 
-生成表工具的前置操作
+- `Imi\Model\Cli\Table\Event\Param\BeforeGenerateTables` - 生成所有表前置事件
+- `Imi\Model\Cli\Table\Event\Param\AfterGenerateTables` - 生成所有表前置事件
+- `Imi\Model\Cli\Table\Event\Param\BeforeGenerateTable` - 生成表前置事件
+- `Imi\Model\Cli\Table\Event\Param\AfterGenerateTable` - 生成表前置事件
 
-**常量：** `Imi\Model\Cli\Table\Event\GenerateModelEvents::BEFORE_GENERATE_MODEL`
+> 事件参数类都是事件名本身
 
-### imi.generate_model.after
+`BeforeGenerateTable`、`AfterGenerateTable` 事件参数：
 
-生成表工具的后置操作
+```php
+/**
+ * 模型类名.
+ */
+public string $className,
 
-**常量：** `Imi\Model\Cli\Table\Event\GenerateModelEvents::AFTER_GENERATE_MODEL`
+/**
+ * 表名.
+ */
+public string $tableName,
+
+/**
+ * 是否跳过.
+ */
+public bool $skip,
+
+/**
+ * DDL 语句.
+ */
+public string $ddl,
+```
diff --git a/src/Components/model/src/Cli/Table/Event/GenerateModelEvents.php b/src/Components/model/src/Cli/Table/Event/GenerateModelEvents.php
deleted file mode 100644
index a1f61c345..000000000
--- a/src/Components/model/src/Cli/Table/Event/GenerateModelEvents.php
+++ /dev/null
@@ -1,22 +0,0 @@
-<?php
-
-declare(strict_types=1);
-
-namespace Imi\Model\Cli\Table\Event;
-
-use Imi\Util\Traits\TStaticClass;
-
-final class GenerateModelEvents
-{
-    use TStaticClass;
-
-    /**
-     * 生成模型前置事件.
-     */
-    public const BEFORE_GENERATE_MODEL = 'imi.generate_model.before';
-
-    /**
-     * 生成模型后置事件.
-     */
-    public const AFTER_GENERATE_MODEL = 'imi.generate_model.after';
-}
diff --git a/src/Components/model/src/Cli/Table/Event/Param/AfterGenerateTable.php b/src/Components/model/src/Cli/Table/Event/Param/AfterGenerateTable.php
new file mode 100644
index 000000000..0be07eebe
--- /dev/null
+++ b/src/Components/model/src/Cli/Table/Event/Param/AfterGenerateTable.php
@@ -0,0 +1,12 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Imi\Model\Cli\Table\Event\Param;
+
+use Imi\Event\CommonEvent;
+
+class AfterGenerateTable extends CommonEvent
+{
+    use TGenerateTable;
+}
diff --git a/src/Components/model/src/Cli/Table/Event/Param/AfterGenerateTables.php b/src/Components/model/src/Cli/Table/Event/Param/AfterGenerateTables.php
new file mode 100644
index 000000000..a81daf8a9
--- /dev/null
+++ b/src/Components/model/src/Cli/Table/Event/Param/AfterGenerateTables.php
@@ -0,0 +1,11 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Imi\Model\Cli\Table\Event\Param;
+
+use Imi\Event\CommonEvent;
+
+class AfterGenerateTables extends CommonEvent
+{
+}
diff --git a/src/Components/model/src/Cli/Table/Event/Param/BeforeGenerateTable.php b/src/Components/model/src/Cli/Table/Event/Param/BeforeGenerateTable.php
new file mode 100644
index 000000000..1532e07d7
--- /dev/null
+++ b/src/Components/model/src/Cli/Table/Event/Param/BeforeGenerateTable.php
@@ -0,0 +1,12 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Imi\Model\Cli\Table\Event\Param;
+
+use Imi\Event\CommonEvent;
+
+class BeforeGenerateTable extends CommonEvent
+{
+    use TGenerateTable;
+}
diff --git a/src/Components/model/src/Cli/Table/Event/Param/BeforeGenerateTables.php b/src/Components/model/src/Cli/Table/Event/Param/BeforeGenerateTables.php
new file mode 100644
index 000000000..47a7ec126
--- /dev/null
+++ b/src/Components/model/src/Cli/Table/Event/Param/BeforeGenerateTables.php
@@ -0,0 +1,11 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Imi\Model\Cli\Table\Event\Param;
+
+use Imi\Event\CommonEvent;
+
+class BeforeGenerateTables extends CommonEvent
+{
+}
diff --git a/src/Components/model/src/Cli/Table/Event/Param/TGenerateTable.php b/src/Components/model/src/Cli/Table/Event/Param/TGenerateTable.php
new file mode 100644
index 000000000..036569bc8
--- /dev/null
+++ b/src/Components/model/src/Cli/Table/Event/Param/TGenerateTable.php
@@ -0,0 +1,32 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Imi\Model\Cli\Table\Event\Param;
+
+trait TGenerateTable
+{
+    public function __construct(
+        /**
+         * 模型类名.
+         */
+        public string $className,
+
+        /**
+         * 表名.
+         */
+        public string $tableName,
+
+        /**
+         * 是否跳过.
+         */
+        public bool $skip,
+
+        /**
+         * DDL 语句.
+         */
+        public string $ddl,
+    ) {
+        parent::__construct(static::class);
+    }
+}
diff --git a/src/Components/model/src/Cli/Table/TableGenerate.php b/src/Components/model/src/Cli/Table/TableGenerate.php
index db1dc83b8..3dd3482d0 100644
--- a/src/Components/model/src/Cli/Table/TableGenerate.php
+++ b/src/Components/model/src/Cli/Table/TableGenerate.php
@@ -12,7 +12,10 @@
 use Imi\Db\Db;
 use Imi\Event\Event;
 use Imi\Model\Annotation\DDL;
-use Imi\Model\Cli\Table\Event\GenerateModelEvents;
+use Imi\Model\Cli\Table\Event\Param\AfterGenerateTable;
+use Imi\Model\Cli\Table\Event\Param\AfterGenerateTables;
+use Imi\Model\Cli\Table\Event\Param\BeforeGenerateTable;
+use Imi\Model\Cli\Table\Event\Param\BeforeGenerateTables;
 use Imi\Util\ClassObject;
 
 #[Command(name: 'generate')]
@@ -31,7 +34,7 @@ class TableGenerate extends BaseCommand
     #[Option(name: 'override', type: \Imi\Cli\ArgType::STRING, default: false, comments: '是否覆盖已存在的表，请慎重！true-全覆盖;false-不覆盖;默认缺省状态为false')]
     public function generate(?string $namespace, ?string $database, ?string $poolName, array $include, array $exclude, string|bool $override): void
     {
-        Event::dispatch(eventName: GenerateModelEvents::BEFORE_GENERATE_MODEL);
+        Event::dispatch(eventName: BeforeGenerateTables::class);
         $override = (bool) json_decode((string) $override, false);
         // 数据库
         if (null === $database)
@@ -85,17 +88,23 @@ public function generate(?string $namespace, ?string $database, ?string $poolNam
                     $tables[] = $table;
                     // 表存在跳过
                     $this->output->writeln('Skip ' . $table);
+                    Event::dispatch(new BeforeGenerateTable($class, $table, true, ''));
+                    Event::dispatch(new AfterGenerateTable($class, $table, true, ''));
                     continue;
                 }
             }
             /** @var \Imi\Model\Annotation\DDL $ddlAnnotation */
             $ddlAnnotation = $point->getAnnotation();
+            $ddl = $ddlAnnotation->getRawSql() . ';';
+            $event = new BeforeGenerateTable($class, $table, false, $ddl);
+            Event::dispatch($event);
             // 创建表
-            Db::getInstance()->batchExec($ddlAnnotation->getRawSql() . ';');
+            Db::getInstance()->batchExec($event->ddl);
+            Event::dispatch(new AfterGenerateTable($class, $table, false, $ddl));
             $tables[] = $table;
             $this->output->writeln('Create <info>' . $table . '</info>');
         }
-        Event::dispatch(eventName: GenerateModelEvents::AFTER_GENERATE_MODEL);
+        Event::dispatch(eventName: AfterGenerateTables::class);
     }
 
     /**