webgme-engine/src/server/util/StringStreamWriter.js

WebGME server and Client API without a GUI

/*eslint-env node*/

/**
 * @author lattmann / https://github.com/lattmann
 */

'use strict';

var Stream = require('stream'),
    util = require('util');

function StringStreamWriter(str, opt) {
    Stream.Writable.call(this, opt);
    this._buffer = Buffer.alloc(0);
}

util.inherits(StringStreamWriter, Stream.Writable);

StringStreamWriter.prototype._write = function (chunk, encoding, callback) {
    // FIXME: This might be slow for big files.
    this._buffer = Buffer.concat([this._buffer, Buffer.from(chunk)]);
    callback(null);
};

StringStreamWriter.prototype.getBuffer = function () {
    return this._buffer;
};

StringStreamWriter.prototype.toString = function () {
    return this._buffer.toString();
};

StringStreamWriter.prototype.toJSON = function () {
    return JSON.parse(this.toString());
};

module.exports = StringStreamWriter;