2021-09-07 18:25:31 +00:00
|
|
|
import { DynamicModule, Module, Provider } from '@nestjs/common';
|
|
|
|
import { randomStringGenerator } from '@nestjs/common/utils/random-string-generator.util';
|
|
|
|
import Axios from 'axios';
|
|
|
|
import {
|
|
|
|
AXIOS_INSTANCE_TOKEN,
|
|
|
|
HTTP_MODULE_ID,
|
|
|
|
HTTP_MODULE_OPTIONS,
|
|
|
|
} from './http.constants';
|
|
|
|
import { HttpService } from './http.service';
|
|
|
|
import {
|
|
|
|
HttpModuleAsyncOptions,
|
|
|
|
HttpModuleOptions,
|
|
|
|
HttpModuleOptionsFactory,
|
|
|
|
} from './interfaces';
|
|
|
|
import axiosRetry from 'axios-retry';
|
|
|
|
|
2021-12-23 18:11:18 +00:00
|
|
|
const createAxiosRetry = (config?: HttpModuleOptions) => {
|
2021-09-07 18:25:31 +00:00
|
|
|
const axiosInstance = Axios.create(config);
|
|
|
|
axiosRetry(axiosInstance, config);
|
|
|
|
return axiosInstance;
|
|
|
|
}
|
|
|
|
|
|
|
|
@Module({
|
|
|
|
providers: [
|
|
|
|
HttpService,
|
|
|
|
{
|
|
|
|
provide: AXIOS_INSTANCE_TOKEN,
|
2021-12-23 18:11:18 +00:00
|
|
|
useValue: createAxiosRetry(),
|
2021-09-07 18:25:31 +00:00
|
|
|
},
|
|
|
|
],
|
|
|
|
exports: [HttpService],
|
|
|
|
})
|
|
|
|
export class HttpModule {
|
|
|
|
static register(config: HttpModuleOptions): DynamicModule {
|
|
|
|
return {
|
|
|
|
module: HttpModule,
|
|
|
|
providers: [
|
|
|
|
{
|
|
|
|
provide: AXIOS_INSTANCE_TOKEN,
|
|
|
|
useValue: createAxiosRetry(config),
|
|
|
|
},
|
|
|
|
{
|
|
|
|
provide: HTTP_MODULE_ID,
|
|
|
|
useValue: randomStringGenerator(),
|
|
|
|
},
|
|
|
|
],
|
|
|
|
};
|
|
|
|
}
|
|
|
|
|
|
|
|
static registerAsync(options: HttpModuleAsyncOptions): DynamicModule {
|
|
|
|
return {
|
|
|
|
module: HttpModule,
|
|
|
|
imports: options.imports,
|
|
|
|
providers: [
|
|
|
|
...this.createAsyncProviders(options),
|
|
|
|
{
|
|
|
|
provide: AXIOS_INSTANCE_TOKEN,
|
|
|
|
useFactory: (config: HttpModuleOptions) => createAxiosRetry(config),
|
|
|
|
inject: [HTTP_MODULE_OPTIONS],
|
|
|
|
},
|
|
|
|
{
|
|
|
|
provide: HTTP_MODULE_ID,
|
|
|
|
useValue: randomStringGenerator(),
|
|
|
|
},
|
|
|
|
...(options.extraProviders || []),
|
|
|
|
],
|
|
|
|
};
|
|
|
|
}
|
|
|
|
|
|
|
|
private static createAsyncProviders(
|
|
|
|
options: HttpModuleAsyncOptions,
|
|
|
|
): Provider[] {
|
|
|
|
if (options.useExisting || options.useFactory) {
|
|
|
|
return [this.createAsyncOptionsProvider(options)];
|
|
|
|
}
|
|
|
|
|
|
|
|
const providers = [
|
|
|
|
this.createAsyncOptionsProvider(options)
|
|
|
|
];
|
|
|
|
|
|
|
|
if(options.useClass)
|
|
|
|
providers.push({
|
|
|
|
provide: options.useClass,
|
|
|
|
useClass: options.useClass,
|
|
|
|
})
|
|
|
|
|
|
|
|
return providers;
|
|
|
|
}
|
|
|
|
|
|
|
|
private static createAsyncOptionsProvider(
|
|
|
|
options: HttpModuleAsyncOptions,
|
|
|
|
): Provider {
|
|
|
|
if (options.useFactory) {
|
|
|
|
return {
|
|
|
|
provide: HTTP_MODULE_OPTIONS,
|
|
|
|
useFactory: options.useFactory,
|
|
|
|
inject: options.inject || [],
|
|
|
|
};
|
|
|
|
}
|
|
|
|
|
|
|
|
let inject;
|
|
|
|
if (options.useExisting)
|
|
|
|
inject = [options.useExisting];
|
|
|
|
else if (options.useClass)
|
|
|
|
inject = [options.useClass];
|
|
|
|
|
|
|
|
return {
|
|
|
|
provide: HTTP_MODULE_OPTIONS,
|
|
|
|
useFactory: async (optionsFactory: HttpModuleOptionsFactory) =>
|
|
|
|
optionsFactory.createHttpOptions(),
|
|
|
|
inject,
|
|
|
|
};
|
|
|
|
}
|
|
|
|
}
|