From a802b87ac76d8e0a565d5355b332a4bcec264eab Mon Sep 17 00:00:00 2001
From: Anthony Fu <anthonyfu117@hotmail.com>
Date: Thu, 9 Jun 2022 04:09:31 +0800
Subject: [PATCH] feat(nuxt): support auto imports within `<template>` (#4904)

---
 packages/nuxt/src/auto-imports/module.ts    | 5 ++++-
 packages/nuxt/src/auto-imports/transform.ts | 2 +-
 test/basic.test.ts                          | 1 +
 test/fixtures/basic/composables/template.ts | 1 +
 test/fixtures/basic/pages/index.vue         | 1 +
 5 files changed, 8 insertions(+), 2 deletions(-)
 create mode 100644 test/fixtures/basic/composables/template.ts

diff --git a/packages/nuxt/src/auto-imports/module.ts b/packages/nuxt/src/auto-imports/module.ts
index 6f5b4208ab..6e4e12c4f6 100644
--- a/packages/nuxt/src/auto-imports/module.ts
+++ b/packages/nuxt/src/auto-imports/module.ts
@@ -36,7 +36,10 @@ export default defineNuxtModule<Partial<AutoImportsOptions>>({
     // Create a context to share state between module internals
     const ctx = createUnimport({
       presets: options.presets,
-      imports: options.imports
+      imports: options.imports,
+      addons: {
+        vueTemplate: true
+      }
     })
 
     // composables/ dirs from all layers
diff --git a/packages/nuxt/src/auto-imports/transform.ts b/packages/nuxt/src/auto-imports/transform.ts
index a184cef0dc..fd5f82a56b 100644
--- a/packages/nuxt/src/auto-imports/transform.ts
+++ b/packages/nuxt/src/auto-imports/transform.ts
@@ -39,7 +39,7 @@ export const TransformPlugin = createUnplugin(({ ctx, options, sourcemap }: {ctx
       }
     },
     async transform (_code, id) {
-      const { code, s } = await ctx.injectImports(_code)
+      const { code, s } = await ctx.injectImports(_code, id)
       if (code === _code) {
         return
       }
diff --git a/test/basic.test.ts b/test/basic.test.ts
index 2379b0baaa..929d250163 100644
--- a/test/basic.test.ts
+++ b/test/basic.test.ts
@@ -41,6 +41,7 @@ describe('pages', () => {
     // composables auto import
     expect(html).toContain('Composable | foo: auto imported from ~/components/foo.ts')
     expect(html).toContain('Composable | bar: auto imported from ~/components/useBar.ts')
+    expect(html).toContain('Composable | template: auto imported from ~/components/template.ts')
     // should import components
     expect(html).toContain('This is a custom component with a named export.')
 
diff --git a/test/fixtures/basic/composables/template.ts b/test/fixtures/basic/composables/template.ts
new file mode 100644
index 0000000000..dc367ffe8e
--- /dev/null
+++ b/test/fixtures/basic/composables/template.ts
@@ -0,0 +1 @@
+export const templateAutoImport = 'auto imported from ~/components/template.ts'
diff --git a/test/fixtures/basic/pages/index.vue b/test/fixtures/basic/pages/index.vue
index 670a5635ca..133f790600 100644
--- a/test/fixtures/basic/pages/index.vue
+++ b/test/fixtures/basic/pages/index.vue
@@ -7,6 +7,7 @@
     <div>RuntimeConfig | testConfig: {{ config.testConfig }}</div>
     <div>Composable | foo: {{ foo }}</div>
     <div>Composable | bar: {{ bar }}</div>
+    <div>Composable | template: {{ templateAutoImport }}</div>
     <div>Path: {{ $route.fullPath }}</div>
     <NuxtLink to="/">
       Link