57 lines
1.2 KiB
TypeScript
57 lines
1.2 KiB
TypeScript
|
import { Arr } from "../types.ts";
|
||
|
import { pushIter, addIterUtils } from "./util/mod.ts";
|
||
|
|
||
|
type Listener<T> = { push: (arg: T) => void; done: () => void };
|
||
|
export type IterSource<T extends Arr> = ReturnType<typeof fromCallback<T>>;
|
||
|
|
||
|
export function fromCallback<T extends Arr>(
|
||
|
source: (listener: (...values: T) => void) => void,
|
||
|
disconnect?: () => void
|
||
|
) {
|
||
|
let isDone = false;
|
||
|
let listeners: Listener<T>[] = [];
|
||
|
|
||
|
function trigger(...values: T) {
|
||
|
if (isDone) {
|
||
|
return;
|
||
|
}
|
||
|
for (const listener of listeners) {
|
||
|
listener.push(values);
|
||
|
}
|
||
|
}
|
||
|
source(trigger);
|
||
|
|
||
|
function done() {
|
||
|
disconnect?.();
|
||
|
if (isDone) {
|
||
|
return;
|
||
|
}
|
||
|
for (const listener of listeners) {
|
||
|
listener.done();
|
||
|
}
|
||
|
listeners = [];
|
||
|
isDone = true;
|
||
|
}
|
||
|
|
||
|
function addListener(listener: Listener<T>) {
|
||
|
listeners.push(listener);
|
||
|
return () => {
|
||
|
removeListener(listener);
|
||
|
};
|
||
|
}
|
||
|
|
||
|
function removeListener(listener: Listener<T>) {
|
||
|
const index = listeners.indexOf(listener);
|
||
|
listeners.splice(index, 1);
|
||
|
}
|
||
|
|
||
|
return {
|
||
|
iterator: () => {
|
||
|
const { push, done, getIterator } = pushIter<T>();
|
||
|
addListener({ push, done });
|
||
|
return addIterUtils<T>(getIterator());
|
||
|
},
|
||
|
disconnect: done,
|
||
|
};
|
||
|
}
|