plugin-build-guide-block 1.0.11 → 1.1.2

package/src/server/actions/getMarkdown.ts

@@ -0,0 +1,26 @@
+import { Context, Next } from '@nocobase/actions';
+import { Repository } from '@nocobase/database';
+
+export async function getMarkdown(ctx: Context, next: Next) {
+  const { filterByTk } = ctx.action.params;
+  const { resourceName } = ctx.action;
+  const repository = ctx.db.getRepository<any>(resourceName) as Repository;
+  const model = await repository.findById(filterByTk);
+
+  if (!model) {
+    ctx.throw(404, 'User Guide not found');
+  }
+
+  if (model.get('status') !== 'completed') {
+    ctx.throw(400, 'User Guide is not ready yet');
+  }
+
+  ctx.body = model.get('generatedMarkdown') || '';
+  ctx.withoutDataWrapping = true;
+
+  ctx.set({
+    'Content-Type': 'text/markdown; charset=UTF-8',
+  });
+
+  await next();
+}

package/src/server/collections/ai-build-guide-pages.ts

@@ -0,0 +1,60 @@
+import { defineCollection } from '@nocobase/database';
+
+export default defineCollection({
+  name: 'aiBuildGuidePages',
+  shared: true,
+  dumpRules: 'required',
+  migrationRules: ['overwrite', 'schema-only'],
+  timestamps: true,
+  fields: [
+    {
+      type: 'uid',
+      name: 'id',
+      primaryKey: true,
+    },
+    {
+      type: 'belongsTo',
+      name: 'space',
+      target: 'aiBuildGuideSpaces',
+      foreignKey: 'spaceId',
+    },
+    {
+      type: 'integer',
+      name: 'sort',
+      defaultValue: 0,
+    },
+    {
+      type: 'string',
+      name: 'title',
+    },
+    {
+      type: 'string',
+      name: 'slug',
+    },
+    {
+      type: 'text',
+      name: 'goal',
+    },
+    {
+      type: 'json',
+      name: 'planItem',
+    },
+    {
+      type: 'string',
+      name: 'status',
+      defaultValue: 'pending',
+    },
+    {
+      type: 'text',
+      name: 'generatedHtml',
+    },
+    {
+      type: 'text',
+      name: 'generatedMarkdown',
+    },
+    {
+      type: 'text',
+      name: 'buildLog',
+    },
+  ],
+});

package/src/server/collections/ai-build-guide-spaces.ts

@@ -28,23 +28,65 @@ export default defineCollection({
       type: 'text',
       name: 'systemPrompt',
     },
-    {
-      type: 'text',
-      name: 'generatedHtml',
-    },
-    {
-      type: 'string',
-      name: 'status',
-      defaultValue: 'draft',
-    },
+    {
+      type: 'string',
+      name: 'outputFormat',
+      defaultValue: 'html',
+    },
+    {
+      type: 'integer',
+      name: 'targetChapterCount',
+      defaultValue: 5,
+    },
+    {
+      type: 'text',
+      name: 'chapterGuidance',
+    },
+    {
+      type: 'text',
+      name: 'generatedHtml',
+    },
+    {
+      type: 'text',
+      name: 'generatedMarkdown',
+    },
+    {
+      type: 'json',
+      name: 'planJson',
+    },
+    {
+      type: 'string',
+      name: 'buildPhase',
+      defaultValue: 'idle',
+    },
+    {
+      type: 'integer',
+      name: 'pageCount',
+      defaultValue: 0,
+    },
+    {
+      type: 'string',
+      name: 'sourceHash',
+    },
+    {
+      type: 'string',
+      name: 'status',
+      defaultValue: 'draft',
+    },
     {
       type: 'text',
       name: 'buildLog',
     },
     {
-      type: 'belongsToMany',
-      name: 'documents',
-      target: 'attachments',
-    },
-  ],
-});
+      type: 'belongsToMany',
+      name: 'documents',
+      target: 'attachments',
+    },
+    {
+      type: 'hasMany',
+      name: 'pages',
+      target: 'aiBuildGuidePages',
+      foreignKey: 'spaceId',
+    },
+  ],
+});

package/src/server/plugin.ts

@@ -1,76 +1,130 @@
-import { InstallOptions, Plugin } from '@nocobase/server';
-import path from 'path';
-import { build } from './actions/build';
-import { getHtml } from './actions/getHtml';
-
-export class PluginBuildGuideBlockServer extends Plugin {
-  afterAdd() {}
-
-  beforeLoad() {}
-
-  async load() {
-    this.app.resourceManager.registerActionHandlers({
-      'aiBuildGuideSpaces:build': build,
-      'aiBuildGuideSpaces:getHtml': getHtml,
-    });
-
-    this.app.acl.allow('aiBuildGuideSpaces', 'getHtml', 'loggedIn');
-    this.app.acl.registerSnippet({
-      name: 'pm.ai-build-guide',
-      actions: [
-        'aiBuildGuideSpaces:create',
-        'aiBuildGuideSpaces:update',
-        'aiBuildGuideSpaces:destroy',
-        'aiBuildGuideSpaces:list',
-        'aiBuildGuideSpaces:get',
-        'aiBuildGuideSpaces:build',
-      ],
-    });
-
-    // Recover stale "building" status after server restart
-    this.app.on('afterStart', async () => {
-      try {
-        const repo = this.db.getRepository('aiBuildGuideSpaces');
-        await repo.update({
-          filter: { status: 'building' },
-          values: {
-            status: 'error',
-            buildLog: 'Build interrupted by server restart',
-          },
-        });
-      } catch (err) {
-        this.app.logger.warn('[plugin-build-guide-block] Failed to recover stale builds', err);
-      }
-    });
-  }
-
-  async install(options?: InstallOptions) {
-    const collection = this.db.getCollection('aiBuildGuideSpaces');
-    if (collection) {
-      await collection.model.sync();
-    }
-    const repo = this.db.getRepository<any>('collections');
-    if (repo) {
-      await repo.db2cm('aiBuildGuideSpaces');
-    }
-  }
-
-  async upgrade() {
-    const collection = this.db.getCollection('aiBuildGuideSpaces');
-    if (collection) {
-      await collection.model.sync();
-    }
-    const repo = this.db.getRepository<any>('collections');
-    if (repo) {
-      await repo.db2cm('aiBuildGuideSpaces');
-    }
-  }
-
-  async afterEnable() {}
-
-  async afterDisable() {}
-
-  async remove() {}
-}
-
-export default PluginBuildGuideBlockServer;
+import { InstallOptions, Plugin } from '@nocobase/server';
+import { resolve } from 'path';
+import { build } from './actions/build';
+import { getHtml } from './actions/getHtml';
+import { getMarkdown } from './actions/getMarkdown';
+
+export class PluginBuildGuideBlockServer extends Plugin {
+  private readonly schemaCollections = ['aiBuildGuideSpaces', 'aiBuildGuidePages'];
+
+  afterAdd() {}
+
+  beforeLoad() {}
+
+  async load() {
+    await this.db.import({
+      directory: resolve(__dirname, 'collections'),
+    });
+
+    this.app.resourceManager.registerActionHandlers({
+      'aiBuildGuideSpaces:build': build,
+      'aiBuildGuideSpaces:getHtml': getHtml,
+      'aiBuildGuideSpaces:getMarkdown': getMarkdown,
+    });
+
+    this.app.acl.allow('aiBuildGuideSpaces', 'getHtml', 'loggedIn');
+    this.app.acl.allow('aiBuildGuideSpaces', 'getMarkdown', 'loggedIn');
+    this.app.acl.registerSnippet({
+      name: 'pm.ai-build-guide',
+      actions: [
+        'aiBuildGuideSpaces:create',
+        'aiBuildGuideSpaces:update',
+        'aiBuildGuideSpaces:destroy',
+        'aiBuildGuideSpaces:list',
+        'aiBuildGuideSpaces:get',
+        'aiBuildGuideSpaces:build',
+        'aiBuildGuidePages:list',
+        'aiBuildGuidePages:get',
+      ],
+    });
+
+    // Recover stale "building" status after server restart
+    this.app.on('afterStart', async () => {
+      try {
+        const repo = this.db.getRepository('aiBuildGuideSpaces');
+        await repo.update({
+          filter: { status: 'building' },
+          values: {
+            status: 'error',
+            buildPhase: 'error',
+            buildLog: 'Build interrupted by server restart',
+          },
+        });
+        const pageRepo = this.db.getRepository('aiBuildGuidePages');
+        await pageRepo.update({
+          filter: { status: 'building' },
+          values: {
+            status: 'error',
+            buildLog: 'Build interrupted by server restart',
+          },
+        });
+      } catch (err) {
+        this.app.logger.warn('[plugin-build-guide-block] Failed to recover stale builds', err);
+      }
+    });
+  }
+
+  private async ensureCollectionSchema(collectionName: string) {
+    const collection = this.db.getCollection(collectionName);
+    if (!collection) {
+      this.app.logger.warn(`[plugin-build-guide-block] Collection "${collectionName}" is not registered`);
+      return;
+    }
+
+    const queryInterface = this.db.sequelize.getQueryInterface();
+    const tableName = collection.getTableNameWithSchema();
+    let columns: Record<string, any> | null = null;
+
+    try {
+      columns = await queryInterface.describeTable(tableName);
+    } catch (error) {
+      await collection.model.sync();
+      columns = await queryInterface.describeTable(tableName);
+    }
+
+    const attributes = collection.model.rawAttributes as Record<string, any>;
+    for (const [attributeName, attribute] of Object.entries(attributes)) {
+      const columnName = attribute.field || attributeName;
+      if (columns[columnName]) {
+        continue;
+      }
+
+      const columnDefinition = { ...attribute };
+      delete columnDefinition.Model;
+      delete columnDefinition.fieldName;
+
+      await queryInterface.addColumn(tableName, columnName, columnDefinition);
+      columns[columnName] = columnDefinition;
+      this.app.logger.info(`[plugin-build-guide-block] Added missing column "${columnName}" to "${collectionName}"`);
+    }
+  }
+
+  private async ensureSchema() {
+    for (const collectionName of this.schemaCollections) {
+      await this.ensureCollectionSchema(collectionName);
+    }
+
+    const repo = this.db.getRepository<any>('collections');
+    if (repo) {
+      for (const collectionName of this.schemaCollections) {
+        await repo.db2cm(collectionName);
+      }
+    }
+  }
+
+  async install(options?: InstallOptions) {
+    await this.ensureSchema();
+  }
+
+  async upgrade() {
+    await this.ensureSchema();
+  }
+
+  async afterEnable() {}
+
+  async afterDisable() {}
+
+  async remove() {}
+}
+
+export default PluginBuildGuideBlockServer;