all files / lib/ writable_streambuffer.js

100% Statements 45/45
100% Branches 22/22
100% Functions 7/7
100% Lines 43/43
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73      21× 21×   21×   21× 21×   21× 21×   21×     21×     21×             21×           21× 14×         21× 14× 14× 14× 14×        
'use strict';
 
var util = require('util');
var stream = require('stream');
var constants = require('./constants');
 
var WritableStreamBuffer = module.exports = function(opts) {
  opts = opts || {};
  opts.decodeStrings = true;
 
  stream.Writable.call(this, opts);
 
  var initialSize = opts.initialSize || constants.DEFAULT_INITIAL_SIZE;
  var incrementAmount = opts.incrementAmount || constants.DEFAULT_INCREMENT_AMOUNT;
 
  var buffer = new Buffer(initialSize);
  var size = 0;
 
  this.size = function() {
    return size;
  };
 
  this.maxSize = function() {
    return buffer.length;
  };
 
  this.getContents = function(length) {
    if(!size) return false;
 
    var data = new Buffer(Math.min(length || size, size));
    buffer.copy(data, 0, 0, data.length);
 
    if(data.length < size)
      buffer.copy(buffer, 0, data.length);
 
    size -= data.length;
 
    return data;
  };
 
  this.getContentsAsString = function(encoding, length) {
    if(!size) return false;
 
    var data = buffer.toString(encoding || 'utf8', 0, Math.min(length || size, size));
    var dataLength = Buffer.byteLength(data);
 
    if(dataLength < size)
      buffer.copy(buffer, 0, dataLength);
 
    size -= dataLength;
    return data;
  };
 
  var increaseBufferIfNecessary = function(incomingDataSize) {
    if((buffer.length - size) < incomingDataSize) {
      var factor = Math.ceil((incomingDataSize - (buffer.length - size)) / incrementAmount);
 
      var newBuffer = new Buffer(buffer.length + (incrementAmount * factor));
      buffer.copy(newBuffer, 0, 0, size);
      buffer = newBuffer;
    }
  };
 
  this._write = function(chunk, encoding, callback) {
    increaseBufferIfNecessary(chunk.length);
    chunk.copy(buffer, size, 0);
    size += chunk.length;
    callback();
  };
};
 
util.inherits(WritableStreamBuffer, stream.Writable);