biz-a-cli 2.1.0 → 2.2.0

package/bin/hub.js

@@ -0,0 +1,114 @@
+#!/usr/bin/env node
+
+const yargs = require('yargs');
+const IDLE_SOCKET_TIMEOUT_MILLISECONDS = 1000 * 30;
+
+let argv = yargs
+    .usage('Usage: $0 --server [string] --subdomain [string] --hostname [string] --port [number]')
+    .options('s', {
+        alias: 'server',
+        describe: '(Required) Tunnel server endpoint'
+    })
+    .options('sub', {
+        alias: 'subdomain',
+        describe: '(Required) Public URL the tunnel server is forwarding to us'
+    })
+    .options('h', {
+        alias: 'hostname',
+        default: '127.0.0.1',
+        describe: 'Address of local server for forwarding over socket-tunnel'
+    })
+    .options('p', {
+        alias: 'port',
+        describe: '(Required) Port of local server for forwarding over socket-tunnel'
+    })
+    .argv;
+
+if (argv.help) {
+    yargs.showHelp();
+    process.exit();
+}
+
+if (!argv['server'] || !argv['subdomain'] || !argv['port']) {
+    for (var key in ['server', 'subdomain', 'port']) {
+        if (argv[key]) continue;
+
+        console.log('Error: Required option, but nothing found');
+
+        yargs.showHelp();
+
+        process.exit();
+    }
+}
+
+return new Promise((resolve, reject) => {
+    const net = require('node:net');
+    const ss = require('socket.io-stream');
+    let socket = require('socket.io-client')(argv['server']);
+
+    socket.on('connect', () => {
+        console.log(new Date() + ': connected to socket server');
+        console.log(new Date() + ': requesting subdomain ' + argv['subdomain'] + ' via ' + argv['server']);
+
+        socket.emit('createTunnel', argv['subdomain'], (err) => {
+            if (err) {
+                console.log(new Date() + ': [error] ' + err);
+                reject(err);
+            } else {
+                console.log(new Date() + ': registered with server successfully');
+
+                // clean and concat requested url
+                let url;
+                let subdomain = argv['subdomain'].toString();
+                let server = argv['server'].toString();
+
+                if (server.includes('https://')) {
+                    url = `https://${subdomain}.${server.slice(8)}`;
+                } else if (server.includes('http://')) {
+                    url = `http://${subdomain}.${server.slice(7)}`;
+                } else {
+                    url = `https://${subdomain}.${server}`;
+                }
+
+                // resolve promise with requested URL
+                resolve(url);
+            }
+        });
+    });
+    socket.on('incomingClient', (clientId) => {
+        // console.log(clientId, 'incoming clientId')
+        // console.log(`connecting to ${argv['hostname']}:${argv['port']}`)
+        let client = net.connect(argv['port'], argv['hostname']);
+        // let client = net.createConnection(argv['port'], argv['hostname']);
+
+        client.on('connect', () => {
+            console.log(`client connected to ${argv['hostname']}:${argv['port']}`)
+            // console.log('client:', client)
+            let s = ss.createStream();
+            s.pipe(client).pipe(s);
+
+            s.on('end', () => {
+                client.destroy();
+            });
+            // console.log(client, '<===client')
+            ss(socket).emit(clientId, s);
+
+            // s.on('data', (chunk)=>{
+            //     console.log(chunk.toString(), 'chunk')
+            // })
+        })
+
+        client.setTimeout(IDLE_SOCKET_TIMEOUT_MILLISECONDS);
+        client.on('timeout', () => {
+            client.end();
+        });
+
+        client.on('error', () => {
+            // handle connection refusal (create a stream and immediately close it)
+            let s = ss.createStream();
+            ss(socket).emit(clientId, s);
+            s.end();
+        });
+    });
+
+})

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "biz-a-cli",
-  "version": "2.1.0",
+  "version": "2.2.0",
   "description": "",
   "main": "bin/index.js",
   "scripts": {
@@ -10,12 +10,16 @@
   "license": "ISC",
   "bin": {
     "uploadbiza": "./bin/index.js",
-    "proxy": "./bin/proxy.js"
+    "proxy": "./bin/proxy.js",
+    "hub": "./bin/hub.js"
   },
   "dependencies": {
     "axios": "^0.27.2",
     "cors": "^2.8.5",
     "express": "^4.18.2",
+    "net": "^1.0.2",
+    "socket.io-client": "^4.7.1",
+    "socket.io-stream": "^0.9.1",
     "yargs": "^17.4.1"
   }
 }