54 lines
1.7 KiB
TypeScript
54 lines
1.7 KiB
TypeScript
|
import { Plugin } from '@nocobase/server';
|
||
|
|
import path from 'path';
|
||
|
|
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,
|
||
|
|
});
|
||
|
|
|
||
|
|
// 要在 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);
|
||
|
|
});
|
||
|
|
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 });
|
||
|
|
}
|
||
|
|
});
|
||
|
|
}
|
||
|
|
|
||
|
|
async load() {
|
||
|
|
await this.app.db.import({
|
||
|
|
directory: path.resolve(__dirname, './collections'),
|
||
|
|
});
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
export default CollectionManagerPlugin;
|