stream-chain/src/utils/takeWhile.js

'use strict';

const {none} = require('../defs');

const takeWhile = (f, finalValue = none) => {
  let test = true;
  return value => {
    if (!test) return finalValue;
    const result = f(value);
    if (result && typeof result.then == 'function') {
      return result.then(result => {
        if (result) return value;
        test = false;
        return finalValue;
      });
    }
    if (result) return value;
    test = false;
    return finalValue;
  };
};

module.exports = takeWhile;