import { INestApplication, Provider, Type } from '@nestjs/common'

type GetApp = () => INestApplication
// eslint-disable-next-line @typescript-eslint/no-unsafe-function-type
type TypeOrToken<T> = string | symbol | Function | Type<T>

export class TestModuleRef {
    static Provider(getApp: GetApp): Provider {
        return {
            provide: TestModuleRef,
            useFactory: () => new TestModuleRef(getApp),
        }
    }

    constructor(protected getApp: GetApp) {}

    public async resolve<T>(typeOrToken: TypeOrToken<T>): Promise<T> {
        return this.getApp()?.resolve(typeOrToken)
    }

    public get<T>(typeOrToken: TypeOrToken<T>): T {
        return this.getApp().get(typeOrToken)
    }
}