feat: add useAffixTargetProvider

chizukicn · Oct 20, 2023 · ce212a9 · ce212a9
1 parent f190456
commit ce212a9
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/packages/core/src/affix/index.ts b/packages/core/src/affix/index.ts
@@ -63,7 +63,7 @@ export const useAffix = defineHookComponent({
   setup(props, { emit }) {
     const wrapperRef = ref<HTMLElement | null>(null);
 
-    const parentRef = inject(AFFIX_TARGET_KEY, undefined);
+    const parentRef = inject(AFFIX_TARGET_KEY, null);
 
     const targetRef = useElement(props.target, parentRef);
 
@@ -164,3 +164,9 @@ export const useAffix = defineHookComponent({
 export function provideAffixTarget(target: MaybeRefOrGetter<Element | null | undefined>) {
   provide(AFFIX_TARGET_KEY, target);
 }
+
+export function useAffixTargetProvider<E extends Element = Element>() {
+  const targetRef = ref<E | null>(null);
+  provideAffixTarget(targetRef);
+  return targetRef;
+}