insomnia/packages/insomnia-app/app/network/network.js

1008 lines
33 KiB
JavaScript
Raw Normal View History

2017-07-18 20:38:19 +00:00
// @flow
2018-06-25 17:42:50 +00:00
import type { ResponseHeader, ResponseTimelineEntry } from '../models/response';
import type { Request, RequestHeader } from '../models/request';
import type { Workspace } from '../models/workspace';
import type { Settings } from '../models/settings';
import type { RenderedRequest } from '../common/render';
import { getRenderedRequestAndContext, RENDER_PURPOSE_SEND } from '../common/render';
import mkdirp from 'mkdirp';
import clone from 'clone';
2018-06-25 17:42:50 +00:00
import { parse as urlParse, resolve as urlResolve } from 'url';
import { Curl } from 'insomnia-libcurl';
import { join as pathJoin } from 'path';
import uuid from 'uuid';
import * as models from '../models';
import {
AUTH_AWS_IAM,
AUTH_BASIC,
AUTH_DIGEST,
AUTH_NETRC,
AUTH_NTLM,
CONTENT_TYPE_FORM_DATA,
CONTENT_TYPE_FORM_URLENCODED,
getAppVersion,
getTempDir,
STATUS_CODE_PLUGIN_ERROR,
} from '../common/constants';
import {
delay,
describeByteSize,
getContentTypeHeader,
getHostHeader,
getLocationHeader,
getSetCookieHeaders,
hasAcceptEncodingHeader,
hasAcceptHeader,
hasAuthHeader,
hasContentTypeHeader,
hasUserAgentHeader,
waitForStreamToFinish,
getDataDirectory,
} from '../common/misc';
2018-06-25 17:42:50 +00:00
import {
buildQueryStringFromParams,
joinUrlAndQueryString,
setDefaultProtocol,
smartEncodeUrl,
2018-06-25 17:42:50 +00:00
} from 'insomnia-url';
import fs from 'fs';
import * as db from '../common/database';
import * as CACerts from './cacert';
import * as plugins from '../plugins/index';
import * as pluginContexts from '../plugins/context/index';
2018-06-25 17:42:50 +00:00
import { getAuthHeader } from './authentication';
import { cookiesFromJar, jarFromCookies } from 'insomnia-cookies';
import { urlMatchesCertHost } from './url-matches-cert-host';
import aws4 from 'aws4';
2018-06-25 17:42:50 +00:00
import { buildMultipart } from './multipart';
export type ResponsePatch = {
statusMessage?: string,
error?: string,
url?: string,
statusCode?: number,
bytesContent?: number,
bodyPath?: string,
bodyCompression?: 'zip' | null,
message?: string,
2017-11-13 23:10:53 +00:00
httpVersion?: string,
headers?: Array<ResponseHeader>,
elapsedTime?: number,
contentType?: string,
bytesRead?: number,
parentId?: string,
settingStoreCookies?: boolean,
settingSendCookies?: boolean,
timeline?: Array<ResponseTimelineEntry>,
2017-07-18 20:38:19 +00:00
};
// Time since user's last keypress to wait before making the request
const MAX_DELAY_TIME = 1000;
// Special header value that will prevent the header being sent
const DISABLE_HEADER_VALUE = '__Di$aB13d__';
let cancelRequestFunction = null;
let lastUserInteraction = Date.now();
2018-06-25 17:42:50 +00:00
export function cancelCurrentRequest() {
if (typeof cancelRequestFunction === 'function') {
cancelRequestFunction();
}
}
2018-06-25 17:42:50 +00:00
export async function _actuallySend(
2017-07-18 22:10:57 +00:00
renderedRequest: RenderedRequest,
renderContext: Object,
2017-07-18 22:10:57 +00:00
workspace: Workspace,
settings: Settings,
): Promise<ResponsePatch> {
return new Promise(async resolve => {
2017-07-18 22:10:57 +00:00
let timeline: Array<ResponseTimelineEntry> = [];
2017-07-11 00:23:40 +00:00
function addTimeline(name, value) {
timeline.push({ name, value, timestamp: Date.now() });
}
function addTimelineText(value) {
addTimeline('TEXT', value);
}
2017-07-19 04:48:28 +00:00
// Initialize the curl handle
const curl = new Curl();
/** Helper function to respond with a success */
2018-06-25 17:42:50 +00:00
async function respond(
patch: ResponsePatch,
bodyPath: string | null,
noPlugins: boolean = false,
2018-06-25 17:42:50 +00:00
): Promise<void> {
const responsePatchBeforeHooks = Object.assign(
({
parentId: renderedRequest._id,
bodyCompression: null, // Will default to .zip otherwise
timeline: timeline,
bodyPath: bodyPath || '',
settingSendCookies: renderedRequest.settingSendCookies,
settingStoreCookies: renderedRequest.settingStoreCookies,
2018-06-25 17:42:50 +00:00
}: ResponsePatch),
patch,
2018-06-25 17:42:50 +00:00
);
2017-07-11 00:23:40 +00:00
if (noPlugins) {
resolve(responsePatchBeforeHooks);
return;
}
let responsePatch: ?ResponsePatch;
try {
2018-06-25 17:42:50 +00:00
responsePatch = await _applyResponsePluginHooks(
responsePatchBeforeHooks,
renderedRequest,
renderContext,
2018-06-25 17:42:50 +00:00
);
} catch (err) {
2018-06-25 17:42:50 +00:00
handleError(
new Error(`[plugin] Response hook failed plugin=${err.plugin.name} err=${err.message}`),
2018-06-25 17:42:50 +00:00
);
return;
}
resolve(responsePatch);
2017-07-18 20:38:19 +00:00
}
2017-07-11 00:23:40 +00:00
2017-07-19 04:48:28 +00:00
/** Helper function to respond with an error */
2018-06-25 17:42:50 +00:00
function handleError(err: Error): void {
respond(
{
url: renderedRequest.url,
parentId: renderedRequest._id,
error: err.message,
elapsedTime: 0,
statusMessage: 'Error',
settingSendCookies: renderedRequest.settingSendCookies,
settingStoreCookies: renderedRequest.settingStoreCookies,
2018-06-25 17:42:50 +00:00
},
null,
true,
2018-06-25 17:42:50 +00:00
);
}
2017-07-19 04:48:28 +00:00
/** Helper function to set Curl options */
2018-06-25 17:42:50 +00:00
function setOpt(opt: number, val: any, optional: boolean = false) {
const name = Object.keys(Curl.option).find(name => Curl.option[name] === opt);
2017-07-19 04:48:28 +00:00
try {
curl.setOpt(opt, val);
} catch (err) {
if (!optional) {
throw new Error(`${err.message} (${opt} ${name || 'n/a'})`);
} else {
console.warn(`Failed to set optional Curl opt (${opt} ${name || 'n/a'})`);
}
2017-07-19 04:48:28 +00:00
}
}
2018-06-25 17:42:50 +00:00
function enable(feature: number) {
2017-11-13 23:10:53 +00:00
curl.enable(feature);
}
2017-07-19 04:48:28 +00:00
try {
// Setup the cancellation logic
cancelRequestFunction = () => {
2018-06-25 17:42:50 +00:00
respond(
{
elapsedTime: curl.getInfo(Curl.info.TOTAL_TIME) * 1000,
bytesRead: curl.getInfo(Curl.info.SIZE_DOWNLOAD),
url: curl.getInfo(Curl.info.EFFECTIVE_URL),
statusMessage: 'Cancelled',
error: 'Request was cancelled',
2018-06-25 17:42:50 +00:00
},
null,
true,
2018-06-25 17:42:50 +00:00
);
// Kill it!
curl.close();
};
// Set all the basic options
setOpt(Curl.option.FOLLOWLOCATION, settings.followRedirects);
setOpt(Curl.option.VERBOSE, true); // True so debug function works
setOpt(Curl.option.NOPROGRESS, false); // False so progress function works
2017-11-12 18:35:01 +00:00
setOpt(Curl.option.ACCEPT_ENCODING, ''); // Auto decode everything
2017-11-13 23:10:53 +00:00
enable(Curl.feature.NO_HEADER_PARSING);
enable(Curl.feature.NO_DATA_PARSING);
// Set maximum amount of redirects allowed
// NOTE: Setting this to -1 breaks some versions of libcurl
if (settings.maxRedirects > 0) {
setOpt(Curl.option.MAXREDIRS, settings.maxRedirects);
}
// Don't rebuild dot sequences in path
if (!renderedRequest.settingRebuildPath) {
setOpt(Curl.option.PATH_AS_IS, true);
}
// Only set CURLOPT_CUSTOMREQUEST if not HEAD or GET. This is because Curl
// See https://curl.haxx.se/libcurl/c/CURLOPT_CUSTOMREQUEST.html
switch (renderedRequest.method.toUpperCase()) {
case 'HEAD':
// This is how you tell Curl to send a HEAD request
setOpt(Curl.option.NOBODY, 1);
break;
case 'POST':
// This is how you tell Curl to send a POST request
setOpt(Curl.option.POST, 1);
break;
default:
// IMPORTANT: Only use CUSTOMREQUEST for all but HEAD and POST
setOpt(Curl.option.CUSTOMREQUEST, renderedRequest.method);
break;
}
// Setup debug handler
2017-07-18 22:10:57 +00:00
setOpt(Curl.option.DEBUGFUNCTION, (infoType: string, content: string) => {
const name = Object.keys(Curl.info.debug).find(k => Curl.info.debug[k] === infoType) || '';
if (infoType === Curl.info.debug.SSL_DATA_IN || infoType === Curl.info.debug.SSL_DATA_OUT) {
return 0;
}
// Ignore the possibly large data messages
2017-03-29 02:21:49 +00:00
if (infoType === Curl.info.debug.DATA_OUT) {
2017-06-21 20:58:41 +00:00
if (content.length === 0) {
// Sometimes this happens, but I'm not sure why. Just ignore it.
} else if (content.length < renderedRequest.settingMaxTimelineDataSize) {
addTimeline(name, content);
2017-03-29 02:21:49 +00:00
} else {
addTimeline(name, `(${describeByteSize(content.length)} hidden)`);
2017-03-29 02:21:49 +00:00
}
return 0;
}
if (infoType === Curl.info.debug.DATA_IN) {
addTimelineText(`Received ${describeByteSize(content.length)} chunk`);
return 0;
}
// Don't show cookie setting because this will display every domain in the jar
if (infoType === Curl.info.debug.TEXT && content.indexOf('Added cookie') === 0) {
return 0;
}
addTimeline(name, content);
return 0; // Must be here
});
// Set the headers (to be modified as we go)
const headers = clone(renderedRequest.headers);
let lastPercent = 0;
// NOTE: This option was added in 7.32.0 so make it optional
2018-06-25 17:42:50 +00:00
setOpt(
Curl.option.XFERINFOFUNCTION,
(dltotal, dlnow, ultotal, ulnow) => {
if (dltotal === 0) {
return 0;
}
2018-06-25 17:42:50 +00:00
const percent = Math.round((dlnow / dltotal) * 100);
if (percent !== lastPercent) {
// console.log(`[network] Request downloaded ${percent}%`);
lastPercent = percent;
}
2018-06-25 17:42:50 +00:00
return 0;
},
true,
2018-06-25 17:42:50 +00:00
);
// Set the URL, including the query parameters
const qs = buildQueryStringFromParams(renderedRequest.parameters);
const url = joinUrlAndQueryString(renderedRequest.url, qs);
2017-06-30 16:00:00 +00:00
const isUnixSocket = url.match(/https?:\/\/unix:\//);
const finalUrl = smartEncodeUrl(url, renderedRequest.settingEncodeUrl);
2017-06-22 18:43:00 +00:00
if (isUnixSocket) {
// URL prep will convert "unix:/path" hostname to "unix/path"
2017-06-22 19:43:11 +00:00
const match = finalUrl.match(/(https?:)\/\/unix:?(\/[^:]+):\/(.+)/);
const protocol = (match && match[1]) || '';
const socketPath = (match && match[2]) || '';
const socketUrl = (match && match[3]) || '';
curl.setUrl(`${protocol}//${socketUrl}`);
2017-06-22 18:43:00 +00:00
setOpt(Curl.option.UNIX_SOCKET_PATH, socketPath);
} else {
curl.setUrl(finalUrl);
2017-06-22 18:43:00 +00:00
}
addTimelineText('Preparing request to ' + finalUrl);
addTimelineText(`Using ${Curl.getVersion()}`);
// Set timeout
if (settings.timeout > 0) {
addTimelineText(`Enable timeout of ${settings.timeout}ms`);
setOpt(Curl.option.TIMEOUT_MS, settings.timeout);
} else {
addTimelineText(`Disable timeout`);
2018-12-14 18:34:48 +00:00
setOpt(Curl.option.TIMEOUT_MS, 0);
}
// log some things
if (renderedRequest.settingEncodeUrl) {
addTimelineText('Enable automatic URL encoding');
} else {
addTimelineText('Disable automatic URL encoding');
}
// SSL Validation
if (settings.validateSSL) {
addTimelineText('Enable SSL validation');
} else {
setOpt(Curl.option.SSL_VERIFYHOST, 0);
setOpt(Curl.option.SSL_VERIFYPEER, 0);
addTimelineText('Disable SSL validation');
}
// Setup CA Root Certificates if not on Mac. Thanks to libcurl, Mac will use
// certificates form the OS.
if (process.platform !== 'darwin') {
2017-11-21 17:52:47 +00:00
const baseCAPath = getTempDir();
const fullCAPath = pathJoin(baseCAPath, CACerts.filename);
try {
fs.statSync(fullCAPath);
} catch (err) {
// Doesn't exist yet, so write it
2017-11-21 17:52:47 +00:00
mkdirp.sync(baseCAPath);
fs.writeFileSync(fullCAPath, CACerts.blob);
2017-11-22 00:07:28 +00:00
console.log('[net] Set CA to', fullCAPath);
}
setOpt(Curl.option.CAINFO, fullCAPath);
}
// Set cookies from jar
if (renderedRequest.settingSendCookies) {
// Tell Curl to store cookies that it receives. This is only important if we receive
// a cookie on a redirect that needs to be sent on the next request in the chain.
curl.setOpt(Curl.option.COOKIEFILE, '');
const cookies = renderedRequest.cookieJar.cookies || [];
for (const cookie of cookies) {
let expiresTimestamp = 0;
if (cookie.expires) {
const expiresDate = new Date(cookie.expires);
expiresTimestamp = Math.round(expiresDate.getTime() / 1000);
}
2018-06-25 17:42:50 +00:00
setOpt(
Curl.option.COOKIELIST,
[
cookie.httpOnly ? `#HttpOnly_${cookie.domain}` : cookie.domain,
cookie.hostOnly ? 'FALSE' : 'TRUE',
cookie.path,
cookie.secure ? 'TRUE' : 'FALSE',
expiresTimestamp,
cookie.key,
cookie.value,
].join('\t'),
2018-06-25 17:42:50 +00:00
);
}
2018-06-25 17:42:50 +00:00
for (const { name, value } of renderedRequest.cookies) {
setOpt(Curl.option.COOKIE, `${name}=${value}`);
}
addTimelineText(
'Enable cookie sending with jar of ' +
`${cookies.length} cookie${cookies.length !== 1 ? 's' : ''}`,
);
} else {
addTimelineText('Disable cookie sending due to user setting');
}
// Set proxy settings if we have them
if (settings.proxyEnabled) {
2018-06-25 17:42:50 +00:00
const { protocol } = urlParse(renderedRequest.url);
const { httpProxy, httpsProxy, noProxy } = settings;
const proxyHost = protocol === 'https:' ? httpsProxy : httpProxy;
const proxy = proxyHost ? setDefaultProtocol(proxyHost) : null;
addTimelineText(`Enable network proxy for ${protocol || ''}`);
if (proxy) {
setOpt(Curl.option.PROXY, proxy);
setOpt(Curl.option.PROXYAUTH, Curl.auth.ANY);
}
2017-06-06 20:21:59 +00:00
if (noProxy) {
setOpt(Curl.option.NOPROXY, noProxy);
}
} else {
setOpt(Curl.option.PROXY, '');
}
// Set client certs if needed
const clientCertificates = await models.clientCertificate.findByParentId(workspace._id);
for (const certificate of clientCertificates) {
if (certificate.disabled) {
continue;
}
const cHostWithProtocol = setDefaultProtocol(certificate.host, 'https:');
if (urlMatchesCertHost(cHostWithProtocol, renderedRequest.url)) {
const ensureFile = blobOrFilename => {
try {
fs.statSync(blobOrFilename);
} catch (err) {
2017-11-21 17:52:47 +00:00
// Certificate file not found!
// LEGACY: Certs used to be stored in blobs (not as paths), so let's write it to
// the temp directory first.
const fullBase = getTempDir();
const name = `${renderedRequest._id}_${renderedRequest.modified}`;
const fullPath = pathJoin(fullBase, name);
fs.writeFileSync(fullPath, Buffer.from(blobOrFilename, 'base64'));
// Set filename to the one we just saved
blobOrFilename = fullPath;
}
return blobOrFilename;
};
2018-06-25 17:42:50 +00:00
const { passphrase, cert, key, pfx } = certificate;
if (cert) {
setOpt(Curl.option.SSLCERT, ensureFile(cert));
setOpt(Curl.option.SSLCERTTYPE, 'PEM');
addTimelineText('Adding SSL PEM certificate');
}
if (pfx) {
setOpt(Curl.option.SSLCERT, ensureFile(pfx));
setOpt(Curl.option.SSLCERTTYPE, 'P12');
addTimelineText('Adding SSL P12 certificate');
}
if (key) {
setOpt(Curl.option.SSLKEY, ensureFile(key));
addTimelineText('Adding SSL KEY certificate');
}
if (passphrase) {
setOpt(Curl.option.KEYPASSWD, passphrase);
}
}
}
// Build the body
2017-04-09 21:53:46 +00:00
let noBody = false;
let requestBody = null;
const expectsBody = ['POST', 'PUT', 'PATCH'].includes(renderedRequest.method.toUpperCase());
if (renderedRequest.body.mimeType === CONTENT_TYPE_FORM_URLENCODED) {
requestBody = buildQueryStringFromParams(renderedRequest.body.params || [], false);
} else if (renderedRequest.body.mimeType === CONTENT_TYPE_FORM_DATA) {
2017-07-18 20:38:19 +00:00
const params = renderedRequest.body.params || [];
const { filePath: multipartBodyPath, boundary, contentLength } = await buildMultipart(
params,
);
2017-10-12 17:32:26 +00:00
// Extend the Content-Type header
const contentTypeHeader = getContentTypeHeader(headers);
if (contentTypeHeader) {
contentTypeHeader.value = `multipart/form-data; boundary=${boundary}`;
} else {
headers.push({
name: 'Content-Type',
value: `multipart/form-data; boundary=${boundary}`,
2017-10-12 17:32:26 +00:00
});
}
const fd = fs.openSync(multipartBodyPath, 'r');
setOpt(Curl.option.INFILESIZE_LARGE, contentLength);
2017-10-12 17:32:26 +00:00
setOpt(Curl.option.UPLOAD, 1);
setOpt(Curl.option.READDATA, fd);
2017-10-12 17:32:26 +00:00
// We need this, otherwise curl will send it as a PUT
setOpt(Curl.option.CUSTOMREQUEST, renderedRequest.method);
const fn = () => {
fs.closeSync(fd);
2018-06-25 17:42:50 +00:00
fs.unlink(multipartBodyPath, () => {});
};
curl.on('end', fn);
curl.on('error', fn);
} else if (renderedRequest.body.fileName) {
2018-06-25 17:42:50 +00:00
const { size } = fs.statSync(renderedRequest.body.fileName);
2017-07-18 20:38:19 +00:00
const fileName = renderedRequest.body.fileName || '';
const fd = fs.openSync(fileName, 'r');
2017-07-26 02:40:32 +00:00
setOpt(Curl.option.INFILESIZE_LARGE, size);
setOpt(Curl.option.UPLOAD, 1);
setOpt(Curl.option.READDATA, fd);
2017-07-26 02:40:32 +00:00
// We need this, otherwise curl will send it as a POST
setOpt(Curl.option.CUSTOMREQUEST, renderedRequest.method);
const fn = () => fs.closeSync(fd);
curl.on('end', fn);
curl.on('error', fn);
} else if (typeof renderedRequest.body.mimeType === 'string' || expectsBody) {
requestBody = renderedRequest.body.text || '';
} else {
// No body
2017-04-09 21:53:46 +00:00
noBody = true;
}
if (!noBody) {
// Don't chunk uploads
2018-06-25 17:42:50 +00:00
headers.push({ name: 'Expect', value: DISABLE_HEADER_VALUE });
headers.push({
name: 'Transfer-Encoding',
value: DISABLE_HEADER_VALUE,
2018-06-25 17:42:50 +00:00
});
}
// If we calculated the body within Insomnia (ie. not computed by Curl)
if (requestBody !== null) {
setOpt(Curl.option.POSTFIELDS, requestBody);
}
// Handle Authorization header
if (!hasAuthHeader(headers) && !renderedRequest.authentication.disabled) {
if (renderedRequest.authentication.type === AUTH_BASIC) {
2018-06-25 17:42:50 +00:00
const { username, password } = renderedRequest.authentication;
setOpt(Curl.option.HTTPAUTH, Curl.auth.BASIC);
setOpt(Curl.option.USERNAME, username || '');
setOpt(Curl.option.PASSWORD, password || '');
} else if (renderedRequest.authentication.type === AUTH_DIGEST) {
2018-06-25 17:42:50 +00:00
const { username, password } = renderedRequest.authentication;
setOpt(Curl.option.HTTPAUTH, Curl.auth.DIGEST);
setOpt(Curl.option.USERNAME, username || '');
setOpt(Curl.option.PASSWORD, password || '');
} else if (renderedRequest.authentication.type === AUTH_NTLM) {
2018-06-25 17:42:50 +00:00
const { username, password } = renderedRequest.authentication;
setOpt(Curl.option.HTTPAUTH, Curl.auth.NTLM);
setOpt(Curl.option.USERNAME, username || '');
setOpt(Curl.option.PASSWORD, password || '');
} else if (renderedRequest.authentication.type === AUTH_AWS_IAM) {
2017-08-01 22:13:24 +00:00
if (!noBody && !requestBody) {
return handleError(
new Error('AWS authentication not supported for provided body type'),
2018-06-25 17:42:50 +00:00
);
}
2018-06-25 17:42:50 +00:00
const { authentication } = renderedRequest;
const credentials = {
2018-04-30 12:48:00 +00:00
accessKeyId: authentication.accessKeyId || '',
secretAccessKey: authentication.secretAccessKey || '',
sessionToken: authentication.sessionToken || '',
};
const extraHeaders = _getAwsAuthHeaders(
credentials,
headers,
2017-08-01 22:13:24 +00:00
requestBody || '',
finalUrl,
2018-04-30 12:48:00 +00:00
renderedRequest.method,
authentication.region || '',
authentication.service || '',
);
for (const header of extraHeaders) {
headers.push(header);
}
} else if (renderedRequest.authentication.type === AUTH_NETRC) {
setOpt(Curl.option.NETRC, Curl.netrc.REQUIRED);
} else {
const authHeader = await getAuthHeader(
renderedRequest._id,
finalUrl,
renderedRequest.method,
renderedRequest.authentication,
);
if (authHeader) {
2017-11-06 20:44:55 +00:00
headers.push({
name: authHeader.name,
value: authHeader.value,
2017-11-06 20:44:55 +00:00
});
}
}
}
2017-11-12 18:35:01 +00:00
// Send a default Accept headers of anything
if (!hasAcceptHeader(headers)) {
2018-06-25 17:42:50 +00:00
headers.push({ name: 'Accept', value: '*/*' }); // Default to anything
2017-11-12 18:35:01 +00:00
}
// Don't auto-send Accept-Encoding header
if (!hasAcceptEncodingHeader(headers)) {
2018-06-25 17:42:50 +00:00
headers.push({ name: 'Accept-Encoding', value: DISABLE_HEADER_VALUE });
2017-11-12 18:35:01 +00:00
}
// Set User-Agent if it't not already in headers
if (!hasUserAgentHeader(headers)) {
setOpt(Curl.option.USERAGENT, `insomnia/${getAppVersion()}`);
}
// Prevent curl from adding default content-type header
if (!hasContentTypeHeader(headers)) {
2018-06-25 17:42:50 +00:00
headers.push({ name: 'content-type', value: DISABLE_HEADER_VALUE });
}
// NOTE: This is last because headers might be modified multiple times
2018-06-25 17:42:50 +00:00
const headerStrings = headers.filter(h => h.name).map(h => {
const value = h.value || '';
if (value === '') {
// Curl needs a semicolon suffix to send empty header values
return `${h.name};`;
} else if (value === DISABLE_HEADER_VALUE) {
// Tell Curl NOT to send the header if value is null
return `${h.name}:`;
} else {
// Send normal header value
return `${h.name}: ${value}`;
}
});
setOpt(Curl.option.HTTPHEADER, headerStrings);
let responseBodyBytes = 0;
const responsesDir = pathJoin(getDataDirectory(), 'responses');
mkdirp.sync(responsesDir);
const responseBodyPath = pathJoin(responsesDir, uuid.v4() + '.response');
const responseBodyWriteStream = fs.createWriteStream(responseBodyPath);
curl.on('end', () => responseBodyWriteStream.end());
curl.on('error', () => responseBodyWriteStream.end());
setOpt(Curl.option.WRITEFUNCTION, (buff: Buffer) => {
responseBodyBytes += buff.length;
responseBodyWriteStream.write(buff);
return buff.length;
});
// Handle the response ending
2017-11-13 23:10:53 +00:00
curl.on('end', async (_1, _2, rawHeaders) => {
const allCurlHeadersObjects = _parseHeaders(rawHeaders);
// Headers are an array (one for each redirect)
const lastCurlHeadersObject = allCurlHeadersObjects[allCurlHeadersObjects.length - 1];
// Collect various things
const httpVersion = lastCurlHeadersObject.version || '';
2017-11-13 23:10:53 +00:00
const statusCode = lastCurlHeadersObject.code || -1;
const statusMessage = lastCurlHeadersObject.reason || '';
// Collect the headers
2017-11-13 23:10:53 +00:00
const headers = lastCurlHeadersObject.headers;
// Calculate the content type
const contentTypeHeader = getContentTypeHeader(headers);
const contentType = contentTypeHeader ? contentTypeHeader.value : '';
// Update Cookie Jar
let currentUrl = finalUrl;
2017-11-13 23:10:53 +00:00
let setCookieStrings: Array<string> = [];
const jar = jarFromCookies(renderedRequest.cookieJar.cookies);
2018-06-25 17:42:50 +00:00
for (const { headers } of allCurlHeadersObjects) {
// Collect Set-Cookie headers
2017-11-13 23:10:53 +00:00
const setCookieHeaders = getSetCookieHeaders(headers);
setCookieStrings = [...setCookieStrings, ...setCookieHeaders.map(h => h.value)];
// Pull out new URL if there is a redirect
2017-11-13 23:10:53 +00:00
const newLocation = getLocationHeader(headers);
if (newLocation !== null) {
2017-11-13 23:10:53 +00:00
currentUrl = urlResolve(currentUrl, newLocation.value);
}
}
// Update jar with Set-Cookie headers
for (const setCookieStr of setCookieStrings) {
try {
jar.setCookieSync(setCookieStr, currentUrl);
} catch (err) {
addTimelineText(`Rejected cookie: ${err.message}`);
}
}
// Update cookie jar if we need to and if we found any cookies
if (renderedRequest.settingStoreCookies && setCookieStrings.length) {
const cookies = await cookiesFromJar(jar);
2018-06-25 17:42:50 +00:00
models.cookieJar.update(renderedRequest.cookieJar, { cookies });
}
// Print informational message
if (setCookieStrings.length > 0) {
const n = setCookieStrings.length;
if (renderedRequest.settingStoreCookies) {
addTimelineText(`Saved ${n} cookie${n === 1 ? '' : 's'}`);
} else {
addTimelineText(`Ignored ${n} cookie${n === 1 ? '' : 's'}`);
}
}
// Return the response data
const responsePatch = {
headers,
contentType,
statusCode,
2017-11-13 23:10:53 +00:00
httpVersion,
statusMessage,
elapsedTime: curl.getInfo(Curl.info.TOTAL_TIME) * 1000,
bytesRead: curl.getInfo(Curl.info.SIZE_DOWNLOAD),
bytesContent: responseBodyBytes,
url: curl.getInfo(Curl.info.EFFECTIVE_URL),
};
// Close the request
2017-11-13 23:10:53 +00:00
curl.close();
// Make sure the response body has been fully written first
await waitForStreamToFinish(responseBodyWriteStream);
respond(responsePatch, responseBodyPath);
});
2018-06-25 17:42:50 +00:00
curl.on('error', function(err, code) {
let error = err + '';
let statusMessage = 'Error';
if (code === Curl.code.CURLE_ABORTED_BY_CALLBACK) {
error = 'Request aborted';
statusMessage = 'Abort';
}
2018-06-25 17:42:50 +00:00
respond({ statusMessage, error }, null, true);
});
curl.perform();
} catch (err) {
handleError(err);
}
});
}
2018-06-25 17:42:50 +00:00
export async function sendWithSettings(
requestId: string,
requestPatch: Object,
): Promise<ResponsePatch> {
const request = await models.request.getById(requestId);
if (!request) {
throw new Error(`Failed to find request: ${requestId}`);
}
const settings = await models.settings.getOrCreate();
const ancestors = await db.withAncestors(request, [
2018-01-17 04:20:45 +00:00
models.request.type,
models.requestGroup.type,
models.workspace.type,
]);
const workspaceDoc = ancestors.find(doc => doc.type === models.workspace.type);
const workspaceId = workspaceDoc ? workspaceDoc._id : 'n/a';
const workspace = await models.workspace.getById(workspaceId);
if (!workspace) {
throw new Error(`Failed to find workspace for: ${requestId}`);
}
const workspaceMeta = await models.workspaceMeta.getOrCreateByParentId(workspace._id);
const environmentId: string = workspaceMeta.activeEnvironmentId || 'n/a';
const newRequest: Request = await models.initModel(models.request.type, requestPatch, {
_id: request._id + '.other',
parentId: request._id,
});
2018-04-30 12:48:00 +00:00
let renderResult: { request: RenderedRequest, context: Object };
try {
renderResult = await getRenderedRequestAndContext(newRequest, environmentId);
} catch (err) {
throw new Error(`Failed to render request: ${requestId}`);
}
return _actuallySend(renderResult.request, renderResult.context, workspace, settings);
}
export async function send(requestId: string, environmentId: string): Promise<ResponsePatch> {
// HACK: wait for all debounces to finish
/*
* TODO: Do this in a more robust way
* The following block adds a "long" delay to let potential debounces and
* database updates finish before making the request. This is done by tracking
* the time of the user's last keypress and making sure the request is sent a
* significant time after the last press.
*/
const timeSinceLastInteraction = Date.now() - lastUserInteraction;
const delayMillis = Math.max(0, MAX_DELAY_TIME - timeSinceLastInteraction);
if (delayMillis > 0) {
await delay(delayMillis);
}
// Fetch some things
const request = await models.request.getById(requestId);
const settings = await models.settings.getOrCreate();
const ancestors = await db.withAncestors(request, [
models.request.type,
models.requestGroup.type,
models.workspace.type,
]);
if (!request) {
throw new Error(`Failed to find request to send for ${requestId}`);
}
const renderResult = await getRenderedRequestAndContext(
request,
environmentId,
RENDER_PURPOSE_SEND,
);
const renderedRequestBeforePlugins = renderResult.request;
const renderedContextBeforePlugins = renderResult.context;
const workspaceDoc = ancestors.find(doc => doc.type === models.workspace.type);
const workspace = await models.workspace.getById(workspaceDoc ? workspaceDoc._id : 'n/a');
if (!workspace) {
throw new Error(`Failed to find workspace for request: ${requestId}`);
}
let renderedRequest: RenderedRequest;
try {
renderedRequest = await _applyRequestPluginHooks(
renderedRequestBeforePlugins,
renderedContextBeforePlugins,
);
} catch (err) {
return {
url: renderedRequestBeforePlugins.url,
parentId: renderedRequestBeforePlugins._id,
error: err.message,
statusCode: STATUS_CODE_PLUGIN_ERROR,
statusMessage: err.plugin ? `Plugin ${err.plugin.name}` : 'Plugin',
settingSendCookies: renderedRequestBeforePlugins.settingSendCookies,
settingStoreCookies: renderedRequestBeforePlugins.settingStoreCookies,
};
}
return _actuallySend(renderedRequest, renderedContextBeforePlugins, workspace, settings);
}
2018-06-25 17:42:50 +00:00
async function _applyRequestPluginHooks(
renderedRequest: RenderedRequest,
renderedContext: Object,
): Promise<RenderedRequest> {
const newRenderedRequest = clone(renderedRequest);
2018-06-25 17:42:50 +00:00
for (const { plugin, hook } of await plugins.getRequestHooks()) {
const context = {
...pluginContexts.app.init(),
...pluginContexts.store.init(plugin),
...pluginContexts.request.init(newRenderedRequest, renderedContext),
};
try {
await hook(context);
} catch (err) {
err.plugin = plugin;
throw err;
}
}
return newRenderedRequest;
}
2018-06-25 17:42:50 +00:00
async function _applyResponsePluginHooks(
response: ResponsePatch,
request: RenderedRequest,
renderContext: Object,
): Promise<ResponsePatch> {
const newResponse = clone(response);
const newRequest = clone(request);
2018-06-25 17:42:50 +00:00
for (const { plugin, hook } of await plugins.getResponseHooks()) {
const context = {
...pluginContexts.app.init(),
...pluginContexts.store.init(plugin),
...pluginContexts.response.init(newResponse),
...pluginContexts.request.init(newRequest, renderContext, true),
};
try {
await hook(context);
} catch (err) {
err.plugin = plugin;
throw err;
}
}
return newResponse;
}
2018-06-25 17:42:50 +00:00
export function _parseHeaders(
buffer: Buffer,
2018-06-25 17:42:50 +00:00
): Array<{
headers: Array<ResponseHeader>,
version: string,
code: number,
reason: string,
2018-06-25 17:42:50 +00:00
}> {
2017-11-13 23:10:53 +00:00
const results = [];
const lines = buffer.toString('utf8').split(/\r?\n|\r/g);
for (let i = 0, currentResult = null; i < lines.length; i++) {
const line = lines[i];
const isEmptyLine = line.trim() === '';
// If we hit an empty line, start parsing the next response
if (isEmptyLine && currentResult) {
results.push(currentResult);
currentResult = null;
continue;
}
if (!currentResult) {
const [version, code, ...other] = line.split(/ +/g);
currentResult = {
version,
code: parseInt(code, 10),
reason: other.join(' '),
headers: [],
2017-11-13 23:10:53 +00:00
};
} else {
const [name, value] = line.split(/:\s(.+)/);
2018-06-25 17:42:50 +00:00
const header: ResponseHeader = { name, value: value || '' };
2017-11-13 23:10:53 +00:00
currentResult.headers.push(header);
}
}
2017-11-13 23:10:53 +00:00
return results;
}
// exported for unit tests only
2018-06-25 17:42:50 +00:00
export function _getAwsAuthHeaders(
2018-04-30 12:48:00 +00:00
credentials: {
accessKeyId: string,
secretAccessKey: string,
sessionToken: string,
2018-04-30 12:48:00 +00:00
},
2017-07-19 02:54:03 +00:00
headers: Array<RequestHeader>,
2017-07-18 22:10:57 +00:00
body: string,
url: string,
2018-04-30 12:48:00 +00:00
method: string,
region?: string,
service?: string,
): Array<{ name: string, value: string, disabled?: boolean }> {
const parsedUrl = urlParse(url);
const contentTypeHeader = getContentTypeHeader(headers);
// AWS uses host header for signing so prioritize that if the user set it manually
const hostHeader = getHostHeader(headers);
const host = hostHeader ? hostHeader.value : parsedUrl.host;
const awsSignOptions = {
2018-04-30 12:48:00 +00:00
service,
region,
host,
body,
method,
path: parsedUrl.path,
headers: contentTypeHeader ? { 'content-type': contentTypeHeader.value } : {},
};
const signature = aws4.sign(awsSignOptions, credentials);
return Object.keys(signature.headers)
.filter(name => name !== 'content-type') // Don't add this because we already have it
2018-06-25 17:42:50 +00:00
.map(name => ({ name, value: signature.headers[name] }));
}
2017-07-18 20:38:19 +00:00
document.addEventListener('keydown', (e: KeyboardEvent) => {
if (e.ctrlKey || e.metaKey || e.altKey) {
return;
}
lastUserInteraction = Date.now();
});
document.addEventListener('paste', (e: Event) => {
lastUserInteraction = Date.now();
});