feat(mysql): add initialization for teach/schedule

koishijs · Oct 15, 2020 · c11add2 · c11add2
1 parent 1078fcd
commit c11add2
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 2 deletions.
diff --git a/packages/plugin-schedule/src/database.ts b/packages/plugin-schedule/src/database.ts
@@ -24,7 +24,7 @@ export interface Schedule {
   session: Session
 }
 
-extendDatabase(MysqlDatabase, (Database) => {
+extendDatabase<typeof MysqlDatabase>('koishi-plugin-mysql', (Database) => {
   Object.assign(Database.tables.schedule = [
     'PRIMARY KEY (`id`) USING BTREE',
   ], {

diff --git a/packages/plugin-teach/src/database/mysql.ts b/packages/plugin-teach/src/database/mysql.ts
@@ -92,8 +92,26 @@ extendDatabase<typeof MysqlDatabase>('koishi-plugin-mysql', {
   },
 })
 
-extendDatabase<typeof MysqlDatabase>('koishi-plugin-mysql', ({ listFields }) => {
+extendDatabase<typeof MysqlDatabase>('koishi-plugin-mysql', ({ listFields, tables }) => {
   listFields.push('dialogue.groups', 'dialogue.predecessors')
+
+  Object.assign(tables.dialogue = [
+    'PRIMARY KEY (`id`) USING BTREE',
+  ], {
+    id: `INT(11) UNSIGNED NOT NULL AUTO_INCREMENT`,
+    flag: `INT(10) UNSIGNED NOT NULL DEFAULT '0'`,
+    probS: `DECIMAL(4,3) UNSIGNED NOT NULL DEFAULT '1.000'`,
+    probA: `DECIMAL(4,3) UNSIGNED NOT NULL DEFAULT '0.000'`,
+    startTime: `INT(10) NOT NULL DEFAULT '0'`,
+    endTime: `INT(10) NOT NULL DEFAULT '0'`,
+    groups: `TINYTEXT NULL DEFAULT NULL COLLATE 'utf8mb4_general_ci'`,
+    original: `TINYTEXT NULL DEFAULT NULL COLLATE 'utf8mb4_general_ci'`,
+    question: `TINYTEXT NULL DEFAULT NULL COLLATE 'utf8mb4_general_ci'`,
+    answer: `TEXT(65535) NULL DEFAULT NULL COLLATE 'utf8mb4_general_ci'`,
+    predecessors: `TINYTEXT NULL DEFAULT NULL COLLATE 'utf8mb4_general_ci'`,
+    successorTimeout: `INT(10) UNSIGNED NOT NULL DEFAULT '0'`,
+    writer: `BIGINT(20) UNSIGNED NOT NULL DEFAULT '0'`,
+  })
 })
 
 export default function apply(ctx: Context, config: Dialogue.Config) {