tachybase_todo/packages/plugin-collection-manager/src/plugin.ts

63 lines
1.9 KiB
TypeScript
Raw Normal View History

import { Plugin } from '@nocobase/server';
import path from 'path';
2022-02-17 01:06:42 +08:00
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,
});
2022-02-17 01:06:42 +08:00
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);
});
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 });
}
});
2022-02-17 01:06:42 +08:00
this.app.on('beforeStart', async () => {
await this.app.db.getRepository<CollectionRepository>('collections').load();
});
}
async load() {
await this.app.db.import({
directory: path.resolve(__dirname, './collections'),
});
}
}
export default CollectionManagerPlugin;