86 lines
2.5 KiB
TypeScript
86 lines
2.5 KiB
TypeScript
import * as https from 'https';
|
|
import * as plugins from './smartrequest.plugins';
|
|
import * as interfaces from './smartrequest.interfaces';
|
|
|
|
import { IncomingMessage } from 'http';
|
|
|
|
export interface extendedIncomingMessage extends IncomingMessage {
|
|
body: any
|
|
};
|
|
|
|
let buildResponse = (incomingMessageArg: IncomingMessage): Promise<extendedIncomingMessage> => {
|
|
let done = plugins.q.defer<extendedIncomingMessage>();
|
|
// Continuously update stream with data
|
|
let body = '';
|
|
incomingMessageArg.on('data', function(chunkArg) {
|
|
body += chunkArg;
|
|
});
|
|
|
|
incomingMessageArg.on('end', function() {
|
|
try {
|
|
(incomingMessageArg as extendedIncomingMessage).body = JSON.parse(body);
|
|
} catch (err) {
|
|
(incomingMessageArg as extendedIncomingMessage).body = body;
|
|
}
|
|
done.resolve(incomingMessageArg as extendedIncomingMessage);
|
|
});
|
|
return done.promise;
|
|
};
|
|
|
|
export let request = async (
|
|
domainArg: string,
|
|
optionsArg: interfaces.ISmartRequestOptions = {},
|
|
streamArg: boolean = false
|
|
): Promise<extendedIncomingMessage> => {
|
|
let done = plugins.q.defer<any>();
|
|
let parsedUrl: plugins.url.Url;
|
|
if (domainArg) {
|
|
parsedUrl = plugins.url.parse(domainArg);
|
|
optionsArg.hostname = parsedUrl.hostname;
|
|
if (parsedUrl.port) {
|
|
optionsArg.port = parseInt(parsedUrl.port);
|
|
}
|
|
optionsArg.path = parsedUrl.path;
|
|
}
|
|
if (!parsedUrl || parsedUrl.protocol === 'https:') {
|
|
let request = plugins.https.request(optionsArg, response => {
|
|
if (streamArg) {
|
|
done.resolve(response);
|
|
} else {
|
|
buildResponse(response).then(done.resolve);
|
|
}
|
|
});
|
|
if (optionsArg.requestBody) {
|
|
if (typeof optionsArg.requestBody !== 'string') {
|
|
optionsArg.requestBody = JSON.stringify(optionsArg.requestBody);
|
|
}
|
|
request.write(optionsArg.requestBody);
|
|
}
|
|
request.on('error', e => {
|
|
console.error(e);
|
|
});
|
|
request.end();
|
|
} else if (parsedUrl.protocol === 'http:') {
|
|
let request = plugins.http.request(optionsArg, response => {
|
|
if (streamArg) {
|
|
done.resolve(response);
|
|
} else {
|
|
buildResponse(response).then(done.resolve);
|
|
}
|
|
});
|
|
if (optionsArg.requestBody) {
|
|
if (typeof optionsArg.requestBody !== 'string') {
|
|
optionsArg.requestBody = JSON.stringify(optionsArg.requestBody);
|
|
}
|
|
request.write(optionsArg.requestBody);
|
|
}
|
|
request.on('error', e => {
|
|
console.error(e);
|
|
});
|
|
request.end();
|
|
} else {
|
|
throw new Error(`unsupported protocol: ${parsedUrl.protocol}`);
|
|
}
|
|
return done.promise;
|
|
};
|