websocket mini

Author: bobzhang

lib/bsb

@@ -13,10 +13,58 @@ var os = require('os');
 var bsconfig = 'bsconfig.json'
 var bsb_exe = __filename + ".exe"
 
+var LAST_SUCCESS_BUILD_STAMP = 0
 
+// All clients of type MiniWebSocket
+var wsClients = []
 var watch_mode = false
 var verbose = false
 var postBuild = undefined
+var useWebSocket = false
+var webSocketPort = 9999
+
+function dlog(str){
+    if(verbose){
+        console.log(str)
+    }
+}
+function notifyClients() {
+    wsClients = wsClients.filter(x => !x.closed && !x.socket.destroyed )
+    var wsClientsLen = wsClients.length
+    dlog(`Alive sockets number: ${wsClientsLen}`)    
+    for (var i = 0; i < wsClientsLen; ++ i ) {
+        var client = wsClients[i]
+        if (!client.closed) {
+            client.sendText(
+                JSON.stringify(
+                    {
+                        LAST_SUCCESS_BUILD_STAMP: LAST_SUCCESS_BUILD_STAMP
+                    }
+                )
+
+            )
+        }
+    }
+}
+
+function setUpWebSocket() {
+    var WebSocket = require('./minisocket.js').MiniWebSocket
+    var id = setInterval(notifyClients, 3000)
+    require('http').createServer()
+        .on('upgrade', function (req, socket, upgradeHead) {
+            dlog("connection opened");
+            var ws = new WebSocket(req, socket, upgradeHead);
+            wsClients.push(ws)
+        })
+        .on('error', function (err) {
+            console.error(err)
+            process.exit(2)
+        })
+        .listen(webSocketPort, "localhost");
+}
+
+
+
 /**
  * @type {string[]}
  */
@@ -28,17 +76,27 @@ for (var i = 2; i < process_argv.length; ++i) {
         // TODO boundary safety check
         // Not really needed
         postBuild = process_argv[++i]
+    } else if (current === "-ws") {
+        var portNumber = parseInt(process_argv[++i])
+        dlog(`WebSocket port number: ${portNumber}`)
+        if (!isNaN (portNumber )){
+            webSocketPort = portNumber
+        }
+        useWebSocket = true
+
     } else {
         delegate_args.push(current)
         if (current === '-w') {
             watch_mode = true
         } else if (current === "-verbose") {
             verbose = true
 
-        } 
+        }
     }
 }
 
+
+
 try {
     child_process.execFileSync(bsb_exe, delegate_args, { stdio: 'inherit' })
 } catch (e) {
@@ -55,7 +113,9 @@ try {
 if (watch_mode) {
     var fs = require('fs')
     var path = require('path')
-
+    if (useWebSocket) {
+        setUpWebSocket()
+    }
     // for column one based error message
     process.env.BS_VSCODE = '1'
     var cwd = process.cwd()
@@ -192,20 +252,12 @@ if (watch_mode) {
     function logFinish(code) {
         if (std_is_tty) {
             if (code === 0) {
-                if (postBuild) {
-                    console.log("running postbuild command", postBuild)
-                    child_process.exec(postBuild, { shell: true })
-                }
                 console.log("\x1b[36m>>>> Finish compiling\x1b[0m")
             } else {
                 console.log("\x1b[1;31m>>>> Finish compiling(exit: " + code + ")\x1b[0m")
             }
         } else {
             if (code === 0) {
-                if (postBuild) {
-                    console.log("running postbuild command", postBuild)
-                    child_process.exec(postBuild, { shell: true })
-                }
                 console.log(">>>> Finish compiling")
             } else {
                 console.log(">>>> Finish compiling(exit: " + code + ")")
@@ -227,6 +279,14 @@ if (watch_mode) {
      * @param signal {string}
      */
     function build_finished_callback(code, signal) {
+        if(code === 0){
+            LAST_SUCCESS_BUILD_STAMP = + new Date() ;
+            notifyClients()
+            if(postBuild){
+                dlog(`running postbuild command: ${postBuild}`)
+                child_process.exec(postBuild, { shell: true })
+            }
+        }
         logFinish(code)
         releaseBuild()
         if (needRebuild()) {

lib/minisocket.js

@@ -0,0 +1,93 @@
+
+var crypto = require("crypto");
+var KEY_SUFFIX = "258EAFA5-E914-47DA-95CA-C5AB0DC85B11";
+
+function hashWebSocketKey(key) {
+    var sha1 = crypto.createHash("sha1");
+    sha1.update(key + KEY_SUFFIX, "ascii");
+    return sha1.digest("base64");
+}
+
+/**
+ * Limitations: the current implementation does not 
+ * care about what the client send:
+ * - we don't know `ws.close` only know `socket.closed` which is later so it has a latency
+ * - ping pong protocol
+ */
+// http://tools.ietf.org/html/rfc6455#section-5.2
+var opcodes = {
+    TEXT: 1,
+    BINARY: 2,
+    CLOSE: 8,
+    PING: 9,
+    PONG: 10
+};
+
+/**
+ * 
+ * @param {number} opcode 
+ * @param {Buffer} payload 
+ */
+function encodeMessage(opcode, payload) {
+    var buf; // Buffer type
+    var b1 = 0x80 | opcode;
+    // always send message as one frame (fin)
+
+    // second byte: maks and length part 1
+    // followed by 0, 2, or 8 additional bytes of continued length
+    var b2 = 0;
+    // server does not mask frames
+    var length = payload.length;
+    if (length < 126) {
+        buf = new Buffer(payload.length + 2 + 0);
+        // zero extra bytes
+        b2 |= length;
+        buf.writeUInt8(b1, 0);
+        buf.writeUInt8(b2, 1);
+        payload.copy(buf, 2);
+    } else if (length < (1 << 16)) {
+        buf = new Buffer(payload.length + 2 + 2);
+        // two bytes extra
+        b2 |= 126;
+        buf.writeUInt8(b1, 0);
+        buf.writeUInt8(b2, 1);
+        // add two byte length
+        buf.writeUInt16BE(length, 2);
+        payload.copy(buf, 4);
+    } else {
+        buf = new Buffer(payload.length + 2 + 8);
+        // eight bytes extra
+        b2 |= 127;
+        buf.writeUInt8(b1, 0);
+        buf.writeUInt8(b2, 1);
+        // add eight byte length
+        // note: this implementation cannot handle lengths greater than 2^32
+        // the 32 bit length is prefixed with 0x0000
+        buf.writeUInt32BE(0, 2);
+        buf.writeUInt32BE(length, 6);
+        payload.copy(buf, 10);
+    }
+    return buf;
+};
+
+var upgradeHeader = 'HTTP/1.1 101 Web Socket Protocol Handshake\r\nUpgrade: WebSocket\r\nConnection: Upgrade\r\nsec-websocket-accept: '
+class MiniWebSocket  {
+    constructor(req, socket, upgradeHead) {
+        this.socket = socket;
+        this.closed = false;
+        var self = this;
+        var key = hashWebSocketKey(req.headers["sec-websocket-key"])
+
+        // http://tools.ietf.org/html/rfc6455#section-4.2.2
+        socket.write( upgradeHeader + key + '\r\n\r\n');
+        socket.on("close", function (hadError) {
+            if (!self.closed) {
+                self.closed = true;
+            }
+        });
+    };
+    sendText(obj) {
+        this.socket.write(encodeMessage(opcodes.TEXT,new Buffer(obj, "utf8")))
+    };
+}
+exports.MiniWebSocket = MiniWebSocket