102 lines
2.9 KiB
TypeScript
102 lines
2.9 KiB
TypeScript
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';
|
|
import axiosBetterStacktrace from 'axios-better-stacktrace';
|
|
|
|
const createAxiosInstance = (config?: HttpModuleOptions) => {
|
|
const axiosInstance = Axios.create(config);
|
|
axiosRetry(axiosInstance, config);
|
|
if (config?.isBetterStackTraceEnabled === undefined || config?.isBetterStackTraceEnabled) {
|
|
axiosBetterStacktrace(axiosInstance);
|
|
}
|
|
return axiosInstance;
|
|
};
|
|
|
|
@Module({
|
|
providers: [
|
|
HttpService,
|
|
{
|
|
provide: AXIOS_INSTANCE_TOKEN,
|
|
useValue: createAxiosInstance(),
|
|
},
|
|
],
|
|
exports: [HttpService],
|
|
})
|
|
export class HttpModule {
|
|
static register(config: HttpModuleOptions): DynamicModule {
|
|
return {
|
|
module: HttpModule,
|
|
providers: [
|
|
{
|
|
provide: AXIOS_INSTANCE_TOKEN,
|
|
useValue: createAxiosInstance(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) => createAxiosInstance(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,
|
|
};
|
|
}
|
|
}
|