plugin-build-guide-block 1.0.9 → 1.0.11

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

@@ -1,50 +1,50 @@
-import { defineCollection } from '@nocobase/database';
-
-export default defineCollection({
-  name: 'aiBuildGuideSpaces',
-  shared: true,
-  dumpRules: 'required',
-  migrationRules: ['overwrite', 'schema-only'],
-  timestamps: true,
-  fields: [
-    {
-      type: 'uid',
-      name: 'id',
-      primaryKey: true,
-    },
-    {
-      type: 'string',
-      name: 'title',
-    },
-    {
-      type: 'string',
-      name: 'llmService',
-    },
-    {
-      type: 'string',
-      name: 'model',
-    },
-    {
-      type: 'text',
-      name: 'systemPrompt',
-    },
-    {
-      type: 'text',
-      name: 'generatedHtml',
-    },
-    {
-      type: 'string',
-      name: 'status',
-      defaultValue: 'draft',
-    },
-    {
-      type: 'text',
-      name: 'buildLog',
-    },
-    {
-      type: 'belongsToMany',
-      name: 'documents',
-      target: 'attachments',
-    },
-  ],
-});
+import { defineCollection } from '@nocobase/database';
+
+export default defineCollection({
+  name: 'aiBuildGuideSpaces',
+  shared: true,
+  dumpRules: 'required',
+  migrationRules: ['overwrite', 'schema-only'],
+  timestamps: true,
+  fields: [
+    {
+      type: 'uid',
+      name: 'id',
+      primaryKey: true,
+    },
+    {
+      type: 'string',
+      name: 'title',
+    },
+    {
+      type: 'string',
+      name: 'llmService',
+    },
+    {
+      type: 'string',
+      name: 'model',
+    },
+    {
+      type: 'text',
+      name: 'systemPrompt',
+    },
+    {
+      type: 'text',
+      name: 'generatedHtml',
+    },
+    {
+      type: 'string',
+      name: 'status',
+      defaultValue: 'draft',
+    },
+    {
+      type: 'text',
+      name: 'buildLog',
+    },
+    {
+      type: 'belongsToMany',
+      name: 'documents',
+      target: 'attachments',
+    },
+  ],
+});

package/src/server/index.ts

@@ -1 +1,2 @@
 export { default } from './plugin';
+export const namespace = 'plugin-build-guide-block';

package/src/server/plugin.ts

@@ -1,60 +1,76 @@
-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',
-      ],
-    });
-  }
-
-  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 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;