updates files to typescript and related test files to use them

Author: smartyandrewb

src/AgentSender.js

@@ -0,0 +1,17 @@
+import { Request, Response, Sender } from "./types";
+
+export default class AgentSender {
+	private sender: Sender;
+
+	constructor(innerSender: Sender) {
+		this.sender = innerSender;
+	}
+
+	send(request: Request): Promise<Response> {
+		request.parameters["agent"] =
+			"smarty (sdk:javascript@" + require("../../package.json").version + ")";
+		return new Promise((resolve, reject) => {
+			this.sender.send(request).then(resolve).catch(reject);
+		});
+	}
+}

src/AgentSender.ts

@@ -0,0 +1,21 @@
+import { Request, Response, Sender } from "./types";
+
+export default class BaseUrlSender {
+	private urlOverride: string;
+	private sender: Sender;
+
+	constructor(innerSender: Sender, urlOverride: string) {
+		this.urlOverride = urlOverride;
+		this.sender = innerSender;
+	}
+
+	send(request: Request): Promise<Response> {
+		return new Promise((resolve, reject) => {
+			request.baseUrl = `${this.urlOverride}${
+				request.baseUrlParam ? `/${request.baseUrlParam}` : ""
+			}`;
+
+			this.sender.send(request).then(resolve).catch(reject);
+		});
+	}
+}

src/BaseUrlSender.js

@@ -8,7 +8,7 @@ const CustomHeaderSender = require("./CustomHeaderSender");
 const StatusCodeSender = require("./StatusCodeSender");
 const LicenseSender = require("./LicenseSender");
 const BadCredentialsError = require("./Errors").BadCredentialsError;
-const RetrySender = require("./RetrySender.js");
+const RetrySender = require("./RetrySender");
 const Sleeper = require("./util/Sleeper.ts");
 
 //TODO: refactor this to work more cleanly with a bundler.
@@ -27,7 +27,8 @@ const US_EXTRACT_API_URL = "https://us-extract.api.smarty.com/";
 const US_STREET_API_URL = "https://us-street.api.smarty.com/street-address";
 const US_ZIP_CODE_API_URL = "https://us-zipcode.api.smarty.com/lookup";
 const US_REVERSE_GEO_API_URL = "https://us-reverse-geo.api.smarty.com/lookup";
-const INTERNATIONAL_ADDRESS_AUTOCOMPLETE_API_URL = "https://international-autocomplete.api.smarty.com/v2/lookup";
+const INTERNATIONAL_ADDRESS_AUTOCOMPLETE_API_URL =
+	"https://international-autocomplete.api.smarty.com/v2/lookup";
 const US_ENRICHMENT_API_URL = "https://us-enrichment.api.smarty.com/lookup";
 
 /**
@@ -201,7 +202,10 @@ class ClientBuilder {
 	}
 
 	buildInternationalAddressAutocompleteClient() {
-		return this.buildClient(INTERNATIONAL_ADDRESS_AUTOCOMPLETE_API_URL, InternationalAddressAutocompleteClient);
+		return this.buildClient(
+			INTERNATIONAL_ADDRESS_AUTOCOMPLETE_API_URL,
+			InternationalAddressAutocompleteClient,
+		);
 	}
 
 	buildUsEnrichmentClient() {

src/BaseUrlSender.ts

@@ -0,0 +1,21 @@
+import { Request, Response, Sender } from "./types";
+
+export default class CustomHeaderSender {
+	private sender: Sender;
+	private customHeaders: Record<string, string>;
+
+	constructor(innerSender: Sender, customHeaders: Record<string, string>) {
+		this.sender = innerSender;
+		this.customHeaders = customHeaders;
+	}
+
+	send(request: Request): Promise<Response> {
+		for (let key in this.customHeaders) {
+			request.headers[key] = this.customHeaders[key];
+		}
+
+		return new Promise((resolve, reject) => {
+			this.sender.send(request).then(resolve).catch(reject);
+		});
+	}
+}

src/ClientBuilder.js

@@ -0,0 +1,59 @@
+import { Request, Response, Sender, Sleeper } from "./types";
+
+export default class RetrySender {
+	private maxRetries: number;
+	private statusToRetry: number[];
+	private statusTooManyRequests: number;
+	private maxBackoffDuration: number;
+	private inner: Sender;
+	private sleeper: Sleeper;
+
+	constructor(maxRetries: number = 5, inner: Sender, sleeper: Sleeper) {
+		this.maxRetries = maxRetries;
+		this.statusToRetry = [408, 429, 500, 502, 503, 504];
+		this.statusTooManyRequests = 429;
+		this.maxBackoffDuration = 10;
+		this.inner = inner;
+		this.sleeper = sleeper;
+	}
+
+	async send(request: Request): Promise<Response> {
+		let response = await this.inner.send(request);
+
+		for (let i = 0; i < this.maxRetries; i++) {
+			const statusCode = parseInt(String(response.statusCode));
+			if (!this.statusToRetry.includes(statusCode)) {
+				break;
+			}
+
+			if (statusCode === this.statusTooManyRequests) {
+				let secondsToBackoff = 10;
+				if (response.headers) {
+					const retryAfterHeader = response.headers["Retry-After"];
+					if (Number.isInteger(Number(retryAfterHeader))) {
+						secondsToBackoff = Number(retryAfterHeader);
+					}
+				}
+				await this.rateLimitBackOff(secondsToBackoff);
+			} else {
+				await this.backoff(i);
+			}
+			response = await this.inner.send(request);
+		}
+
+		return response;
+	}
+
+	private async backoff(attempt: number): Promise<void> {
+		const backoffDuration = Math.min(attempt, this.maxBackoffDuration);
+		console.log(
+			`There was an error processing the request. Retrying in ${backoffDuration} seconds...`,
+		);
+		await this.sleeper.sleep(backoffDuration);
+	}
+
+	private async rateLimitBackOff(backoffDuration: number): Promise<void> {
+		console.log(`Rate limit reached. Retrying in ${backoffDuration} seconds...`);
+		await this.sleeper.sleep(backoffDuration);
+	}
+}

src/CustomHeaderSender.js

@@ -1,13 +1,14 @@
-class SharedCredentials {
-	constructor(authId, hostName) {
+export default class SharedCredentials {
+	private authId: string;
+	private hostName: string | undefined;
+
+	constructor(authId: string, hostName?: string) {
 		this.authId = authId;
 		this.hostName = hostName;
 	}
 
-	sign(request) {
+	sign(request: any): void {
 		request.parameters["key"] = this.authId;
 		if (this.hostName) request.headers["Referer"] = "https://" + this.hostName;
 	}
 }
-
-module.exports = SharedCredentials;