@vincentbanana/bananacli 0.3.0 → 0.4.0

package/dist/server/ui-server.js

@@ -1,6 +1,7 @@
 import { createServer } from 'node:http';
 import { randomBytes } from 'node:crypto';
 import { mkdir, readdir, readFile, stat, writeFile } from 'node:fs/promises';
+import { existsSync } from 'node:fs';
 import { extname, join, resolve, dirname } from 'node:path';
 import { spawn } from 'node:child_process';
 import { fileURLToPath } from 'node:url';
@@ -41,13 +42,30 @@ const SKIP_DIRS = new Set([
     'node_modules', '.git', 'dist', 'build', 'out', '.next', '.nuxt', 'coverage', '__pycache__',
     '.venv', 'venv', 'target', '.cache', '.idea', '.gradle',
 ]);
-/** 静态资源目录：默认是 CLI 包自带的 ui/，GUI 包启动时可以传自己的目录进来。 */
-function defaultAssetsDir() {
-    return join(dirname(fileURLToPath(import.meta.url)), '..', 'ui');
+/**
+ * 静态资源目录。同一个包在两种形态下路径不同，所以按候选顺序找第一个真实存在的：
+ * - 源码运行：src/server/ -> src/ui
+ * - 编译产物：dist/server/ -> dist/ui（构建时由 scripts/copy-ui.mjs 复制）
+ * - 兜底：包根 ui/ 或环境变量指定的目录（GUI 独立包用这个）
+ */
+export function resolveAssetsDir() {
+    const here = dirname(fileURLToPath(import.meta.url));
+    const candidates = [
+        process.env.BANANA_UI_DIR,
+        join(here, '..', 'ui'),
+        join(here, '..', '..', 'ui'),
+        join(here, '..', '..', 'src', 'ui'),
+    ].filter((item) => Boolean(item));
+    for (const candidate of candidates) {
+        if (existsSync(join(candidate, 'index.html')))
+            return candidate;
+    }
+    // 都找不到时返回默认位置，让 404 报错能指向一个具体路径
+    return join(here, '..', 'ui');
 }
 let activeAssetsDir = null;
 function assetRoot() {
-    return activeAssetsDir ?? defaultAssetsDir();
+    return activeAssetsDir ?? resolveAssetsDir();
 }
 /** 启动只监听回环地址的本地 GUI 服务；token 保护所有接口。 */
 export async function startUiServer(options) {
@@ -55,7 +73,7 @@ export async function startUiServer(options) {
     const token = randomBytes(24).toString('hex');
     const host = options.host ?? '127.0.0.1';
     const prefix = `/t/${token}`;
-    const assets = options.assetsDir ?? defaultAssetsDir();
+    const assets = options.assetsDir ?? resolveAssetsDir();
     activeAssetsDir = assets;
     let turn = null;
     const provider = () => createProvider(config.current, config);