local-traffic 0.0.31 → 0.0.34

package/README.md

@@ -62,3 +62,4 @@ npx local-traffic [location-of-the-local-traffic-config-file]
 - "replaceResponseBodyUrls": (boolean) replace every matching string from the mapping in the response body.
 - "dontUseHttp2Downstream": (boolean) force calling downstream services in http1.1 only (to save some time)
 - "simpleLogs": (boolean) disable colored logs for text terminals
+- "websocket": (boolean) true to activate websocket connections proxying via sockets

package/dist/localTraffic.js

@@ -1,2 +1,2 @@
 #!/usr/bin/env node
-"use strict";Object.defineProperty(exports,"__esModule",{value:!0});const e=require("http2"),t=require("http"),r=require("https"),n=require("url"),o=require("fs"),s=require("zlib"),i=require("path");var a;!function(e){e[e.ERROR=124]="ERROR",e[e.SUCCESS=35]="SUCCESS",e[e.INFO=21]="INFO",e[e.WARNING=172]="WARNING"}(a||(a={}));const l=(0,i.resolve)(process.env.HOME,".local-traffic.json"),p=(0,i.resolve)(process.cwd(),process.argv.slice(-1)[0].endsWith(".json")?process.argv.slice(-1)[0]:l),c={mapping:{},port:8080,replaceResponseBodyUrls:!1,dontUseHttp2Downstream:!1,simpleLogs:!1};let d,h;const u=(e,t,r)=>{console.log(`${(e=>{const t=new Date;return`${e?"":""}${`${t.getHours()}`.padStart(2,"0")}${e?":":":"}${`${t.getMinutes()}`.padStart(2,"0")}${e?":":":"}${`${t.getSeconds()}`.padStart(2,"0")}${e?"":""}`})(d.simpleLogs)} ${d.simpleLogs?e.replace(/⎸/g,"|").replace(/⎹/g,"|").replace(/\u001b\[[^m]*m/g,"").replace(/↘️/g,"inbound").replace(/☎️/g,"port").replace(/↗️/g,"outbound"):t?`[48;5;${t}m⎸    ${process.stdout.isTTY&&r||""}  ${e.padEnd(36)} ⎹`:e}`)},m=async(e=!0)=>new Promise((t=>(0,o.readFile)(p,((r,n)=>{r&&!e&&u("config error. Using default value",a.ERROR,"❌");try{d=Object.assign({},c,JSON.parse((n||"{}").toString()))}catch(e){return u("config syntax incorrect, aborting",a.ERROR,"⛈️"),d=d||{...c},void t(d)}d.mapping[""]||u('default mapping "" not provided.',a.WARNING,"☢️"),r&&"ENOENT"===r.code&&e&&p===l?(0,o.writeFile)(p,JSON.stringify(c),(e=>{e?u("config file NOT created",a.ERROR,"⁉️"):u("config file created",a.SUCCESS,"✨"),t(d)})):t(d)})))).then((()=>{e&&(0,o.watchFile)(p,f)})),g=e=>{u(`⎸ ↘️  : ${e.ssl?"HTTP/2  ":"HTTP 1.1"} ⎸ ☎️  : ${e.port.toString().padStart(5)} ⎸ ↗️  : ${e.dontUseHttp2Downstream?"HTTP 1.1":"HTTP/2  "} ⎹`)},f=async()=>{const e={...d};return await m(!1),isNaN(d.port)||d.port>65535||d.port<0?(d=e,void u("port number invalid. Not refreshing",a.ERROR,"☎️")):"object"!=typeof d.mapping?(d=e,void u("mapping should be an object. Aborting",a.ERROR,"⚡")):(d.replaceResponseBodyUrls&&!e.replaceResponseBodyUrls&&u("response body url replacement",a.INFO,"✔️"),!d.replaceResponseBodyUrls&&e.replaceResponseBodyUrls&&u("response body url NO replacement",a.INFO,"✖️"),u(`${Object.keys(d.mapping).length.toString().padStart(5)} loaded mapping rules`,a.SUCCESS,"↻"),void(d.port!==e.port||JSON.stringify(d.ssl)!==JSON.stringify(e.ssl)?(await new Promise((e=>h?h.close(e):e(void 0))),v()):d.dontUseHttp2Downstream!==e.dontUseHttp2Downstream&&g(d)))},$=e=>""==e?"":(0,i.normalize)(e).replace(/\\/g,"/"),b=e=>{const t=(0,i.resolve)("/",e.hostname,...e.pathname.replace(/[?#].*$/,"").replace(/^\/+/,"").split("/"));return{error:null,data:null,hasRun:!1,run:function(){return this.hasRun?Promise.resolve():new Promise((r=>(0,o.readFile)(t,((n,s)=>{if(this.hasRun=!0,!n||"EISDIR"!==n.code)return this.error=n,this.data=s,void r(void 0);(0,o.readdir)(t,((t,n)=>{this.error=t,this.data=n,t?r(void 0):Promise.all(n.map((t=>new Promise((r=>(0,o.lstat)((0,i.resolve)(e.pathname,t),((e,n)=>r([t,n,e])))))))).then((t=>{const n=t.filter((e=>!e[2]&&e[1].isDirectory())).concat(t.filter((e=>!e[2]&&e[1].isFile())));this.data=`${y(128194,"directory",e.href)}\n                      <p>Directory content of <i>${e.href.replace(/\//g,"&#x002F;")}</i></p>\n                      <ul class="list-group">\n                        <li class="list-group-item">&#x1F4C1;<a href="${e.pathname.endsWith("/")?"..":"."}">&lt;parent&gt;</a></li>\n                        ${n.filter((e=>!e[2])).map((t=>`<li class="list-group-item">&#x${(t[1].isDirectory()?128193:128196).toString(16)};<a href="${e.pathname.endsWith("/")?"":`${e.pathname.split("/").slice(-1)[0]}/`}${t[0]}">${t[0]}</a></li>`)).join("\n")}\n                        </li>\n                      </ul>\n                      </body></html>`,r(void 0)}))}))}))))},events:{},on:function(e,r){return this.events[e]=r,this.run().then((()=>{"response"===e&&this.events.response({Server:"local","Content-Type":t.endsWith(".svg")?"image/svg+xml":null},0),"data"===e&&this.data&&(this.events.data(this.data),this.events.end()),"error"===e&&this.error&&this.events.error(this.error)})),this},end:function(){return this},request:function(){return this}}},y=(e,t,r)=>`<!doctype html>\n<html lang="en">\n<head>\n<title>&#x${e.toString(16)}; local-traffic ${t} | ${r}</title>\n<link href="https://cdn.jsdelivr.net/npm/bootstrap@latest/dist/css/bootstrap.min.css" rel="stylesheet"/>\n<script src="https://cdn.jsdelivr.net/npm/jquery@latest/dist/jquery.min.js"><\/script>\n<script src="https://cdn.jsdelivr.net/npm/bootstrap@latest/dist/js/bootstrap.bundle.min.js"><\/script>\n</head>\n<body><div class="container"><h1>&#x${e.toString(16)}; local-traffic ${t}</h1>\n<br/>`,w=(e,t,r,n)=>`${y(128163,"error",e.message)}\n<p>An error happened while trying to proxy a remote exchange</p>\n<div class="alert alert-warning" role="alert">\n  &#x24D8;&nbsp;This is not an error from the downstream service.\n</div>\n<div class="alert alert-danger" role="alert">\n<pre><code>${e.stack||`<i>${e.name} : ${e.message}</i>`}${e.errno?`<br/>(code : ${e.errno})`:""}</code></pre>\n</div>\nMore information about the request :\n<table class="table">\n  <tbody>\n    <tr>\n      <td>phase</td>\n      <td>${t}</td>\n    </tr>\n    <tr>\n      <td>requested URL</td>\n      <td>${r}</td>\n    </tr>\n    <tr>\n      <td>downstream URL</td>\n      <td>${n||"&lt;no-target-url&gt;"}</td>\n    </tr>\n  </tbody>\n</table>\n</div></body></html>`,R=(e,t,r)=>{t.writeHead(e,void 0,{"content-type":"text/html","content-length":r.length}),t.end(r)},v=()=>{h=(d.ssl?e.createSecureServer.bind(null,{...d.ssl,allowHTTP1:!0}):t.createServer)((async(o,i)=>{if(!o.headers.host&&!o.headers[":authority"])return void R(400,i,Buffer.from(w(new Error("client must supply a 'host' header"),"proxy",new n.URL(`http${d.ssl?"s":""}://unknowndomain${o.url}`))));const a=o.headers[":authority"]||`${o.headers.host}${o.headers.host.match(/:[0-9]+$/)?"":80!==d.port||d.ssl?443===d.port&&d.ssl?"":`:${d.port}`:""}`,l=new n.URL(`http${d.ssl?"s":""}://${a}${o.url}`),c=l.href.substring(l.origin.length),[h,u]=Object.entries({...Object.assign({},...Object.entries(d.mapping).map((([e,t])=>({[e]:new n.URL($(t))}))))}).find((([e])=>c.match(RegExp(e))))||[];if(!u)return void R(502,i,Buffer.from(w(new Error(`No mapping found in config file ${p}`),"proxy",l)));const m=u.host.replace(RegExp(/\/+$/),""),g=`${u.href.substring("https://".length+u.host.length)}${$(c.replace(RegExp($(h)),""))}`.replace(/^\/*/,"/"),f=new n.URL(`${u.protocol}//${m}${g}`);let y=null,v=!d.dontUseHttp2Downstream;const S="file:"===u.protocol?b(f):v?await Promise.race([new Promise((t=>{const r=(0,e.connect)(f,{rejectUnauthorized:!1,protocol:u.protocol},((e,n)=>{v=v&&!!n.alpnProtocol,t(v?r:null)}));r.on("error",(e=>{y=v&&Buffer.from(w(e,"connection",l,f))}))})),new Promise((e=>setTimeout((()=>{v=!1,e(null)}),3e3)))]):null;!v&&y&&(y=null);const O={...[...Object.entries(o.headers)].filter((([e])=>!["host","connection"].includes(e.toLowerCase()))).reduce(((e,[t,r])=>(e[t]=(e[t]||"")+(Array.isArray(r)?r:[r]).map((e=>e.replace(l.hostname,m))).join(", "),e)),{}),origin:u.href,referer:f.toString(),":authority":m,":method":o.method,":path":g,":scheme":u.protocol.replace(":","")},E=S&&!y&&S.request(O,{endStream:d.ssl?!(o?.stream?.readableLength??1):!o.readableLength});E&&E.on("error",(e=>{const t=-505===e.errno;y=Buffer.from(w(e,"stream"+(t?" (error -505 usually means that the downstream service does not support this http version)":""),l,f))}));const j={hostname:u.hostname,path:g,port:u.port,protocol:u.protocol,rejectUnauthorized:!1,method:o.method,headers:{...Object.assign({},...Object.entries(O).filter((([e])=>!e.startsWith(":")&&"transfer-encoding"!==e.toLowerCase())).map((([e,t])=>({[e]:t})))),host:u.hostname}},U=!y&&!v&&"file:"!==u.protocol&&await new Promise((e=>{const n="https:"===u.protocol?(0,r.request)(j,e):(0,t.request)(j,e);n.on("error",(t=>{y=Buffer.from(w(t,"request",l,f)),e(null)})),o.on("data",(e=>n.write(e))),o.on("end",(()=>n.end()))}));if(y)return void R(502,i,y);d.ssl&&o.stream&&o.stream.readableLength&&E&&(o.stream.on("data",(e=>E.write(e))),o.stream.on("end",(()=>E.end()))),!d.ssl&&o.readableLength&&E&&(o.on("data",(e=>E.write(e))),o.on("end",(()=>E.end())));const{outboundResponseHeaders:x}=await new Promise((e=>E?E.on("response",(t=>{e({outboundResponseHeaders:t})})):e(!E&&U?{outboundResponseHeaders:U.headers}:{outboundResponseHeaders:{}}))),N=x.location?new n.URL(x.location.startsWith("/")?`${u.href}${x.location.replace(/^\/+/,"")}`:x.location):null,L=N?N.href.substring(N.origin.length):null,P=l.origin,C=N?`${P}${L}`:null,B=E||U,H=y??await new Promise((e=>{let t=Buffer.alloc(0);B?(B.on("data",(e=>t=Buffer.concat([t,"string"==typeof e?Buffer.from(e):e]))),B.on("end",(()=>{e(t)}))):e(t)})).then((e=>d.replaceResponseBodyUrls&&e.length?(x["content-encoding"]||"").split(",").reduce((async(e,t)=>{const r=t.trim().toLowerCase(),n="gzip"===r||"x-gzip"===r?s.gunzip:"deflate"===r?s.inflate:"br"===r?s.brotliDecompress:"identity"===r||""===r?(e,t)=>{t(null,e)}:null;if(null===n)return void R(502,i,Buffer.from(w(new Error(`${r} compression not supported by the proxy`),"stream",l,f)));const o=await e;return await new Promise((e=>n(o,((t,r)=>{if(t)return R(502,i,Buffer.from(w(t,"stream",l,f))),void e("");e(r)}))))}),Promise.resolve(e)).then((e=>e.length>1e6||/[^\x00-\x7F]/.test(e.toString())&&!(x["content-type"]??"").includes("text/html")?e:d.replaceResponseBodyUrls?Object.entries(d.mapping).reduce(((e,[t,r])=>t.match(/^[-a-zA-Z0-9()@:%_\+.~#?&//=]*$/)||""===t?e.replace(new RegExp(r.replace(/^file:\/\//,"").replace(/[*+?^${}()|[\]\\]/g,"").replace(/^https/,"https?"),"ig"),`https://${a}${t.replace(/\/+$/,"")}/`):e),e.toString()).split(`${a}/:`).join(`${a}:`):e.toString())).then((e=>(x["content-encoding"]||"").split(",").reduce(((e,t)=>{const r=t.trim().toLowerCase(),n="gzip"===r||"x-gzip"===r?s.gzip:"deflate"===r?s.deflate:"br"===r?s.brotliCompress:"identity"===r||""===r?(e,t)=>{t(null,e)}:null;if(null===n)throw new Error(`${r} compression not supported by the proxy`);return e.then((e=>new Promise((t=>n(e,((e,r)=>{if(e)throw e;t(r)}))))))}),Promise.resolve(Buffer.from(e))))):e)),T={...Object.entries({...x,...d.replaceResponseBodyUrls?{"content-length":`${H.byteLength}`}:{}}).filter((([e])=>!e.startsWith(":")&&"transfer-encoding"!==e.toLowerCase()&&"connection"!==e.toLowerCase())).reduce(((e,[t,r])=>{const n=m.split("").map(((e,t)=>m.substring(t).startsWith(".")&&m.substring(t))).filter((e=>e)),o=[m].concat(n).reduce(((e,t)=>(Array.isArray(e)?e:[e]).map((e=>"string"==typeof e?e.replace(`Domain=${t}`,`Domain=${l.hostname}`):e))),r);return e[t]=(e[t]||[]).concat(o),e}),{}),...C?{location:[C]}:{}};try{Object.entries(T).forEach((([e,t])=>t&&i.setHeader(e,t)))}catch(e){}i.writeHead(x[":status"]||U.statusCode||200,d.ssl?void 0:U.statusMessage||"Status read from http/2",T),H?i.end(H):i.end()})).addListener("error",(e=>{"EACCES"===e.code&&u("permission denied for this port",a.ERROR,"⛔"),"EADDRINUSE"===e.code&&u("port is already used. NOT started",a.ERROR,"☠️")})).addListener("listening",(()=>{g(d)})).listen(d.port)};m().then(v);
+"use strict";Object.defineProperty(exports,"__esModule",{value:!0});const e=require("http2"),t=require("http"),r=require("https"),o=require("url"),n=require("fs"),s=require("zlib"),a=require("path");var i;!function(e){e[e.ERROR=124]="ERROR",e[e.SUCCESS=35]="SUCCESS",e[e.INFO=21]="INFO",e[e.WARNING=172]="WARNING"}(i||(i={}));const l=(0,a.resolve)(process.env.HOME,".local-traffic.json"),p=(0,a.resolve)(process.cwd(),process.argv.slice(-1)[0].endsWith(".json")?process.argv.slice(-1)[0]:l),c={mapping:{},port:8080,replaceResponseBodyUrls:!1,dontUseHttp2Downstream:!1,simpleLogs:!1,websocket:!1};let d,h;const u=(e,t,r)=>{console.log(`${(e=>{const t=new Date;return`${e?"":""}${`${t.getHours()}`.padStart(2,"0")}${e?":":":"}${`${t.getMinutes()}`.padStart(2,"0")}${e?":":":"}${`${t.getSeconds()}`.padStart(2,"0")}${e?"":""}`})(d.simpleLogs)} ${d.simpleLogs?e.replace(/⎸/g,"|").replace(/⎹/g,"|").replace(/\u001b\[[^m]*m/g,"").replace(/↘️/g,"inbound").replace(/☎️/g,"port").replace(/↗️/g,"outbound"):t?`[48;5;${t}m⎸    ${process.stdout.isTTY&&r||""}  ${e.padEnd(36)} ⎹`:e}`)},m=async(e=!0)=>new Promise((t=>(0,n.readFile)(p,((r,o)=>{r&&!e&&u("config error. Using default value",i.ERROR,"❌");try{d=Object.assign({},c,JSON.parse((o||"{}").toString()))}catch(e){return u("config syntax incorrect, aborting",i.ERROR,"⛈️"),d=d||{...c},void t(d)}d.mapping[""]||u('default mapping "" not provided.',i.WARNING,"☢️"),r&&"ENOENT"===r.code&&e&&p===l?(0,n.writeFile)(p,JSON.stringify(c),(e=>{e?u("config file NOT created",i.ERROR,"⁉️"):u("config file created",i.SUCCESS,"✨"),t(d)})):t(d)})))).then((()=>{e&&(0,n.watchFile)(p,f)})),g=e=>{u(`⎸ ↘️  : ${e.ssl?"HTTP/2  ":"HTTP 1.1"} ⎸ ☎️  : ${e.port.toString().padStart(5)} ⎸ ↗️  : ${e.dontUseHttp2Downstream?"HTTP 1.1":"HTTP/2  "} ⎹`)},f=async()=>{const e={...d};return await m(!1),isNaN(d.port)||d.port>65535||d.port<0?(d=e,void u("port number invalid. Not refreshing",i.ERROR,"☎️")):"object"!=typeof d.mapping?(d=e,void u("mapping should be an object. Aborting",i.ERROR,"⚡")):(d.replaceResponseBodyUrls&&!e.replaceResponseBodyUrls&&u("response body url replacement",i.INFO,"✔️"),!d.replaceResponseBodyUrls&&e.replaceResponseBodyUrls&&u("response body url NO replacement",i.INFO,"✖️"),d.websocket&&!e.websocket&&u("websocket activated",i.INFO,"☄️"),!d.websocket&&e.websocket&&u("websocket deactivated",i.INFO,"☄️"),u(`${Object.keys(d.mapping).length.toString().padStart(5)} loaded mapping rules`,i.SUCCESS,"↻"),void(d.port!==e.port||JSON.stringify(d.ssl)!==JSON.stringify(e.ssl)?(await new Promise((e=>h?h.close(e):e(void 0))),S()):d.dontUseHttp2Downstream!==e.dontUseHttp2Downstream&&g(d)))},$=e=>""==e?"":(0,a.normalize)(e).replace(/\\/g,"/"),w=e=>{const t=(0,a.resolve)("/",e.hostname,...e.pathname.replace(/[?#].*$/,"").replace(/^\/+/,"").split("/"));return{error:null,data:null,hasRun:!1,run:function(){return this.hasRun?Promise.resolve():new Promise((r=>(0,n.readFile)(t,((o,s)=>{if(this.hasRun=!0,!o||"EISDIR"!==o.code)return this.error=o,this.data=s,void r(void 0);(0,n.readdir)(t,((t,o)=>{this.error=t,this.data=o,t?r(void 0):Promise.all(o.map((t=>new Promise((r=>(0,n.lstat)((0,a.resolve)(e.pathname,t),((e,o)=>r([t,o,e])))))))).then((t=>{const o=t.filter((e=>!e[2]&&e[1].isDirectory())).concat(t.filter((e=>!e[2]&&e[1].isFile())));this.data=`${b(128194,"directory",e.href)}\n                      <p>Directory content of <i>${e.href.replace(/\//g,"&#x002F;")}</i></p>\n                      <ul class="list-group">\n                        <li class="list-group-item">&#x1F4C1;<a href="${e.pathname.endsWith("/")?"..":"."}">&lt;parent&gt;</a></li>\n                        ${o.filter((e=>!e[2])).map((t=>`<li class="list-group-item">&#x${(t[1].isDirectory()?128193:128196).toString(16)};<a href="${e.pathname.endsWith("/")?"":`${e.pathname.split("/").slice(-1)[0]}/`}${t[0]}">${t[0]}</a></li>`)).join("\n")}\n                        </li>\n                      </ul>\n                      </body></html>`,r(void 0)}))}))}))))},events:{},on:function(e,r){return this.events[e]=r,this.run().then((()=>{"response"===e&&this.events.response({Server:"local","Content-Type":t.endsWith(".svg")?"image/svg+xml":null},0),"data"===e&&this.data&&(this.events.data(this.data),this.events.end()),"error"===e&&this.error&&this.events.error(this.error)})),this},end:function(){return this},request:function(){return this}}},b=(e,t,r)=>`<!doctype html>\n<html lang="en">\n<head>\n<title>&#x${e.toString(16)}; local-traffic ${t} | ${r}</title>\n<link href="https://cdn.jsdelivr.net/npm/bootstrap@latest/dist/css/bootstrap.min.css" rel="stylesheet"/>\n<script src="https://cdn.jsdelivr.net/npm/jquery@latest/dist/jquery.min.js"><\/script>\n<script src="https://cdn.jsdelivr.net/npm/bootstrap@latest/dist/js/bootstrap.bundle.min.js"><\/script>\n</head>\n<body><div class="container"><h1>&#x${e.toString(16)}; local-traffic ${t}</h1>\n<br/>`,y=(e,t,r,o)=>`${b(128163,"error",e.message)}\n<p>An error happened while trying to proxy a remote exchange</p>\n<div class="alert alert-warning" role="alert">\n  &#x24D8;&nbsp;This is not an error from the downstream service.\n</div>\n<div class="alert alert-danger" role="alert">\n<pre><code>${e.stack||`<i>${e.name} : ${e.message}</i>`}${e.errno?`<br/>(code : ${e.errno})`:""}</code></pre>\n</div>\nMore information about the request :\n<table class="table">\n  <tbody>\n    <tr>\n      <td>phase</td>\n      <td>${t}</td>\n    </tr>\n    <tr>\n      <td>requested URL</td>\n      <td>${r}</td>\n    </tr>\n    <tr>\n      <td>downstream URL</td>\n      <td>${o||"&lt;no-target-url&gt;"}</td>\n    </tr>\n  </tbody>\n</table>\n</div></body></html>`,R=(e,t,r)=>{t.writeHead(e,void 0,{"content-type":"text/html","content-length":r.length}),t.end(r)},v=e=>{const t=(e.headers[":authority"]?.toString()??e.headers.host??"localhost").replace(/:.*/,""),r=e.headers[":authority"]||`${e.headers.host}${e.headers.host.match(/:[0-9]+$/)?"":80!==d.port||d.ssl?443===d.port&&d.ssl?"":`:${d.port}`:""}`,n=new o.URL(`http${d.ssl?"s":""}://${r}${e.url}`),s=n.href.substring(n.origin.length),[a,i]=Object.entries({...Object.assign({},...Object.entries(d.mapping).map((([e,t])=>({[e]:new o.URL($(t))}))))}).find((([e])=>s.match(RegExp(e))))||[];return{proxyHostname:t,proxyHostnameAndPort:r,url:n,path:s,key:a,target:i}},S=()=>{h=(d.ssl?e.createSecureServer.bind(null,{...d.ssl,allowHTTP1:!0}):t.createServer)((async(n,a)=>{if(!n.headers.host&&!n.headers[":authority"])return void R(400,a,Buffer.from(y(new Error("client must supply a 'host' header"),"proxy",new o.URL(`http${d.ssl?"s":""}://unknowndomain${n.url}`))));const{proxyHostname:i,proxyHostnameAndPort:l,url:c,path:h,key:u,target:m}=v(n);if(!m)return void R(502,a,Buffer.from(y(new Error(`No mapping found in config file ${p}`),"proxy",c)));const g=m.host.replace(RegExp(/\/+$/),""),f=`${m.href.substring("https://".length+m.host.length)}${$(h.replace(RegExp($(u)),""))}`.replace(/^\/*/,"/"),b=new o.URL(`${m.protocol}//${g}${f}`);let S=null,O=!d.dontUseHttp2Downstream;const j="file:"===m.protocol?w(b):O?await Promise.race([new Promise((t=>{const r=(0,e.connect)(b,{rejectUnauthorized:!1,protocol:m.protocol},((e,o)=>{O=O&&!!o.alpnProtocol,t(O?r:null)}));r.on("error",(e=>{S=O&&Buffer.from(y(e,"connection",c,b))}))})),new Promise((e=>setTimeout((()=>{O=!1,e(null)}),3e3)))]):null;S instanceof Buffer||(S=null);const E={...[...Object.entries(n.headers)].filter((([e])=>!["host","connection"].includes(e.toLowerCase()))).reduce(((e,[t,r])=>(e[t]=(e[t]||"")+(Array.isArray(r)?r:[r]).map((e=>e.replace(c.hostname,g))).join(", "),e)),{}),origin:m.href,referer:b.toString(),":authority":g,":method":n.method,":path":f,":scheme":m.protocol.replace(":","")},N=j&&!S&&j.request(E,{endStream:d.ssl?!(n?.stream?.readableLength??1):!n.readableLength});N&&N.on("error",(e=>{const t=-505===e.errno;S=Buffer.from(y(e,"stream"+(t?" (error -505 usually means that the downstream service does not support this http version)":""),c,b))}));const U={hostname:m.hostname,path:f,port:m.port,protocol:m.protocol,rejectUnauthorized:!1,method:n.method,headers:{...Object.assign({},...Object.entries(E).filter((([e])=>!e.startsWith(":")&&"transfer-encoding"!==e.toLowerCase())).map((([e,t])=>({[e]:t})))),host:m.hostname}},x=!S&&!O&&"file:"!==m.protocol&&await new Promise((e=>{const o="https:"===m.protocol?(0,r.request)(U,e):(0,t.request)(U,e);o.on("error",(t=>{S=Buffer.from(y(t,"request",c,b)),e(null)})),n.on("data",(e=>o.write(e))),n.on("end",(()=>o.end()))}));if(S)return void R(502,a,S);S=null,d.ssl&&n.stream&&n.stream.readableLength&&N&&(n.stream.on("data",(e=>N.write(e))),n.stream.on("end",(()=>N.end()))),!d.ssl&&n.readableLength&&N&&(n.on("data",(e=>N.write(e))),n.on("end",(()=>N.end())));const{outboundResponseHeaders:H}=await new Promise((e=>N?N.on("response",(t=>{e({outboundResponseHeaders:t})})):e(!N&&x?{outboundResponseHeaders:x.headers}:{outboundResponseHeaders:{}}))),P=H.location?new o.URL(H.location.startsWith("/")?`${m.href}${H.location.replace(/^\/+/,"")}`:H.location):null,L=P?P.href.substring(P.origin.length):null,C=c.origin,T=P?`${C}${L}`:null,A=N||x,B=S??await new Promise((e=>{let t=Buffer.alloc(0);A?(A.on("data",(e=>t=Buffer.concat([t,"string"==typeof e?Buffer.from(e):e]))),A.on("end",(()=>{e(t)}))):e(t)})).then((e=>d.replaceResponseBodyUrls&&e.length?(H["content-encoding"]||"").split(",").reduce((async(e,t)=>{const r=t.trim().toLowerCase(),o="gzip"===r||"x-gzip"===r?s.gunzip:"deflate"===r?s.inflate:"br"===r?s.brotliDecompress:"identity"===r||""===r?(e,t)=>{t(null,e)}:null;if(null===o)return void R(502,a,Buffer.from(y(new Error(`${r} compression not supported by the proxy`),"stream",c,b)));const n=await e;return await new Promise((e=>o(n,((t,r)=>{if(t)return R(502,a,Buffer.from(y(t,"stream",c,b))),void e("");e(r)}))))}),Promise.resolve(e)).then((e=>{const t=e.length>1e7,r=["text/html","application/javascript","application/json"].some((e=>(H["content-type"]??"").includes(e)));return!t&&(r||!/[^\x00-\x7F]/.test(e.toString()))?d.replaceResponseBodyUrls?Object.entries(d.mapping).reduce(((e,[t,r])=>""===t||t.match(/^[-a-zA-Z0-9()@:%_\+.~#?&//=]*$/)?e.replace(new RegExp(r.replace(/^file:\/\//,"").replace(/[*+?^${}()|[\]\\]/g,"").replace(/^https/,"https?")+"/*","ig"),`https://${l}${t.replace(/\/+$/,"")}/`):e),e.toString()).split(`${l}/:`).join(`${l}:`).replace(/\?protocol=wss?%3A&hostname=[^&]+&port=[0-9]+&pathname=/g,`?protocol=ws${d.ssl?"s":""}%3A&hostname=${i}&port=${d.port}&pathname=${encodeURIComponent(u.replace(/\/+$/,""))}`):e.toString():e})).then((e=>(H["content-encoding"]||"").split(",").reduce(((e,t)=>{const r=t.trim().toLowerCase(),o="gzip"===r||"x-gzip"===r?s.gzip:"deflate"===r?s.deflate:"br"===r?s.brotliCompress:"identity"===r||""===r?(e,t)=>{t(null,e)}:null;if(null===o)throw new Error(`${r} compression not supported by the proxy`);return e.then((e=>new Promise((t=>o(e,((e,r)=>{if(e)throw e;t(r)}))))))}),Promise.resolve(Buffer.from(e))))):e)),q={...Object.entries({...H,...d.replaceResponseBodyUrls?{"content-length":`${B.byteLength}`}:{}}).filter((([e])=>!e.startsWith(":")&&"transfer-encoding"!==e.toLowerCase()&&"connection"!==e.toLowerCase())).reduce(((e,[t,r])=>{const o=g.split("").map(((e,t)=>g.substring(t).startsWith(".")&&g.substring(t))).filter((e=>e)),n=[g].concat(o).reduce(((e,t)=>(Array.isArray(e)?e:[e]).map((e=>"string"==typeof e?e.replace(`Domain=${t}`,`Domain=${c.hostname}`):e))),r);return e[t]=(e[t]||[]).concat(n),e}),{}),...T?{location:[T]}:{}};try{Object.entries(q).forEach((([e,t])=>t&&a.setHeader(e,t)))}catch(e){}a.writeHead(H[":status"]||x.statusCode||200,d.ssl?void 0:x.statusMessage||"Status read from http/2",q),B?a.end(B):a.end()})).addListener("error",(e=>{"EACCES"===e.code&&u("permission denied for this port",i.ERROR,"⛔"),"EADDRINUSE"===e.code&&u("port is already used. NOT started",i.ERROR,"☠️")})).addListener("listening",(()=>{g(d)})).on("upgrade",((e,n)=>{if(!d.websocket)return void n.end("HTTP/1.1 503 Service Unavailable\r\n\r\n");const{key:s,target:a}=v(e),l=new o.URL(`${a.protocol}//${a.host}${e.url.replace(new RegExp(`^${s}`,"g"),"").replace(/^\/*/,"/")}`),p={hostname:l.hostname,path:l.pathname,port:l.port,protocol:l.protocol,rejectUnauthorized:!1,method:e.method,headers:e.headers,host:l.hostname},c="https:"===l.protocol?(0,r.request)(p):(0,t.request)(p);c.end(),c.on("error",(e=>{u("websocket request has errored "+(e.errno?`(${e.errno})`:""),i.WARNING,"☄️")})),c.on("upgrade",((e,t)=>{const r=`HTTP/${e.httpVersion} ${e.statusCode} ${e.statusMessage}\r\n${Object.entries(e.headers).flatMap((([e,t])=>(Array.isArray(t)?t:[t]).map((t=>[e,t])))).map((([e,t])=>`${e}: ${t}\r\n`)).join("")}\r\n`;n.write(r),n.allowHalfOpen=!0,t.allowHalfOpen=!0,t.on("data",(e=>n.write(e))),n.on("data",(e=>t.write(e))),t.on("error",(e=>{u("downstream socket has errored "+(e.errno?`(${e.errno})`:""),i.WARNING,"☄️")})),n.on("error",(e=>{u("upstream socket has errored "+(e.errno?`(${e.errno})`:""),i.WARNING,"☄️")}))}))})).listen(d.port)};m().then(S);

package/index.ts

@@ -33,6 +33,8 @@ import {
 import { resolve, normalize } from "path";
 import type { Duplex } from "stream";
 
+type ErrorWithErrno = NodeJS.ErrnoException;
+
 enum LogLevel {
   ERROR = 124,
   SUCCESS = 35,
@@ -47,6 +49,7 @@ interface LocalConfiguration {
   replaceResponseBodyUrls?: boolean;
   dontUseHttp2Downstream?: boolean;
   simpleLogs?: boolean;
+  websocket?: boolean;
 }
 
 const userHomeConfigFile = resolve(process.env.HOME, ".local-traffic.json");
@@ -62,6 +65,7 @@ const defaultConfig: LocalConfiguration = {
   replaceResponseBodyUrls: false,
   dontUseHttp2Downstream: false,
   simpleLogs: false,
+  websocket: false,
 };
 
 let config: LocalConfiguration;
@@ -172,6 +176,18 @@ const onWatch = async () => {
   ) {
     log("response body url NO replacement", LogLevel.INFO, "✖️");
   }
+  if (
+    config.websocket &&
+    !previousConfig.websocket
+  ) {
+    log("websocket activated", LogLevel.INFO, "☄️");
+  }
+  if (
+    !config.websocket &&
+    previousConfig.websocket
+  ) {
+    log("websocket deactivated", LogLevel.INFO, "☄️");
+  }
   log(
     `${Object.keys(config.mapping)
       .length.toString()
@@ -348,7 +364,7 @@ const errorPage = (
 </div>
 <div class="alert alert-danger" role="alert">
 <pre><code>${thrown.stack || `<i>${thrown.name} : ${thrown.message}</i>`}${
-  (thrown as any).errno ? `<br/>(code : ${(thrown as any).errno})` : ""
+  (thrown as ErrorWithErrno).errno ? `<br/>(code : ${(thrown as ErrorWithErrno).errno})` : ""
 }</code></pre>
 </div>
 More information about the request :
@@ -386,8 +402,39 @@ const send = (
   inboundResponse.end(errorBuffer);
 };
 
+const determineMapping = (inboundRequest: Http2ServerRequest | IncomingMessage): {
+  proxyHostname: string,
+  proxyHostnameAndPort: string,
+  url: URL,
+  path: string,
+  key: string,
+  target: URL
+} => {
+
+  const proxyHostname =
+    (inboundRequest.headers[":authority"]?.toString() ??
+      inboundRequest.headers.host ?? 'localhost').replace(/:.*/, '');
+  const proxyHostnameAndPort =
+    inboundRequest.headers[":authority"] as string ||
+    `${inboundRequest.headers.host}${inboundRequest.headers.host.match(/:[0-9]+$/)
+      ? ""
+      : config.port === 80 && !config.ssl
+        ? ""
+        : config.port === 443 && config.ssl
+          ? ""
+          : `:${config.port}`
+    }`;
+  const url = new URL(
+    `http${config.ssl ? "s" : ""}://${proxyHostnameAndPort}${inboundRequest.url}`
+  );
+  const path = url.href.substring(url.origin.length);
+  const [key, target] =
+    Object.entries(envs()).find(([key]) => path.match(RegExp(key))) || [];
+  return { proxyHostname, proxyHostnameAndPort, url, path, key, target };
+}
+
 const start = () => {
-  server = (config.ssl
+  server = ((config.ssl
     ? createSecureServer.bind(null, { ...config.ssl, allowHTTP1: true })
     : createServer)(
     async (
@@ -409,23 +456,8 @@ const start = () => {
         );
         return;
       }
-      const proxyHostname =
-        inboundRequest.headers[":authority"] ||
-        `${inboundRequest.headers.host}${
-          inboundRequest.headers.host.match(/:[0-9]+$/)
-            ? ""
-            : config.port === 80 && !config.ssl
-            ? ""
-            : config.port === 443 && config.ssl
-            ? ""
-            : `:${config.port}`
-        }`;
-      const url = new URL(
-        `http${config.ssl ? "s" : ""}://${proxyHostname}${inboundRequest.url}`
-      );
-      const path = url.href.substring(url.origin.length);
-      const [key, target] =
-        Object.entries(envs()).find(([key]) => path.match(RegExp(key))) || [];
+      const { proxyHostname, proxyHostnameAndPort, url, path, key, target } =
+        determineMapping(inboundRequest);
       if (!target) {
         send(
           502,
@@ -489,7 +521,7 @@ const start = () => {
                 }, 3000)
               ),
             ]);
-      if (!http2IsSupported && error) error = null;
+      if (!(error instanceof Buffer)) error = null;
 
       const outboundHeaders: OutgoingHttpHeaders = {
         ...[...Object.entries(inboundRequest.headers)]
@@ -526,7 +558,7 @@ const start = () => {
         ((outboundExchange as unknown) as Http2Stream).on(
           "error",
           (thrown: Error) => {
-            const httpVersionSupported = (thrown as any).errno === -505;
+            const httpVersionSupported = (thrown as ErrorWithErrno).errno === -505;
             error = Buffer.from(
               errorPage(
                 thrown,
@@ -581,11 +613,11 @@ const start = () => {
           );
           inboundRequest.on("end", () => outboundHttp1Request.end());
         }));
-
+      // intriguingly, error is reset to "false" at this point, even if it was null
       if (error) {
         send(502, inboundResponse, error);
         return;
-      }
+      } else error = null;
 
       // phase : request body
       if (
@@ -728,37 +760,44 @@ const start = () => {
                 })
               );
             }, Promise.resolve(payloadBuffer))
-            .then((uncompressedBuffer: Buffer) =>
-              (uncompressedBuffer.length > 1E6 ||
-              /[^\x00-\x7F]/.test(uncompressedBuffer.toString()) &&
-              !(outboundResponseHeaders["content-type"] ?? "").includes('text/html')) ?
+            .then((uncompressedBuffer: Buffer) => {
+              const fileTooBig = uncompressedBuffer.length > 1E7;
+              const fileHasSpecialChars = () => /[^\x00-\x7F]/.test(uncompressedBuffer.toString());
+              const contentTypeCanBeProcessed =
+              ['text/html', 'application/javascript', 'application/json'].some(allowedContentType =>
+                (outboundResponseHeaders["content-type"] ?? "").includes(allowedContentType));
+              const willReplace = !fileTooBig && (contentTypeCanBeProcessed || !fileHasSpecialChars());
+              return !willReplace ?
               uncompressedBuffer :
               !config.replaceResponseBodyUrls
                 ? uncompressedBuffer.toString()
                 : Object.entries(config.mapping)
                     .reduce(
                       (inProgress, [path, mapping]) =>
-                        !path.match(/^[-a-zA-Z0-9()@:%_\+.~#?&//=]*$/) &&
-                        path !== ''
+                        path !== '' && !path.match(/^[-a-zA-Z0-9()@:%_\+.~#?&//=]*$/)
                           ? inProgress
                           : inProgress.replace(
                               new RegExp(
                                 mapping
                                   .replace(/^file:\/\//, "")
                                   .replace(/[*+?^${}()|[\]\\]/g, "")
-                                  .replace(/^https/, 'https?'),
+                                  .replace(/^https/, 'https?') + '/*',
                                 "ig"
                               ),
-                              `https://${proxyHostname}${path.replace(
+                              `https://${proxyHostnameAndPort}${path.replace(
                                 /\/+$/,
                                 ""
                               )}/`
                             ),
                       uncompressedBuffer.toString()
                     )
-                    .split(`${proxyHostname}/:`)
-                    .join(`${proxyHostname}:`)
-            )
+                    .split(`${proxyHostnameAndPort}/:`)
+                    .join(`${proxyHostnameAndPort}:`)
+                    .replace(/\?protocol=wss?%3A&hostname=[^&]+&port=[0-9]+&pathname=/g,
+                      `?protocol=ws${config.ssl ? 
+                        "s" : ""}%3A&hostname=${proxyHostname}&port=${config.port}&pathname=${
+                        encodeURIComponent(key.replace(/\/+$/, ''))}`)
+            })
             .then((updatedBody: Buffer | string) =>
               (outboundResponseHeaders["content-encoding"] || "")
                 .split(",")
@@ -859,16 +898,70 @@ const start = () => {
       if (payload) inboundResponse.end(payload);
       else inboundResponse.end();
     }
-  )
+  ) as Server)
     .addListener("error", (err: Error) => {
-      if ((err as any).code === "EACCES")
+      if ((err as ErrorWithErrno).code === "EACCES")
         log(`permission denied for this port`, LogLevel.ERROR, "⛔");
-      if ((err as any).code === "EADDRINUSE")
+      if ((err as ErrorWithErrno).code === "EADDRINUSE")
         log(`port is already used. NOT started`, LogLevel.ERROR, "☠️");
     })
     .addListener("listening", () => {
       logProtocols(config);
     })
+    .on("upgrade", (request: IncomingMessage, upstreamSocket: Duplex) => {
+      if (!config.websocket) {
+        upstreamSocket.end(`HTTP/1.1 503 Service Unavailable\r\n\r\n`)
+        return;
+      }
+
+      const { key, target: targetWithForcedPrefix } = determineMapping(request);
+      const target = new URL(`${targetWithForcedPrefix.protocol}//${
+        targetWithForcedPrefix.host}${request.url.replace(
+          new RegExp(`^${key}`, 'g'), '').replace(/^\/*/, '/')}`);
+      const downstreamRequestOptions: RequestOptions = {
+        hostname: target.hostname,
+        path: target.pathname,
+        port: target.port,
+        protocol: target.protocol,
+        rejectUnauthorized: false,
+        method: request.method,
+        headers: request.headers,
+        host: target.hostname,
+      };
+
+      const downstreamRequest = target.protocol === "https:"
+        ? httpsRequest(downstreamRequestOptions)
+        : httpRequest(downstreamRequestOptions);
+      downstreamRequest.end();
+      downstreamRequest.on('error', (error) => {
+        log(`websocket request has errored ${
+          (error as ErrorWithErrno).errno ?
+          `(${(error as ErrorWithErrno).errno})` : ''}`, LogLevel.WARNING, "☄️")
+    });
+      downstreamRequest.on('upgrade', (response, downstreamSocket) => {
+        const upgradeResponse = `HTTP/${response.httpVersion} ${response.statusCode} ${
+          response.statusMessage}\r\n${Object.entries(response.headers)
+            .flatMap(([key, value]) => (!Array.isArray(value) ? [value] : value)
+              .map(oneValue => [key, oneValue]))
+            .map(([key, value]) =>
+              `${key}: ${value}\r\n`).join('')}\r\n`;
+        upstreamSocket.write(upgradeResponse);
+        upstreamSocket.allowHalfOpen = true;
+        downstreamSocket.allowHalfOpen = true;
+        downstreamSocket.on('data', (data) => upstreamSocket.write(data));
+        upstreamSocket.on('data', (data) => downstreamSocket.write(data));
+        downstreamSocket.on('error', (error) => {
+          log(`downstream socket has errored ${
+            (error as ErrorWithErrno).errno ?
+            `(${(error as ErrorWithErrno).errno})` : ''}`, LogLevel.WARNING, "☄️")
+        })
+        upstreamSocket.on('error', (error) => {
+          log(`upstream socket has errored ${
+            (error as ErrorWithErrno).errno ?
+            `(${(error as ErrorWithErrno).errno})` : ''}`, LogLevel.WARNING, "☄️")
+        })
+      });
+    })
     .listen(config.port);
 };
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "local-traffic",
-  "version": "0.0.31",
+  "version": "0.0.34",
   "main": "index.ts",
   "private": false,
   "keywords": [
@@ -8,7 +8,8 @@
     "proxy",
     "h2",
     "http2",
-    "https"
+    "https",
+    "websocket"
   ],
   "license": "MIT",
   "repository": "git@github.com:libetl/local-traffic.git",
@@ -16,16 +17,22 @@
   "scripts": {
     "start": "./dist/localTraffic.js",
     "typescript": "tsc",
-    "terser": "echo '#!/usr/bin/env node\n' \"$(terser ./dist/index.js -c -m --toplevel)\" > ./dist/localTraffic.js",
+    "shebang": "echo '#!/usr/bin/env node' | cat - ./dist/terser.js > ./dist/localTraffic.js",
+    "terser": "terser ./dist/index.js -c -m --toplevel > ./dist/terser.js",
     "chmod": "chmod a+x ./dist/localTraffic.js",
-    "build": "yarn typescript && yarn terser && yarn chmod"
+    "clean": "rm -rf dist",
+    "build": "npm run clean && npm run typescript && npm run terser && npm run shebang && npm run chmod"
   },
   "devDependencies": {
-    "@types/node": "^17.0.14",
-    "terser": "^5.10.0",
-    "typescript": "^4.5.5"
+    "@types/node": "^17.0.33",
+    "terser": "^5.13.1",
+    "typescript": "^4.6.4"
   },
   "bin": {
     "local-traffic": "./dist/localTraffic.js"
+  },
+  "volta": {
+    "node": "18.1.0",
+    "npm": "8.10.0"
   }
 }