|
1 | 1 | import type { EventHook } from '@vueuse/core'
|
2 | 2 | import { tryOnScopeDispose } from '@vueuse/core'
|
3 | 3 |
|
4 |
| -/** |
5 |
| - * Source code taken from https://github.com/vueuse/vueuse/blob/main/packages/shared/createEventHook/index.ts |
6 |
| - * |
7 |
| - * Modified to be able to check if there are any event listeners |
8 |
| - */ |
9 | 4 | export interface EventHookExtended<T> extends EventHook<T> {
|
| 5 | + /** true if any user listeners are registered (emitter ignored) */ |
10 | 6 | hasListeners: () => boolean
|
11 |
| - fns: Set<(param: T) => void> |
| 7 | + /** current user listeners (read-only; do not mutate externally) */ |
| 8 | + listeners: ReadonlySet<(param: T) => void> |
| 9 | + /** wire a single external emitter (e.g., for `emit`) */ |
| 10 | + setEmitter: (fn: (param: T) => void) => void |
| 11 | + /** remove the external emitter */ |
| 12 | + removeEmitter: () => void |
| 13 | + /** wire a function to detect if any emit listeners exist (e.g., for `$listeners` in Vue 2) */ |
| 14 | + setHasEmitListeners: (fn: () => boolean) => void |
| 15 | + /** remove the emit listeners detector */ |
| 16 | + removeHasEmitListeners: () => void |
12 | 17 | }
|
13 | 18 |
|
14 |
| -export function createExtendedEventHook<T = any>(defaultHandler?: (param: T) => void): EventHookExtended<T> { |
15 |
| - const fns = new Set<(param: T) => void>() |
| 19 | +type Handler<T = any> = (param: T) => any | Promise<any> |
| 20 | + |
| 21 | +const noop: Handler = () => {} |
16 | 22 |
|
17 |
| - let hasDefaultHandler = false |
| 23 | +export function createExtendedEventHook<T = any>(defaultHandler?: (param: T) => void): EventHookExtended<T> { |
| 24 | + const listeners = new Set<Handler>() |
| 25 | + let emitter: Handler = noop |
| 26 | + let hasEmitListeners = () => false |
18 | 27 |
|
19 |
| - const hasListeners = () => fns.size > 0 |
| 28 | + const hasListeners = () => listeners.size > 0 || hasEmitListeners() |
20 | 29 |
|
21 |
| - if (defaultHandler) { |
22 |
| - hasDefaultHandler = true |
23 |
| - fns.add(defaultHandler) |
| 30 | + const setEmitter = (fn: Handler) => { |
| 31 | + emitter = fn |
24 | 32 | }
|
25 | 33 |
|
26 |
| - const off = (fn: (param: T) => void) => { |
27 |
| - fns.delete(fn) |
| 34 | + const removeEmitter = () => { |
| 35 | + emitter = noop |
28 | 36 | }
|
29 | 37 |
|
30 |
| - const on = (fn: (param: T) => void) => { |
31 |
| - if (defaultHandler && hasDefaultHandler) { |
32 |
| - fns.delete(defaultHandler) |
33 |
| - } |
| 38 | + const setHasEmitListeners = (fn: () => boolean) => { |
| 39 | + hasEmitListeners = fn |
| 40 | + } |
34 | 41 |
|
35 |
| - fns.add(fn) |
| 42 | + const removeHasEmitListeners = () => { |
| 43 | + hasEmitListeners = () => false |
| 44 | + } |
36 | 45 |
|
37 |
| - const offFn = () => { |
38 |
| - off(fn) |
| 46 | + const off = (fn: Handler) => { |
| 47 | + listeners.delete(fn) |
| 48 | + } |
39 | 49 |
|
40 |
| - if (defaultHandler && hasDefaultHandler) { |
41 |
| - fns.add(defaultHandler) |
42 |
| - } |
43 |
| - } |
| 50 | + const on = (fn: Handler) => { |
| 51 | + listeners.add(fn) |
44 | 52 |
|
| 53 | + const offFn = () => off(fn) |
45 | 54 | tryOnScopeDispose(offFn)
|
46 | 55 |
|
47 |
| - return { |
48 |
| - off: offFn, |
49 |
| - } |
| 56 | + return { off: offFn } |
50 | 57 | }
|
51 | 58 |
|
| 59 | + /** |
| 60 | + * Trigger order: |
| 61 | + * 1) If any user listeners OR an emitter exist -> call all of those (defaultHandler is skipped) |
| 62 | + * 2) Else (no listeners and no emitter) -> call defaultHandler (if provided) |
| 63 | + * |
| 64 | + * Errors are isolated via allSettled so one failing handler doesn't break others. |
| 65 | + */ |
52 | 66 | const trigger = (param: T) => {
|
53 |
| - return Promise.all(Array.from(fns).map((fn) => fn(param))) |
| 67 | + const queue: Handler[] = [emitter] |
| 68 | + |
| 69 | + if (hasListeners()) { |
| 70 | + queue.push(...listeners) |
| 71 | + } else if (defaultHandler) { |
| 72 | + queue.push(defaultHandler) |
| 73 | + } |
| 74 | + |
| 75 | + return Promise.allSettled(queue.map((fn) => fn(param))) |
54 | 76 | }
|
55 | 77 |
|
56 | 78 | return {
|
57 | 79 | on,
|
58 | 80 | off,
|
59 | 81 | trigger,
|
60 | 82 | hasListeners,
|
61 |
| - fns, |
| 83 | + listeners, |
| 84 | + setEmitter, |
| 85 | + removeEmitter, |
| 86 | + setHasEmitListeners, |
| 87 | + removeHasEmitListeners, |
62 | 88 | }
|
63 | 89 | }
|
0 commit comments