tachybase_todo/packages/plugins/@nocobase/plugin-error-handler/src/server/server.ts

60 lines
1.8 KiB
TypeScript

import { Schema } from '@formily/json-schema';
import { BaseError } from '@nocobase/database';
import { Plugin } from '@nocobase/server';
import lodash from 'lodash';
import { ErrorHandler } from './error-handler';
import enUS from './locale/en_US';
import zhCN from './locale/zh_CN';
export class PluginErrorHandler extends Plugin {
errorHandler: ErrorHandler = new ErrorHandler();
i18nNs = 'error-handler';
beforeLoad() {
this.registerSequelizeValidationErrorHandler();
}
registerSequelizeValidationErrorHandler() {
const findFieldTitle = (instance, path, tFunc, ctx) => {
if (!instance) {
return path;
}
const model = instance.constructor;
const dataSourceKey = ctx.get('x-data-source');
const dataSource = ctx.app.dataSourceManager.dataSources.get(dataSourceKey);
const database = dataSource ? dataSource.collectionManager.db : ctx.db;
const collection = database.modelCollection.get(model);
const field = collection.getField(path);
const fieldOptions = Schema.compile(field?.options, { t: tFunc });
const title = lodash.get(fieldOptions, 'uiSchema.title', path);
return title;
};
this.errorHandler.register(
(err) => err?.errors?.length && err instanceof BaseError,
(err, ctx) => {
ctx.body = {
errors: err.errors.map((err) => {
return {
message: ctx.i18n.t(err.type, {
ns: this.i18nNs,
field: findFieldTitle(err.instance, err.path, ctx.i18n.t, ctx),
}),
};
}),
};
ctx.status = 400;
},
);
}
async load() {
this.app.i18n.addResources('zh-CN', this.i18nNs, zhCN);
this.app.i18n.addResources('en-US', this.i18nNs, enUS);
this.app.use(this.errorHandler.middleware(), { before: 'cors', tag: 'errorHandler' });
}
}