smartrequest/ts/smartrequest.request.ts

124 lines
3.3 KiB
TypeScript
Raw Normal View History

2018-07-16 21:39:25 +00:00
import * as https from "https";
import * as plugins from "./smartrequest.plugins";
import * as interfaces from "./smartrequest.interfaces";
2017-01-28 23:51:47 +00:00
2018-07-16 21:39:25 +00:00
import { IncomingMessage } from "http";
2018-06-13 20:34:49 +00:00
2018-07-16 21:39:25 +00:00
export interface IExtendedIncomingMessage extends IncomingMessage {
body: any;
}
2018-06-13 20:34:49 +00:00
2018-07-16 21:39:25 +00:00
const buildUtf8Response = (
incomingMessageArg: IncomingMessage
): Promise<IExtendedIncomingMessage> => {
let done = plugins.smartpromise.defer<IExtendedIncomingMessage>();
2017-06-05 17:09:40 +00:00
// Continuously update stream with data
2018-07-16 21:39:25 +00:00
let body = "";
incomingMessageArg.on("data", function(chunkArg) {
2018-06-13 20:34:49 +00:00
body += chunkArg;
});
2018-07-16 21:39:25 +00:00
incomingMessageArg.on("end", function() {
2017-06-05 17:09:40 +00:00
try {
2018-07-16 21:39:25 +00:00
(incomingMessageArg as IExtendedIncomingMessage).body = JSON.parse(body);
2017-06-05 17:09:40 +00:00
} catch (err) {
2018-07-16 21:39:25 +00:00
(incomingMessageArg as IExtendedIncomingMessage).body = body;
2017-06-05 17:09:40 +00:00
}
2018-07-16 21:39:25 +00:00
done.resolve(incomingMessageArg as IExtendedIncomingMessage);
2018-06-13 20:34:49 +00:00
});
return done.promise;
};
2017-01-28 23:51:47 +00:00
2018-07-16 21:39:25 +00:00
/**
* determine wether a url is a unix sock
* @param urlArg
*/
const testForUnixSock = (urlArg: string): boolean => {
const unixRegex = /^(http:\/\/|https:\/\/|)unix:/;
return unixRegex.test(urlArg);
};
/**
* determine socketPath and path for unixsock
*/
const parseSocketPathAndRoute = (stringToParseArg: string) => {
const parseRegex = /(.*):(.*)/;
const result = parseRegex.exec(stringToParseArg);
return {
socketPath: result[1],
path: result[2]
}
}
2018-06-13 20:34:49 +00:00
export let request = async (
domainArg: string,
optionsArg: interfaces.ISmartRequestOptions = {},
streamArg: boolean = false
2018-07-16 21:39:25 +00:00
): Promise<IExtendedIncomingMessage> => {
let done = plugins.smartpromise.defer<any>();
2018-07-16 21:39:25 +00:00
// parse url
2018-06-13 20:34:49 +00:00
let parsedUrl: plugins.url.Url;
2018-07-16 21:39:25 +00:00
parsedUrl = plugins.url.parse(domainArg);
optionsArg.hostname = parsedUrl.hostname;
if (parsedUrl.port) {
optionsArg.port = parseInt(parsedUrl.port);
}
optionsArg.path = parsedUrl.path;
// determine if unixsock
if(testForUnixSock(domainArg)) {
const detailedUnixPath = parseSocketPathAndRoute(optionsArg.path)
optionsArg.socketPath = detailedUnixPath.socketPath;
optionsArg.path = detailedUnixPath.path;
2017-06-05 17:09:40 +00:00
}
2018-07-16 21:39:25 +00:00
// lets determine the request module to use
const requestModule = (() => {
if (parsedUrl.protocol === "https:") {
return plugins.https;
} else if (parsedUrl.protocol === "http:") {
return plugins.http;
} else {
throw new Error(`unsupported protocol: ${parsedUrl.protocol}`);
2017-01-28 23:51:47 +00:00
}
2018-07-16 21:39:25 +00:00
})() as typeof plugins.https;
// lets perform the actual request
2018-07-19 14:16:02 +00:00
let request = requestModule.request(optionsArg);
2018-07-16 21:39:25 +00:00
// lets write the requestBody
if (optionsArg.requestBody) {
2018-07-19 14:16:02 +00:00
if (
typeof optionsArg.requestBody !== "string"
&& !(optionsArg.requestBody instanceof plugins.formData)
) {
2018-07-16 21:39:25 +00:00
optionsArg.requestBody = JSON.stringify(optionsArg.requestBody);
2018-07-19 14:16:02 +00:00
request.write(optionsArg.requestBody);
} else if (optionsArg.requestBody instanceof plugins.formData) {
optionsArg.requestBody.pipe(request);
2017-01-28 23:51:47 +00:00
}
2017-06-05 17:09:40 +00:00
}
2018-07-19 14:16:02 +00:00
// lets handle an error
2018-07-16 21:39:25 +00:00
request.on("error", e => {
console.error(e);
});
2018-07-19 14:16:02 +00:00
// lets handle the response
request.on("response", async response => {
if (streamArg) {
done.resolve(response);
} else {
const builtResponse = await buildUtf8Response(response)
done.resolve(builtResponse);
}
});
2018-07-16 21:39:25 +00:00
request.end();
const result = await done.promise;
return result;
2018-06-13 20:34:49 +00:00
};