From b29c3c6fdd37da2b2016161f5e54cccb6526b9b7 Mon Sep 17 00:00:00 2001
From: Aaron Klinker <aklinker1@users.noreply.github.com>
Date: Mon, 10 Jul 2023 23:19:13 -0500
Subject: [PATCH] fix: Only add imports to JS files

---
 src/core/vite-plugins/unimport.ts | 13 ++++++++++++-
 1 file changed, 12 insertions(+), 1 deletion(-)

diff --git a/src/core/vite-plugins/unimport.ts b/src/core/vite-plugins/unimport.ts
index 16818ac3..9b8c2818 100644
--- a/src/core/vite-plugins/unimport.ts
+++ b/src/core/vite-plugins/unimport.ts
@@ -2,6 +2,16 @@ import { createUnimport } from 'unimport';
 import { InternalConfig } from '../types';
 import { getUnimportOptions } from '../utils/auto-imports';
 import { Plugin } from 'vite';
+import { extname } from 'path';
+
+const ENABLED_EXTENSIONS: Record<string, boolean | undefined> = {
+  '.js': true,
+  '.jsx': true,
+  '.ts': true,
+  '.tsx': true,
+  '.vue': true,
+  '.svelte': true,
+};
 
 /**
  * Inject any global imports defined by unimport
@@ -16,7 +26,8 @@ export function unimport(config: InternalConfig): Plugin {
       await unimport.scanImportsFromDir(undefined, { cwd: config.srcDir });
     },
     async transform(code, id) {
-      return unimport.injectImports(code, id);
+      const ext = extname(id);
+      if (ENABLED_EXTENSIONS[ext]) return unimport.injectImports(code, id);
     },
   };
 }