feat: support specifying allowed keys via generic argument in useTemplateRef()

diff --git a/packages/runtime-core/src/helpers/useTemplateRef.ts b/packages/runtime-core/src/helpers/useTemplateRef.ts
index 69e5669b6f6c6b6356319d11a4a9ca827e369622..12663962a8e4c340c6a512386e234ca1e4239aa7 100644 (file)
--- a/packages/runtime-core/src/helpers/useTemplateRef.ts
+++ b/packages/runtime-core/src/helpers/useTemplateRef.ts
@@ -3,8 +3,8 @@ import { getCurrentInstance } from '../component'
 import { warn } from '../warning'
 import { EMPTY_OBJ } from '@vue/shared'
 
-export function useTemplateRef<T = unknown>(
-  key: string,
+export function useTemplateRef<T = unknown, Keys extends string = string>(
+  key: Keys,
 ): Readonly<ShallowRef<T | null>> {
   const i = getCurrentInstance()
   const r = shallowRef(null)