insomnia/app/backend/export/curl.js
Gregory Schier 318c35c2cb Move a bunch of stuff to async/await (#39)
* Some minor implementations

* Some more

* Lots more

* Removed 'backend' alias

* removed all promises

* Removed a bunch of module exports stuff

* Some morE'

* Fix

* custom DNS

* Tests for DNS

* bug fix

* Some small adjustments

* Small stuff
2016-10-02 13:57:00 -07:00

193 lines
5.1 KiB
JavaScript

'use strict';
const FLAGS = [
'cacert', 'capath', 'E', 'cert', 'cert-type', 'ciphers', 'K', 'config',
'connect-timeout', 'C', 'continue-at', 'b',
'cookie', // TODO: Handle this
'c', 'cookie-jar', 'crlfile', 'd', 'data', 'data-ascii', 'data-binary',
'data-urlencode', 'delegation', 'D', 'dump-header', 'egd-file',
'engine', 'F', 'form', 'form-string', 'ftp-account', 'ftp-method',
'ftp-port', 'H', 'header', 'hostpubmd5', 'interface', 'keepalive-time',
'key', 'key-type', 'krb', 'lib-curl', 'limit-rate', 'local-port',
'mail-from', 'mail-rcpt', 'mail-auth', 'max-filesize', 'max-redirs',
'max-time', 'netrtc-file', 'output', 'pass', 'proto', 'proto-redir',
'proxy', 'proxy-user', 'proxy1.0', 'pubkey', 'Q', 'quote',
'random-file', 'range', 'X', 'request', 'resolve', 'retry',
'retry-delay', 'retry-max-time', 'socks4', 'socks4a', 'socks5',
'socks5-hostname', 'socks5-gssapi-service', 'Y', 'speed-limit',
'y', 'speed-time', 'stderr', 'tftp-blksize', 'z', 'time-cond', 'trace',
'trace-ascii', 'T', 'upload-file',
'url', // TODO: Handle this
'u', 'user', 'tlsuser', 'tlspassword', 'tlsauthtype',
'A', 'user-agent', // TODO: Handle this
'w', 'write-out'
];
const FLAG_REGEXES = [
/\s--([\w\-]{2,})\s+"((?:[^"\\]|\\.)*)"/, // --my-flag "hello"
/\s--([\w\-]{2,})\s+'((?:[^'\\]|\\.)*)'/, // --my-flag 'hello'
/\s--([\w\-]{2,})\s+([^\s]+)/, // --my-flag hello
/\s-([\w])\s*'((?:[^'\\]|\\.)*)'/, // -X 'hello'
/\s-([\w])\s*"((?:[^"\\]|\\.)*)"/, // -X "hello"
/\s-([\w])\s*([^\s]+)/, // -X hello
/\s--([\w\-]+)/ // --switch (cleanup at the end)
];
function getFlags (cmd) {
const flags = {};
const switches = {};
for (var i = 0; i < FLAG_REGEXES.length; i++) {
var matches = [];
var match;
var key;
var val;
var matchedFlag;
// Stop at 1000 to prevent infinite loops
// TODO: Make this recursive
for (var j = 0; (matches = FLAG_REGEXES[i].exec(cmd)); j++) {
if (j > 1000) {
console.error('INFINITE LOOP');
break;
}
match = matches[0];
key = matches[1];
val = matches[2];
matchedFlag = !!val;
if (matchedFlag) {
if (isFlag(key)) {
// Matched a flag
flags[key] = flags[key] || [];
flags[key].push(val);
cmd = cmd.replace(match, '');
} else {
// Matched a flag that was actually a switch
cmd = cmd.replace('--' + key, '');
}
} else {
// Matched a switch directly without a value
switches[key] = true;
cmd = cmd.replace(match, '');
}
}
}
return {cmd, flags, switches};
}
function isFlag (key) {
for (var i = 0; i < FLAGS.length; i++) {
if (key === FLAGS[i]) {
return true;
}
}
return false;
}
function splitHeaders (flags) {
var headers = (flags['H'] || []).concat(flags['header'] || []);
var parsed = [];
for (var i = 0; i < headers.length; i++) {
var header = headers[i].split(':');
var name = header[0].trim();
var value = header[1].trim();
parsed.push({name: name, value: value});
}
return parsed;
}
function getContentType (headers) {
for (var i = 0; i < headers.length; i++) {
var header = headers[i];
if (header.name.toLowerCase() === 'content-type') {
return header.value;
}
}
return null;
}
function getHttpMethod (flags, hasBody) {
var method = (flags['X'] || flags['request'] || [])[0];
// If there is no method specified, but there is a body, default
// to POST, else default to GET
if (!method) {
method = hasBody ? 'POST' : 'GET';
}
return method;
}
function getBasicAuth (flags) {
var authString = flags.u || flags.user;
var auth = {
username: '',
password: ''
};
if (authString) {
var authSplit = authString[0].split(':');
auth.username = (authSplit[0] || '').trim();
auth.password = (authSplit[1] || '').trim();
}
return auth;
}
export function importCurl (blob) {
if (!blob || blob.toLowerCase().indexOf('curl ') !== 0) {
return false;
}
// Rip out the flags
let {cmd, flags, switches} = getFlags(blob);
// Final values
const headers = splitHeaders(flags);
let body = (
flags.d ||
flags.data ||
flags['data-binary'] ||
flags['data-ascii'] ||
[]
)[0] || '';
const contentType = getContentType(headers) || null;
if (contentType && contentType.toLowerCase() === 'application/json') {
try {
body = JSON.stringify(JSON.parse(body), null, '\t');
} catch (e) {
}
}
const httpMethod = getHttpMethod(flags, !!body);
const authentication = getBasicAuth(flags);
// Clean up the remaining URL
cmd = (cmd + ' ')
.replace(/\\ /g, ' ').replace(/ \\/g, ' ') // slashes
.replace(/\s/g, ' ') // whitespaces
.replace(/ "/g, ' ').replace(/" /g, ' ') // double-quotes
.replace(/ '/g, ' ').replace(/' /g, ' '); // single-quotes
const url = /curl\s+['"]?((?!('|")).*)['"]?/.exec(cmd)[1].trim();
return {
url: url,
body: body,
headers: headers,
contentType: contentType,
method: httpMethod,
authentication: authentication
};
}