78 lines
2.4 KiB
TypeScript
78 lines
2.4 KiB
TypeScript
import { Plugin } from '@nocobase/server';
|
|
import path from 'path';
|
|
import { CollectionRepository } from '.';
|
|
import {
|
|
afterCreateForReverseField,
|
|
beforeCreateForChildrenCollection,
|
|
beforeCreateForReverseField,
|
|
beforeInitOptions
|
|
} from './hooks';
|
|
import { CollectionModel, FieldModel } from './models';
|
|
|
|
export class CollectionManagerPlugin extends Plugin {
|
|
async beforeLoad() {
|
|
this.app.db.registerModels({
|
|
CollectionModel,
|
|
FieldModel,
|
|
});
|
|
|
|
this.app.db.registerRepositories({
|
|
CollectionRepository,
|
|
});
|
|
|
|
// 要在 beforeInitOptions 之前处理
|
|
this.app.db.on('fields.beforeCreate', beforeCreateForReverseField(this.app.db));
|
|
this.app.db.on('fields.beforeCreate', beforeCreateForChildrenCollection(this.app.db));
|
|
this.app.db.on('fields.beforeCreate', async (model, options) => {
|
|
const type = model.get('type');
|
|
await this.app.db.emitAsync(`fields.${type}.beforeInitOptions`, model, {
|
|
...options,
|
|
database: this.app.db,
|
|
});
|
|
});
|
|
for (const key in beforeInitOptions) {
|
|
if (Object.prototype.hasOwnProperty.call(beforeInitOptions, key)) {
|
|
const fn = beforeInitOptions[key];
|
|
this.app.db.on(`fields.${key}.beforeInitOptions`, fn);
|
|
}
|
|
}
|
|
this.app.db.on('fields.afterCreate', afterCreateForReverseField(this.app.db));
|
|
|
|
this.app.db.on('collections.afterCreateWithAssociations', async (model, { context, transaction }) => {
|
|
if (context) {
|
|
await model.migrate({ transaction });
|
|
}
|
|
});
|
|
|
|
this.app.db.on('fields.afterCreate', async (model, { context, transaction }) => {
|
|
if (context) {
|
|
await model.migrate({ transaction });
|
|
}
|
|
});
|
|
|
|
this.app.on('beforeStart', async () => {
|
|
await this.app.db.getRepository<CollectionRepository>('collections').load();
|
|
});
|
|
|
|
this.app.resourcer.use(async (ctx, next) => {
|
|
const { resourceName, actionName } = ctx.action;
|
|
if (resourceName === 'collections.fields' && actionName === 'update') {
|
|
const { updateAssociationValues = [] } = ctx.action.params;
|
|
updateAssociationValues.push('uiSchema');
|
|
ctx.action.mergeParams({
|
|
updateAssociationValues,
|
|
});
|
|
}
|
|
await next();
|
|
});
|
|
}
|
|
|
|
async load() {
|
|
await this.app.db.import({
|
|
directory: path.resolve(__dirname, './collections'),
|
|
});
|
|
}
|
|
}
|
|
|
|
export default CollectionManagerPlugin;
|