Merge pull request #18 from Will-create/main

Added proxy client to cli

Author: petersirka

bin/total5

@@ -36,12 +36,13 @@ function load(args) {
 setTimeout(load, 10, ARGS);
 
 FUNC.help = function() {
-	console.log('translate                : it makes a resource file with a localized dictionary from the current directory');
-	console.log('minify <filename>        : it minifies .js, .css and .html files');
-	console.log('bundle <filename>        : it makes a bundle from the current directory');
-	console.log('extract <filename>       : it extracts a bundle into the current directory');
-	console.log('edit <url?id=project>    : it opens remote editing of the current directory with the Total.js Code Editor');
-	console.log('8000                     : it starts a web server on port "8000" for the current directory');
+	console.log('translate                                   : it makes a resource file with a localized dictionary from the current directory');
+	console.log('minify <filename>                           : it minifies .js, .css and .html files');
+	console.log('bundle <filename>                           : it makes a bundle from the current directory');
+	console.log('extract <filename>                          : it extracts a bundle into the current directory');
+	console.log('edit <url?id=project>                       : it opens remote editing of the current directory with the Total.js Code Editor');
+	console.log('proxyclient <server_endpoint> <local_port>  : It connects a local port with a remote totaljs proxy server');
+	console.log('8000                                        : it starts a web server on port "8000" for the current directory');
 	done();
 };
 
@@ -241,6 +242,10 @@ FUNC.server = function(port) {
 	F.http({ load: 'none', port: port || 8000 });
 };
 
+FUNC.proxyclient = function(endpoint, port) {
+	require('../proxyclient').init(endpoint, port);
+};
+
 FUNC.edit = function(url) {
 	if (!url)
 		throw new Error('Invalid URL address');

proxyclient.js

@@ -0,0 +1,146 @@
+
+// Total.js client for Proxy
+// The MIT License
+// Copyright 2017-2023 (c) Peter Širka <petersirka@gmail.com>
+
+const BUFFER_INIT = Buffer.alloc(1);
+BUFFER_INIT.writeUInt8(0);
+
+const BUFFER_DATA = Buffer.alloc(1);
+BUFFER_DATA.writeUInt8(1);
+
+const BUFFER_END = Buffer.alloc(1);
+BUFFER_END.writeUInt8(2);
+
+const BUFFER_ERROR = Buffer.alloc(1);
+BUFFER_END.writeUInt8(3);
+
+const BUFFER_ABORT = Buffer.alloc(1);
+BUFFER_END.writeUInt8(4);
+
+const Pending = {};
+
+function help() {
+	console.log('Proxy client arguments:');
+	console.log('client [WSS proxy server] [port or ip:port]');
+}
+
+function log() {
+
+	let arr = [new Date().format('yyyy-MM-dd HH:mm:ss')];
+
+	for (let index in arguments)
+		arr.push(arguments[index]);
+
+	console.log.apply(console, arr);
+}
+
+function connect(Source, Target) {
+
+	F.websocketclient(function(client) {
+
+		client.options.type = 'binary';
+		client.connect(Target.replace(/^http/i, 'ws'));
+
+		client.on('open', function() {
+			log('\x1b[42m[OK]\x1b[0m', 'Proxy client successfully connected to \x1b[41m{0}\x1b[0m'.format(Target));
+		});
+
+		client.on('close', function() {
+			log('\x1b[41m[NO]\x1b[0m', 'Proxy client is disconnected.');
+		});
+
+		client.on('message', function(msg) {
+
+			let type = msg.readInt8(0);
+			let id = msg.readInt32BE(1);
+			let data = msg.slice(5);
+			let req = Pending[id];
+
+			switch (type) {
+				case 0: // init
+
+					let tmp = JSON.parse(data.toString('utf8'));
+					let opt = {};
+					opt.hostname = Source[0];
+					opt.port = Source[1];
+					opt.headers = tmp.headers;
+					opt.headers['x-proxy'] = 'Total.js';
+					opt.headers.host = Source[0] + ':' + Source[1];
+					opt.method = tmp.method;
+					opt.path = tmp.url;
+
+					log('\x1b[43m[' + opt.method + ']\x1b[0m', '\x1b[34m' + tmp.ip + '\x1b[0m', opt.path);
+
+					req = Total.Http.request(opt);
+					req.ID = Buffer.alloc(4);
+					req.ID.writeUInt32BE(id);
+
+					Pending[id] = req;
+
+					req.on('error', function(err) {
+						delete Pending[id];
+						client.send(Buffer.concat([BUFFER_ERROR, req.ID, Buffer.from(err.toString(), 'utf8')]));
+					});
+
+					req.on('abort', function() {
+						delete Pending[id];
+						client.send(Buffer.concat([BUFFER_ABORT, req.ID]));
+					});
+
+					req.on('response', function(res) {
+
+						let obj = {};
+						obj.status = res.statusCode;
+						obj.headers = res.headers;
+
+						delete Pending[id];
+						res.req = req;
+						client.send(Buffer.concat([BUFFER_INIT, req.ID, Buffer.from(JSON.stringify(obj))]));
+						res.on('data', chunk => client.send(Buffer.concat([BUFFER_DATA, req.ID, chunk])));
+
+						res.on('error', function(err) {
+							delete Pending[id];
+							client.send(Buffer.concat([BUFFER_ERROR, req.ID, Buffer.from(err.toString(), 'utf8')]));
+						});
+
+						res.on('abort', function() {
+							delete Pending[id];
+							client.send(Buffer.concat([BUFFER_ABORT, req.ID]));
+						});
+
+						res.on('end', () => client.send(Buffer.concat([BUFFER_END, req.ID])));
+					});
+					break;
+
+				case 1: // data
+					req && req.write(data);
+					break;
+
+				case 2: // end
+					req && req.end();
+					break;
+			}
+
+		});
+	});
+}
+
+exports.init = function (endpoint, port) { 
+    let Target = endpoint;
+    let Source = port;
+    
+    if (!Target || !Source) {
+        help();
+        return;
+    }
+    
+    Source = Source.split(':');
+    
+    if (!Source[1]) {
+        Source[1] = +Source[0];
+        Source[0] = '127.0.0.1';
+    }
+
+    connect(Source, Target);
+};