import { setTimeout } from 'node:timers/promises'; import { Global, Inject, Module } from '@nestjs/common'; import * as Redis from 'ioredis'; import { DataSource } from 'typeorm'; import { MeiliSearch } from 'meilisearch'; import { DI } from './di-symbols.js'; import { Config, loadConfig } from './config.js'; import { createPostgresDataSource } from './postgres.js'; import { RepositoryModule } from './models/RepositoryModule.js'; import type { Provider, OnApplicationShutdown } from '@nestjs/common'; const $config: Provider = { provide: DI.config, useValue: loadConfig(), }; const $db: Provider = { provide: DI.db, useFactory: async (config) => { const db = createPostgresDataSource(config); return await db.initialize(); }, inject: [DI.config], }; const $meilisearch: Provider = { provide: DI.meilisearch, useFactory: (config: Config) => { if (config.meilisearch) { return new MeiliSearch({ host: `${config.meilisearch.ssl ? 'https' : 'http' }://${config.meilisearch.host}:${config.meilisearch.port}`, apiKey: config.meilisearch.apiKey, }); } else { return null; } }, inject: [DI.config], }; const $redis: Provider = { provide: DI.redis, useFactory: (config: Config) => { return new Redis.Redis({ port: config.redis.port, host: config.redis.host, family: config.redis.family == null ? 0 : config.redis.family, password: config.redis.pass, keyPrefix: `${config.redis.prefix}:`, db: config.redis.db ?? 0, }); }, inject: [DI.config], }; const $redisForPub: Provider = { provide: DI.redisForPub, useFactory: (config: Config) => { const redis = new Redis.Redis({ port: config.redisForPubsub.port, host: config.redisForPubsub.host, family: config.redisForPubsub.family == null ? 0 : config.redisForPubsub.family, password: config.redisForPubsub.pass, keyPrefix: `${config.redisForPubsub.prefix}:`, db: config.redisForPubsub.db ?? 0, }); return redis; }, inject: [DI.config], }; const $redisForSub: Provider = { provide: DI.redisForSub, useFactory: (config: Config) => { const redis = new Redis.Redis({ port: config.redisForPubsub.port, host: config.redisForPubsub.host, family: config.redisForPubsub.family == null ? 0 : config.redisForPubsub.family, password: config.redisForPubsub.pass, keyPrefix: `${config.redisForPubsub.prefix}:`, db: config.redisForPubsub.db ?? 0, }); redis.subscribe(config.host); return redis; }, inject: [DI.config], }; @Global() @Module({ imports: [RepositoryModule], providers: [$config, $db, $meilisearch, $redis, $redisForPub, $redisForSub], exports: [$config, $db, $meilisearch, $redis, $redisForPub, $redisForSub, RepositoryModule], }) export class GlobalModule implements OnApplicationShutdown { constructor( @Inject(DI.db) private db: DataSource, @Inject(DI.redis) private redisClient: Redis.Redis, @Inject(DI.redisForPub) private redisForPub: Redis.Redis, @Inject(DI.redisForSub) private redisForSub: Redis.Redis, ) {} async onApplicationShutdown(signal: string): Promise { if (process.env.NODE_ENV === 'test') { // XXX: // Shutting down the existing connections causes errors on Jest as // Misskey has asynchronous postgres/redis connections that are not // awaited. // Let's wait for some random time for them to finish. await setTimeout(5000); } await Promise.all([ this.db.destroy(), this.redisClient.disconnect(), this.redisForPub.disconnect(), this.redisForSub.disconnect(), ]); } }