aboutsummaryrefslogtreecommitdiff
path: root/source/utility/pubsub.ts
blob: bb8679a7b1c60ed2bff9dac2be4882eafaa83768 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
export class PubSub {
  #target = new EventTarget();

  publish(trigger: string, payload: unknown): Promise<void> {
    this.#target.dispatchEvent(new CustomEvent(trigger, { detail: payload }));
    return Promise.resolve();
  }

  asyncIterator<T>(triggers: string | string[]): AsyncIterableIterator<T> {
    const names = Array.isArray(triggers) ? triggers : [triggers];
    const pulls: Array<(result: IteratorResult<T>) => void> = [];
    const pushes: T[] = [];
    let done = false;

    const listener = (event: Event) => {
      const value = (event as CustomEvent<T>).detail;
      const pull = pulls.shift();

      if (pull)
        pull({ done: false, value });
      else
        pushes.push(value);
    };

    for (const name of names)
      this.#target.addEventListener(name, listener);

    const close = (): Promise<IteratorResult<T>> => {
      if (!done) {
        done = true;

        for (const name of names)
          this.#target.removeEventListener(name, listener);

        for (const pull of pulls.splice(0))
          pull({ done: true, value: undefined });

        pushes.length = 0;
      }

      return Promise.resolve({ done: true, value: undefined });
    };

    return {
      next: () => {
        if (done)
          return Promise.resolve({ done: true, value: undefined });

        if (pushes.length > 0)
          return Promise.resolve({ done: false, value: pushes.shift() as T });

        return new Promise((resolve) => pulls.push(resolve));
      },
      return: close,
      throw: (error: unknown) => {
        close();
        return Promise.reject(error);
      },
      [Symbol.asyncIterator]() {
        return this;
      }
    };
  }
}