@jasmith79/sequable/dist/take.js

Library functions for working with generators

"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
exports.takeWhile = exports.take = void 0;
const sequable_js_1 = require("./sequable.js");
/**
 * Yields the first n items of the given Iterator, Iterable, or generator
 * function.
 *
 * @param n The number of items to yield.
 * @param sequable The Iterator, Iterable, or generator function sequence
 * to yield from.
 */
function* take(n, sequable) {
    const iter = (0, sequable_js_1.toIterator)(sequable);
    let { value, done } = iter.next();
    while (n-- && !done) {
        yield value;
        ({ value, done } = iter.next());
    }
}
exports.take = take;
/**
 * Yields the items of the given Iterator, Iterable, or generator function
 * until one item fails the supplied predicate function.
 *
 * @param pred The predicate function.
 * @param sequable The Iterator, Iterable, or generator function sequence
 * to yield from.
 */
function* takeWhile(pred, sequable) {
    const iter = (0, sequable_js_1.toIterator)(sequable);
    let { value, done } = iter.next();
    while (!done && pred(value)) {
        yield value;
        ({ value, done } = iter.next());
    }
    if (value !== undefined && pred(value))
        return value;
}
exports.takeWhile = takeWhile;