stream-chain/utils/fold.js

'use strict';

const {Transform} = require('stream');

const defaultInitial = 0;
const defaultReducer = (acc, value) => value;

class Fold extends Transform {
  constructor(options) {
    super(Object.assign({}, options, {writableObjectMode: true, readableObjectMode: true}));
    this._accumulator = defaultInitial;
    this._reducer = defaultReducer;
    if (options) {
      'initial' in options && (this._accumulator = options.initial);
      'reducer' in options && (this._reducer = options.reducer);
    }
  }
  _transform(chunk, encoding, callback) {
    this._accumulator = this._reducer.call(this, this._accumulator, chunk);
    callback(null);
  }
  _final(callback) {
    this.push(this._accumulator);
    callback(null);
  }
  static make(reducer, initial) {
    return new Fold(typeof reducer == 'object' ? reducer : {reducer, initial});
  }
}
Fold.make.Constructor = Fold;

module.exports = Fold.make;