|
| 1 | +import { on } from 'svelte/events'; |
| 2 | + |
| 3 | +/** |
| 4 | + * A way to keep track of things you've "on"ed so you can "off" them too. |
| 5 | + * |
| 6 | + * This is needed because Svelte doesn't have an "off" function, it returns |
| 7 | + * unlisten functions from its "on" function, and MDC-Web expects register and |
| 8 | + * deregister functions in the adapters. |
| 9 | + */ |
| 10 | +export class SvelteEventManager { |
| 11 | + elementMap = new Map< |
| 12 | + Element | Window | Document, |
| 13 | + { [k: string]: Map<Function, Function> } |
| 14 | + >(); |
| 15 | + |
| 16 | + /** |
| 17 | + * Listen to an event on an element. |
| 18 | + */ |
| 19 | + on<T extends Event>( |
| 20 | + element: Element | Window | Document, |
| 21 | + event: string, |
| 22 | + handler: (evt: T) => void, |
| 23 | + options?: AddEventListenerOptions, |
| 24 | + ) { |
| 25 | + if (!this.elementMap.has(element)) { |
| 26 | + this.elementMap.set(element, {}); |
| 27 | + } |
| 28 | + |
| 29 | + const eventMap = this.elementMap.get(element); |
| 30 | + |
| 31 | + if (eventMap == null) { |
| 32 | + throw new Error("Event map couldn't be created."); |
| 33 | + } |
| 34 | + |
| 35 | + if (!(event in eventMap)) { |
| 36 | + eventMap[event] = new Map<Function, Function>(); |
| 37 | + } |
| 38 | + |
| 39 | + const handlerMap = eventMap[event]; |
| 40 | + |
| 41 | + handlerMap.set( |
| 42 | + handler, |
| 43 | + on(element, event, handler as EventListener, options), |
| 44 | + ); |
| 45 | + } |
| 46 | + |
| 47 | + /** |
| 48 | + * Unlisten to an event on an element. |
| 49 | + */ |
| 50 | + off<T extends Event>( |
| 51 | + element: Element | Window | Document, |
| 52 | + event: string, |
| 53 | + handler: (evt: T) => void, |
| 54 | + ) { |
| 55 | + const eventMap = this.elementMap.get(element); |
| 56 | + |
| 57 | + if (eventMap == null || !(event in eventMap)) { |
| 58 | + return; |
| 59 | + } |
| 60 | + |
| 61 | + const handlerMap = eventMap[event]; |
| 62 | + const unlisten = handlerMap.get(handler); |
| 63 | + |
| 64 | + if (unlisten != null) { |
| 65 | + unlisten(); |
| 66 | + handlerMap.delete(handler); |
| 67 | + if (handlerMap.size === 0) { |
| 68 | + delete eventMap[event]; |
| 69 | + |
| 70 | + if (Object.keys(eventMap).length === 0) { |
| 71 | + this.elementMap.delete(element); |
| 72 | + } |
| 73 | + } |
| 74 | + } |
| 75 | + } |
| 76 | + |
| 77 | + /** |
| 78 | + * Unlisten to all events managed by this instance. |
| 79 | + */ |
| 80 | + clear() { |
| 81 | + this.elementMap.forEach((eventMaps, _element) => { |
| 82 | + for (let [_event, eventMap] of Object.entries(eventMaps)) { |
| 83 | + eventMap.forEach((unlisten, _handler) => { |
| 84 | + unlisten(); |
| 85 | + }); |
| 86 | + } |
| 87 | + }); |
| 88 | + this.elementMap.clear(); |
| 89 | + } |
| 90 | +} |
0 commit comments