|
| 1 | +import { DynamicModule, Global, Module, Provider } from '@nestjs/common' |
| 2 | +import { ImapflowModuleAsyncOptions, ImapflowModuleOptions, ImapflowModuleOptionsFactory } from './imapflow.interfaces' |
| 3 | +import { createImapflowConnection, getImapflowConnectionToken, getImapflowOptionsToken } from './imapflow.utils' |
| 4 | + |
| 5 | +@Global() |
| 6 | +@Module({}) |
| 7 | +export class ImapflowCoreModule { |
| 8 | + public static forRoot(options: ImapflowModuleOptions, connection?: string): DynamicModule { |
| 9 | + const imapflowOptionsProvider: Provider = { |
| 10 | + provide: getImapflowOptionsToken(connection), |
| 11 | + useValue: options, |
| 12 | + } |
| 13 | + |
| 14 | + const imapflowConnectionProvider: Provider = { |
| 15 | + provide: getImapflowConnectionToken(connection), |
| 16 | + useValue: createImapflowConnection(options), |
| 17 | + } |
| 18 | + |
| 19 | + return { |
| 20 | + module: ImapflowCoreModule, |
| 21 | + providers: [imapflowOptionsProvider, imapflowConnectionProvider], |
| 22 | + exports: [imapflowOptionsProvider, imapflowConnectionProvider], |
| 23 | + } |
| 24 | + } |
| 25 | + |
| 26 | + public static forRootAsync(options: ImapflowModuleAsyncOptions, connection: string): DynamicModule { |
| 27 | + const imapflowConnectionProvider: Provider = { |
| 28 | + provide: getImapflowConnectionToken(connection), |
| 29 | + useFactory(options: ImapflowModuleOptions) { |
| 30 | + return createImapflowConnection(options) |
| 31 | + }, |
| 32 | + inject: [getImapflowOptionsToken(connection)], |
| 33 | + } |
| 34 | + |
| 35 | + return { |
| 36 | + module: ImapflowCoreModule, |
| 37 | + imports: options.imports, |
| 38 | + providers: [...this.createAsyncProviders(options, connection), imapflowConnectionProvider], |
| 39 | + exports: [imapflowConnectionProvider], |
| 40 | + } |
| 41 | + } |
| 42 | + |
| 43 | + public static createAsyncProviders(options: ImapflowModuleAsyncOptions, connection?: string): Provider[] { |
| 44 | + if (!(options.useExisting || options.useFactory || options.useClass)) { |
| 45 | + throw new Error('Invalid configuration. Must provide useFactory, useClass or useExisting') |
| 46 | + } |
| 47 | + |
| 48 | + if (options.useExisting || options.useFactory) { |
| 49 | + return [this.createAsyncOptionsProvider(options, connection)] |
| 50 | + } |
| 51 | + |
| 52 | + return [ |
| 53 | + this.createAsyncOptionsProvider(options, connection), |
| 54 | + { provide: options.useClass, useClass: options.useClass }, |
| 55 | + ] |
| 56 | + } |
| 57 | + |
| 58 | + public static createAsyncOptionsProvider(options: ImapflowModuleAsyncOptions, connection?: string): Provider { |
| 59 | + if (!(options.useExisting || options.useFactory || options.useClass)) { |
| 60 | + throw new Error('Invalid configuration. Must provide useFactory, useClass or useExisting') |
| 61 | + } |
| 62 | + |
| 63 | + if (options.useFactory) { |
| 64 | + return { |
| 65 | + provide: getImapflowOptionsToken(connection), |
| 66 | + useFactory: options.useFactory, |
| 67 | + inject: options.inject || [], |
| 68 | + } |
| 69 | + } |
| 70 | + |
| 71 | + return { |
| 72 | + provide: getImapflowOptionsToken(connection), |
| 73 | + async useFactory(optionsFactory: ImapflowModuleOptionsFactory): Promise<ImapflowModuleOptions> { |
| 74 | + // noinspection ES6RedundantAwait |
| 75 | + return await optionsFactory.createImapflowModuleOptions() |
| 76 | + }, |
| 77 | + inject: [options.useClass || options.useExisting], |
| 78 | + } |
| 79 | + } |
| 80 | +} |
0 commit comments