import type { KobpCustomization, KobpModule } from 'kobp'

import { MikroORM, MikroORMOptions, RequestContext } from '@mikro-orm/core'

import { createDI, DI } from '../di'

export class MikroormModule implements KobpModule {
  constructor(protected initOrmOrConfig: MikroORMOptions | (() => Promise<MikroORM>)) {}

  public customization(): KobpCustomization {
    return {
      onInit: async () => {
        const orm =
          typeof this.initOrmOrConfig === 'function'
            ? await this.initOrmOrConfig()
            : await MikroORM.init(this.initOrmOrConfig)
        createDI(orm)
      },
      middlewares: async (app) => {
        app.use((ctx, next) => RequestContext.create(DI.orm.em, next))
        app.use(async (ctx, next) => {
          DI.em = RequestContext.getEntityManager()
          await next()
        })
      },
    }
  }
}