@klodianimeri/pipejs/src/pipes/intersection.ts

Pipe functions that provide convenient and efficient ways to work with iterators.

import { Pipe } from "../pipe.js";
import { assertSync } from "../util/index.js";

export function intersection(source: Iterable<any, any, unknown>): Pipe {
    assertSync(source);

    return () => {
        let sourceSet: Set<any> = source instanceof Set ? source : new Set<any>(source);

        return (result: IteratorResult<any>): IteratorResult<any> => {
            if (result?.done) {
                return result;
            }

            if (sourceSet.has(result.value)) {
                sourceSet.delete(result.value);
                return result;
            }
        };
    }
}