ethers.js/misc/admin/src.ts/geturl.ts

138 lines
3.9 KiB
TypeScript
Raw Normal View History

import http from "http";
import https from "https";
import { parse } from "url"
export type GetUrlResponse = {
statusCode: number,
statusMessage: string;
headers: { [ key: string] : string };
body: Uint8Array;
};
export type Options = {
method?: string,
body?: Uint8Array
2020-10-03 09:20:50 +03:00
headers?: { [ key: string] : string },
2020-10-03 09:20:50 +03:00
user?: string,
password?: string,
};
function getResponse(request: http.ClientRequest): Promise<GetUrlResponse> {
return new Promise((resolve, reject) => {
request.once("response", (resp: http.IncomingMessage) => {
const response: GetUrlResponse = {
statusCode: resp.statusCode,
statusMessage: resp.statusMessage,
headers: Object.keys(resp.headers).reduce((accum, name) => {
let value = resp.headers[name];
if (Array.isArray(value)) {
value = value.join(", ");
}
accum[name] = value;
return accum;
}, <{ [ name: string ]: string }>{ }),
body: null
};
//resp.setEncoding("utf8");
resp.on("data", (chunk: Uint8Array) => {
if (response.body == null) { response.body = new Uint8Array(0); }
const body = new Uint8Array(response.body.length + chunk.length);
body.set(response.body, 0);
body.set(chunk, response.body.length);
response.body = body;
});
resp.on("end", () => {
resolve(response);
});
resp.on("error", (error) => {
/* istanbul ignore next */
(<any>error).response = response;
reject(error);
});
});
request.on("error", (error) => { reject(error); });
});
}
// The URL.parse uses null instead of the empty string
function nonnull(value: string): string {
if (value == null) { return ""; }
return value;
}
function staller(duration: number): Promise<void> {
return new Promise((resolve) => {
const timer = setTimeout(resolve, duration);
timer.unref();
});
}
async function _getUrl(href: string, options?: Options): Promise<GetUrlResponse> {
if (options == null) { options = { }; }
// @TODO: Once we drop support for node 8, we can pass the href
// firectly into request and skip adding the components
// to this request object
const url = parse(href);
2020-10-03 09:20:50 +03:00
const request: http.ClientRequestArgs = {
protocol: nonnull(url.protocol),
hostname: nonnull(url.hostname),
port: nonnull(url.port),
path: (nonnull(url.pathname) + nonnull(url.search)),
method: (options.method || "GET"),
headers: (options.headers || { }),
};
2020-10-03 09:20:50 +03:00
if (options.user && options.password) {
request.auth = `${ options.user }:${ options.password }`;
}
let req: http.ClientRequest = null;
switch (nonnull(url.protocol)) {
case "http:":
req = http.request(request);
break;
case "https:":
req = https.request(request);
break;
default:
/* istanbul ignore next */
throw new Error(`unsupported protocol ${ url.protocol }`);
}
if (options.body) {
req.write(Buffer.from(options.body));
}
req.end();
const response = await getResponse(req);
return response;
}
export async function getUrl(href: string, options?: Options): Promise<GetUrlResponse> {
let error: Error = null;
for (let i = 0; i < 3; i++) {
try {
const result = await Promise.race([
_getUrl(href, options),
staller(30000).then((result) => { throw new Error("timeout") })
]);
return result;
} catch (e) {
error = e;
}
await staller(1000);
}
throw error;
}