2.0.11

package-lock.json

@@ -1,6 +1,6 @@
 {
   "name": "@pushrocks/smartproxy",
-  "version": "2.0.4",
+  "version": "2.0.11",
   "lockfileVersion": 1,
   "requires": true,
   "dependencies": {

package.json

@@ -1,6 +1,6 @@
 {
   "name": "@pushrocks/smartproxy",
-  "version": "2.0.4",
+  "version": "2.0.11",
   "private": false,
   "description": "a proxy for handling high workloads of proxying",
   "main": "dist_ts/index.js",

ts/smartproxy.classes.proxyworker.ts

@@ -8,6 +8,7 @@ export class ProxyWorker {
   public port = 8001;
   public router = new SmartproxyRouter();
   public socketMap = new plugins.lik.ObjectMap<plugins.net.Socket>();
+  public defaultHeaders: {[key: string]: string} = {};
 
   /**
    * starts the proxyInstance
@@ -161,6 +162,7 @@ JNj2Dr5H0XoLFFnvuvzcRbhlJ9J67JzR+7g=
           {
             method: req.method,
             headers: req.headers,
+            keepAlive: true
           },
           true, // lets make this streaming
           (request) => {
@@ -174,6 +176,9 @@ JNj2Dr5H0XoLFFnvuvzcRbhlJ9J67JzR+7g=
         );
         res.statusCode = response.statusCode;
         console.log(response.statusCode);
+        for (const defaultHeader of Object.keys(this.defaultHeaders)) {
+          res.setHeader(defaultHeader, this.defaultHeaders[defaultHeader]);
+        } 
         for (const header of Object.keys(response.headers)) {
           res.setHeader(header, response.headers[header]);
         }
@@ -270,6 +275,7 @@ const proxyWorkerCalls = {
   updateReverseConfigs: async (configArray: plugins.tsclass.network.IReverseProxyConfig[]) => {
     await proxyWorkerInstance.updateProxyConfigs(configArray);
   },
+  addDefaultHeaders: async (headers: {[key: string]: string}) => {}
 };
 
 export type TProxyWorkerCalls = typeof proxyWorkerCalls;

ts/smartproxy.classes.smartproxy.ts

@@ -45,6 +45,10 @@ export class SmartProxy {
     console.log('successfully spawned portproxy and proxyworkers!');
   }
 
+  public async updateDefaultHeaders(defaultHeadersArg: {[key: string]: string}) {
+    await this.proxyWorkerFunctions.addDefaultHeaders(defaultHeadersArg);
+  }
+
   public async stop() {
     await this.proxyWorkerFunctions.stop();
     await plugins.smartspawn.Thread.terminate(this.proxyWorkerFunctions);

ts/smartproxy.portproxy.ts

@@ -18,23 +18,49 @@ const portProxyCalls = {
       response.end();
     });
     httpServer.listen(7999);
+    const cleanUpSockets = (from: plugins.net.Socket, to: plugins.net.Socket) => {
+      from.end();
+      to.end();
+      from.removeAllListeners();
+      to.removeAllListeners();
+      from.unpipe();
+      to.unpipe();
+      from.destroy();
+      to.destroy();
+    }
     netServer = net
       .createServer((from) => {
         const to = net.createConnection({
           host: 'localhost',
           port: 8001,
         });
+        from.setTimeout(120000);
         from.pipe(to);
         to.pipe(from);
         from.on('error', () => {
-          from.end(), to.end();
+          cleanUpSockets(from, to);
         });
         to.on('error', () => {
-          from.end(), to.end();
+          cleanUpSockets(from, to);
         });
         from.on('close', () => {
-          to.end();
+          cleanUpSockets(from, to);
         });
+        to.on('close', () => {
+          cleanUpSockets(from, to);
+        });
+        from.on('timeout', () => {
+          cleanUpSockets(from, to);
+        });
+        to.on('timeout', () => {
+          cleanUpSockets(from, to);
+        });
+        from.on('end', () => {
+          cleanUpSockets(from, to);
+        })
+        to.on('end', () => {
+          cleanUpSockets(from, to);
+        })
       })
       .listen(portArg);
     console.log(`PortProxy -> OK: Now listening on port ${portArg}`);