Initial commit: OneOS frontend based on yudao-ui-admin-vben

2026-03-11 22:18:23 +08:00
commit 2650d1cdf0
5166 changed files with 641242 additions and 0 deletions
--- a/internal/vite-config/src/plugins/inject-app-loading/index.ts
+++ b/internal/vite-config/src/plugins/inject-app-loading/index.ts
@@ -0,0 +1,66 @@
+import type { PluginOption } from 'vite';
+
+import fs from 'node:fs';
+import fsp from 'node:fs/promises';
+import { join } from 'node:path';
+import { fileURLToPath } from 'node:url';
+
+import { readPackageJSON } from '@vben/node-utils';
+
+/**
+ * 用于生成将loading样式注入到项目中
+ * 为多app提供loading样式，无需在每个 app -> index.html单独引入
+ */
+async function viteInjectAppLoadingPlugin(
+  isBuild: boolean,
+  env: Record<string, any> = {},
+  loadingTemplate = 'loading.html',
+): Promise<PluginOption | undefined> {
+  const loadingHtml = await getLoadingRawByHtmlTemplate(loadingTemplate);
+  const { version } = await readPackageJSON(process.cwd());
+  const envRaw = isBuild ? 'prod' : 'dev';
+  const cacheName = `'${env.VITE_APP_NAMESPACE}-${version}-${envRaw}-preferences-theme'`;
+
+  // 获取缓存的主题
+  // 保证黑暗主题下，刷新页面时，loading也是黑暗主题
+  const injectScript = `
+  <script data-app-loading="inject-js">
+  var theme = localStorage.getItem(${cacheName});
+  document.documentElement.classList.toggle('dark', /dark/.test(theme));
+</script>
+`;
+
+  if (!loadingHtml) {
+    return;
+  }
+
+  return {
+    enforce: 'pre',
+    name: 'vite:inject-app-loading',
+    transformIndexHtml: {
+      handler(html) {
+        const re = /<body\s*>/;
+        html = html.replace(re, `<body>${injectScript}${loadingHtml}`);
+        return html;
+      },
+      order: 'pre',
+    },
+  };
+}
+
+/**
+ * 用于获取loading的html模板
+ */
+async function getLoadingRawByHtmlTemplate(loadingTemplate: string) {
+  // 支持在app内自定义loading模板，模版参考default-loading.html即可
+  let appLoadingPath = join(process.cwd(), loadingTemplate);
+
+  if (!fs.existsSync(appLoadingPath)) {
+    const __dirname = fileURLToPath(new URL('.', import.meta.url));
+    appLoadingPath = join(__dirname, './default-loading.html');
+  }
+
+  return await fsp.readFile(appLoadingPath, 'utf8');
+}
+
+export { viteInjectAppLoadingPlugin };