'use strict'; const networkRequest = require('request'); const {parse: urlParse, format: urlFormat} = require('url'); const db = require('./database'); const querystring = require('./querystring'); const {DEBOUNCE_MILLIS, STATUS_CODE_PEBKAC} = require('./constants'); const {jarFromCookies, cookiesFromJar} = require('./cookies'); const {setDefaultProtocol} = require('./util'); const {getRenderedRequest} = require('./render'); let cancelRequestFunction = null; module.exports.cancelCurrentRequest = () => { if (typeof cancelRequestFunction === 'function') { cancelRequestFunction(); } }; module.exports._buildRequestConfig = (renderedRequest, patch = {}) => { const config = { method: renderedRequest.method, body: renderedRequest.body, headers: {}, // Setup redirect rules followAllRedirects: true, maxRedirects: 20, timeout: 0, // Unzip gzipped responses gzip: true, // Time the request time: true, // SSL Checking rejectUnauthorized: true, // Proxy proxy: null }; // Set the URL, including the query parameters const qs = querystring.buildFromParams(renderedRequest.parameters); const url = querystring.joinURL(renderedRequest.url, qs); // Encode path portion of URL const parsedUrl = urlParse(url); parsedUrl.pathname = encodeURI(parsedUrl.pathname || ''); config.url = urlFormat(parsedUrl); config.headers.host = parsedUrl.host; for (let i = 0; i < renderedRequest.headers.length; i++) { let header = renderedRequest.headers[i]; if (header.name) { config.headers[header.name] = header.value; } } return Object.assign(config, patch); }; module.exports._actuallySend = (renderedRequest, settings) => { return new Promise((resolve, reject) => { const cookieJar = renderedRequest.cookieJar; const jar = jarFromCookies(cookieJar.cookies); // Detect and set the proxy based on the request protocol // NOTE: request does not have a separate settings for http/https proxies const {protocol} = urlParse(renderedRequest.url); const proxyHost = protocol === 'https:' ? settings.httpsProxy : settings.httpProxy; const proxy = proxyHost ? setDefaultProtocol(proxyHost) : null; let config = module.exports._buildRequestConfig(renderedRequest, { jar: jar, proxy: proxy, followAllRedirects: settings.followRedirects, timeout: settings.timeout > 0 ? settings.timeout : null, rejectUnauthorized: settings.validateSSL }, true); const startTime = Date.now(); // TODO: Handle redirects ourselves const req = networkRequest(config, function (err, networkResponse) { if (err) { db.responseCreate({ parentId: renderedRequest._id, elapsedTime: Date.now() - startTime, error: err.toString() }); return reject(err); } // TODO: Add image support to Insomnia const contentType = networkResponse.headers['content-type']; if (contentType && contentType.toLowerCase().indexOf('image/') === 0) { const err = new Error(`Content-Type ${contentType} not supported`); db.responseCreate({ parentId: renderedRequest._id, elapsedTime: Date.now() - startTime, error: err.toString(), statusMessage: 'UNSUPPORTED' }); return reject(err); } // Update the cookie jar cookiesFromJar(jar).then(cookies => { db.cookieJarUpdate(cookieJar, {cookies}); }); // Format the headers into Insomnia format // TODO: Move this to a better place const headers = []; for (const name of Object.keys(networkResponse.headers)) { const tmp = networkResponse.headers[name]; const values = Array.isArray(tmp) ? tmp : [tmp]; for (const value of values) { headers.push({name, value}); } } const responsePatch = { parentId: renderedRequest._id, statusCode: networkResponse.statusCode, statusMessage: networkResponse.statusMessage, contentType: networkResponse.headers['content-type'], url: config.url, // TODO: Handle redirects somehow elapsedTime: networkResponse.elapsedTime, bytesRead: networkResponse.connection.bytesRead, body: networkResponse.body, headers: headers }; db.responseCreate(responsePatch).then(resolve, reject); }); // Kind of hacky, but this is how we cancel a request. cancelRequestFunction = () => { req.abort(); db.responseCreate({ parentId: renderedRequest._id, elapsedTime: Date.now() - startTime, statusMessage: 'Cancelled', error: 'The request was cancelled' }); return reject('Cancelled'); } }) }; module.exports.send = requestId => { return new Promise((resolve, reject) => { // First, lets wait for all debounces to finish setTimeout(() => { Promise.all([ db.requestGetById(requestId), db.settingsGetOrCreate() ]).then(([request, settings]) => { getRenderedRequest(request).then(renderedRequest => { module.exports._actuallySend(renderedRequest, settings).then(resolve, reject); }, err => { db.responseCreate({ parentId: request._id, statusCode: STATUS_CODE_PEBKAC, error: err.message }).then(resolve, reject); }); }) }, DEBOUNCE_MILLIS); }) };