diff --git a/src/request-helper.ts b/src/request-helper.ts index 16f1490..5f4b65b 100644 --- a/src/request-helper.ts +++ b/src/request-helper.ts @@ -1,9 +1,11 @@ import * as rp from 'request-promise-native'; +import * as http from 'http'; +import * as https from 'https'; import { AccessProvider } from './access-provider'; export class RequestHelper { - constructor(private accessProvider: AccessProvider) {} + constructor(private accessProvider: AccessProvider, private agent?: http.Agent | https.Agent) {} get basicRequest() { const headers = this.accessProvider.trustpilotApiConfig.defaultHeaders || {}; @@ -12,6 +14,7 @@ export class RequestHelper { baseUrl: this.accessProvider.trustpilotApiConfig.baseUrl, headers, json: true, + agent: this.agent, }); } diff --git a/src/trustpilot-api.ts b/src/trustpilot-api.ts index 1a7d5b0..8aa0e6f 100644 --- a/src/trustpilot-api.ts +++ b/src/trustpilot-api.ts @@ -1,13 +1,15 @@ import { AccessProvider } from './access-provider'; import { ITrustpilotApiConfig } from './models'; import { RequestHelper } from './request-helper'; +import * as http from 'http'; +import * as https from 'https'; export class TrustpilotApi { private requestHelper: RequestHelper; - constructor(config: Readonly) { + constructor(config: Readonly, agent?: http.Agent | https.Agent) { const accessProvider = new AccessProvider(Object.freeze(config)); - this.requestHelper = new RequestHelper(accessProvider); + this.requestHelper = new RequestHelper(accessProvider, agent); } get apiRequest() {