node-redis/lib/parsers/javascript.js

'use strict';

var util   = require("util");

function Packet(type, size) {
    this.type = type;
    this.size = +size;
}

function ReplyParser(return_buffers) {
    this.name = exports.name;
    this.return_buffers = return_buffers;

    this._buffer            = null;
    this._offset            = 0;
    this._encoding          = "utf-8";
}

function IncompleteReadBuffer(message) {
    this.name = "IncompleteReadBuffer";
    this.message = message;
}
util.inherits(IncompleteReadBuffer, Error);

ReplyParser.prototype._parseResult = function (type) {
    var start, end, offset, packetHeader;

    if (type === 43 || type === 45) { // + or -
        // up to the delimiter
        end = this._packetEndOffset() - 1;
        start = this._offset;

        if (end > this._buffer.length) {
            throw new IncompleteReadBuffer("Wait for more data.");
        }

        // include the delimiter
        this._offset = end + 2;

        if (type === 45) {
            return new Error(this._buffer.toString(this._encoding, start, end));
        } else if (this.return_buffers) {
            return this._buffer.slice(start, end);
        }
        return this._buffer.toString(this._encoding, start, end);
    } else if (type === 58) { // :
        // up to the delimiter
        end = this._packetEndOffset() - 1;
        start = this._offset;

        if (end > this._buffer.length) {
            throw new IncompleteReadBuffer("Wait for more data.");
        }

        // include the delimiter
        this._offset = end + 2;

        // return the coerced numeric value
        return +this._buffer.toString('ascii', start, end);
    } else if (type === 36) { // $
        // set a rewind point, as the packet could be larger than the
        // buffer in memory
        offset = this._offset - 1;

        packetHeader = new Packet(type, this.parseHeader());

        // packets with a size of -1 are considered null
        if (packetHeader.size === -1) {
            return null;
        }

        end = this._offset + packetHeader.size;
        start = this._offset;

        if (end > this._buffer.length) {
            throw new IncompleteReadBuffer("Wait for more data.");
        }

        // set the offset to after the delimiter
        this._offset = end + 2;

        if (this.return_buffers) {
            return this._buffer.slice(start, end);
        }
        return this._buffer.toString(this._encoding, start, end);
    } else { // *
        offset = this._offset;
        packetHeader = new Packet(type, this.parseHeader());

        if (packetHeader.size < 0) {
            return null;
        }

        if (packetHeader.size > this._bytesRemaining()) {
            this._offset = offset - 1;
            throw new IncompleteReadBuffer("Wait for more data.");
        }

        var reply = [];
        var ntype, i, res;

        offset = this._offset - 1;

        for (i = 0; i < packetHeader.size; i++) {
            ntype = this._buffer[this._offset++];

            if (this._offset > this._buffer.length) {
                throw new IncompleteReadBuffer("Wait for more data.");
            }
            res = this._parseResult(ntype);
            reply.push(res);
        }

        return reply;
    }
};

ReplyParser.prototype.execute = function (buffer) {
    this.append(buffer);

    var type, ret, offset;

    while (true) {
        offset = this._offset;
        // at least 4 bytes: :1\r\n
        if (this._bytesRemaining() < 4) {
            break;
        }

        try {
            type = this._buffer[this._offset++];

            if (type === 43) { // Strings +
                ret = this._parseResult(type);

                this.send_reply(ret);
            } else  if (type === 45) { // Errors -
                ret = this._parseResult(type);

                this.send_error(ret);
            } else if (type === 58) { // Integers :
                ret = this._parseResult(type);

                this.send_reply(ret);
            } else if (type === 36) { // Bulk strings $
                ret = this._parseResult(type);

                this.send_reply(ret);
            } else if (type === 42) { // Arrays *
                // set a rewind point. if a failure occurs,
                // wait for the next execute()/append() and try again
                offset = this._offset - 1;

                ret = this._parseResult(type);

                this.send_reply(ret);
            } else if (type === 10 || type === 13) {
                break;
            } else {
                var err = new Error('Protocol error, got "' + String.fromCharCode(type) + '" as reply type byte');
                this.send_error(err);
            }
        } catch (err) {
            // catch the error (not enough data), rewind, and wait
            // for the next packet to appear
            this._offset = offset;
            break;
        }
    }
};

ReplyParser.prototype.append = function (newBuffer) {

    // first run
    if (this._buffer === null) {
        this._buffer = newBuffer;

        return;
    }

    // out of data
    if (this._offset >= this._buffer.length) {
        this._buffer = newBuffer;
        this._offset = 0;

        return;
    }

    this._buffer = Buffer.concat([this._buffer.slice(this._offset), newBuffer]);
    this._offset = 0;
};

ReplyParser.prototype.parseHeader = function () {
    var end   = this._packetEndOffset(),
        value = this._buffer.toString('ascii', this._offset, end - 1);

    this._offset = end + 1;

    return value;
};

ReplyParser.prototype._packetEndOffset = function () {
    var offset = this._offset;

    while (this._buffer[offset] !== 0x0d && this._buffer[offset + 1] !== 0x0a) {
        offset++;

        /* istanbul ignore if: activate the js parser out of memory test to test this */
        if (offset >= this._buffer.length) {
            throw new IncompleteReadBuffer("didn't see LF after NL reading multi bulk count (" + offset + " => " + this._buffer.length + ", " + this._offset + ")");
        }
    }

    offset++;
    return offset;
};

ReplyParser.prototype._bytesRemaining = function () {
    return (this._buffer.length - this._offset) < 0 ? 0 : (this._buffer.length - this._offset);
};

exports.Parser = ReplyParser;
exports.name = "javascript";