From f0a5931b6e41d1ee9092bba9c9840234d65328b9 Mon Sep 17 00:00:00 2001
From: neverland <chenjiahan@youzan.com>
Date: Mon, 12 Jul 2021 20:40:30 +0800
Subject: [PATCH] types(@vant/use): useRelation support InjectionKey (#9015)

---
 packages/vant-use/src/useRelation/useChildren.ts | 9 ++++++---
 packages/vant-use/src/useRelation/useParent.ts   | 5 +++--
 2 files changed, 9 insertions(+), 5 deletions(-)

diff --git a/packages/vant-use/src/useRelation/useChildren.ts b/packages/vant-use/src/useRelation/useChildren.ts
index d9afd36c4..3ef15d4a9 100644
--- a/packages/vant-use/src/useRelation/useChildren.ts
+++ b/packages/vant-use/src/useRelation/useChildren.ts
@@ -3,6 +3,7 @@ import {
   isVNode,
   provide,
   reactive,
+  InjectionKey,
   getCurrentInstance,
   VNodeNormalizedChildren,
   ComponentPublicInstance,
@@ -57,13 +58,15 @@ export function sortChildren(
 }
 
 export function useChildren<
-  Child extends ComponentPublicInstance = ComponentPublicInstance
->(key: string | symbol) {
+  // eslint-disable-next-line
+  Child extends ComponentPublicInstance = ComponentPublicInstance<{}, any>,
+  ProvideValue = never
+>(key: InjectionKey<ProvideValue>) {
   const publicChildren: Child[] = reactive([]);
   const internalChildren: ComponentInternalInstance[] = reactive([]);
   const parent = getCurrentInstance()!;
 
-  const linkChildren = (value?: any) => {
+  const linkChildren = (value?: ProvideValue) => {
     const link = (child: ComponentInternalInstance) => {
       if (child.proxy) {
         internalChildren.push(child);
diff --git a/packages/vant-use/src/useRelation/useParent.ts b/packages/vant-use/src/useRelation/useParent.ts
index e439926f3..4cf77317f 100644
--- a/packages/vant-use/src/useRelation/useParent.ts
+++ b/packages/vant-use/src/useRelation/useParent.ts
@@ -3,6 +3,7 @@ import {
   inject,
   computed,
   onUnmounted,
+  InjectionKey,
   getCurrentInstance,
   ComponentPublicInstance,
   ComponentInternalInstance,
@@ -15,8 +16,8 @@ type ParentProvide<T> = T & {
   internalChildren: ComponentInternalInstance[];
 };
 
-export function useParent<T>(key: string | symbol) {
-  const parent = inject<ParentProvide<T> | null>(key, null);
+export function useParent<T>(key: InjectionKey<ParentProvide<T>>) {
+  const parent = inject(key, null);
 
   if (parent) {
     const instance = getCurrentInstance()!;