import { ALL, Body, Controller, Inject, Post, Provide, Query } from "@midwayjs/core"; import { AddonDefine, AddonRequestHandleReq, AddonService, Constants, CrudController, newAddon, ValidateException } from "@certd/lib-server"; import { AuthService } from "../../../modules/sys/authority/service/auth-service.js"; import { checkPlus } from "@certd/plus-core"; import { http, logger, utils } from "@certd/basic"; import { TaskServiceBuilder } from "../../../modules/pipeline/service/getter/task-service-getter.js"; import { ApiTags } from "@midwayjs/swagger"; import { AuditType } from "../../../modules/sys/enterprise/service/audit-constants.js"; /** * Addon */ @Provide() @Controller("/api/addon") @ApiTags(["addon"]) export class AddonController extends CrudController { @Inject() service: AddonService; @Inject() authService: AuthService; @Inject() taskServiceBuilder: TaskServiceBuilder; getService(): AddonService { return this.service; } getAuditType(): string { return AuditType.addon; } @Post("/page", { description: Constants.per.authOnly, summary: "查询Addon分页列表" }) async page(@Body(ALL) body) { const { projectId, userId } = await this.getProjectUserIdRead(); body.query = body.query ?? {}; delete body.query.userId; body.query.projectId = projectId; const buildQuery = qb => { qb.andWhere("user_id = :userId", { userId }); }; const res = await this.service.page({ query: body.query, page: body.page, sort: body.sort, buildQuery, }); return this.ok(res); } @Post("/list", { description: Constants.per.authOnly, summary: "查询Addon列表" }) async list(@Body(ALL) body) { const { projectId, userId } = await this.getProjectUserIdRead(); body.query = body.query ?? {}; body.query.userId = userId; body.query.projectId = projectId; return super.list(body); } @Post("/add", { description: Constants.per.authOnly, summary: "添加Addon" }) async add(@Body(ALL) bean) { const { userId, projectId } = await this.getProjectUserIdRead(); bean.userId = userId; bean.projectId = projectId; const type = bean.type; const addonType = bean.addonType; if (!type || !addonType) { throw new ValidateException("请选择Addon类型"); } const define: AddonDefine = this.service.getDefineByType(type, addonType); if (!define) { throw new ValidateException("Addon类型不存在"); } if (define.needPlus) { checkPlus(); } const res = await super.add(bean); this.auditLog({ content: `新增了Addon「${bean.name}」(ID:${res.data}, 类型:${bean.type})` }); return res; } @Post("/update", { description: Constants.per.authOnly, summary: "更新Addon" }) async update(@Body(ALL) bean) { await this.checkOwner(this.getService(), bean.id, "write"); const old = await this.service.info(bean.id); if (!old) { throw new ValidateException("Addon配置不存在"); } if (old.type !== bean.type) { const addonType = old.type; const type = bean.type; const define: AddonDefine = this.service.getDefineByType(type, addonType); if (!define) { throw new ValidateException("Addon类型不存在"); } if (define.needPlus) { checkPlus(); } } delete bean.userId; delete bean.projectId; const res = await super.update(bean); this.auditLog({ content: `修改了Addon「${bean.name}」(ID:${bean.id})` }); return res; } @Post("/info", { description: Constants.per.authOnly, summary: "查询Addon详情" }) async info(@Query("id") id: number) { await this.checkOwner(this.getService(), id, "read"); return super.info(id); } @Post("/delete", { description: Constants.per.authOnly, summary: "删除Addon" }) async delete(@Query("id") id: number) { await this.checkOwner(this.getService(), id, "write"); const res = await super.delete(id); this.auditLog({ content: `删除了Addon(ID:${id})` }); return res; } @Post("/define", { description: Constants.per.authOnly, summary: "查询Addon插件定义" }) async define(@Query("type") type: string, @Query("addonType") addonType: string) { const notification = this.service.getDefineByType(type, addonType); return this.ok(notification); } @Post("/getTypeDict", { description: Constants.per.authOnly, summary: "查询Addon插件类型字典" }) async getTypeDict(@Query("addonType") addonType: string) { const list: any = this.service.getDefineList(addonType); let dict = []; for (const item of list) { dict.push({ value: item.name, label: item.title, needPlus: item.needPlus ?? false, icon: item.icon, }); } dict = dict.sort(a => { return a.needPlus ? 0 : -1; }); return this.ok(dict); } @Post("/simpleInfo", { description: Constants.per.authOnly, summary: "查询Addon插件简单信息" }) async simpleInfo(@Query("addonType") addonType: string, @Query("id") id: number) { if (id === 0) { //获取默认 const { projectId, userId } = await this.getProjectUserIdRead(); const res = await this.service.getDefault(userId, addonType, projectId); if (!res) { throw new ValidateException("默认Addon配置不存在"); } const simple = await this.service.getSimpleInfo(res.id); return this.ok(simple); } await this.checkOwner(this.getService(), id, "read", true); const res = await this.service.getSimpleInfo(id); return this.ok(res); } @Post("/getDefaultId", { description: Constants.per.authOnly, summary: "查询Addon插件默认配置ID" }) async getDefaultId(@Query("addonType") addonType: string) { const { projectId, userId } = await this.getProjectUserIdRead(); const res = await this.service.getDefault(userId, addonType, projectId); return this.ok(res?.id); } @Post("/setDefault", { description: Constants.per.authOnly, summary: "设置Addon插件默认配置" }) async setDefault(@Query("addonType") addonType: string, @Query("id") id: number) { const { projectId, userId } = await this.checkOwner(this.getService(), id, "write", true); const res = await this.service.setDefault(id, userId, addonType, projectId); this.auditLog({ content: `设置了默认Addon(ID:${id})` }); return this.ok(res); } @Post("/options", { description: Constants.per.authOnly, summary: "查询Addon插件配置字典" }) async options(@Query("addonType") addonType: string) { const { projectId, userId } = await this.getProjectUserIdRead(); const res = await this.service.list({ query: { userId, addonType, projectId, }, }); for (const item of res) { delete item.setting; } return this.ok(res); } @Post("/handle", { description: Constants.per.authOnly, summary: "Addon插件处理请求" }) async handle(@Body(ALL) body: AddonRequestHandleReq) { let inputAddon = body.input.addon; if (body.input.id > 0) { await this.checkOwner(this.getService(), body.input.id, "write", true); const oldEntity = await this.service.info(body.input.id); if (oldEntity) { inputAddon = JSON.parse(oldEntity.setting); } } const { projectId, userId } = await this.getProjectUserIdRead(); const serviceGetter = this.taskServiceBuilder.create({ userId, projectId }); const ctx = { http: http, logger: logger, utils: utils, serviceGetter, }; const addon = await newAddon(body.addonType, body.typeName, inputAddon, ctx); const res = await addon.onRequest(body); return this.ok(res); } }