feat: cut over rust web main shell

scripts/desktop-hot.js

@@ -1,6 +1,6 @@
-#!/usr/bin/env node
-
-/**
+#!/usr/bin/env node
+
+/**
  * 同时热启动前端、FastAPI 与 Celery。
  * 可使用以下环境变量调整行为：
  * - FRONTEND_CMD：覆盖前端启动命令，默认为 "pnpm dev"
@@ -11,14 +11,17 @@
  * - CELERY_BIN：只在 CELERY_CMD 未覆盖时，设置 Celery 可执行文件，默认 "celery"
  * - REDIS_URL：仅用于探测 Redis 是否就绪，默认 "redis://localhost:6379/0"
  * - SKIP_CELERY：设为 "1" or "true" 可跳过 Celery。
+ * - MNOTE_WEB_SKIP_GATEWAY：设为 "1" or "true" 临时恢复旧 Next 3000 入口。
+ * - NEXT_LEGACY_PORT：Rust gateway 模式下 Next legacy upstream 端口，默认 3100。
+ * - SKIP_NEXT_LEGACY：设为 "1" or "true" 时只启动 Rust gateway，不启动 Next legacy upstream。
  */
-
+
 const { spawn, execSync } = require("child_process");
 const path = require("path");
 const net = require("net");
 const { URL } = require("url");
 const fs = require("fs");
-
+
 const rootDir = path.resolve(__dirname, "..");
 const frontendDir = path.join(rootDir, "wolai-frontend");
 const backendDir = path.join(rootDir, "wolai-backend");
@@ -59,22 +62,85 @@ const celeryCmdFromEnv = process.env.CELERY_CMD;
 const redisUrl = process.env.REDIS_URL || "redis://localhost:6379/0";
 const frontendPortFromEnv = Number(process.env.FRONTEND_PORT || 3000);
 const backendPortFromEnv = Number(process.env.BACKEND_PORT || 8000);
+const nextLegacyPortFromEnv = Number(process.env.NEXT_LEGACY_PORT || 3100);
+
+function isEnabledEnv(value) {
+  const normalized = String(value || "").toLowerCase();
+  return normalized === "1" || normalized === "true";
+}
+
+function resolveRuntimePlan(env = process.env) {
+  const frontendPort = Number(env.FRONTEND_PORT || 3000);
+  const nextLegacyPort = Number(env.NEXT_LEGACY_PORT || 3100);
+  const skipGateway = isEnabledEnv(env.MNOTE_WEB_SKIP_GATEWAY);
+  const skipNextLegacy = !skipGateway && isEnabledEnv(env.SKIP_NEXT_LEGACY);
+  const publicPort = Number.isFinite(frontendPort) ? Math.floor(frontendPort) : 3000;
+  const legacyPort = Number.isFinite(nextLegacyPort) ? Math.floor(nextLegacyPort) : 3100;
+  const legacyUrl = `http://127.0.0.1:${legacyPort}`;
+  const legacyCompatEnabled = skipNextLegacy ? "0" : env.MNOTE_WEB_ENABLE_LEGACY_NEXT_COMPAT || "1";
+
+  return {
+    skipGateway,
+    skipNextLegacy,
+    publicPort,
+    legacyPort,
+    publicUrl: `http://localhost:${publicPort}`,
+    legacyUrl,
+    frontendTaskName: skipGateway ? "frontend" : skipNextLegacy ? null : "next-legacy",
+    frontendCommand: env.FRONTEND_CMD || `pnpm dev -p ${skipGateway ? publicPort : legacyPort}`,
+    mnoteWebCommand: env.MNOTE_WEB_CMD || "cargo run -p mnote-web --bin mnote-web",
+    mnoteWebEnv: skipGateway
+      ? {}
+      : {
+          MNOTE_WEB_BIND: env.MNOTE_WEB_BIND || `127.0.0.1:${publicPort}`,
+          MNOTE_WEB_PUBLIC_BIND: env.MNOTE_WEB_PUBLIC_BIND || `127.0.0.1:${publicPort}`,
+          ...(skipNextLegacy
+            ? {}
+            : { MNOTE_WEB_LEGACY_NEXT_BASE_URL: env.MNOTE_WEB_LEGACY_NEXT_BASE_URL || legacyUrl }),
+          MNOTE_WEB_ENABLE_LEGACY_NEXT_COMPAT: legacyCompatEnabled,
+        },
+  };
+}
+
+const runtimePlan = resolveRuntimePlan(process.env);
+const skipMnoteWebGateway = runtimePlan.skipGateway;
+
+const frontendTasks = runtimePlan.frontendTaskName
+  ? [
+      {
+        name: runtimePlan.frontendTaskName,
+        command: runtimePlan.frontendCommand,
+        cwd: frontendDir,
+      },
+    ]
+  : [];
 
 const tasks = [
-  {
-    name: "frontend",
-    command: process.env.FRONTEND_CMD || "pnpm dev",
-    cwd: frontendDir,
-  },
+  ...frontendTasks,
+  ...(skipMnoteWebGateway
+    ? []
+    : [
+        {
+          name: "mnote-web",
+          command:
+            process.env.MNOTE_WEB_CMD ||
+            runtimePlan.mnoteWebCommand,
+          cwd: path.join(rootDir, "rust"),
+        },
+      ]),
   {
     name: "backend",
-    command:
-      process.env.BACKEND_CMD ||
-      `${pythonBin} -m uvicorn app.main:app --reload --port 8000`,
-    cwd: backendDir,
-  },
-];
-
+    command:
+      process.env.BACKEND_CMD ||
+      `${pythonBin} -m uvicorn app.main:app --reload --port 8000`,
+    cwd: backendDir,
+  },
+];
+
+function findTask(name) {
+  return tasks.find((task) => task.name === name);
+}
+
 const children = [];
 let shuttingDown = false;
 
@@ -288,16 +354,16 @@ function loadEnvFile(filePath) {
   if (!fs.existsSync(filePath)) return {};
   const content = fs.readFileSync(filePath, "utf8");
   return content
-    .split(/\r?\n/)
-    .filter((line) => line.trim() && !line.trim().startsWith("#"))
-    .reduce((acc, line) => {
-      const idx = line.indexOf("=");
-      if (idx === -1) return acc;
-      const key = line.slice(0, idx).trim();
-      const value = line.slice(idx + 1).trim();
-      acc[key] = value;
-      return acc;
-    }, {});
+    .split(/\r?\n/)
+    .filter((line) => line.trim() && !line.trim().startsWith("#"))
+    .reduce((acc, line) => {
+      const idx = line.indexOf("=");
+      if (idx === -1) return acc;
+      const key = line.slice(0, idx).trim();
+      const value = line.slice(idx + 1).trim();
+      acc[key] = value;
+      return acc;
+    }, {});
 }
 
 // 约定：全局仅使用仓库根目录的 .env.all 作为配置来源（生产优先）。
@@ -311,7 +377,7 @@ const mergedEnv = {
   ...process.env,
   ...envFromAll,
 };
-
+
 function logPrefix(name, message) {
   console.log(`[${name}] ${message}`);
 }
@@ -330,31 +396,31 @@ function buildDefaultCeleryCommand() {
 
 function startTask(task) {
   logPrefix(task.name, `启动命令：${task.command}`);
-  const child = spawn(task.command, {
-    cwd: task.cwd,
-    stdio: "inherit",
-    shell: true,
-    env: mergedEnv,
-  });
-
-  child.on("exit", (code, signal) => {
-    if (shuttingDown) {
-      return;
-    }
-    const status =
-      signal !== null ? `因信号 ${signal} 退出` : `退出码 ${code ?? "null"}`;
-    logPrefix(task.name, `进程结束（${status}），准备清理其它任务。`);
-    shutdown(code ?? 0);
-  });
-
-  child.on("error", (err) => {
-    logPrefix(task.name, `启动失败：${err.message}`);
-    shutdown(1);
-  });
-
-  children.push(child);
-}
-
+  const child = spawn(task.command, {
+    cwd: task.cwd,
+    stdio: "inherit",
+    shell: true,
+    env: mergedEnv,
+  });
+
+  child.on("exit", (code, signal) => {
+    if (shuttingDown) {
+      return;
+    }
+    const status =
+      signal !== null ? `因信号 ${signal} 退出` : `退出码 ${code ?? "null"}`;
+    logPrefix(task.name, `进程结束（${status}），准备清理其它任务。`);
+    shutdown(code ?? 0);
+  });
+
+  child.on("error", (err) => {
+    logPrefix(task.name, `启动失败：${err.message}`);
+    shutdown(1);
+  });
+
+  children.push(child);
+}
+
 function shutdown(code) {
   if (shuttingDown) {
     return;
@@ -388,40 +454,40 @@ function shutdown(code) {
 
   setTimeout(() => process.exit(code), 200);
 }
-
-process.on("SIGINT", () => shutdown(0));
-process.on("SIGTERM", () => shutdown(0));
-
-async function checkRedisReachable(urlString, timeoutMs = 2000) {
-  try {
-    const url = new URL(urlString);
-    const host = url.hostname || "localhost";
-    const port = Number(url.port) || 6379;
-
-    return await new Promise((resolve) => {
-      const socket = net.createConnection({ host, port });
-      const timer = setTimeout(() => {
-        socket.destroy();
-        resolve(false);
-      }, timeoutMs);
-
-      socket.once("connect", () => {
-        clearTimeout(timer);
-        socket.end();
-        resolve(true);
-      });
-
-      socket.once("error", () => {
-        clearTimeout(timer);
-        resolve(false);
-      });
-    });
-  } catch (error) {
-    logPrefix("celery", `REDIS_URL (${urlString}) 解析失败：${error.message}，跳过连通性检查。`);
-    return true;
-  }
-}
-
+
+process.on("SIGINT", () => shutdown(0));
+process.on("SIGTERM", () => shutdown(0));
+
+async function checkRedisReachable(urlString, timeoutMs = 2000) {
+  try {
+    const url = new URL(urlString);
+    const host = url.hostname || "localhost";
+    const port = Number(url.port) || 6379;
+
+    return await new Promise((resolve) => {
+      const socket = net.createConnection({ host, port });
+      const timer = setTimeout(() => {
+        socket.destroy();
+        resolve(false);
+      }, timeoutMs);
+
+      socket.once("connect", () => {
+        clearTimeout(timer);
+        socket.end();
+        resolve(true);
+      });
+
+      socket.once("error", () => {
+        clearTimeout(timer);
+        resolve(false);
+      });
+    });
+  } catch (error) {
+    logPrefix("celery", `REDIS_URL (${urlString}) 解析失败：${error.message}，跳过连通性检查。`);
+    return true;
+  }
+}
+
 async function main() {
   // 说明：Next dev 在异常退出时可能残留 `.next/dev/lock`，会导致后续启动直接失败。
   // 这里在启动前做一次“安全清理”，避免重启后仍卡住。
@@ -437,8 +503,9 @@ async function main() {
 
   // 说明：你外网绑定了 3000 端口，这里默认强制使用 3000。
   // 如果检测到 3000 被占用，则自动结束旧进程后重启，以保证始终跑在 3000。
-  const desiredFrontendPort = frontendPortFromEnv;
-  const frontendPortOk = await ensurePortFree(desiredFrontendPort, "frontend");
+  const desiredFrontendPort = runtimePlan.publicPort;
+  const frontendOwnerName = skipMnoteWebGateway ? "frontend" : "mnote-web";
+  const frontendPortOk = await ensurePortFree(desiredFrontendPort, frontendOwnerName);
   if (!frontendPortOk) {
     console.error(`前端端口 ${desiredFrontendPort} 无法释放，已中止启动。`);
     process.exit(1);
@@ -446,11 +513,36 @@ async function main() {
   const frontendPort = desiredFrontendPort;
   const frontendUrl = `http://localhost:${frontendPort}`;
 
+  let nextLegacyPort = null;
+  if (!skipMnoteWebGateway) {
+    nextLegacyPort = runtimePlan.legacyPort;
+    const nextLegacyPortOk = await ensurePortFree(nextLegacyPort, "next-legacy");
+    if (!nextLegacyPortOk) {
+      console.error(`Next legacy 端口 ${nextLegacyPort} 无法释放，已中止启动。`);
+      process.exit(1);
+    }
+  }
+
   // 说明：在 Windows 的 cmd.exe 下，`pnpm dev -- -p 3000` 会把 `--` 原样传给 next，导致 next 把 `-p` 误当成目录。
   // 用 `pnpm dev -p 3000` 在 PowerShell/cmd.exe 下都能正确传参。
-  tasks[0].command = process.env.FRONTEND_CMD || `pnpm dev -p ${frontendPort}`;
-  logPrefix("frontend", `前端目录：${frontendDir}`);
-  logPrefix("frontend", `前端地址：${frontendUrl}`);
+  const frontendTask = skipMnoteWebGateway ? findTask("frontend") : findTask("next-legacy");
+  if (!frontendTask) {
+    throw new Error("缺少前端任务配置");
+  }
+  frontendTask.command = runtimePlan.frontendCommand;
+  logPrefix(frontendTask.name, `前端目录：${frontendDir}`);
+  if (skipMnoteWebGateway) {
+    logPrefix("frontend", `前端地址：${frontendUrl}`);
+  } else {
+    const legacyUrl = runtimePlan.legacyUrl;
+    logPrefix("mnote-web", `Rust gateway 公开入口：${frontendUrl}`);
+    logPrefix("next-legacy", `Next legacy upstream：${legacyUrl}`);
+    const gatewayTask = tasks.find((task) => task.name === "mnote-web");
+    if (gatewayTask) {
+      gatewayTask.command = runtimePlan.mnoteWebCommand;
+    }
+    Object.assign(mergedEnv, runtimePlan.mnoteWebEnv);
+  }
 
   const desiredBackendPort = backendPortFromEnv;
   if (!process.env.BACKEND_CMD) {
@@ -459,7 +551,11 @@ async function main() {
       console.error(`后端端口 ${desiredBackendPort} 无法释放，已中止启动。`);
       process.exit(1);
     }
-    tasks[1].command = `${pythonBin} -m uvicorn app.main:app --reload --port ${desiredBackendPort}`;
+    const backendTask = findTask("backend");
+    if (!backendTask) {
+      throw new Error("缺少后端任务配置");
+    }
+    backendTask.command = `${pythonBin} -m uvicorn app.main:app --reload --port ${desiredBackendPort}`;
   }
 
   if (!skipCelery) {
@@ -480,23 +576,23 @@ async function main() {
       tasks.push(celeryTask);
     } else {
       // Redis 未就绪时直接跳过 Celery，避免热调试流程整体退出。
-      logPrefix(
-        "celery",
-        `检测到 ${redisUrl} 无法连接，自动跳过 Celery。请先启动 Redis 或设置 SKIP_CELERY=1 显式跳过。`,
-      );
-    }
-  }
-
-  if (tasks.length === 0) {
-    console.error("未配置任何可运行的任务，检查环境变量设置。");
-    process.exit(1);
-  }
-
-  for (const task of tasks) {
-    startTask(task);
-  }
-}
-
+      logPrefix(
+        "celery",
+        `检测到 ${redisUrl} 无法连接，自动跳过 Celery。请先启动 Redis 或设置 SKIP_CELERY=1 显式跳过。`,
+      );
+    }
+  }
+
+  if (tasks.length === 0) {
+    console.error("未配置任何可运行的任务，检查环境变量设置。");
+    process.exit(1);
+  }
+
+  for (const task of tasks) {
+    startTask(task);
+  }
+}
+
 if (require.main === module) {
   main().catch((error) => {
     logPrefix("system", `启动失败：${error.message}`);
@@ -509,6 +605,7 @@ module.exports = {
   getListeningPidsByPort,
   getProcessNameByPid,
   isPortFree,
+  resolveRuntimePlan,
   resolveBackendExecutable,
   terminatePid,
 };