types: fix + test inject API typing

ref: #2052

diff --git a/packages/runtime-core/src/apiInject.ts b/packages/runtime-core/src/apiInject.ts
index 02dc69e9d298d8e2fabce896131d233ca130a4cb..c12b6490849e7a262eae74afc3cb88dcc6ef2533 100644 (file)
--- a/packages/runtime-core/src/apiInject.ts
+++ b/packages/runtime-core/src/apiInject.ts
@@ -31,7 +31,12 @@ export function inject<T>(key: InjectionKey<T> | string): T | undefined
 export function inject<T>(
   key: InjectionKey<T> | string,
   defaultValue: T,
-  treatDefaultAsFactory?: boolean
+  treatDefaultAsFactory?: false
+): T
+export function inject<T>(
+  key: InjectionKey<T> | string,
+  defaultValue: T | (() => T),
+  treatDefaultAsFactory: true
 ): T
 export function inject(
   key: InjectionKey<any> | string,

diff --git a/test-dts/inject.test-d.ts b/test-dts/inject.test-d.ts
new file mode 100644 (file)
index 0000000..69e06f1
--- /dev/null
+++ b/test-dts/inject.test-d.ts
@@ -0,0 +1,15 @@
+import { provide, inject, InjectionKey, expectType } from './index'
+
+const key: InjectionKey<number> = Symbol()
+
+provide(key, 1)
+// @ts-expect-error
+provide(key, 'foo')
+
+expectType<number | undefined>(inject(key))
+expectType<number>(inject(key, 1))
+expectType<number>(inject(key, () => 1, true /* treatDefaultAsFactory */))
+
+expectType<() => number>(inject('foo', () => 1))
+expectType<() => number>(inject('foo', () => 1, false))
+expectType<number>(inject('foo', () => 1, true))