@jasmith79/sequable/dist/esm/take.mjs

Library functions for working with generators

import { toIterator } from "./sequable.mjs";
/**
 * 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.
 */
export function* take(n, sequable) {
    const iter = toIterator(sequable);
    let { value, done } = iter.next();
    while (n-- && !done) {
        yield value;
        ({ value, done } = iter.next());
    }
}
/**
 * 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.
 */
export function* takeWhile(pred, sequable) {
    const iter = toIterator(sequable);
    let { value, done } = iter.next();
    while (!done && pred(value)) {
        yield value;
        ({ value, done } = iter.next());
    }
    if (value !== undefined && pred(value))
        return value;
}