bigcapital/packages/server/src/modules/System/SystemModels/SystemModels.module.ts

import { Knex } from 'knex';
import { Model } from 'objection';
import { Global, Module } from '@nestjs/common';
import { PlanSubscription } from '@/modules/Subscription/models/PlanSubscription';
import { TenantModel } from '@/modules/System/models/TenantModel';
import { SystemKnexConnection } from '../SystemDB/SystemDB.constants';
import { SystemModelsConnection } from './SystemModels.constants';
import { SystemUser } from '../models/SystemUser';
import { TenantMetadata } from '../models/TenantMetadataModel';
import { TenantRepository } from '../repositories/Tenant.repository';

const models = [SystemUser, PlanSubscription, TenantModel, TenantMetadata];

const modelProviders = models.map((model) => {
  return {
    provide: model.name,
    useValue: model,
  };
});

export const InjectSystemModel = (model: typeof Model) => ({
  useValue: model,
  provide: model.name,
});

const providers = [
  ...modelProviders,
  {
    provide: SystemModelsConnection,
    inject: [SystemKnexConnection],
    useFactory: async (systemKnex: Knex) => {
      Model.knex(systemKnex);
    },
  },
];

@Global()
@Module({
  providers: [...providers, TenantRepository],
  exports: [...providers, TenantRepository],
})
export class SystemModelsModule {}