Merge pull request #2 from web-tunnel/multiple-clients

Author: embbnux

README.md

@@ -57,6 +57,34 @@ Please replace your_local_server_port with your local HTTP server port, eg: `300
 
 After that you can access your local HTTP server by access `your_public_server_domain`.
 
+## Multiple Clients
+
+The server steams HTTP request to WebSocket connection which has same host value in request headers.
+
+So if you have multiple domains for the proxy server, you can have multiple clients.
+
+For example, you have `https://app1.test.com` and `https://app2.test.com` for this proxy server.
+
+In client 1:
+
+```
+$ lite-http-tunnel config server https://app1.test.com -p app1
+$ lite-http-tunnel start your_local_server_port -p app1
+```
+
+In client 2:
+
+```
+$ lite-http-tunnel config server https://app2.test.com -p app2
+$ lite-http-tunnel start your_local_server_port -p app2
+```
+
 ## Related
 
-A introduce article: [Building a HTTP Tunnel with WebSocket and Node.JS](https://medium.com/@embbnux/building-a-http-tunnel-with-websocket-and-node-js-98068b0225d3)
+A introduce article: [Building a HTTP Tunnel with WebSocket and Node.JS](https://medium.com/@embbnux/building-a-http-tunnel-with-websocket-and-node-js-98068b0225d3?source=friends_link&sk=985d90ec9f512928b34ed38b7ddcb378)
+
+## TODO
+
+- [ ] Add tests
+- [ ] Support multiple clients based on request path prefix
+- [ ] Support to stream WebSocket request

server.js

@@ -13,11 +13,12 @@ const app = express();
 const httpServer = http.createServer(app);
 const io = new Server(httpServer);
 
-let connectedSocket = null;
+let connectedSockets = {};
 
 io.use((socket, next) => {
-  if (connectedSocket) {
-    return next(new Error('Connected error'));
+  const connectHost = socket.handshake.headers.host;
+  if (connectedSockets[connectHost]) {
+    return next(new Error(`${connectHost} has a existing connection`));
   }
   if (!socket.handshake.auth || !socket.handshake.auth.token){
     next(new Error('Authentication error'));
@@ -34,21 +35,22 @@ io.use((socket, next) => {
 });
 
 io.on('connection', (socket) => {
-  console.log('client connected');
-  connectedSocket = socket;
+  const connectHost = socket.handshake.headers.host;
+  connectedSockets[connectHost] = socket;
+  console.log(`client connected at ${connectHost}`);
   const onMessage = (message) => {
     if (message === 'ping') {
       socket.send('pong');
     }
   }
   const onDisconnect = (reason) => {
     console.log('client disconnected: ', reason);
-    connectedSocket = null;
+    delete connectedSockets[connectHost];
     socket.off('message', onMessage);
     socket.off('error', onError);
   };
   const onError = (e) => {
-    connectedSocket = null;
+    delete connectedSockets[connectHost];
     socket.off('message', onMessage);
     socket.off('disconnect', onDisconnect);
   };
@@ -78,6 +80,7 @@ app.get('/tunnel_jwt_generator', (req, res) => {
 });
 
 app.use('/', (req, res) => {
+  const connectedSocket = connectedSockets[req.headers.host];
   if (!connectedSocket) {
     res.status(404);
     res.send('Not Found');