feat(useCurrentElement): try to infer type from $el (#4202)

KazariEX · web-flow · commit 8e141e54247f · 2024-09-12T14:07:49.000Z
diff --git a/packages/core/useCurrentElement/index.ts b/packages/core/useCurrentElement/index.ts
@@ -1,19 +1,20 @@
 // eslint-disable-next-line no-restricted-imports
 import { getCurrentInstance, onMounted, onUpdated } from 'vue-demi'
-import { computedWithControl } from '@vueuse/shared'
+import { type IsAny, computedWithControl } from '@vueuse/shared'
 import type { MaybeElement, MaybeElementRef, VueInstance } from '../unrefElement'
 import { unrefElement } from '../unrefElement'
 
 export function useCurrentElement<
   T extends MaybeElement = MaybeElement,
   R extends VueInstance = VueInstance,
+  E extends MaybeElement = MaybeElement extends T ? IsAny<R['$el']> extends false ? R['$el'] : T : T,
 >(
   rootComponent?: MaybeElementRef<R>,
 ) {
   const vm = getCurrentInstance()!
   const currentElement = computedWithControl(
     () => null,
-    () => (rootComponent ? unrefElement(rootComponent) : vm.proxy!.$el) as T,
+    () => (rootComponent ? unrefElement(rootComponent) : vm.proxy!.$el) as E,
   )
 
   onUpdated(currentElement.trigger)
