useEventListener.ts 1.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960
  1. import type { Ref } from 'vue';
  2. import { ref, watch, unref } from 'vue';
  3. import { useThrottleFn, useDebounceFn } from '@vueuse/core';
  4. export type RemoveEventFn = () => void;
  5. export interface UseEventParams {
  6. el?: Element | Ref<Element | undefined> | Window | any;
  7. name: string;
  8. listener: EventListener;
  9. options?: boolean | AddEventListenerOptions;
  10. autoRemove?: boolean;
  11. isDebounce?: boolean;
  12. wait?: number;
  13. }
  14. export function useEventListener({
  15. el = window,
  16. name,
  17. listener,
  18. options,
  19. autoRemove = true,
  20. isDebounce = true,
  21. wait = 80,
  22. }: UseEventParams): { removeEvent: RemoveEventFn } {
  23. /* eslint-disable-next-line */
  24. let remove: RemoveEventFn = () => {};
  25. const isAddRef = ref(false);
  26. if (el) {
  27. const element: Ref<Element> = ref(el as Element);
  28. const handler = isDebounce ? useDebounceFn(listener, wait) : useThrottleFn(listener, wait);
  29. const realHandler = wait ? handler : listener;
  30. const removeEventListener = (e: Element) => {
  31. isAddRef.value = true;
  32. e.removeEventListener(name, realHandler, options);
  33. };
  34. const addEventListener = (e: Element) => e.addEventListener(name, realHandler, options);
  35. const removeWatch = watch(
  36. element,
  37. (v, _ov, cleanUp) => {
  38. if (v) {
  39. !unref(isAddRef) && addEventListener(v);
  40. cleanUp(() => {
  41. autoRemove && removeEventListener(v);
  42. });
  43. }
  44. },
  45. { immediate: true }
  46. );
  47. remove = () => {
  48. removeEventListener(element.value);
  49. removeWatch();
  50. };
  51. }
  52. return { removeEvent: remove };
  53. }