smartrequest/ts/smartrequest.request.ts

180 lines
4.9 KiB
TypeScript
Raw Normal View History

2018-08-13 23:47:54 +00:00
import * as plugins from './smartrequest.plugins';
import * as interfaces from './smartrequest.interfaces';
2017-01-28 23:51:47 +00:00
2018-08-13 23:47:54 +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 = (
2019-06-12 13:16:27 +00:00
incomingMessageArg: IncomingMessage,
autoJsonParse = true
2018-07-16 21:39:25 +00:00
): Promise<IExtendedIncomingMessage> => {
2019-08-22 10:38:11 +00:00
const done = plugins.smartpromise.defer<IExtendedIncomingMessage>();
2017-06-05 17:09:40 +00:00
// Continuously update stream with data
2018-08-13 23:47:54 +00:00
let body = '';
2019-09-08 15:49:32 +00:00
incomingMessageArg.on('data', chunkArg => {
2018-06-13 20:34:49 +00:00
body += chunkArg;
});
2019-08-22 10:38:55 +00:00
incomingMessageArg.on('end', () => {
2019-06-12 13:16:27 +00:00
if (autoJsonParse) {
try {
(incomingMessageArg as IExtendedIncomingMessage).body = JSON.parse(body);
} catch (err) {
(incomingMessageArg as IExtendedIncomingMessage).body = body;
}
} else {
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-08-13 23:47:54 +00:00
};
};
2018-07-16 21:39:25 +00:00
2019-08-22 10:40:19 +00:00
/**
* a custom http agent to make sure we can set custom keepAlive options for speedy subsequent calls
*/
2019-08-21 10:55:19 +00:00
const httpAgent = new plugins.http.Agent({
keepAlive: true,
keepAliveMsecs: 600000
});
2019-09-08 15:47:30 +00:00
/**
* a custom http agent to make sure we can set custom keepAlive options for speedy subsequent calls
*/
const httpAgentKeepAliveFalse = new plugins.http.Agent({
2019-09-08 15:57:28 +00:00
maxFreeSockets: 0,
2019-09-08 15:47:30 +00:00
keepAlive: false,
2019-09-08 15:57:28 +00:00
keepAliveMsecs: 0
2019-09-08 15:47:30 +00:00
});
2019-08-22 10:40:19 +00:00
/**
* a custom https agent to make sure we can set custom keepAlive options for speedy subsequent calls
*/
2019-08-21 10:55:19 +00:00
const httpsAgent = new plugins.https.Agent({
keepAlive: true,
keepAliveMsecs: 600000
});
2019-09-08 15:47:30 +00:00
/**
* a custom https agent to make sure we can set custom keepAlive options for speedy subsequent calls
*/
const httpsAgentKeepAliveFalse = new plugins.https.Agent({
2019-09-08 15:57:28 +00:00
maxFreeSockets: 0,
2019-09-08 15:47:30 +00:00
keepAlive: false,
2019-09-08 15:57:28 +00:00
keepAliveMsecs: 0
2019-09-08 15:47:30 +00:00
});
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> => {
2019-08-22 10:38:11 +00:00
const done = plugins.smartpromise.defer<any>();
2018-07-16 21:39:25 +00:00
2019-06-12 13:16:27 +00:00
// merge options
const defaultOptions: interfaces.ISmartRequestOptions = {
2019-08-21 10:55:19 +00:00
// agent: agent,
2019-09-08 15:47:30 +00:00
autoJsonParse: true,
2019-09-08 15:49:32 +00:00
keepAlive: true
2019-08-16 19:38:50 +00:00
};
2019-06-12 13:16:27 +00:00
optionsArg = {
...defaultOptions,
...optionsArg
2019-08-16 19:38:50 +00:00
};
2019-06-12 13:16:27 +00:00
2018-07-16 21:39:25 +00:00
// parse url
2019-08-22 10:38:11 +00:00
const parsedUrl = plugins.url.parse(domainArg);
2018-07-16 21:39:25 +00:00
optionsArg.hostname = parsedUrl.hostname;
if (parsedUrl.port) {
2019-08-22 10:38:11 +00:00
optionsArg.port = parseInt(parsedUrl.port, 10);
2018-07-16 21:39:25 +00:00
}
optionsArg.path = parsedUrl.path;
// determine if unixsock
2018-08-13 23:47:54 +00:00
if (testForUnixSock(domainArg)) {
const detailedUnixPath = parseSocketPathAndRoute(optionsArg.path);
2018-07-16 21:39:25 +00:00
optionsArg.socketPath = detailedUnixPath.socketPath;
optionsArg.path = detailedUnixPath.path;
2017-06-05 17:09:40 +00:00
}
2018-08-13 23:47:54 +00:00
2019-08-16 20:00:01 +00:00
// TODO: support tcp sockets
2018-07-16 21:39:25 +00:00
// lets determine the request module to use
const requestModule = (() => {
2019-09-08 15:47:30 +00:00
switch (true) {
case parsedUrl.protocol === 'https:' && optionsArg.keepAlive:
optionsArg.agent = httpsAgent;
return plugins.https;
2019-09-08 16:00:48 +00:00
case parsedUrl.protocol === 'https:' && (!optionsArg.keepAlive):
2019-09-08 15:47:30 +00:00
optionsArg.agent = httpsAgentKeepAliveFalse;
return plugins.https;
case parsedUrl.protocol === 'http:' && optionsArg.keepAlive:
optionsArg.agent = httpAgent;
return plugins.http;
2019-09-08 16:00:48 +00:00
case parsedUrl.protocol === 'http:' && (!optionsArg.keepAlive):
2019-09-08 15:47:30 +00:00
optionsArg.agent = httpAgentKeepAliveFalse;
return plugins.http;
}
2018-07-16 21:39:25 +00:00
})() as typeof plugins.https;
// lets perform the actual request
2019-09-28 19:33:13 +00:00
const requestToFire = requestModule.request(optionsArg, async response => {
if (streamArg) {
done.resolve(response);
} else {
const builtResponse = await buildUtf8Response(response, optionsArg.autoJsonParse);
done.resolve(builtResponse);
}
});
2018-07-16 21:39:25 +00:00
// lets write the requestBody
if (optionsArg.requestBody) {
2019-09-28 19:33:13 +00:00
if (optionsArg.requestBody instanceof plugins.formData) {
optionsArg.requestBody.pipe(requestToFire).on('finish', event => {
requestToFire.end();
});
} else {
2018-08-13 23:47:54 +00:00
if (typeof optionsArg.requestBody !== 'string') {
optionsArg.requestBody = JSON.stringify(optionsArg.requestBody);
}
2019-08-22 10:38:11 +00:00
requestToFire.write(optionsArg.requestBody);
requestToFire.end();
2017-01-28 23:51:47 +00:00
}
2018-07-19 21:22:11 +00:00
} else {
2019-08-22 10:38:11 +00:00
requestToFire.end();
2017-06-05 17:09:40 +00:00
}
2018-07-19 14:16:02 +00:00
// lets handle an error
2019-08-22 10:38:11 +00:00
requestToFire.on('error', e => {
2018-07-16 21:39:25 +00:00
console.error(e);
});
2018-07-19 14:16:02 +00:00
2018-07-16 21:39:25 +00:00
const result = await done.promise;
return result;
2018-06-13 20:34:49 +00:00
};