tachybase_todo/packages/database/src/model-hook.ts

63 lines
1.4 KiB
TypeScript

import lodash from 'lodash';
import type { SequelizeHooks } from 'sequelize/types/lib/hooks';
import Database from './database';
import { Model } from './model';
const { hooks } = require('sequelize/lib/hooks');
export class ModelHook {
database: Database;
boundEvent = new Set<string>();
constructor(database: Database) {
this.database = database;
}
isModelHook(eventName: string | symbol): keyof SequelizeHooks | false {
if (lodash.isString(eventName)) {
const hookType = eventName.split('.').pop();
if (hooks[hookType]) {
return <keyof SequelizeHooks>hookType;
}
}
return false;
}
findModelName(hookArgs) {
for (const arg of hookArgs) {
if (arg instanceof Model) {
return (<Model>arg).constructor.name;
}
if (lodash.isPlainObject(arg) && arg['model']) {
return arg['model'].name;
}
}
return null;
}
bindEvent(eventName) {
this.boundEvent.add(eventName);
}
hasBindEvent(eventName) {
return this.boundEvent.has(eventName);
}
sequelizeHookBuilder(eventName) {
return async (...args: any[]) => {
const modelName = this.findModelName(args);
if (modelName) {
// emit model event
await this.database.emitAsync(`${modelName}.${eventName}`, ...args);
}
// emit sequelize global event
await this.database.emitAsync(eventName, ...args);
};
}
}