tedious 1.12.3 → 1.14.0

package/lib/token/stream-parser.js

@@ -1,344 +1,464 @@
-var _regeneratorRuntime = require('babel-runtime/regenerator').default;
+'use strict';
 
-// Generated by CoffeeScript 1.10.0
-var Parser,
-    StreamParser,
-    TYPE,
-    tokenParsers,
-    extend = function (child, parent) {
-  for (var key in parent) {
-    if (hasProp.call(parent, key)) child[key] = parent[key];
-  }function ctor() {
-    this.constructor = child;
-  }ctor.prototype = parent.prototype;child.prototype = new ctor();child.__super__ = parent.prototype;return child;
-},
-    hasProp = ({}).hasOwnProperty;
+var _get = require('babel-runtime/helpers/get')['default'];
 
-StreamParser = require("../stream-parser");
+var _inherits = require('babel-runtime/helpers/inherits')['default'];
 
-TYPE = require('./token').TYPE;
+var _createClass = require('babel-runtime/helpers/create-class')['default'];
 
-tokenParsers = {};
+var _classCallCheck = require('babel-runtime/helpers/class-call-check')['default'];
 
-tokenParsers[TYPE.COLMETADATA] = require('./colmetadata-token-parser');
+var Transform = require('readable-stream').Transform;
+var TYPE = require('./token').TYPE;
 
+var tokenParsers = {};
+tokenParsers[TYPE.COLMETADATA] = require('./colmetadata-token-parser');
 tokenParsers[TYPE.DONE] = require('./done-token-parser').doneParser;
-
 tokenParsers[TYPE.DONEINPROC] = require('./done-token-parser').doneInProcParser;
-
 tokenParsers[TYPE.DONEPROC] = require('./done-token-parser').doneProcParser;
-
 tokenParsers[TYPE.ENVCHANGE] = require('./env-change-token-parser');
-
 tokenParsers[TYPE.ERROR] = require('./infoerror-token-parser').errorParser;
-
 tokenParsers[TYPE.INFO] = require('./infoerror-token-parser').infoParser;
-
 tokenParsers[TYPE.LOGINACK] = require('./loginack-token-parser');
-
 tokenParsers[TYPE.ORDER] = require('./order-token-parser');
-
 tokenParsers[TYPE.RETURNSTATUS] = require('./returnstatus-token-parser');
-
 tokenParsers[TYPE.RETURNVALUE] = require('./returnvalue-token-parser');
-
 tokenParsers[TYPE.ROW] = require('./row-token-parser');
-
 tokenParsers[TYPE.NBCROW] = require('./nbcrow-token-parser');
-
 tokenParsers[TYPE.SSPI] = require('./sspi-token-parser');
 
-module.exports = Parser = (function (superClass) {
-  extend(Parser, superClass);
+module.exports = (function (_Transform) {
+  _inherits(Parser, _Transform);
 
   function Parser(debug, colMetadata, options) {
+    _classCallCheck(this, Parser);
+
+    _get(Object.getPrototypeOf(Parser.prototype), 'constructor', this).call(this, { objectMode: true });
+
     this.debug = debug;
     this.colMetadata = colMetadata;
     this.options = options;
-    Parser.__super__.constructor.call(this);
-  }
 
-  Parser.prototype.parser = _regeneratorRuntime.mark(function callee$1$0() {
-    var token, type;
-    return _regeneratorRuntime.wrap(function callee$1$0$(context$2$0) {
-      while (1) switch (context$2$0.prev = context$2$0.next) {
-        case 0:
-          if (!true) {
-            context$2$0.next = 13;
-            break;
-          }
-
-          context$2$0.next = 3;
-          return this.readUInt8();
-
-        case 3:
-          type = context$2$0.sent;
+    this.buffer = new Buffer(0);
+    this.position = 0;
+    this.suspended = false;
+    this.await = undefined;
+    this.next = undefined;
+  }
 
-          if (!tokenParsers[type]) {
-            context$2$0.next = 10;
-            break;
-          }
+  _createClass(Parser, [{
+    key: '_transform',
+    value: function _transform(input, encoding, done) {
+      if (this.position === this.buffer.length) {
+        this.buffer = input;
+      } else {
+        this.buffer = Buffer.concat([this.buffer.slice(this.position), input]);
+      }
+      this.position = 0;
 
-          return context$2$0.delegateYield(tokenParsers[type](this, this.colMetadata, this.options), 't0', 6);
+      // This will be called once we need to wait for more data to
+      // become available
+      this.await = done;
 
-        case 6:
-          token = context$2$0.t0;
+      if (this.suspended) {
+        // Unsuspend and continue from where ever we left off.
+        this.suspended = false;
+        this.next.call(null);
+      }
 
-          if (token) {
-            switch (token.name) {
-              case 'COLMETADATA':
-                this.colMetadata = token.columns;
-            }
-            this.push(token);
+      // If we're no longer suspended, parse new tokens
+      if (!this.suspended) {
+        // Start the parser
+        this.parseTokens();
+      }
+    }
+  }, {
+    key: 'parseTokens',
+    value: function parseTokens() {
+      var _this = this;
+
+      var doneParsing = function doneParsing(token) {
+        if (token) {
+          switch (token.name) {
+            case 'COLMETADATA':
+              _this.colMetadata = token.columns;
           }
-          context$2$0.next = 11;
-          break;
 
-        case 10:
-          throw new Error("Token type " + type + " not implemented");
+          _this.push(token);
+        }
+      };
 
-        case 11:
-          context$2$0.next = 0;
-          break;
+      while (!this.suspended && this.position + 1 <= this.buffer.length) {
+        var type = this.buffer.readUInt8(this.position, true);
 
-        case 13:
-          return context$2$0.abrupt('return', void 0);
+        this.position += 1;
 
-        case 14:
-        case 'end':
-          return context$2$0.stop();
+        if (tokenParsers[type]) {
+          tokenParsers[type](this, this.colMetadata, this.options, doneParsing);
+        } else {
+          this.emit('error', new Error('Unknown type: ' + type));
+        }
       }
-    }, callee$1$0, this);
-  });
-
-  Parser.prototype.readBVarChar = _regeneratorRuntime.mark(function callee$1$0(name) {
-    var data, length;
-    return _regeneratorRuntime.wrap(function callee$1$0$(context$2$0) {
-      while (1) switch (context$2$0.prev = context$2$0.next) {
-        case 0:
-          context$2$0.next = 2;
-          return this.readUInt8();
-
-        case 2:
-          length = context$2$0.sent;
-          context$2$0.next = 5;
-          return this.readBuffer(length * 2);
-
-        case 5:
-          data = context$2$0.sent;
-          return context$2$0.abrupt('return', data.toString("ucs2"));
-
-        case 7:
-        case 'end':
-          return context$2$0.stop();
-      }
-    }, callee$1$0, this);
-  });
-
-  Parser.prototype.readUsVarChar = _regeneratorRuntime.mark(function callee$1$0(name) {
-    var data, length;
-    return _regeneratorRuntime.wrap(function callee$1$0$(context$2$0) {
-      while (1) switch (context$2$0.prev = context$2$0.next) {
-        case 0:
-          context$2$0.next = 2;
-          return this.readUInt16LE();
-
-        case 2:
-          length = context$2$0.sent;
-          context$2$0.next = 5;
-          return this.readBuffer(length * 2);
-
-        case 5:
-          data = context$2$0.sent;
-          return context$2$0.abrupt('return', data.toString("ucs2"));
-
-        case 7:
-        case 'end':
-          return context$2$0.stop();
-      }
-    }, callee$1$0, this);
-  });
-
-  Parser.prototype.readBVarByte = _regeneratorRuntime.mark(function callee$1$0(name) {
-    var length;
-    return _regeneratorRuntime.wrap(function callee$1$0$(context$2$0) {
-      while (1) switch (context$2$0.prev = context$2$0.next) {
-        case 0:
-          context$2$0.next = 2;
-          return this.readUInt8();
-
-        case 2:
-          length = context$2$0.sent;
-          context$2$0.next = 5;
-          return this.readBuffer(length);
-
-        case 5:
-          return context$2$0.abrupt('return', context$2$0.sent);
-
-        case 6:
-        case 'end':
-          return context$2$0.stop();
-      }
-    }, callee$1$0, this);
-  });
-
-  Parser.prototype.readUsVarByte = _regeneratorRuntime.mark(function callee$1$0(name) {
-    var length;
-    return _regeneratorRuntime.wrap(function callee$1$0$(context$2$0) {
-      while (1) switch (context$2$0.prev = context$2$0.next) {
-        case 0:
-          context$2$0.next = 2;
-          return this.readUInt16LE();
-
-        case 2:
-          length = context$2$0.sent;
-          context$2$0.next = 5;
-          return this.readBuffer(length);
-
-        case 5:
-          return context$2$0.abrupt('return', context$2$0.sent);
-
-        case 6:
-        case 'end':
-          return context$2$0.stop();
-      }
-    }, callee$1$0, this);
-  });
-
-  Parser.prototype.readUInt24LE = _regeneratorRuntime.mark(function callee$1$0(name) {
-    var high, low;
-    return _regeneratorRuntime.wrap(function callee$1$0$(context$2$0) {
-      while (1) switch (context$2$0.prev = context$2$0.next) {
-        case 0:
-          context$2$0.next = 2;
-          return this.readUInt16LE();
-
-        case 2:
-          low = context$2$0.sent;
-          context$2$0.next = 5;
-          return this.readUInt8();
-
-        case 5:
-          high = context$2$0.sent;
-          return context$2$0.abrupt('return', low | high << 16);
-
-        case 7:
-        case 'end':
-          return context$2$0.stop();
-      }
-    }, callee$1$0, this);
-  });
-
-  Parser.prototype.readUInt40LE = _regeneratorRuntime.mark(function callee$1$0(name) {
-    var high, low;
-    return _regeneratorRuntime.wrap(function callee$1$0$(context$2$0) {
-      while (1) switch (context$2$0.prev = context$2$0.next) {
-        case 0:
-          context$2$0.next = 2;
-          return this.readUInt32LE();
-
-        case 2:
-          low = context$2$0.sent;
-          context$2$0.next = 5;
-          return this.readUInt8();
-
-        case 5:
-          high = context$2$0.sent;
-          return context$2$0.abrupt('return', 0x100000000 * high + low);
-
-        case 7:
-        case 'end':
-          return context$2$0.stop();
-      }
-    }, callee$1$0, this);
-  });
-
-  Parser.prototype.readUNumeric64LE = _regeneratorRuntime.mark(function callee$1$0(name) {
-    var high, low;
-    return _regeneratorRuntime.wrap(function callee$1$0$(context$2$0) {
-      while (1) switch (context$2$0.prev = context$2$0.next) {
-        case 0:
-          context$2$0.next = 2;
-          return this.readUInt32LE();
-
-        case 2:
-          low = context$2$0.sent;
-          context$2$0.next = 5;
-          return this.readUInt32LE();
-
-        case 5:
-          high = context$2$0.sent;
-          return context$2$0.abrupt('return', 0x100000000 * high + low);
-
-        case 7:
-        case 'end':
-          return context$2$0.stop();
-      }
-    }, callee$1$0, this);
-  });
-
-  Parser.prototype.readUNumeric96LE = _regeneratorRuntime.mark(function callee$1$0(name) {
-    var dword1, dword2, dword3;
-    return _regeneratorRuntime.wrap(function callee$1$0$(context$2$0) {
-      while (1) switch (context$2$0.prev = context$2$0.next) {
-        case 0:
-          context$2$0.next = 2;
-          return this.readUInt32LE();
-
-        case 2:
-          dword1 = context$2$0.sent;
-          context$2$0.next = 5;
-          return this.readUInt32LE();
-
-        case 5:
-          dword2 = context$2$0.sent;
-          context$2$0.next = 8;
-          return this.readUInt32LE();
-
-        case 8:
-          dword3 = context$2$0.sent;
-          return context$2$0.abrupt('return', dword1 + 0x100000000 * dword2 + 0x100000000 * 0x100000000 * dword3);
-
-        case 10:
-        case 'end':
-          return context$2$0.stop();
+
+      if (!this.suspended && this.position === this.buffer.length) {
+        // If we reached the end of the buffer, we can stop parsing now.
+        return this.await.call(null);
       }
-    }, callee$1$0, this);
-  });
-
-  Parser.prototype.readUNumeric128LE = _regeneratorRuntime.mark(function callee$1$0(name) {
-    var dword1, dword2, dword3, dword4;
-    return _regeneratorRuntime.wrap(function callee$1$0$(context$2$0) {
-      while (1) switch (context$2$0.prev = context$2$0.next) {
-        case 0:
-          context$2$0.next = 2;
-          return this.readUInt32LE();
-
-        case 2:
-          dword1 = context$2$0.sent;
-          context$2$0.next = 5;
-          return this.readUInt32LE();
-
-        case 5:
-          dword2 = context$2$0.sent;
-          context$2$0.next = 8;
-          return this.readUInt32LE();
-
-        case 8:
-          dword3 = context$2$0.sent;
-          context$2$0.next = 11;
-          return this.readUInt32LE();
-
-        case 11:
-          dword4 = context$2$0.sent;
-          return context$2$0.abrupt('return', dword1 + 0x100000000 * dword2 + 0x100000000 * 0x100000000 * dword3 + 0x100000000 * 0x100000000 * 0x100000000 * dword4);
-
-        case 13:
-        case 'end':
-          return context$2$0.stop();
+    }
+  }, {
+    key: 'suspend',
+    value: function suspend(next) {
+      this.suspended = true;
+      this.next = next;
+      this.await.call(null);
+    }
+  }, {
+    key: 'awaitData',
+    value: function awaitData(length, callback) {
+      var _this2 = this;
+
+      if (this.position + length <= this.buffer.length) {
+        callback();
+      } else {
+        this.suspend(function () {
+          _this2.awaitData(length, callback);
+        });
       }
-    }, callee$1$0, this);
-  });
+    }
+  }, {
+    key: 'readInt8',
+    value: function readInt8(callback) {
+      var _this3 = this;
+
+      this.awaitData(1, function () {
+        var data = _this3.buffer.readInt8(_this3.position);
+        _this3.position += 1;
+        callback(data);
+      });
+    }
+  }, {
+    key: 'readUInt8',
+    value: function readUInt8(callback) {
+      var _this4 = this;
+
+      this.awaitData(1, function () {
+        var data = _this4.buffer.readUInt8(_this4.position);
+        _this4.position += 1;
+        callback(data);
+      });
+    }
+  }, {
+    key: 'readInt16LE',
+    value: function readInt16LE(callback) {
+      var _this5 = this;
+
+      this.awaitData(2, function () {
+        var data = _this5.buffer.readInt16LE(_this5.position);
+        _this5.position += 2;
+        callback(data);
+      });
+    }
+  }, {
+    key: 'readInt16BE',
+    value: function readInt16BE(callback) {
+      var _this6 = this;
+
+      this.awaitData(2, function () {
+        var data = _this6.buffer.readInt16BE(_this6.position);
+        _this6.position += 2;
+        callback(data);
+      });
+    }
+  }, {
+    key: 'readUInt16LE',
+    value: function readUInt16LE(callback) {
+      var _this7 = this;
+
+      this.awaitData(2, function () {
+        var data = _this7.buffer.readUInt16LE(_this7.position);
+        _this7.position += 2;
+        callback(data);
+      });
+    }
+  }, {
+    key: 'readUInt16BE',
+    value: function readUInt16BE(callback) {
+      var _this8 = this;
+
+      this.awaitData(2, function () {
+        var data = _this8.buffer.readUInt16BE(_this8.position);
+        _this8.position += 2;
+        callback(data);
+      });
+    }
+  }, {
+    key: 'readInt32LE',
+    value: function readInt32LE(callback) {
+      var _this9 = this;
+
+      this.awaitData(4, function () {
+        var data = _this9.buffer.readInt32LE(_this9.position);
+        _this9.position += 4;
+        callback(data);
+      });
+    }
+  }, {
+    key: 'readInt32BE',
+    value: function readInt32BE(callback) {
+      var _this10 = this;
+
+      this.awaitData(4, function () {
+        var data = _this10.buffer.readInt32BE(_this10.position);
+        _this10.position += 4;
+        callback(data);
+      });
+    }
+  }, {
+    key: 'readUInt32LE',
+    value: function readUInt32LE(callback) {
+      var _this11 = this;
+
+      this.awaitData(4, function () {
+        var data = _this11.buffer.readUInt32LE(_this11.position);
+        _this11.position += 4;
+        callback(data);
+      });
+    }
+  }, {
+    key: 'readUInt32BE',
+    value: function readUInt32BE(callback) {
+      var _this12 = this;
+
+      this.awaitData(4, function () {
+        var data = _this12.buffer.readUInt32BE(_this12.position);
+        _this12.position += 4;
+        callback(data);
+      });
+    }
+  }, {
+    key: 'readInt64LE',
+    value: function readInt64LE(callback) {
+      var _this13 = this;
+
+      this.awaitData(8, function () {
+        var data = Math.pow(2, 32) * _this13.buffer.readInt32LE(_this13.position + 4) + (_this13.buffer[_this13.position + 4] & 0x80 === 0x80 ? 1 : -1) * _this13.buffer.readUInt32LE(_this13.position);
+        _this13.position += 8;
+        callback(data);
+      });
+    }
+  }, {
+    key: 'readInt64BE',
+    value: function readInt64BE(callback) {
+      var _this14 = this;
+
+      this.awaitData(8, function () {
+        var data = Math.pow(2, 32) * _this14.buffer.readInt32BE(_this14.position) + (_this14.buffer[_this14.position] & 0x80 === 0x80 ? 1 : -1) * _this14.buffer.readUInt32BE(_this14.position + 4);
+        _this14.position += 8;
+        callback(data);
+      });
+    }
+  }, {
+    key: 'readUInt64LE',
+    value: function readUInt64LE(callback) {
+      var _this15 = this;
+
+      this.awaitData(8, function () {
+        var data = Math.pow(2, 32) * _this15.buffer.readUInt32LE(_this15.position + 4) + _this15.buffer.readUInt32LE(_this15.position);
+        _this15.position += 8;
+        callback(data);
+      });
+    }
+  }, {
+    key: 'readUInt64BE',
+    value: function readUInt64BE(callback) {
+      var _this16 = this;
+
+      this.awaitData(8, function () {
+        var data = Math.pow(2, 32) * _this16.buffer.readUInt32BE(_this16.position) + _this16.buffer.readUInt32BE(_this16.position + 4);
+        _this16.position += 8;
+        callback(data);
+      });
+    }
+  }, {
+    key: 'readFloatLE',
+    value: function readFloatLE(callback) {
+      var _this17 = this;
+
+      this.awaitData(4, function () {
+        var data = _this17.buffer.readFloatLE(_this17.position);
+        _this17.position += 4;
+        callback(data);
+      });
+    }
+  }, {
+    key: 'readFloatBE',
+    value: function readFloatBE(callback) {
+      var _this18 = this;
+
+      this.awaitData(4, function () {
+        var data = _this18.buffer.readFloatBE(_this18.position);
+        _this18.position += 4;
+        callback(data);
+      });
+    }
+  }, {
+    key: 'readDoubleLE',
+    value: function readDoubleLE(callback) {
+      var _this19 = this;
+
+      this.awaitData(8, function () {
+        var data = _this19.buffer.readDoubleLE(_this19.position);
+        _this19.position += 8;
+        callback(data);
+      });
+    }
+  }, {
+    key: 'readDoubleBE',
+    value: function readDoubleBE(callback) {
+      var _this20 = this;
+
+      this.awaitData(8, function () {
+        var data = _this20.buffer.readDoubleBE(_this20.position);
+        _this20.position += 8;
+        callback(data);
+      });
+    }
+  }, {
+    key: 'readUInt24LE',
+    value: function readUInt24LE(callback) {
+      var _this21 = this;
+
+      this.awaitData(3, function () {
+        var low = _this21.buffer.readUInt16LE(_this21.position);
+        var high = _this21.buffer.readUInt8(_this21.position + 2);
+
+        _this21.position += 3;
+
+        callback(low | high << 16);
+      });
+    }
+  }, {
+    key: 'readUInt40LE',
+    value: function readUInt40LE(callback) {
+      var _this22 = this;
+
+      this.awaitData(5, function () {
+        var low = _this22.buffer.readUInt32LE(_this22.position);
+        var high = _this22.buffer.readUInt8(_this22.position + 4);
+
+        _this22.position += 5;
+
+        callback(0x100000000 * high + low);
+      });
+    }
+  }, {
+    key: 'readUNumeric64LE',
+    value: function readUNumeric64LE(callback) {
+      var _this23 = this;
+
+      this.awaitData(8, function () {
+        var low = _this23.buffer.readUInt32LE(_this23.position);
+        var high = _this23.buffer.readUInt32LE(_this23.position + 4);
+
+        _this23.position += 8;
+
+        callback(0x100000000 * high + low);
+      });
+    }
+  }, {
+    key: 'readUNumeric96LE',
+    value: function readUNumeric96LE(callback) {
+      var _this24 = this;
+
+      this.awaitData(12, function () {
+        var dword1 = _this24.buffer.readUInt32LE(_this24.position);
+        var dword2 = _this24.buffer.readUInt32LE(_this24.position + 4);
+        var dword3 = _this24.buffer.readUInt32LE(_this24.position + 8);
+
+        _this24.position += 12;
+
+        callback(dword1 + 0x100000000 * dword2 + 0x100000000 * 0x100000000 * dword3);
+      });
+    }
+  }, {
+    key: 'readUNumeric128LE',
+    value: function readUNumeric128LE(callback) {
+      var _this25 = this;
+
+      this.awaitData(16, function () {
+        var dword1 = _this25.buffer.readUInt32LE(_this25.position);
+        var dword2 = _this25.buffer.readUInt32LE(_this25.position + 4);
+        var dword3 = _this25.buffer.readUInt32LE(_this25.position + 8);
+        var dword4 = _this25.buffer.readUInt32LE(_this25.position + 12);
+
+        _this25.position += 16;
+
+        callback(dword1 + 0x100000000 * dword2 + 0x100000000 * 0x100000000 * dword3 + 0x100000000 * 0x100000000 * 0x100000000 * dword4);
+      });
+    }
+
+    // Variable length data
+
+  }, {
+    key: 'readBuffer',
+    value: function readBuffer(length, callback) {
+      var _this26 = this;
+
+      this.awaitData(length, function () {
+        var data = _this26.buffer.slice(_this26.position, _this26.position + length);
+        _this26.position += length;
+        callback(data);
+      });
+    }
+
+    // Read a Unicode String (BVARCHAR)
+  }, {
+    key: 'readBVarChar',
+    value: function readBVarChar(callback) {
+      var _this27 = this;
+
+      this.readUInt8(function (length) {
+        _this27.readBuffer(length * 2, function (data) {
+          callback(data.toString('ucs2'));
+        });
+      });
+    }
+
+    // Read a Unicode String (USVARCHAR)
+  }, {
+    key: 'readUsVarChar',
+    value: function readUsVarChar(callback) {
+      var _this28 = this;
+
+      this.readUInt16LE(function (length) {
+        _this28.readBuffer(length * 2, function (data) {
+          callback(data.toString('ucs2'));
+        });
+      });
+    }
+
+    // Read binary data (BVARBYTE)
+  }, {
+    key: 'readBVarByte',
+    value: function readBVarByte(callback) {
+      var _this29 = this;
+
+      this.readUInt8(function (length) {
+        _this29.readBuffer(length, callback);
+      });
+    }
+
+    // Read binary data (USVARBYTE)
+  }, {
+    key: 'readUsVarByte',
+    value: function readUsVarByte(callback) {
+      var _this30 = this;
+
+      this.readUInt16LE(function (length) {
+        _this30.readBuffer(length, callback);
+      });
+    }
+  }]);
 
   return Parser;
-})(StreamParser);
+})(Transform);