From 5ca38fb55f7601f4c66742ea284ce207d876c633 Mon Sep 17 00:00:00 2001
From: haoziqaq <357229046@qq.com>
Date: Wed, 4 Dec 2024 00:43:22 +0800
Subject: [PATCH] feat: enhance popover reference

---
 .../varlet-ui/src/menu-select/MenuSelect.vue  |   9 +-
 packages/varlet-ui/src/menu/Menu.vue          |  23 ++--
 packages/varlet-ui/src/menu/props.ts          |   4 +-
 packages/varlet-ui/src/menu/usePopover.ts     | 109 ++++++++++++------
 packages/varlet-ui/src/tooltip/Tooltip.vue    |  23 ++--
 packages/varlet-use/src/useEventListener.ts   |   1 +
 6 files changed, 96 insertions(+), 73 deletions(-)

diff --git a/packages/varlet-ui/src/menu-select/MenuSelect.vue b/packages/varlet-ui/src/menu-select/MenuSelect.vue
index 40e4ede4b6c..76ca18af85c 100644
--- a/packages/varlet-ui/src/menu-select/MenuSelect.vue
+++ b/packages/varlet-ui/src/menu-select/MenuSelect.vue
@@ -56,10 +56,11 @@ import { props } from './props'
 import { createNamespace, formatElevation } from '../utils/components'
 import { useMenuOptions, type MenuSelectProvider } from './provide'
 import { useSelectController } from '../select/useSelectController'
-import { type MenuOptionProvider } from '../menu-option/provide'
 import { call, preventDefault } from '@varlet/shared'
 import { useEventListener, useVModel } from '@varlet/use'
 import { focusChildElementByKey } from '../utils/elements'
+import { type MenuOptionProvider } from '../menu-option/provide'
+import { type Reference } from '../menu/usePopover'
 
 const { name, n, classes } = createNamespace('menu-select')
 
@@ -139,6 +140,11 @@ export default defineComponent({
       menu.value?.resize()
     }
 
+    // expose
+    function setReference(reference: Reference) {
+      menu.value?.setReference(reference)
+    }
+
     return {
       show,
       menu,
@@ -149,6 +155,7 @@ export default defineComponent({
       open,
       close,
       resize,
+      setReference,
     }
   },
 })
diff --git a/packages/varlet-ui/src/menu/Menu.vue b/packages/varlet-ui/src/menu/Menu.vue
index c0b588aafed..5bbff9daf04 100644
--- a/packages/varlet-ui/src/menu/Menu.vue
+++ b/packages/varlet-ui/src/menu/Menu.vue
@@ -1,11 +1,5 @@
 <template>
-  <div
-    ref="host"
-    :class="classes(n(), n('$--box'))"
-    @click="handleHostClick"
-    @mouseenter="handleHostMouseenter"
-    @mouseleave="handleHostMouseleave"
-  >
+  <div ref="host" :class="classes(n(), n('$--box'))">
     <slot />
 
     <Teleport :to="teleport === false ? undefined : teleport" :disabled="teleportDisabled || teleport === false">
@@ -14,7 +8,7 @@
           ref="popover"
           :style="{
             zIndex,
-            width: sameWidth ? toSizeUnit(Math.ceil(hostSize.width)) : undefined,
+            width: sameWidth ? toSizeUnit(Math.ceil(referenceSize.width)) : undefined,
           }"
           :class="
             classes(
@@ -54,12 +48,9 @@ export default defineComponent({
     const {
       popover,
       host,
-      hostSize,
+      referenceSize,
       show,
       zIndex,
-      handleHostClick,
-      handleHostMouseenter,
-      handleHostMouseleave,
       handlePopoverMouseenter,
       handlePopoverMouseleave,
       handlePopoverClose,
@@ -70,12 +61,14 @@ export default defineComponent({
       close,
       // expose
       resize,
+      // expose
+      setReference,
     } = usePopover(props)
 
     return {
       popover,
       host,
-      hostSize,
+      referenceSize,
       show,
       zIndex,
       teleportDisabled,
@@ -83,9 +76,6 @@ export default defineComponent({
       toSizeUnit,
       n,
       classes,
-      handleHostClick,
-      handleHostMouseenter,
-      handleHostMouseleave,
       handlePopoverMouseenter,
       handlePopoverMouseleave,
       handlePopoverClose,
@@ -93,6 +83,7 @@ export default defineComponent({
       resize,
       open,
       close,
+      setReference,
     }
   },
 })
diff --git a/packages/varlet-ui/src/menu/props.ts b/packages/varlet-ui/src/menu/props.ts
index f24d288bc13..507f3272911 100644
--- a/packages/varlet-ui/src/menu/props.ts
+++ b/packages/varlet-ui/src/menu/props.ts
@@ -1,5 +1,5 @@
 import { type PropType, type TeleportProps } from 'vue'
-import { type Placement, type Trigger } from './usePopover'
+import { type Placement, type Trigger, type Reference } from './usePopover'
 import { type PositioningStrategy } from '@popperjs/core'
 import { defineListenerProp } from '../utils/components'
 
@@ -10,7 +10,7 @@ export const props = {
     type: String as PropType<Trigger>,
     default: 'click',
   },
-  reference: String,
+  reference: [String, Object] as PropType<Reference>,
   placement: {
     type: String as PropType<Placement>,
     default: 'cover-top-start',
diff --git a/packages/varlet-ui/src/menu/usePopover.ts b/packages/varlet-ui/src/menu/usePopover.ts
index 583b5d07bab..7de1f328be0 100644
--- a/packages/varlet-ui/src/menu/usePopover.ts
+++ b/packages/varlet-ui/src/menu/usePopover.ts
@@ -1,8 +1,8 @@
 import flip from '@popperjs/core/lib/modifiers/flip.js'
 import offset from '@popperjs/core/lib/modifiers/offset.js'
 import computeStyles from '@popperjs/core/lib/modifiers/computeStyles.js'
-import { onWindowResize, useClickOutside, useEventListener, useVModel } from '@varlet/use'
-import { doubleRaf, getStyle, call, preventDefault } from '@varlet/shared'
+import { onWindowResize, useEventListener, useVModel } from '@varlet/use'
+import { doubleRaf, getStyle, call, preventDefault, isString } from '@varlet/shared'
 import { toPxNum } from '../utils/elements'
 import { type ListenerProp } from '../utils/components'
 import { onMounted, onUnmounted, ref, watch, type Ref } from 'vue'
@@ -20,11 +20,13 @@ export interface Position {
   distance: number
 }
 
-export interface HostSize {
+export interface ReferenceSize {
   width: number
   height: number
 }
 
+export type Reference = string | HTMLElement
+
 export type Placement =
   | NeededPopperPlacement
   | 'cover-top'
@@ -46,7 +48,7 @@ export interface UsePopoverOptions {
   disabled: boolean
   offsetX: string | number
   offsetY: string | number
-  reference?: string
+  reference?: string | HTMLElement
   closeOnClickReference?: boolean
   closeOnKeyEscape?: boolean
   onOpen?: ListenerProp<() => void>
@@ -59,7 +61,7 @@ export interface UsePopoverOptions {
 export function usePopover(options: UsePopoverOptions) {
   const host: Ref<null | HTMLElement> = ref(null)
   const popover: Ref<null | HTMLElement> = ref(null)
-  const hostSize: Ref<HostSize> = ref({ width: 0, height: 0 })
+  const referenceSize: Ref<ReferenceSize> = ref({ width: 0, height: 0 })
   const show = useVModel(options, 'show', {
     passive: true,
     defaultValue: false,
@@ -76,11 +78,11 @@ export function usePopover(options: UsePopoverOptions) {
   useStack(() => show.value, zIndex)
 
   let popoverInstance: Instance | null = null
+  let reference: string | HTMLElement
   let enterPopover = false
   let enterHost = false
 
   useEventListener(() => window, 'keydown', handleKeydown)
-  useClickOutside(getReference, 'click', handleClickOutside)
   onWindowResize(resize)
   watch(() => [options.offsetX, options.offsetY, options.placement, options.strategy], resize)
   watch(() => options.disabled, close)
@@ -93,21 +95,38 @@ export function usePopover(options: UsePopoverOptions) {
     }
   )
 
-  onMounted(() => {
-    popoverInstance = createPopper(getReference() ?? host.value!, popover.value!, getPopperOptions())
-  })
-  onUnmounted(() => {
+  onMounted(createPopperInstance)
+  onUnmounted(destroyPopperInstance)
+
+  function createPopperInstance() {
+    const reference = getReference()!
+
+    popoverInstance = createPopper(reference, popover.value!, getPopperOptions())
+    reference.addEventListener('mouseenter', handleHostMouseenter)
+    reference.addEventListener('mouseleave', handleHostMouseleave)
+    reference.addEventListener('click', handleReferenceClick)
+    document.addEventListener('click', handleClickOutside)
+  }
+
+  function destroyPopperInstance() {
+    const reference = getReference()!
+
     popoverInstance!.destroy()
-  })
+    reference.removeEventListener('mouseenter', handleHostMouseenter)
+    reference.removeEventListener('mouseleave', handleHostMouseleave)
+    reference.removeEventListener('click', handleReferenceClick)
+    document.removeEventListener('click', handleClickOutside)
+  }
 
-  function computeHostSize() {
-    if (!host.value) {
+  function computeReferenceSize() {
+    const reference = getReference()
+    if (!reference) {
       return
     }
 
-    const { width, height } = getStyle(host.value)
+    const { width, height } = getStyle(reference)
 
-    hostSize.value = {
+    referenceSize.value = {
       width: toPxNum(width),
       height: toPxNum(height),
     }
@@ -203,7 +222,7 @@ export function usePopover(options: UsePopoverOptions) {
     close()
   }
 
-  function handleHostClick() {
+  function handleReferenceClick() {
     if (options.trigger !== 'click') {
       return
     }
@@ -216,17 +235,21 @@ export function usePopover(options: UsePopoverOptions) {
     open()
   }
 
-  function handlePopoverClose() {
-    close()
-  }
-
   function handleClickOutside(e: Event) {
-    if (options.trigger !== 'click') {
-      return
+    const reference = getReference()
+
+    if (reference && !reference.contains(e.target as Node)) {
+      if (options.trigger !== 'click') {
+        return
+      }
+
+      handlePopoverClose()
+      call(options.onClickOutside, e)
     }
+  }
 
-    handlePopoverClose()
-    call(options.onClickOutside, e)
+  function handlePopoverClose() {
+    close()
   }
 
   function handleClosed() {
@@ -237,7 +260,7 @@ export function usePopover(options: UsePopoverOptions) {
   function getPosition(): Position {
     const { offsetX, offsetY, placement } = options
 
-    computeHostSize()
+    computeReferenceSize()
 
     const offset = {
       x: toPxNum(offsetX),
@@ -249,56 +272,56 @@ export function usePopover(options: UsePopoverOptions) {
         return {
           placement: 'bottom',
           skidding: offset.x,
-          distance: offset.y - hostSize.value.height,
+          distance: offset.y - referenceSize.value.height,
         }
 
       case 'cover-top-start':
         return {
           placement: 'bottom-start',
           skidding: offset.x,
-          distance: offset.y - hostSize.value.height,
+          distance: offset.y - referenceSize.value.height,
         }
 
       case 'cover-top-end':
         return {
           placement: 'bottom-end',
           skidding: offset.x,
-          distance: offset.y - hostSize.value.height,
+          distance: offset.y - referenceSize.value.height,
         }
 
       case 'cover-bottom':
         return {
           placement: 'top',
           skidding: offset.x,
-          distance: -offset.y - hostSize.value.height,
+          distance: -offset.y - referenceSize.value.height,
         }
 
       case 'cover-bottom-start':
         return {
           placement: 'top-start',
           skidding: offset.x,
-          distance: -offset.y - hostSize.value.height,
+          distance: -offset.y - referenceSize.value.height,
         }
 
       case 'cover-bottom-end':
         return {
           placement: 'top-end',
           skidding: offset.x,
-          distance: -offset.y - hostSize.value.height,
+          distance: -offset.y - referenceSize.value.height,
         }
 
       case 'cover-left':
         return {
           placement: 'right',
           skidding: offset.y,
-          distance: offset.x - hostSize.value.width,
+          distance: offset.x - referenceSize.value.width,
         }
 
       case 'cover-right':
         return {
           placement: 'left',
           skidding: offset.y,
-          distance: -offset.x - hostSize.value.width,
+          distance: -offset.x - referenceSize.value.width,
         }
 
       case 'left':
@@ -378,7 +401,19 @@ export function usePopover(options: UsePopoverOptions) {
   }
 
   function getReference() {
-    return options.reference ? host.value!.querySelector(options.reference)! : host.value!
+    const targetReference = reference ?? options.reference ?? host.value
+
+    if (isString(targetReference)) {
+      return host.value?.querySelector(targetReference)
+    }
+
+    return targetReference
+  }
+
+  function setReference(newReference: Reference) {
+    destroyPopperInstance()
+    reference = newReference
+    createPopperInstance()
   }
 
   function handleKeydown(event: KeyboardEvent) {
@@ -415,14 +450,12 @@ export function usePopover(options: UsePopoverOptions) {
     popover,
     zIndex,
     host,
-    hostSize,
-    handleHostClick,
-    handleHostMouseenter,
-    handleHostMouseleave,
+    referenceSize,
     handlePopoverClose,
     handlePopoverMouseenter,
     handlePopoverMouseleave,
     handleClosed,
+    setReference,
     resize,
     open,
     close,
diff --git a/packages/varlet-ui/src/tooltip/Tooltip.vue b/packages/varlet-ui/src/tooltip/Tooltip.vue
index 75caf0919e5..bc369d85f59 100644
--- a/packages/varlet-ui/src/tooltip/Tooltip.vue
+++ b/packages/varlet-ui/src/tooltip/Tooltip.vue
@@ -1,11 +1,5 @@
 <template>
-  <div
-    ref="host"
-    :class="classes(n(), n('$--box'))"
-    @click="handleHostClick"
-    @mouseenter="handleHostMouseenter"
-    @mouseleave="handleHostMouseleave"
-  >
+  <div ref="host" :class="classes(n(), n('$--box'))">
     <slot />
 
     <Teleport :to="teleport === false ? undefined : teleport" :disabled="teleportDisabled || teleport === false">
@@ -23,7 +17,7 @@
             :style="{
               background: color,
               color: textColor,
-              width: sameWidth ? toSizeUnit(Math.ceil(hostSize.width)) : undefined,
+              width: sameWidth ? toSizeUnit(Math.ceil(referenceSize.width)) : undefined,
             }"
             :class="classes(n('content-container'), n(`--${type}`))"
             role="tooltip"
@@ -53,12 +47,9 @@ export default defineComponent({
     const {
       popover,
       host,
-      hostSize,
+      referenceSize,
       show,
       zIndex,
-      handleHostClick,
-      handleHostMouseenter,
-      handleHostMouseleave,
       handlePopoverMouseenter,
       handlePopoverMouseleave,
       handlePopoverClose,
@@ -69,28 +60,28 @@ export default defineComponent({
       close,
       // expose
       resize,
+      // expose
+      setReference,
     } = usePopover(props)
 
     return {
       popover,
       host,
-      hostSize,
+      referenceSize,
       show,
       zIndex,
       teleportDisabled,
       toSizeUnit,
       n,
       classes,
-      handleHostClick,
       handlePopoverClose,
-      handleHostMouseenter,
-      handleHostMouseleave,
       handlePopoverMouseenter,
       handlePopoverMouseleave,
       handleClosed,
       resize,
       open,
       close,
+      setReference,
     }
   },
 })
diff --git a/packages/varlet-use/src/useEventListener.ts b/packages/varlet-use/src/useEventListener.ts
index e2933d38996..96d5c3ccc39 100644
--- a/packages/varlet-use/src/useEventListener.ts
+++ b/packages/varlet-use/src/useEventListener.ts
@@ -98,6 +98,7 @@ export function useEventListener(
   onBeforeUnmount(() => {
     remove(target)
   })
+
   onDeactivated(() => {
     remove(target)
   })