From 37d17f9f657c3db91a1b4839d21a330c2787066f Mon Sep 17 00:00:00 2001
From: xiaohuoni <xiaohuoni@dingtalk.com>
Date: Mon, 30 Dec 2024 10:19:15 +0800
Subject: [PATCH] fix: frameworkName when vite dev

---
 .../preset-umi/src/commands/dev/plugins/ViteHtmlPlugin.ts     | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/packages/preset-umi/src/commands/dev/plugins/ViteHtmlPlugin.ts b/packages/preset-umi/src/commands/dev/plugins/ViteHtmlPlugin.ts
index 3561d51f0888..7f236f6ac9b1 100644
--- a/packages/preset-umi/src/commands/dev/plugins/ViteHtmlPlugin.ts
+++ b/packages/preset-umi/src/commands/dev/plugins/ViteHtmlPlugin.ts
@@ -21,7 +21,9 @@ export default function ViteHtmlPlugin(api: IApi): Plugin {
             try {
               // 处理通用html
               const viteScripts: IOpts['scripts'] = [
-                api.appData.hasSrcDir ? '/src/.umi/umi.ts' : '/.umi/umi.ts',
+                `${api.appData.hasSrcDir ? '/src' : ''}/.${
+                  api.service.frameworkName
+                }/umi.ts`,
               ];
 
               const markupArgs = (await getMarkupArgs({ api })) as any;