@eggjs/tegg-orm-plugin 1.2.0 → 2.1.0

package/CHANGELOG.md

@@ -3,6 +3,28 @@
 All notable changes to this project will be documented in this file.
 See [Conventional Commits](https://conventionalcommits.org) for commit guidelines.
 
+# [2.1.0](https://github.com/eggjs/tegg/compare/@eggjs/tegg-orm-plugin@2.0.0...@eggjs/tegg-orm-plugin@2.1.0) (2022-07-20)
+
+
+### Features
+
+* impl Inject Model ([#43](https://github.com/eggjs/tegg/issues/43)) ([ced2ce2](https://github.com/eggjs/tegg/commit/ced2ce2134964dcb410410c0192a34f77507c42d))
+
+
+
+
+
+# [1.3.0](https://github.com/eggjs/tegg/compare/v1.2.0...v1.3.0) (2022-07-01)
+
+
+### Features
+
+* support leoric hooks ([#41](https://github.com/eggjs/tegg/issues/41)) ([9bdbc2c](https://github.com/eggjs/tegg/commit/9bdbc2cbe96df9f66f96b4f8e208883e99957946))
+
+
+
+
+
 # [0.2.0](https://github.com/eggjs/tegg/compare/v0.1.19...v0.2.0) (2022-01-20)
 
 **Note:** Version bump only for package @eggjs/tegg-orm-plugin

package/app.js

@@ -1,9 +1,15 @@
 "use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
 Object.defineProperty(exports, "__esModule", { value: true });
 const DataSourceManager_1 = require("./lib/DataSourceManager");
 const LeoricRegister_1 = require("./lib/LeoricRegister");
 const ModelProtoManager_1 = require("./lib/ModelProtoManager");
 const ModelProtoHook_1 = require("./lib/ModelProtoHook");
+const tegg_orm_decorator_1 = require("@eggjs/tegg-orm-decorator");
+const ContextModelProto_1 = __importDefault(require("./lib/ContextModelProto"));
+const ContextModeObject_1 = require("./lib/ContextModeObject");
 class OrmAppBootHook {
     constructor(app) {
         this.app = app;
@@ -11,9 +17,12 @@ class OrmAppBootHook {
         this.modelProtoManager = new ModelProtoManager_1.ModelProtoManager();
         this.leoricRegister = new LeoricRegister_1.LeoricRegister(this.modelProtoManager, this.dataSourceManager);
         this.modelProtoHook = new ModelProtoHook_1.ModelProtoHook(this.modelProtoManager);
+        this.app.eggPrototypeCreatorFactory.registerPrototypeCreator(tegg_orm_decorator_1.MODEL_PROTO_IMPL_TYPE, ContextModelProto_1.default.createProto);
+        this.app.leoricRegister = this.leoricRegister;
     }
     configWillLoad() {
         this.app.eggPrototypeLifecycleUtil.registerLifecycle(this.modelProtoHook);
+        this.app.eggObjectFactory.registerEggObjectCreateMethod(ContextModelProto_1.default, ContextModeObject_1.ContextModeObject.createObject);
     }
     configDidLoad() {
         const config = this.app.config.orm;
@@ -35,4 +44,4 @@ class OrmAppBootHook {
     }
 }
 exports.default = OrmAppBootHook;
-//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoiYXBwLmpzIiwic291cmNlUm9vdCI6IiIsInNvdXJjZXMiOlsiYXBwLnRzIl0sIm5hbWVzIjpbXSwibWFwcGluZ3MiOiI7O0FBQ0EsK0RBQTREO0FBQzVELHlEQUFzRDtBQUN0RCwrREFBNEQ7QUFDNUQseURBQXNEO0FBRXRELE1BQXFCLGNBQWM7SUFPakMsWUFBWSxHQUFHO1FBQ2IsSUFBSSxDQUFDLEdBQUcsR0FBRyxHQUFHLENBQUM7UUFDZixJQUFJLENBQUMsaUJBQWlCLEdBQUcsSUFBSSxxQ0FBaUIsRUFBRSxDQUFDO1FBQ2pELElBQUksQ0FBQyxpQkFBaUIsR0FBRyxJQUFJLHFDQUFpQixFQUFFLENBQUM7UUFDakQsSUFBSSxDQUFDLGNBQWMsR0FBRyxJQUFJLCtCQUFjLENBQUMsSUFBSSxDQUFDLGlCQUFpQixFQUFFLElBQUksQ0FBQyxpQkFBaUIsQ0FBQyxDQUFDO1FBQ3pGLElBQUksQ0FBQyxjQUFjLEdBQUcsSUFBSSwrQkFBYyxDQUFDLElBQUksQ0FBQyxpQkFBaUIsQ0FBQyxDQUFDO0lBQ25FLENBQUM7SUFFRCxjQUFjO1FBQ1osSUFBSSxDQUFDLEdBQUcsQ0FBQyx5QkFBeUIsQ0FBQyxpQkFBaUIsQ0FBQyxJQUFJLENBQUMsY0FBYyxDQUFDLENBQUM7SUFDNUUsQ0FBQztJQUVELGFBQWE7UUFDWCxNQUFNLE1BQU0sR0FBRyxJQUFJLENBQUMsR0FBRyxDQUFDLE1BQU0sQ0FBQyxHQUFHLENBQUM7UUFDbkMsSUFBSSxNQUFNLENBQUMsV0FBVyxFQUFFO1lBQ3RCLEtBQUssTUFBTSxVQUFVLElBQUksTUFBTSxDQUFDLFdBQVcsRUFBRTtnQkFDM0MsSUFBSSxDQUFDLGlCQUFpQixDQUFDLFNBQVMsQ0FBQyxVQUFVLENBQUMsQ0FBQzthQUM5QztTQUNGO2FBQU07WUFDTCxJQUFJLENBQUMsaUJBQWlCLENBQUMsZ0JBQWdCLENBQUMsTUFBTSxDQUFDLENBQUM7U0FDakQ7SUFDSCxDQUFDO0lBRUQsS0FBSyxDQUFDLE9BQU87UUFDWCxNQUFNLElBQUksQ0FBQyxHQUFHLENBQUMsYUFBYSxDQUFDLEtBQUssRUFBRSxDQUFDO1FBQ3JDLE1BQU0sSUFBSSxDQUFDLGNBQWMsQ0FBQyxRQUFRLEVBQUUsQ0FBQztJQUN2QyxDQUFDO0lBRUQsS0FBSyxDQUFDLFdBQVc7UUFDZixJQUFJLENBQUMsR0FBRyxDQUFDLHlCQUF5QixDQUFDLGVBQWUsQ0FBQyxJQUFJLENBQUMsY0FBYyxDQUFDLENBQUM7SUFDMUUsQ0FBQztDQUNGO0FBdENELGlDQXNDQyJ9
+//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoiYXBwLmpzIiwic291cmNlUm9vdCI6IiIsInNvdXJjZXMiOlsiYXBwLnRzIl0sIm5hbWVzIjpbXSwibWFwcGluZ3MiOiI7Ozs7O0FBQ0EsK0RBQTREO0FBQzVELHlEQUFzRDtBQUN0RCwrREFBNEQ7QUFDNUQseURBQXNEO0FBQ3RELGtFQUFrRTtBQUNsRSxnRkFBd0Q7QUFDeEQsK0RBQTREO0FBRTVELE1BQXFCLGNBQWM7SUFPakMsWUFBWSxHQUFHO1FBQ2IsSUFBSSxDQUFDLEdBQUcsR0FBRyxHQUFHLENBQUM7UUFDZixJQUFJLENBQUMsaUJBQWlCLEdBQUcsSUFBSSxxQ0FBaUIsRUFBRSxDQUFDO1FBQ2pELElBQUksQ0FBQyxpQkFBaUIsR0FBRyxJQUFJLHFDQUFpQixFQUFFLENBQUM7UUFDakQsSUFBSSxDQUFDLGNBQWMsR0FBRyxJQUFJLCtCQUFjLENBQUMsSUFBSSxDQUFDLGlCQUFpQixFQUFFLElBQUksQ0FBQyxpQkFBaUIsQ0FBQyxDQUFDO1FBQ3pGLElBQUksQ0FBQyxjQUFjLEdBQUcsSUFBSSwrQkFBYyxDQUFDLElBQUksQ0FBQyxpQkFBaUIsQ0FBQyxDQUFDO1FBQ2pFLElBQUksQ0FBQyxHQUFHLENBQUMsMEJBQTBCLENBQUMsd0JBQXdCLENBQUMsMENBQXFCLEVBQUUsMkJBQWlCLENBQUMsV0FBVyxDQUFDLENBQUM7UUFDbkgsSUFBSSxDQUFDLEdBQUcsQ0FBQyxjQUFjLEdBQUcsSUFBSSxDQUFDLGNBQWMsQ0FBQztJQUNoRCxDQUFDO0lBRUQsY0FBYztRQUNaLElBQUksQ0FBQyxHQUFHLENBQUMseUJBQXlCLENBQUMsaUJBQWlCLENBQUMsSUFBSSxDQUFDLGNBQWMsQ0FBQyxDQUFDO1FBQzFFLElBQUksQ0FBQyxHQUFHLENBQUMsZ0JBQWdCLENBQUMsNkJBQTZCLENBQUMsMkJBQWlCLEVBQUUscUNBQWlCLENBQUMsWUFBWSxDQUFDLENBQUM7SUFDN0csQ0FBQztJQUVELGFBQWE7UUFDWCxNQUFNLE1BQU0sR0FBRyxJQUFJLENBQUMsR0FBRyxDQUFDLE1BQU0sQ0FBQyxHQUFHLENBQUM7UUFDbkMsSUFBSSxNQUFNLENBQUMsV0FBVyxFQUFFO1lBQ3RCLEtBQUssTUFBTSxVQUFVLElBQUksTUFBTSxDQUFDLFdBQVcsRUFBRTtnQkFDM0MsSUFBSSxDQUFDLGlCQUFpQixDQUFDLFNBQVMsQ0FBQyxVQUFVLENBQUMsQ0FBQzthQUM5QztTQUNGO2FBQU07WUFDTCxJQUFJLENBQUMsaUJBQWlCLENBQUMsZ0JBQWdCLENBQUMsTUFBTSxDQUFDLENBQUM7U0FDakQ7SUFDSCxDQUFDO0lBRUQsS0FBSyxDQUFDLE9BQU87UUFDWCxNQUFNLElBQUksQ0FBQyxHQUFHLENBQUMsYUFBYSxDQUFDLEtBQUssRUFBRSxDQUFDO1FBQ3JDLE1BQU0sSUFBSSxDQUFDLGNBQWMsQ0FBQyxRQUFRLEVBQUUsQ0FBQztJQUN2QyxDQUFDO0lBRUQsS0FBSyxDQUFDLFdBQVc7UUFDZixJQUFJLENBQUMsR0FBRyxDQUFDLHlCQUF5QixDQUFDLGVBQWUsQ0FBQyxJQUFJLENBQUMsY0FBYyxDQUFDLENBQUM7SUFDMUUsQ0FBQztDQUNGO0FBekNELGlDQXlDQyJ9

package/lib/ContextModeObject.d.ts

@@ -0,0 +1,20 @@
+import { EggContext, EggObject, EggObjectLifeCycleContext } from '@eggjs/tegg-runtime';
+import { EggPrototype } from '@eggjs/tegg-metadata';
+import { EggPrototypeName, EggObjectName } from '@eggjs/tegg';
+import { Id } from '@eggjs/tegg-lifecycle';
+import { Bone } from 'leoric';
+import ContextModelProto from './ContextModelProto';
+export declare class ContextModeObject implements EggObject {
+    private status;
+    id: Id;
+    readonly name: EggPrototypeName;
+    private _obj;
+    readonly proto: ContextModelProto;
+    readonly ctx: EggContext;
+    constructor(name: EggObjectName, proto: ContextModelProto, ctx: EggContext);
+    init(): Promise<void>;
+    injectProperty(): void;
+    get isReady(): boolean;
+    get obj(): typeof Bone;
+    static createObject(name: EggObjectName, proto: EggPrototype, _: EggObjectLifeCycleContext, ctx?: EggContext): Promise<ContextModeObject>;
+}

package/lib/ContextModeObject.js

@@ -0,0 +1,55 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ContextModeObject = void 0;
+const assert_1 = __importDefault(require("assert"));
+const tegg_runtime_1 = require("@eggjs/tegg-runtime");
+const tegg_lifecycle_1 = require("@eggjs/tegg-lifecycle");
+const egg_module_common_1 = require("@eggjs/egg-module-common");
+class ContextModeObject {
+    constructor(name, proto, ctx) {
+        this.status = tegg_runtime_1.EggObjectStatus.PENDING;
+        this.name = name;
+        this.proto = proto;
+        this.ctx = ctx;
+        this.id = tegg_lifecycle_1.IdenticalUtil.createObjectId(this.proto.id, this.ctx.id);
+    }
+    async init() {
+        const ctx = this.ctx;
+        const clazz = class ContextModelClass extends this.proto.model {
+            // eslint-disable-next-line @typescript-eslint/ban-ts-comment
+            // @ts-ignore
+            static get name() {
+                return super.name;
+            }
+            static get ctx() {
+                return ctx.get(egg_module_common_1.EGG_CONTEXT);
+            }
+            // custom setter always execute before define [CTX] when new Instance(super(opts) calling), if custom setter requires ctx, it should not be undefined
+            get ctx() {
+                return ctx.get(egg_module_common_1.EGG_CONTEXT);
+            }
+        };
+        this._obj = clazz;
+        this.status = tegg_runtime_1.EggObjectStatus.READY;
+    }
+    injectProperty() {
+        throw new Error('never call ModelObject#injectProperty');
+    }
+    get isReady() {
+        return this.status === tegg_runtime_1.EggObjectStatus.READY;
+    }
+    get obj() {
+        return this._obj;
+    }
+    static async createObject(name, proto, _, ctx) {
+        (0, assert_1.default)(ctx, 'ctx must be defined for ContextModelObject');
+        const modelObject = new ContextModeObject(name, proto, ctx);
+        await modelObject.init();
+        return modelObject;
+    }
+}
+exports.ContextModeObject = ContextModeObject;
+//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoiQ29udGV4dE1vZGVPYmplY3QuanMiLCJzb3VyY2VSb290IjoiIiwic291cmNlcyI6WyJDb250ZXh0TW9kZU9iamVjdC50cyJdLCJuYW1lcyI6W10sIm1hcHBpbmdzIjoiOzs7Ozs7QUFBQSxvREFBNEI7QUFDNUIsc0RBSzZCO0FBRzdCLDBEQUEwRDtBQUcxRCxnRUFBdUQ7QUFFdkQsTUFBYSxpQkFBaUI7SUFRNUIsWUFBWSxJQUFtQixFQUFFLEtBQXdCLEVBQUUsR0FBZTtRQVBsRSxXQUFNLEdBQW9CLDhCQUFlLENBQUMsT0FBTyxDQUFDO1FBUXhELElBQUksQ0FBQyxJQUFJLEdBQUcsSUFBSSxDQUFDO1FBQ2pCLElBQUksQ0FBQyxLQUFLLEdBQUcsS0FBSyxDQUFDO1FBQ25CLElBQUksQ0FBQyxHQUFHLEdBQUcsR0FBRyxDQUFDO1FBQ2YsSUFBSSxDQUFDLEVBQUUsR0FBRyw4QkFBYSxDQUFDLGNBQWMsQ0FBQyxJQUFJLENBQUMsS0FBSyxDQUFDLEVBQUUsRUFBRSxJQUFJLENBQUMsR0FBRyxDQUFDLEVBQUUsQ0FBQyxDQUFDO0lBQ3JFLENBQUM7SUFFRCxLQUFLLENBQUMsSUFBSTtRQUNSLE1BQU0sR0FBRyxHQUFHLElBQUksQ0FBQyxHQUFHLENBQUM7UUFDckIsTUFBTSxLQUFLLEdBQUcsTUFBTSxpQkFBa0IsU0FBUSxJQUFJLENBQUMsS0FBSyxDQUFDLEtBQUs7WUFDNUQsNkRBQTZEO1lBQzdELGFBQWE7WUFDYixNQUFNLEtBQUssSUFBSTtnQkFDYixPQUFPLEtBQUssQ0FBQyxJQUFJLENBQUM7WUFDcEIsQ0FBQztZQUVELE1BQU0sS0FBSyxHQUFHO2dCQUNaLE9BQU8sR0FBRyxDQUFDLEdBQUcsQ0FBQywrQkFBVyxDQUFDLENBQUM7WUFDOUIsQ0FBQztZQUVELHFKQUFxSjtZQUNySixJQUFJLEdBQUc7Z0JBQ0wsT0FBTyxHQUFHLENBQUMsR0FBRyxDQUFDLCtCQUFXLENBQUMsQ0FBQztZQUM5QixDQUFDO1NBQ0YsQ0FBQztRQUNGLElBQUksQ0FBQyxJQUFJLEdBQUcsS0FBSyxDQUFDO1FBQ2xCLElBQUksQ0FBQyxNQUFNLEdBQUcsOEJBQWUsQ0FBQyxLQUFLLENBQUM7SUFDdEMsQ0FBQztJQUVELGNBQWM7UUFDWixNQUFNLElBQUksS0FBSyxDQUFDLHVDQUF1QyxDQUFDLENBQUM7SUFDM0QsQ0FBQztJQUVELElBQUksT0FBTztRQUNULE9BQU8sSUFBSSxDQUFDLE1BQU0sS0FBSyw4QkFBZSxDQUFDLEtBQUssQ0FBQztJQUMvQyxDQUFDO0lBRUQsSUFBSSxHQUFHO1FBQ0wsT0FBTyxJQUFJLENBQUMsSUFBSSxDQUFDO0lBQ25CLENBQUM7SUFFRCxNQUFNLENBQUMsS0FBSyxDQUFDLFlBQVksQ0FBQyxJQUFtQixFQUFFLEtBQW1CLEVBQUUsQ0FBNEIsRUFBRSxHQUFnQjtRQUNoSCxJQUFBLGdCQUFNLEVBQUMsR0FBRyxFQUFFLDRDQUE0QyxDQUFDLENBQUM7UUFDMUQsTUFBTSxXQUFXLEdBQUcsSUFBSSxpQkFBaUIsQ0FBQyxJQUFJLEVBQUUsS0FBMEIsRUFBRSxHQUFHLENBQUMsQ0FBQztRQUNqRixNQUFNLFdBQVcsQ0FBQyxJQUFJLEVBQUUsQ0FBQztRQUN6QixPQUFPLFdBQVcsQ0FBQztJQUNyQixDQUFDO0NBQ0Y7QUF2REQsOENBdURDIn0=

package/lib/ContextModelProto.d.ts

@@ -0,0 +1,21 @@
+import { EggPrototype, LoadUnit, EggPrototypeLifecycleContext } from '@eggjs/tegg-metadata';
+import { AccessLevel, EggPrototypeName, ObjectInitType, QualifierInfo, MetaDataKey } from '@eggjs/tegg';
+import { Id } from '@eggjs/tegg-lifecycle';
+import { Bone } from 'leoric';
+export default class ContextModelProto implements EggPrototype {
+    private readonly qualifiers;
+    readonly accessLevel = AccessLevel.PUBLIC;
+    id: Id;
+    readonly initType = ObjectInitType.CONTEXT;
+    readonly injectObjects: never[];
+    readonly loadUnitId: string;
+    readonly moduleName: string;
+    readonly name: EggPrototypeName;
+    readonly model: typeof Bone;
+    constructor(loadUnit: LoadUnit, model: typeof Bone);
+    constructEggObject(): object;
+    getMetaData<T>(metadataKey: MetaDataKey): T | undefined;
+    verifyQualifier(qualifier: QualifierInfo): boolean;
+    verifyQualifiers(qualifiers: QualifierInfo[]): boolean;
+    static createProto(ctx: EggPrototypeLifecycleContext): ContextModelProto;
+}

package/lib/ContextModelProto.js

@@ -0,0 +1,40 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const tegg_1 = require("@eggjs/tegg");
+const tegg_lifecycle_1 = require("@eggjs/tegg-lifecycle");
+class ContextModelProto {
+    constructor(loadUnit, model) {
+        this.accessLevel = tegg_1.AccessLevel.PUBLIC;
+        this.initType = tegg_1.ObjectInitType.CONTEXT;
+        this.injectObjects = [];
+        this.model = model;
+        this.id = tegg_lifecycle_1.IdenticalUtil.createProtoId(loadUnit.id, `leoric:${model.name}`);
+        this.loadUnitId = loadUnit.id;
+        this.moduleName = loadUnit.name;
+        this.name = model.name;
+        this.qualifiers = tegg_1.QualifierUtil.getProtoQualifiers(model);
+    }
+    constructEggObject() {
+        return {};
+    }
+    getMetaData(metadataKey) {
+        return tegg_1.MetadataUtil.getMetaData(metadataKey, this.model);
+    }
+    verifyQualifier(qualifier) {
+        const selfQualifiers = this.qualifiers.find(t => t.attribute === qualifier.attribute);
+        return (selfQualifiers === null || selfQualifiers === void 0 ? void 0 : selfQualifiers.value) === qualifier.value;
+    }
+    verifyQualifiers(qualifiers) {
+        for (const qualifier of qualifiers) {
+            if (!this.verifyQualifier(qualifier)) {
+                return false;
+            }
+        }
+        return true;
+    }
+    static createProto(ctx) {
+        return new ContextModelProto(ctx.loadUnit, ctx.clazz);
+    }
+}
+exports.default = ContextModelProto;
+//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoiQ29udGV4dE1vZGVsUHJvdG8uanMiLCJzb3VyY2VSb290IjoiIiwic291cmNlcyI6WyJDb250ZXh0TW9kZWxQcm90by50cyJdLCJuYW1lcyI6W10sIm1hcHBpbmdzIjoiOztBQUNBLHNDQVFxQjtBQUNyQiwwREFBMEQ7QUFHMUQsTUFBcUIsaUJBQWlCO0lBV3BDLFlBQVksUUFBa0IsRUFBRSxLQUFrQjtRQVR6QyxnQkFBVyxHQUFHLGtCQUFXLENBQUMsTUFBTSxDQUFDO1FBRWpDLGFBQVEsR0FBRyxxQkFBYyxDQUFDLE9BQU8sQ0FBQztRQUNsQyxrQkFBYSxHQUFHLEVBQUUsQ0FBQztRQU8xQixJQUFJLENBQUMsS0FBSyxHQUFHLEtBQUssQ0FBQztRQUNuQixJQUFJLENBQUMsRUFBRSxHQUFHLDhCQUFhLENBQUMsYUFBYSxDQUFDLFFBQVEsQ0FBQyxFQUFFLEVBQUUsVUFBVSxLQUFLLENBQUMsSUFBSSxFQUFFLENBQUMsQ0FBQztRQUMzRSxJQUFJLENBQUMsVUFBVSxHQUFHLFFBQVEsQ0FBQyxFQUFFLENBQUM7UUFDOUIsSUFBSSxDQUFDLFVBQVUsR0FBRyxRQUFRLENBQUMsSUFBSSxDQUFDO1FBQ2hDLElBQUksQ0FBQyxJQUFJLEdBQUcsS0FBSyxDQUFDLElBQUksQ0FBQztRQUN2QixJQUFJLENBQUMsVUFBVSxHQUFHLG9CQUFhLENBQUMsa0JBQWtCLENBQUMsS0FBSyxDQUFDLENBQUM7SUFDNUQsQ0FBQztJQUVELGtCQUFrQjtRQUNoQixPQUFPLEVBQUUsQ0FBQztJQUNaLENBQUM7SUFFRCxXQUFXLENBQUksV0FBd0I7UUFDckMsT0FBTyxtQkFBWSxDQUFDLFdBQVcsQ0FBQyxXQUFXLEVBQUUsSUFBSSxDQUFDLEtBQUssQ0FBQyxDQUFDO0lBQzNELENBQUM7SUFFRCxlQUFlLENBQUMsU0FBd0I7UUFDdEMsTUFBTSxjQUFjLEdBQUcsSUFBSSxDQUFDLFVBQVUsQ0FBQyxJQUFJLENBQUMsQ0FBQyxDQUFDLEVBQUUsQ0FBQyxDQUFDLENBQUMsU0FBUyxLQUFLLFNBQVMsQ0FBQyxTQUFTLENBQUMsQ0FBQztRQUN0RixPQUFPLENBQUEsY0FBYyxhQUFkLGNBQWMsdUJBQWQsY0FBYyxDQUFFLEtBQUssTUFBSyxTQUFTLENBQUMsS0FBSyxDQUFDO0lBQ25ELENBQUM7SUFFRCxnQkFBZ0IsQ0FBQyxVQUEyQjtRQUMxQyxLQUFLLE1BQU0sU0FBUyxJQUFJLFVBQVUsRUFBRTtZQUNsQyxJQUFJLENBQUMsSUFBSSxDQUFDLGVBQWUsQ0FBQyxTQUFTLENBQUMsRUFBRTtnQkFDcEMsT0FBTyxLQUFLLENBQUM7YUFDZDtTQUNGO1FBQ0QsT0FBTyxJQUFJLENBQUM7SUFDZCxDQUFDO0lBRUQsTUFBTSxDQUFDLFdBQVcsQ0FBQyxHQUFpQztRQUNsRCxPQUFPLElBQUksaUJBQWlCLENBQUMsR0FBRyxDQUFDLFFBQVEsRUFBRSxHQUFHLENBQUMsS0FBb0IsQ0FBQyxDQUFDO0lBQ3ZFLENBQUM7Q0FDRjtBQTdDRCxvQ0E2Q0MifQ==

package/lib/LeoricRegister.d.ts

@@ -1,10 +1,11 @@
+import Base from 'sdk-base';
 import { ModelProtoManager } from './ModelProtoManager';
 import { DataSourceManager } from './DataSourceManager';
 import { ModelMetadata } from '@eggjs/tegg-orm-decorator';
-export declare class LeoricRegister {
+export declare class LeoricRegister extends Base {
     private readonly modelProtoManager;
     private readonly dataSourceManager;
-    private readonly realmMap;
+    readonly realmMap: Map<string, any>;
     constructor(modelProtoManager: ModelProtoManager, dataSourceManager: DataSourceManager);
     getOrCreateRealm(datasource: string | undefined): any;
     generateLeoricAttributes(metadata: ModelMetadata): {};

package/lib/LeoricRegister.js

@@ -4,10 +4,13 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
 };
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.LeoricRegister = void 0;
+const sdk_base_1 = __importDefault(require("sdk-base"));
 const leoric_1 = __importDefault(require("leoric"));
+const setup_hooks_1 = require("leoric/src/setup_hooks");
 const tegg_orm_decorator_1 = require("@eggjs/tegg-orm-decorator");
-class LeoricRegister {
+class LeoricRegister extends sdk_base_1.default {
     constructor(modelProtoManager, dataSourceManager) {
+        super();
         this.modelProtoManager = modelProtoManager;
         this.dataSourceManager = dataSourceManager;
         this.realmMap = new Map();
@@ -47,20 +50,28 @@ class LeoricRegister {
     }
     async register() {
         for (const { proto, clazz } of this.modelProtoManager.getProtos()) {
-            const metadata = tegg_orm_decorator_1.ModelMetadataUtil.getControllerMetadata(clazz);
+            const metadata = tegg_orm_decorator_1.ModelMetadataUtil.getModelMetadata(clazz);
             if (!metadata)
                 throw new Error(`not found metadata for model ${proto.id}`);
             const realm = this.getOrCreateRealm(metadata.dataSource);
             realm.models[clazz.name] = clazz;
             realm[clazz.name] = clazz;
             const attributes = this.generateLeoricAttributes(metadata);
+            const hooks = {};
+            for (const hookName of setup_hooks_1.hookNames) {
+                if (clazz[hookName]) {
+                    hooks[hookName] = clazz[hookName];
+                }
+            }
             clazz.init(attributes, {
                 tableName: metadata.tableName,
+                hooks,
             }, {});
         }
         await Promise.all(Array.from(this.realmMap.values())
             .map(realm => realm.connect()));
+        this.ready(true);
     }
 }
 exports.LeoricRegister = LeoricRegister;
-//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoiTGVvcmljUmVnaXN0ZXIuanMiLCJzb3VyY2VSb290IjoiIiwic291cmNlcyI6WyJMZW9yaWNSZWdpc3Rlci50cyJdLCJuYW1lcyI6W10sIm1hcHBpbmdzIjoiOzs7Ozs7QUFFQSxvREFBMkI7QUFDM0Isa0VBQTZFO0FBRTdFLE1BQWEsY0FBYztJQUt6QixZQUFZLGlCQUFvQyxFQUFFLGlCQUFvQztRQUNwRixJQUFJLENBQUMsaUJBQWlCLEdBQUcsaUJBQWlCLENBQUM7UUFDM0MsSUFBSSxDQUFDLGlCQUFpQixHQUFHLGlCQUFpQixDQUFDO1FBQzNDLElBQUksQ0FBQyxRQUFRLEdBQUcsSUFBSSxHQUFHLEVBQUUsQ0FBQztJQUM1QixDQUFDO0lBRUQsZ0JBQWdCLENBQUMsVUFBOEI7UUFDN0MsSUFBSSxNQUE2QixDQUFDO1FBQ2xDLElBQUksQ0FBQyxVQUFVLEVBQUU7WUFDZixNQUFNLEdBQUcsSUFBSSxDQUFDLGlCQUFpQixDQUFDLGdCQUFnQixFQUFFLENBQUM7U0FDcEQ7YUFBTTtZQUNMLE1BQU0sR0FBRyxJQUFJLENBQUMsaUJBQWlCLENBQUMsU0FBUyxDQUFDLFVBQVUsQ0FBQyxDQUFDO1NBQ3ZEO1FBQ0QsSUFBSSxDQUFDLE1BQU0sRUFBRTtZQUNYLE1BQU0sSUFBSSxLQUFLLENBQUMsNEJBQTRCLFVBQVUsRUFBRSxDQUFDLENBQUM7U0FDM0Q7UUFDRCxJQUFJLEtBQUssR0FBRyxJQUFJLENBQUMsUUFBUSxDQUFDLEdBQUcsQ0FBQyxNQUFNLENBQUMsUUFBUSxDQUFDLENBQUM7UUFDL0MsSUFBSSxLQUFLLEVBQUU7WUFDVCxPQUFPLEtBQUssQ0FBQztTQUNkO1FBQ0QsS0FBSyxHQUFHLElBQUssZ0JBQWEsQ0FBQyxFQUFFLEdBQUcsTUFBTSxFQUFFLENBQUMsQ0FBQztRQUMxQyxJQUFJLENBQUMsUUFBUSxDQUFDLEdBQUcsQ0FBQyxNQUFNLENBQUMsUUFBUSxFQUFFLEtBQUssQ0FBQyxDQUFDO1FBQzFDLE9BQU8sS0FBSyxDQUFDO0lBQ2YsQ0FBQztJQUVELHdCQUF3QixDQUFDLFFBQXVCO1FBQzlDLE1BQU0sVUFBVSxHQUFHLEVBQUUsQ0FBQztRQUN0QixLQUFLLE1BQU0sU0FBUyxJQUFJLFFBQVEsQ0FBQyxVQUFVLEVBQUU7WUFDM0MsVUFBVSxDQUFDLFNBQVMsQ0FBQyxZQUFZLENBQUMsR0FBRztnQkFDbkMsVUFBVSxFQUFFLFNBQVMsQ0FBQyxhQUFhO2dCQUNuQyxJQUFJLEVBQUUsU0FBUyxDQUFDLFFBQVE7Z0JBQ3hCLFNBQVMsRUFBRSxTQUFTLENBQUMsU0FBUztnQkFDOUIsVUFBVSxFQUFFLFNBQVMsQ0FBQyxPQUFPO2dCQUM3QixNQUFNLEVBQUUsU0FBUyxDQUFDLE1BQU07Z0JBQ3hCLGFBQWEsRUFBRSxTQUFTLENBQUMsYUFBYTthQUN2QyxDQUFDO1NBQ0g7UUFDRCxPQUFPLFVBQVUsQ0FBQztJQUNwQixDQUFDO0lBRUQsS0FBSyxDQUFDLFFBQVE7UUFDWixLQUFLLE1BQU0sRUFBRSxLQUFLLEVBQUUsS0FBSyxFQUFFLElBQUksSUFBSSxDQUFDLGlCQUFpQixDQUFDLFNBQVMsRUFBRSxFQUFFO1lBQ2pFLE1BQU0sUUFBUSxHQUFHLHNDQUFpQixDQUFDLHFCQUFxQixDQUFDLEtBQUssQ0FBQyxDQUFDO1lBQ2hFLElBQUksQ0FBQyxRQUFRO2dCQUFFLE1BQU0sSUFBSSxLQUFLLENBQUMsZ0NBQWdDLEtBQUssQ0FBQyxFQUFFLEVBQUUsQ0FBQyxDQUFDO1lBQzNFLE1BQU0sS0FBSyxHQUFHLElBQUksQ0FBQyxnQkFBZ0IsQ0FBQyxRQUFRLENBQUMsVUFBVSxDQUFDLENBQUM7WUFDekQsS0FBSyxDQUFDLE1BQU0sQ0FBQyxLQUFLLENBQUMsSUFBSSxDQUFDLEdBQUcsS0FBSyxDQUFDO1lBQ2pDLEtBQUssQ0FBQyxLQUFLLENBQUMsSUFBSSxDQUFDLEdBQUcsS0FBSyxDQUFDO1lBQzFCLE1BQU0sVUFBVSxHQUFHLElBQUksQ0FBQyx3QkFBd0IsQ0FBQyxRQUFRLENBQUMsQ0FBQztZQUMxRCxLQUFhLENBQUMsSUFBSSxDQUFDLFVBQVUsRUFBRTtnQkFDOUIsU0FBUyxFQUFFLFFBQVEsQ0FBQyxTQUFTO2FBQzlCLEVBQUUsRUFBRSxDQUFDLENBQUM7U0FDUjtRQUNELE1BQU0sT0FBTyxDQUFDLEdBQUcsQ0FBQyxLQUFLLENBQUMsSUFBSSxDQUFDLElBQUksQ0FBQyxRQUFRLENBQUMsTUFBTSxFQUFFLENBQUM7YUFDakQsR0FBRyxDQUFDLEtBQUssQ0FBQyxFQUFFLENBQUMsS0FBSyxDQUFDLE9BQU8sRUFBRSxDQUFDLENBQUMsQ0FBQztJQUNwQyxDQUFDO0NBQ0Y7QUE1REQsd0NBNERDIn0=
+//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoiTGVvcmljUmVnaXN0ZXIuanMiLCJzb3VyY2VSb290IjoiIiwic291cmNlcyI6WyJMZW9yaWNSZWdpc3Rlci50cyJdLCJuYW1lcyI6W10sIm1hcHBpbmdzIjoiOzs7Ozs7QUFBQSx3REFBNEI7QUFHNUIsb0RBQTJCO0FBQzNCLHdEQUFtRDtBQUNuRCxrRUFBNkU7QUFFN0UsTUFBYSxjQUFlLFNBQVEsa0JBQUk7SUFLdEMsWUFBWSxpQkFBb0MsRUFBRSxpQkFBb0M7UUFDcEYsS0FBSyxFQUFFLENBQUM7UUFDUixJQUFJLENBQUMsaUJBQWlCLEdBQUcsaUJBQWlCLENBQUM7UUFDM0MsSUFBSSxDQUFDLGlCQUFpQixHQUFHLGlCQUFpQixDQUFDO1FBQzNDLElBQUksQ0FBQyxRQUFRLEdBQUcsSUFBSSxHQUFHLEVBQUUsQ0FBQztJQUM1QixDQUFDO0lBRUQsZ0JBQWdCLENBQUMsVUFBOEI7UUFDN0MsSUFBSSxNQUE2QixDQUFDO1FBQ2xDLElBQUksQ0FBQyxVQUFVLEVBQUU7WUFDZixNQUFNLEdBQUcsSUFBSSxDQUFDLGlCQUFpQixDQUFDLGdCQUFnQixFQUFFLENBQUM7U0FDcEQ7YUFBTTtZQUNMLE1BQU0sR0FBRyxJQUFJLENBQUMsaUJBQWlCLENBQUMsU0FBUyxDQUFDLFVBQVUsQ0FBQyxDQUFDO1NBQ3ZEO1FBQ0QsSUFBSSxDQUFDLE1BQU0sRUFBRTtZQUNYLE1BQU0sSUFBSSxLQUFLLENBQUMsNEJBQTRCLFVBQVUsRUFBRSxDQUFDLENBQUM7U0FDM0Q7UUFDRCxJQUFJLEtBQUssR0FBRyxJQUFJLENBQUMsUUFBUSxDQUFDLEdBQUcsQ0FBQyxNQUFNLENBQUMsUUFBUSxDQUFDLENBQUM7UUFDL0MsSUFBSSxLQUFLLEVBQUU7WUFDVCxPQUFPLEtBQUssQ0FBQztTQUNkO1FBQ0QsS0FBSyxHQUFHLElBQUssZ0JBQWEsQ0FBQyxFQUFFLEdBQUcsTUFBTSxFQUFFLENBQUMsQ0FBQztRQUMxQyxJQUFJLENBQUMsUUFBUSxDQUFDLEdBQUcsQ0FBQyxNQUFNLENBQUMsUUFBUSxFQUFFLEtBQUssQ0FBQyxDQUFDO1FBQzFDLE9BQU8sS0FBSyxDQUFDO0lBQ2YsQ0FBQztJQUVELHdCQUF3QixDQUFDLFFBQXVCO1FBQzlDLE1BQU0sVUFBVSxHQUFHLEVBQUUsQ0FBQztRQUN0QixLQUFLLE1BQU0sU0FBUyxJQUFJLFFBQVEsQ0FBQyxVQUFVLEVBQUU7WUFDM0MsVUFBVSxDQUFDLFNBQVMsQ0FBQyxZQUFZLENBQUMsR0FBRztnQkFDbkMsVUFBVSxFQUFFLFNBQVMsQ0FBQyxhQUFhO2dCQUNuQyxJQUFJLEVBQUUsU0FBUyxDQUFDLFFBQVE7Z0JBQ3hCLFNBQVMsRUFBRSxTQUFTLENBQUMsU0FBUztnQkFDOUIsVUFBVSxFQUFFLFNBQVMsQ0FBQyxPQUFPO2dCQUM3QixNQUFNLEVBQUUsU0FBUyxDQUFDLE1BQU07Z0JBQ3hCLGFBQWEsRUFBRSxTQUFTLENBQUMsYUFBYTthQUN2QyxDQUFDO1NBQ0g7UUFDRCxPQUFPLFVBQVUsQ0FBQztJQUNwQixDQUFDO0lBRUQsS0FBSyxDQUFDLFFBQVE7UUFDWixLQUFLLE1BQU0sRUFBRSxLQUFLLEVBQUUsS0FBSyxFQUFFLElBQUksSUFBSSxDQUFDLGlCQUFpQixDQUFDLFNBQVMsRUFBRSxFQUFFO1lBQ2pFLE1BQU0sUUFBUSxHQUFHLHNDQUFpQixDQUFDLGdCQUFnQixDQUFDLEtBQUssQ0FBQyxDQUFDO1lBQzNELElBQUksQ0FBQyxRQUFRO2dCQUFFLE1BQU0sSUFBSSxLQUFLLENBQUMsZ0NBQWdDLEtBQUssQ0FBQyxFQUFFLEVBQUUsQ0FBQyxDQUFDO1lBQzNFLE1BQU0sS0FBSyxHQUFHLElBQUksQ0FBQyxnQkFBZ0IsQ0FBQyxRQUFRLENBQUMsVUFBVSxDQUFDLENBQUM7WUFDekQsS0FBSyxDQUFDLE1BQU0sQ0FBQyxLQUFLLENBQUMsSUFBSSxDQUFDLEdBQUcsS0FBSyxDQUFDO1lBQ2pDLEtBQUssQ0FBQyxLQUFLLENBQUMsSUFBSSxDQUFDLEdBQUcsS0FBSyxDQUFDO1lBQzFCLE1BQU0sVUFBVSxHQUFHLElBQUksQ0FBQyx3QkFBd0IsQ0FBQyxRQUFRLENBQUMsQ0FBQztZQUMzRCxNQUFNLEtBQUssR0FBRyxFQUFFLENBQUM7WUFDakIsS0FBSyxNQUFNLFFBQVEsSUFBSSx1QkFBUyxFQUFFO2dCQUNoQyxJQUFJLEtBQUssQ0FBQyxRQUFRLENBQUMsRUFBRTtvQkFDbkIsS0FBSyxDQUFDLFFBQVEsQ0FBQyxHQUFHLEtBQUssQ0FBQyxRQUFRLENBQUMsQ0FBQztpQkFDbkM7YUFDRjtZQUVBLEtBQWEsQ0FBQyxJQUFJLENBQUMsVUFBVSxFQUFFO2dCQUM5QixTQUFTLEVBQUUsUUFBUSxDQUFDLFNBQVM7Z0JBQzdCLEtBQUs7YUFDTixFQUFFLEVBQUUsQ0FBQyxDQUFDO1NBQ1I7UUFDRCxNQUFNLE9BQU8sQ0FBQyxHQUFHLENBQUMsS0FBSyxDQUFDLElBQUksQ0FBQyxJQUFJLENBQUMsUUFBUSxDQUFDLE1BQU0sRUFBRSxDQUFDO2FBQ2pELEdBQUcsQ0FBQyxLQUFLLENBQUMsRUFBRSxDQUFDLEtBQUssQ0FBQyxPQUFPLEVBQUUsQ0FBQyxDQUFDLENBQUM7UUFDbEMsSUFBSSxDQUFDLEtBQUssQ0FBQyxJQUFJLENBQUMsQ0FBQztJQUNuQixDQUFDO0NBQ0Y7QUF0RUQsd0NBc0VDIn0=

package/lib/ModelProtoHook.js

@@ -12,9 +12,9 @@ class ModelProtoHook {
         }
         const builder = new tegg_orm_decorator_1.ModelMetaBuilder(ctx.clazz);
         const metadata = builder.build();
-        tegg_orm_decorator_1.ModelMetadataUtil.setControllerMetadata(ctx.clazz, metadata);
+        tegg_orm_decorator_1.ModelMetadataUtil.setModelMetadata(ctx.clazz, metadata);
         this.modelProtoManager.addProto(ctx.clazz, obj);
     }
 }
 exports.ModelProtoHook = ModelProtoHook;
-//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoiTW9kZWxQcm90b0hvb2suanMiLCJzb3VyY2VSb290IjoiIiwic291cmNlcyI6WyJNb2RlbFByb3RvSG9vay50cyJdLCJuYW1lcyI6W10sIm1hcHBpbmdzIjoiOzs7QUFFQSxrRUFBMEY7QUFHMUYsTUFBYSxjQUFjO0lBR3pCLFlBQVksaUJBQW9DO1FBQzlDLElBQUksQ0FBQyxpQkFBaUIsR0FBRyxpQkFBaUIsQ0FBQztJQUM3QyxDQUFDO0lBRUQsS0FBSyxDQUFDLFVBQVUsQ0FBQyxHQUFpQyxFQUFFLEdBQWlCO1FBQ25FLElBQUksQ0FBQyxHQUFHLENBQUMsV0FBVyxDQUFDLDZCQUFRLENBQUMsRUFBRTtZQUM5QixPQUFPO1NBQ1I7UUFDRCxNQUFNLE9BQU8sR0FBRyxJQUFJLHFDQUFnQixDQUFDLEdBQUcsQ0FBQyxLQUFLLENBQUMsQ0FBQztRQUNoRCxNQUFNLFFBQVEsR0FBRyxPQUFPLENBQUMsS0FBSyxFQUFFLENBQUM7UUFDakMsc0NBQWlCLENBQUMscUJBQXFCLENBQUMsR0FBRyxDQUFDLEtBQUssRUFBRSxRQUFRLENBQUMsQ0FBQztRQUM3RCxJQUFJLENBQUMsaUJBQWlCLENBQUMsUUFBUSxDQUFDLEdBQUcsQ0FBQyxLQUFLLEVBQUUsR0FBRyxDQUFDLENBQUM7SUFDbEQsQ0FBQztDQUNGO0FBaEJELHdDQWdCQyJ9
+//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoiTW9kZWxQcm90b0hvb2suanMiLCJzb3VyY2VSb290IjoiIiwic291cmNlcyI6WyJNb2RlbFByb3RvSG9vay50cyJdLCJuYW1lcyI6W10sIm1hcHBpbmdzIjoiOzs7QUFFQSxrRUFBMEY7QUFHMUYsTUFBYSxjQUFjO0lBR3pCLFlBQVksaUJBQW9DO1FBQzlDLElBQUksQ0FBQyxpQkFBaUIsR0FBRyxpQkFBaUIsQ0FBQztJQUM3QyxDQUFDO0lBRUQsS0FBSyxDQUFDLFVBQVUsQ0FBQyxHQUFpQyxFQUFFLEdBQWlCO1FBQ25FLElBQUksQ0FBQyxHQUFHLENBQUMsV0FBVyxDQUFDLDZCQUFRLENBQUMsRUFBRTtZQUM5QixPQUFPO1NBQ1I7UUFDRCxNQUFNLE9BQU8sR0FBRyxJQUFJLHFDQUFnQixDQUFDLEdBQUcsQ0FBQyxLQUFLLENBQUMsQ0FBQztRQUNoRCxNQUFNLFFBQVEsR0FBRyxPQUFPLENBQUMsS0FBSyxFQUFFLENBQUM7UUFDakMsc0NBQWlCLENBQUMsZ0JBQWdCLENBQUMsR0FBRyxDQUFDLEtBQUssRUFBRSxRQUFRLENBQUMsQ0FBQztRQUN4RCxJQUFJLENBQUMsaUJBQWlCLENBQUMsUUFBUSxDQUFDLEdBQUcsQ0FBQyxLQUFLLEVBQUUsR0FBRyxDQUFDLENBQUM7SUFDbEQsQ0FBQztDQUNGO0FBaEJELHdDQWdCQyJ9

package/package.json

@@ -6,7 +6,7 @@
       "tegg"
     ]
   },
-  "version": "1.2.0",
+  "version": "2.1.0",
   "description": "orm decorator for egg",
   "keywords": [
     "egg",
@@ -49,21 +49,22 @@
   },
   "dependencies": {
     "@eggjs/egg-module-common": "^1.0.0",
-    "@eggjs/tegg": "^1.2.0",
+    "@eggjs/tegg": "^1.3.2",
     "@eggjs/tegg-common-util": "^1.1.1",
-    "@eggjs/tegg-loader": "^1.1.1",
-    "@eggjs/tegg-metadata": "^1.1.1",
-    "@eggjs/tegg-orm-decorator": "^1.1.1",
-    "@eggjs/tegg-runtime": "^1.2.0",
+    "@eggjs/tegg-lifecycle": "^1.0.0",
+    "@eggjs/tegg-loader": "^1.3.0",
+    "@eggjs/tegg-metadata": "^1.3.0",
+    "@eggjs/tegg-orm-decorator": "^1.4.0",
+    "@eggjs/tegg-runtime": "^1.3.0",
     "@types/koa-router": "^7.0.40",
     "koa-compose": "^3.2.1",
-    "leoric": "^1.6.7"
+    "leoric": "^2.6.1"
   },
   "devDependencies": {
-    "@eggjs/module-test-util": "^1.2.0",
+    "@eggjs/module-test-util": "^1.3.0",
     "@eggjs/router": "^2.0.0",
     "@eggjs/tegg-config": "^1.1.1",
-    "@eggjs/tegg-plugin": "^1.2.0",
+    "@eggjs/tegg-plugin": "^1.3.2",
     "egg": "^2.26.0",
     "egg-mock": "^3.25.1",
     "egg-tracer": "^1.1.0",
@@ -73,5 +74,5 @@
   "publishConfig": {
     "access": "public"
   },
-  "gitHead": "8e2a7384017df27fb28362defec21cd40e628bdb"
+  "gitHead": "f43f1f05dd4d89cc6ff890b667d220e837b59679"
 }

package/typings/index.d.ts

@@ -3,7 +3,17 @@ import '@eggjs/tegg-plugin';
 
 import { DataType } from 'leoric';
 import { AttributeOptions } from '@eggjs/tegg-orm-decorator';
+import { LeoricRegister } from '../lib/LeoricRegister';
 
 declare module '@eggjs/tegg-orm-decorator' {
   export declare function Attribute(dataType: DataType, options?: AttributeOptions): (target: any, propertyKey: PropertyKey) => void;
 }
+
+declare module 'egg' {
+  export interface TeggOrmApplication {
+    leoricRegister: LeoricRegister;
+  }
+
+  interface Application extends TeggOrmApplication {
+  }
+}