This closes #1544

examples/protocols/stomp/stomp-websockets/chat/stomp.js

@@ -17,10 +17,10 @@
  * under the License.
  */
 
-// Generated by CoffeeScript 1.3.3
 (function() {
   var Byte, Client, Frame, Stomp,
-    __hasProp = {}.hasOwnProperty;
+    __hasProp = {}.hasOwnProperty,
+    __slice = [].slice;
 
   Byte = {
     LF: '\x0A',
@@ -28,6 +28,7 @@
   };
 
   Frame = (function() {
+    var unmarshallSingle;
 
     function Frame(command, headers, body) {
       this.command = command;
@@ -36,23 +37,35 @@
     }
 
     Frame.prototype.toString = function() {
-      var lines, name, value, _ref;
+      var lines, name, skipContentLength, value, _ref;
       lines = [this.command];
+      skipContentLength = this.headers['content-length'] === false ? true : false;
+      if (skipContentLength) {
+        delete this.headers['content-length'];
+      }
       _ref = this.headers;
       for (name in _ref) {
         if (!__hasProp.call(_ref, name)) continue;
         value = _ref[name];
         lines.push("" + name + ":" + value);
       }
-      if (this.body) {
+      if (this.body && !skipContentLength) {
-        lines.push("content-length:" + ('' + this.body).length);
+        lines.push("content-length:" + (Frame.sizeOfUTF8(this.body)));
       }
       lines.push(Byte.LF + this.body);
       return lines.join(Byte.LF);
     };
 
-    Frame._unmarshallSingle = function(data) {
+    Frame.sizeOfUTF8 = function(s) {
-      var body, chr, command, divider, headerLines, headers, i, idx, len, line, start, trim, _i, _j, _ref, _ref1;
+      if (s) {
+        return encodeURI(s).match(/%..|./g).length;
+      } else {
+        return 0;
+      }
+    };
+
+    unmarshallSingle = function(data) {
+      var body, chr, command, divider, headerLines, headers, i, idx, len, line, start, trim, _i, _j, _len, _ref, _ref1;
       divider = data.search(RegExp("" + Byte.LF + Byte.LF));
       headerLines = data.substring(0, divider).split(Byte.LF);
       command = headerLines.shift();
@@ -60,9 +73,9 @@
       trim = function(str) {
         return str.replace(/^\s+|\s+$/g, '');
       };
-      line = idx = null;
+      _ref = headerLines.reverse();
-      for (i = _i = 0, _ref = headerLines.length; 0 <= _ref ? _i < _ref : _i > _ref; i = 0 <= _ref ? ++_i : --_i) {
+      for (_i = 0, _len = _ref.length; _i < _len; _i++) {
-        line = headerLines[i];
+        line = _ref[_i];
         idx = line.indexOf(':');
         headers[trim(line.substring(0, idx))] = trim(line.substring(idx + 1));
       }
@@ -85,19 +98,29 @@
     };
 
     Frame.unmarshall = function(datas) {
-      var data;
+      var frame, frames, last_frame, r;
-      return (function() {
+      frames = datas.split(RegExp("" + Byte.NULL + Byte.LF + "*"));
+      r = {
+        frames: [],
+        partial: ''
+      };
+      r.frames = (function() {
         var _i, _len, _ref, _results;
-        _ref = datas.split(RegExp("" + Byte.NULL + Byte.LF + "*"));
+        _ref = frames.slice(0, -1);
         _results = [];
         for (_i = 0, _len = _ref.length; _i < _len; _i++) {
-          data = _ref[_i];
+          frame = _ref[_i];
-          if ((data != null ? data.length : void 0) > 0) {
+          _results.push(unmarshallSingle(frame));
-            _results.push(Frame._unmarshallSingle(data));
-          }
         }
         return _results;
       })();
+      last_frame = frames.slice(-1)[0];
+      if (last_frame === Byte.LF || (last_frame.search(RegExp("" + Byte.NULL + Byte.LF + "*$"))) !== -1) {
+        r.frames.push(unmarshallSingle(last_frame));
+      } else {
+        r.partial = last_frame;
+      }
+      return r;
     };
 
     Frame.marshall = function(command, headers, body) {
@@ -111,6 +134,7 @@
   })();
 
   Client = (function() {
+    var now;
 
     function Client(ws) {
       this.ws = ws;
@@ -121,21 +145,45 @@
         outgoing: 10000,
         incoming: 10000
       };
+      this.maxWebSocketFrameSize = 16 * 1024;
       this.subscriptions = {};
+      this.partialData = '';
     }
 
+    Client.prototype.debug = function(message) {
+      var _ref;
+      return typeof window !== "undefined" && window !== null ? (_ref = window.console) != null ? _ref.log(message) : void 0 : void 0;
+    };
+
+    now = function() {
+      if (Date.now) {
+        return Date.now();
+      } else {
+        return new Date().valueOf;
+      }
+    };
+
     Client.prototype._transmit = function(command, headers, body) {
       var out;
       out = Frame.marshall(command, headers, body);
       if (typeof this.debug === "function") {
         this.debug(">>> " + out);
       }
+      while (true) {
+        if (out.length > this.maxWebSocketFrameSize) {
+          this.ws.send(out.substring(0, this.maxWebSocketFrameSize));
+          out = out.substring(this.maxWebSocketFrameSize);
+          if (typeof this.debug === "function") {
+            this.debug("remaining = " + out.length);
+          }
+        } else {
           return this.ws.send(out);
+        }
+      }
     };
 
     Client.prototype._setupHeartbeat = function(headers) {
-      var serverIncoming, serverOutgoing, ttl, v, _ref, _ref1,
+      var serverIncoming, serverOutgoing, ttl, v, _ref, _ref1;
-        _this = this;
       if ((_ref = headers.version) !== Stomp.VERSIONS.V1_1 && _ref !== Stomp.VERSIONS.V1_2) {
         return;
       }
@@ -154,37 +202,68 @@
         if (typeof this.debug === "function") {
           this.debug("send PING every " + ttl + "ms");
         }
-        this.pinger = typeof window !== "undefined" && window !== null ? window.setInterval(function() {
+        this.pinger = Stomp.setInterval(ttl, (function(_this) {
+          return function() {
             _this.ws.send(Byte.LF);
             return typeof _this.debug === "function" ? _this.debug(">>> PING") : void 0;
-        }, ttl) : void 0;
+          };
+        })(this));
       }
       if (!(this.heartbeat.incoming === 0 || serverOutgoing === 0)) {
         ttl = Math.max(this.heartbeat.incoming, serverOutgoing);
         if (typeof this.debug === "function") {
           this.debug("check PONG every " + ttl + "ms");
         }
-        return this.ponger = typeof window !== "undefined" && window !== null ? window.setInterval(function() {
+        return this.ponger = Stomp.setInterval(ttl, (function(_this) {
+          return function() {
             var delta;
-          delta = Date.now() - _this.serverActivity;
+            delta = now() - _this.serverActivity;
             if (delta > ttl * 2) {
               if (typeof _this.debug === "function") {
                 _this.debug("did not receive server activity for the last " + delta + "ms");
               }
-            return _this._cleanUp();
+              return _this.ws.close();
             }
-        }, ttl) : void 0;
+          };
+        })(this));
       }
     };
 
-    Client.prototype.connect = function(login, passcode, connectCallback, errorCallback, vhost) {
+    Client.prototype._parseConnect = function() {
-      var _this = this;
+      var args, connectCallback, errorCallback, headers;
-      this.connectCallback = connectCallback;
+      args = 1 <= arguments.length ? __slice.call(arguments, 0) : [];
+      headers = {};
+      switch (args.length) {
+        case 2:
+          headers = args[0], connectCallback = args[1];
+          break;
+        case 3:
+          if (args[1] instanceof Function) {
+            headers = args[0], connectCallback = args[1], errorCallback = args[2];
+          } else {
+            headers.login = args[0], headers.passcode = args[1], connectCallback = args[2];
+          }
+          break;
+        case 4:
+          headers.login = args[0], headers.passcode = args[1], connectCallback = args[2], errorCallback = args[3];
+          break;
+        default:
+          headers.login = args[0], headers.passcode = args[1], connectCallback = args[2], errorCallback = args[3], headers.host = args[4];
+      }
+      return [headers, connectCallback, errorCallback];
+    };
+
+    Client.prototype.connect = function() {
+      var args, errorCallback, headers, out;
+      args = 1 <= arguments.length ? __slice.call(arguments, 0) : [];
+      out = this._parseConnect.apply(this, args);
+      headers = out[0], this.connectCallback = out[1], errorCallback = out[2];
       if (typeof this.debug === "function") {
         this.debug("Opening Web Socket...");
       }
-      this.ws.onmessage = function(evt) {
+      this.ws.onmessage = (function(_this) {
-        var arr, c, data, frame, onreceive, _i, _len, _ref, _results;
+        return function(evt) {
+          var arr, c, client, data, frame, messageID, onreceive, subscription, unmarshalledData, _i, _len, _ref, _results;
           data = typeof ArrayBuffer !== 'undefined' && evt.data instanceof ArrayBuffer ? (arr = new Uint8Array(evt.data), typeof _this.debug === "function" ? _this.debug("--- got data length: " + arr.length) : void 0, ((function() {
             var _i, _len, _results;
             _results = [];
@@ -194,7 +273,7 @@
             }
             return _results;
           })()).join('')) : evt.data;
-        _this.serverActivity = Date.now();
+          _this.serverActivity = now();
           if (data === Byte.LF) {
             if (typeof _this.debug === "function") {
               _this.debug("<<< PONG");
@@ -204,7 +283,9 @@
           if (typeof _this.debug === "function") {
             _this.debug("<<< " + data);
           }
-        _ref = Frame.unmarshall(data);
+          unmarshalledData = Frame.unmarshall(_this.partialData + data);
+          _this.partialData = unmarshalledData.partial;
+          _ref = unmarshalledData.frames;
           _results = [];
           for (_i = 0, _len = _ref.length; _i < _len; _i++) {
             frame = _ref[_i];
@@ -218,8 +299,27 @@
                 _results.push(typeof _this.connectCallback === "function" ? _this.connectCallback(frame) : void 0);
                 break;
               case "MESSAGE":
-              onreceive = _this.subscriptions[frame.headers.subscription];
+                subscription = frame.headers.subscription;
-              _results.push(typeof onreceive === "function" ? onreceive(frame) : void 0);
+                onreceive = _this.subscriptions[subscription] || _this.onreceive;
+                if (onreceive) {
+                  client = _this;
+                  messageID = frame.headers["message-id"];
+                  frame.ack = function(headers) {
+                    if (headers == null) {
+                      headers = {};
+                    }
+                    return client.ack(messageID, subscription, headers);
+                  };
+                  frame.nack = function(headers) {
+                    if (headers == null) {
+                      headers = {};
+                    }
+                    return client.nack(messageID, subscription, headers);
+                  };
+                  _results.push(onreceive(frame));
+                } else {
+                  _results.push(typeof _this.debug === "function" ? _this.debug("Unhandled received MESSAGE: " + frame) : void 0);
+                }
                 break;
               case "RECEIPT":
                 _results.push(typeof _this.onreceipt === "function" ? _this.onreceipt(frame) : void 0);
@@ -233,53 +333,48 @@
           }
           return _results;
         };
-      this.ws.onclose = function() {
+      })(this);
+      this.ws.onclose = (function(_this) {
+        return function() {
           var msg;
           msg = "Whoops! Lost connection to " + _this.ws.url;
           if (typeof _this.debug === "function") {
             _this.debug(msg);
           }
+          _this._cleanUp();
           return typeof errorCallback === "function" ? errorCallback(msg) : void 0;
         };
-      return this.ws.onopen = function() {
+      })(this);
-        var headers;
+      return this.ws.onopen = (function(_this) {
+        return function() {
           if (typeof _this.debug === "function") {
             _this.debug('Web Socket Opened...');
           }
-        headers = {
+          headers["accept-version"] = Stomp.VERSIONS.supportedVersions();
-          "accept-version": Stomp.VERSIONS.supportedVersions(),
+          headers["heart-beat"] = [_this.heartbeat.outgoing, _this.heartbeat.incoming].join(',');
-          "heart-beat": [_this.heartbeat.outgoing, _this.heartbeat.incoming].join(',')
-        };
-        if (vhost) {
-          headers.host = vhost;
-        }
-        if (login) {
-          headers.login = login;
-        }
-        if (passcode) {
-          headers.passcode = passcode;
-        }
           return _this._transmit("CONNECT", headers);
         };
+      })(this);
     };
 
-    Client.prototype.disconnect = function(disconnectCallback) {
+    Client.prototype.disconnect = function(disconnectCallback, headers) {
-      this._transmit("DISCONNECT");
+      if (headers == null) {
+        headers = {};
+      }
+      this._transmit("DISCONNECT", headers);
       this.ws.onclose = null;
+      this.ws.close();
       this._cleanUp();
       return typeof disconnectCallback === "function" ? disconnectCallback() : void 0;
     };
 
     Client.prototype._cleanUp = function() {
-      this.ws.close();
       this.connected = false;
       if (this.pinger) {
-        if (typeof window !== "undefined" && window !== null) {
+        Stomp.clearInterval(this.pinger);
-          window.clearInterval(this.pinger);
-        }
       }
       if (this.ponger) {
-        return typeof window !== "undefined" && window !== null ? window.clearInterval(this.ponger) : void 0;
+        return Stomp.clearInterval(this.ponger);
       }
     };
 
@@ -295,6 +390,7 @@
     };
 
     Client.prototype.subscribe = function(destination, callback, headers) {
+      var client;
       if (headers == null) {
         headers = {};
       }
@@ -304,7 +400,13 @@
       headers.destination = destination;
       this.subscriptions[headers.id] = callback;
       this._transmit("SUBSCRIBE", headers);
-      return headers.id;
+      client = this;
+      return {
+        id: headers.id,
+        unsubscribe: function() {
+          return client.unsubscribe(headers.id);
+        }
+      };
     };
 
     Client.prototype.unsubscribe = function(id) {
@@ -315,9 +417,21 @@
     };
 
     Client.prototype.begin = function(transaction) {
-      return this._transmit("BEGIN", {
+      var client, txid;
-        transaction: transaction
+      txid = transaction || "tx-" + this.counter++;
+      this._transmit("BEGIN", {
+        transaction: txid
       });
+      client = this;
+      return {
+        id: txid,
+        commit: function() {
+          return client.commit(txid);
+        },
+        abort: function() {
+          return client.abort(txid);
+        }
+      };
     };
 
     Client.prototype.commit = function(transaction) {
@@ -355,7 +469,6 @@
   })();
 
   Stomp = {
-    libVersion: "2.0.0-next",
     VERSIONS: {
       V1_0: '1.0',
       V1_1: '1.1',
@@ -379,11 +492,20 @@
     Frame: Frame
   };
 
-  if (typeof window !== "undefined" && window !== null) {
+  if (typeof exports !== "undefined" && exports !== null) {
-    window.Stomp = Stomp;
-  } else {
     exports.Stomp = Stomp;
-    Stomp.WebSocketClass = require('./test/server.mock.js').StompServerMock;
+  }
+
+  if (typeof window !== "undefined" && window !== null) {
+    Stomp.setInterval = function(interval, f) {
+      return window.setInterval(f, interval);
+    };
+    Stomp.clearInterval = function(id) {
+      return window.clearInterval(id);
+    };
+    window.Stomp = Stomp;
+  } else if (!exports) {
+    self.Stomp = Stomp;
   }
 
 }).call(this);