Compare commits

...

2 Commits

Author SHA1 Message Date
kingecg 9748d89079 add plugin related 2025-06-09 23:19:43 +08:00
kingecg 5103327563 add migration 2025-06-09 23:11:02 +08:00
6 changed files with 126 additions and 4 deletions

View File

@ -22,15 +22,16 @@
"pg": "^8.16.0",
"reflect-metadata": "^0.2.2",
"rxjs": "^7.8.2",
"typeorm": "^0.3.24",
"uid": "^2.0.2"
},
"devDependencies": {
"@types/bcrypt": "^5.0.0",
"@types/node": "^22.15.30",
"@types/passport-jwt": "^4.0.1",
"rimraf": "^6.0.1",
"ts-node": "^10.9.2",
"tsconfig-paths": "^4.2.0",
"typeorm": "^0.3.24",
"typescript": "^5.8.3"
}
},

View File

@ -7,8 +7,11 @@
"build": "rimraf dist && tsc",
"start": "npm run build && node dist/main.js",
"dev": "ts-node -r tsconfig-paths/register src/main.ts",
"typeorm": "typeorm-ts-node-commonjs -d dist/data-source.js",
"test": "echo \"Error: no test specified\" && exit 1"
"typeorma": "typeorm-ts-node-commonjs -d dist/data-source.js",
"test": "echo \"Error: no test specified\" && exit 1",
"typeorm": "ts-node -r ts-node/register ./node_modules/typeorm/cli.js",
"migration:generate": "npm run typeorm migration:generate -d dist/data-source.js",
"migration:run": "npm run typeorm migration:run -- --dataSource dist/data-source.js"
},
"keywords": [],
"author": "",
@ -27,15 +30,16 @@
"pg": "^8.16.0",
"reflect-metadata": "^0.2.2",
"rxjs": "^7.8.2",
"typeorm": "^0.3.24",
"uid": "^2.0.2"
},
"devDependencies": {
"@types/bcrypt": "^5.0.0",
"@types/node": "^22.15.30",
"@types/passport-jwt": "^4.0.1",
"rimraf": "^6.0.1",
"ts-node": "^10.9.2",
"tsconfig-paths": "^4.2.0",
"typeorm": "^0.3.24",
"typescript": "^5.8.3"
}
}

View File

@ -0,0 +1,57 @@
import { MigrationInterface, QueryRunner } from 'typeorm';
export class AddForeignKeys1690 implements MigrationInterface {
async up(queryRunner: QueryRunner): Promise<void> {
// 1. User与Workspace的关联
await queryRunner.query(`
ALTER TABLE "user"
ADD CONSTRAINT "fk_user_workspace"
FOREIGN KEY ("workspaceId")
REFERENCES "workspace"("id")
ON DELETE CASCADE
`);
// 2. AiConfig与Workspace的关联
await queryRunner.query(`
ALTER TABLE "ai_config"
ADD CONSTRAINT "fk_ai_config_workspace"
FOREIGN KEY ("workspaceId")
REFERENCES "workspace"("id")
ON DELETE CASCADE
`);
// 3. Project与Workspace的关联
await queryRunner.query(`
ALTER TABLE "project"
ADD CONSTRAINT "fk_project_workspace"
FOREIGN KEY ("workspaceId")
REFERENCES "workspace"("id")
ON DELETE CASCADE
`);
// 4. PluginConfig与Workspace的关联
await queryRunner.query(`
ALTER TABLE "plugin_config"
ADD CONSTRAINT "fk_plugin_config_workspace"
FOREIGN KEY ("workspaceId")
REFERENCES "workspace"("id")
ON DELETE CASCADE
`);
}
async down(queryRunner: QueryRunner): Promise<void> {
// 删除外键约束
const foreignKeys = [
'fk_user_workspace',
'fk_ai_config_workspace',
'fk_project_workspace',
'fk_plugin_config_workspace'
];
for (const fk of foreignKeys) {
await queryRunner.query(
`ALTER TABLE ${fk.split('_').slice(2).join('_')} DROP CONSTRAINT ${fk}`
);
}
}
}

View File

@ -0,0 +1,16 @@
import { Controller, Post, Body, Param } from '@nestjs/common';
import { PluginService } from './plugin.service';
@Controller('plugins')
export class PluginController {
constructor(private readonly pluginService: PluginService) {}
@Post(':id/execute')
async execute(
@Param('id') pluginId: string,
@Body() executionArgs: Record<string, any>
) {
return this.pluginService.executePlugin(pluginId, executionArgs);
}
}

View File

@ -0,0 +1,7 @@
import { EntityRepository, Repository } from 'typeorm';
import { PluginConfig } from './plugin-config.entity';
@EntityRepository(PluginConfig)
export class PluginRepository extends Repository<PluginConfig> {
// 可以在这里添加自定义查询方法
}

View File

@ -0,0 +1,37 @@
import { Injectable, NotFoundException } from '@nestjs/common';
import { PluginConfig } from './plugin-config.entity';
import { PluginRepository } from './plugin.repository';
@Injectable()
export class PluginService {
constructor(private readonly pluginRepository: PluginRepository) {}
async executePlugin(pluginId: string, args: Record<string, any>): Promise<any> {
// 1. 获取插件配置
const pluginConfig = await this.pluginRepository.findOne({ where: { pluginId } });
if (!pluginConfig) {
throw new NotFoundException(`Plugin ${pluginId} not found`);
}
try {
// 2. 动态加载插件模块假设插件以npm包形式安装
const pluginModule = require.resolve(pluginId);
const plugin = require(pluginModule);
// 3. 验证插件接口假设插件必须实现execute方法
if (typeof plugin.execute !== 'function') {
throw new Error(`Plugin ${pluginId} missing required execute method`);
}
// 4. 执行插件并返回结果
return await plugin.execute({
...args,
config: pluginConfig.settings // 传递插件配置
});
} catch (error) {
throw new Error(`Failed to execute plugin ${pluginId}: ${error}`);
}
}
}