feat(runtime-core): add app.inject() API

LinusBorg · LinusBorg · commit 7177cf3ba8a5 · 2021-10-15T15:43:44.000+02:00
diff --git a/packages/runtime-core/__tests__/apiCreateApp.spec.ts b/packages/runtime-core/__tests__/apiCreateApp.spec.ts
@@ -93,6 +93,14 @@ describe('api: createApp', () => {
     app.provide('foo', 1)
     app.provide('bar', 2)
 
+    const foo = app.inject('foo')
+    const bar = app.inject('bar')
+    app.inject('non-existant')
+
+    expect('[Vue warn]: injection "non-existant" not found.').toHaveBeenWarned()
+    expect(foo).toBe(1)
+    expect(bar).toBe(2)
+
     const root = nodeOps.createElement('div')
     app.mount(root)
     expect(serializeInner(root)).toBe(`3,2`)
diff --git a/packages/runtime-core/src/apiCreateApp.ts b/packages/runtime-core/src/apiCreateApp.ts
@@ -41,6 +41,7 @@ export interface App<HostElement = any> {
   ): ComponentPublicInstance
   unmount(): void
   provide<T>(key: InjectionKey<T> | string, value: T): this
+  inject<T>(key: InjectionKey<T> | string): T | undefined
 
   // internal, but we need to expose these for the server-renderer and devtools
   _uid: number
@@ -346,6 +347,14 @@ export function createAppAPI<HostElement>(
         context.provides[key as string] = value
 
         return app
+      },
+
+      inject<T>(key: InjectionKey<T> | string) {
+        const value = context.provides[key as string]
+        if (__DEV__ && !((key as string) in context.provides)) {
+          warn(`injection "${String(key)}" not found.`)
+        }
+        return value
       }
     })
 
