whistle/lib/rules/protocols.js

var protocols = [
  'G',
  'style',
  'host',
  'rule',
  'pipe',
  'weinre',
  'proxy',
  'https2http-proxy',
  'http2https-proxy',
  'internal-proxy',
  'pac',
  'filter',
  'ignore',
  'enable',
  'disable',
  'delete',
  'log',
  'plugin',
  'referer',
  'auth',
  'ua',
  'urlParams',
  'params',
  'resMerge',
  'replaceStatus',
  'method',
  'cache',
  'attachment',
  'forwardedFor',
  'responseFor',
  'rulesFile',
  'resScript',
  'frameScript',
  'reqDelay',
  'resDelay',
  'headerReplace',
  'reqSpeed',
  'resSpeed',
  'reqType',
  'resType',
  'reqCharset',
  'resCharset',
  'reqCookies',
  'resCookies',
  'reqCors',
  'resCors',
  'reqHeaders',
  'resHeaders',
  'trailers',
  'reqPrepend',
  'resPrepend',
  'reqBody',
  'resBody',
  'reqAppend',
  'resAppend',
  'urlReplace',
  'reqReplace',
  'resReplace',
  'reqWrite',
  'resWrite',
  'reqWriteRaw',
  'resWriteRaw',
  'cssAppend',
  'htmlAppend',
  'jsAppend',
  'cssBody',
  'htmlBody',
  'jsBody',
  'cssPrepend',
  'htmlPrepend',
  'jsPrepend',
  'cipher',
  'sniCallback'
];
var toolProtocols = ['log', 'weinre'];

var RULE_RE = /^(?:|x|xs)(?:file|rawfile|dust|tpl|jsonp):/;
var LOC_RE = /^locationHref:/;
var PROXY_RE =
  /^x?(?:socks|proxy|https?-proxy|internal-proxy|internal-https?-proxy|https2http-proxy|http2https-proxy)$/;
var pureResProtocols = [
  'replaceStatus',
  'cache',
  'attachment',
  'resMerge',
  'resDelay',
  'resSpeed',
  'resType',
  'resCharset',
  'resCookies',
  'resCors',
  'resHeaders',
  'trailers',
  'resPrepend',
  'resBody',
  'resAppend',
  'resReplace',
  'resWrite',
  'resWriteRaw',
  'cssAppend',
  'htmlAppend',
  'jsAppend',
  'cssBody',
  'htmlBody',
  'jsBody',
  'cssPrepend',
  'htmlPrepend',
  'jsPrepend',
  'responseFor'
].concat(toolProtocols);
var resProtocols = [
  'filter',
  'enable',
  'disable',
  'ignore',
  'style',
  'delete',
  'headerReplace'
].concat(pureResProtocols);
var binProtocols = [
  'reqPrepend',
  'resPrepend',
  'reqBody',
  'resBody',
  'reqAppend',
  'resAppend'
];
var jsProtocols = ['jsAppend', 'jsBody', 'jsPrepend'];
var cssProtocols = ['cssAppend', 'cssBody', 'cssPrepend'];
var aliasProtocols = {
  ruleFile: 'rulesFile',
  ruleScript: 'rulesFile',
  rulesScript: 'rulesFile',
  reqScript: 'rulesFile',
  reqRules: 'rulesFile',
  resRules: 'resScript',
  pathReplace: 'urlReplace',
  download: 'attachment',
  skip: 'ignore',
  'http-proxy': 'proxy',
  'xhttp-proxy': 'xproxy',
  status: 'statusCode',
  hosts: 'host',
  xhost: 'host',
  html: 'htmlAppend',
  js: 'jsAppend',
  reqMerge: 'params',
  tlsOptions: 'cipher',
  css: 'cssAppend',
  excludeFilter: 'filter',
  includeFilter: 'filter',
  P: 'G'
};
var protocolsWithoutG = protocols.slice(1);
var reqProtocols = protocols.filter(function (name) {
  return pureResProtocols.indexOf(name) === -1;
}).concat(toolProtocols);
var reqProtosWithoutG = reqProtocols.filter(function (name) {
  return name !== 'G';
});

var RULE_PROTO_RE = /^([\w.-]+):\/\//;

exports.getRuleProto = function(rule) {
  if (!RULE_PROTO_RE.test(rule.matcher)) {
    return;
  }
  var proto = RegExp.$1;
  var ruleProto = aliasProtocols[proto];
  if (!ruleProto || ruleProto === 'filter') {
    return proto;
  }
  if (ruleProto === 'rulesFile') {
    return 'reqScript';
  }
  if (ruleProto === 'urlReplace') {
    return 'pathReplace';
  }
  if (ruleProto === 'params') {
    return 'reqMerge';
  }
  return ruleProto;
};

exports.multiMatchs = [
  'G',
  'ignore',
  'enable',
  'filter',
  'disable',
  'plugin',
  'delete',
  'style',
  'cipher',
  'trailers',
  'urlParams',
  'params',
  'headerReplace',
  'reqHeaders',
  'resHeaders',
  'reqCors',
  'resCors',
  'reqCookies',
  'resCookies',
  'reqReplace',
  'urlReplace',
  'resReplace',
  'resMerge',
  'reqBody',
  'reqPrepend',
  'resPrepend',
  'reqAppend',
  'resAppend',
  'resBody',
  'htmlAppend',
  'jsAppend',
  'cssAppend',
  'htmlBody',
  'jsBody',
  'cssBody',
  'htmlPrepend',
  'jsPrepend',
  'cssPrepend',
  'rulesFile',
  'resScript'
];
exports.protocols = protocols;
exports.protocolsWithoutG = protocolsWithoutG;
exports.pureResProtocols = pureResProtocols;
exports.reqProtocols = reqProtocols;
exports.reqProtosWithoutG = reqProtosWithoutG;
exports.resProtocols = resProtocols;
exports.aliasProtocols = aliasProtocols;

function getRules() {
  return resetRules({});
}

exports.getRules = getRules;

function isBinProtocol(protocol) {
  if (binProtocols.indexOf(protocol) != -1) {
    return 1;
  }
  if (jsProtocols.indexOf(protocol) != -1) {
    return 2;
  }
  if (cssProtocols.indexOf(protocol) != -1) {
    return 3;
  }
}

exports.isBinProtocol = isBinProtocol;

function resetRules(rules) {
  protocols.forEach(function (protocol) {
    rules[protocol] = [];
  });
  rules._localRule = [];
  rules._bodyFilters = [];
  rules._clientCerts = [];
  rules.rule.isRuleProto = true;
  return rules;
}

exports.resetRules = resetRules;

function isWebProtocol(protocol) {
  return protocol == 'http:' || protocol == 'https:';
}

exports.isWebProtocol = isWebProtocol;

function isWebsocketProtocol(protocol) {
  return protocol == 'ws:' || protocol == 'wss:';
}

exports.isWebsocketProtocol = isWebsocketProtocol;

function isFileProxy(protocol) {
  return RULE_RE.test(protocol) || LOC_RE.test(protocol);
}

exports.isFileProxy = isFileProxy;

exports.isLocHref = function(protocol) {
  return LOC_RE.test(protocol);
};

function contains(name) {
  if (
    protocols.indexOf(name) != -1 ||
    aliasProtocols[name] ||
    PROXY_RE.test(name)
  ) {
    return true;
  }
  name += ':';
  return (
    isWebsocketProtocol(name) ||
    isWebProtocol(name) ||
    isFileProxy(name) ||
    name == 'tunnel:'
  );
}

exports.contains = contains;