codex-endpoint-switcher 1.4.0 → 1.5.0

package/README.md

@@ -50,6 +50,7 @@ codex-switcher status
 codex-switcher restart
 codex-switcher stop
 codex-switcher check-update
+codex-switcher self-update
 codex-switcher sync-server
 codex-switcher install-access
 codex-switcher remove-access
@@ -63,6 +64,7 @@ codex-switcher remove-access
 - `restart`：重启本地网页服务
 - `stop`：关闭本地网页服务
 - `check-update`：检查 npm 是否有新版本
+- `self-update`：自动更新 npm 包并自动重启控制台
 - `sync-server`：启动账号同步服务，服务端使用 SQLite 单文件数据库
 - `install-access`：创建桌面快捷方式和开机启动项
 - `remove-access`：移除桌面快捷方式和开机启动项
@@ -140,7 +142,8 @@ codex-switcher sync-server
 
 - 控制台打开后会自动检查 npm 最新版本
 - 如果你发布了新版本，其他用户打开控制台后会看到“发现新版本”的提示
-- 用户可以直接复制升级命令执行
+- 用户可以直接点 `立即更新`，程序会自动关闭、更新并重新打开
+- 也可以继续使用“复制升级命令”手动更新
 
 也可以用 CLI 手动检查：
 
@@ -148,6 +151,12 @@ codex-switcher sync-server
 codex-switcher check-update
 ```
 
+或者直接自动更新：
+
+```bash
+codex-switcher self-update
+```
+
 ## 更新 npm 包
 
 先进入项目目录：

package/bin/codex-switcher.js

@@ -19,6 +19,7 @@ function printUsage() {
   codex-switcher restart
   codex-switcher stop
   codex-switcher check-update
+  codex-switcher self-update
   codex-switcher sync-server
   codex-switcher install-access
   codex-switcher remove-access
@@ -30,6 +31,7 @@ function printUsage() {
   restart         重启本地网页服务
   stop            关闭本地网页服务
   check-update    检查 npm 最新版本
+  self-update     自动更新到最新版本并重启控制台
   sync-server     启动账号同步服务（SQLite）
   install-access  创建桌面快捷方式与开机启动项
   remove-access   删除桌面快捷方式与开机启动项
@@ -110,6 +112,25 @@ async function main() {
       console.log(`当前已是最新版：${status.currentVersion}`);
       return;
     }
+    case "self-update": {
+      const status = await updateService.getUpdateStatus({ force: true });
+      if (!status.hasUpdate) {
+        if (status.lastError) {
+          console.log(`当前版本：${status.currentVersion}`);
+          console.log(`暂时无法检查更新：${status.lastError}`);
+          return;
+        }
+
+        console.log(`当前已是最新版：${status.currentVersion}`);
+        return;
+      }
+
+      const result = updateService.scheduleAutoUpdate({ reopen: true });
+      console.log(`已开始自动更新到 ${status.latestVersion}`);
+      console.log(`日志文件：${result.logPath}`);
+      await handleCommand("stop");
+      return;
+    }
     case "sync-server": {
       await runNodeScript(cloudSyncServerPath, ["--experimental-sqlite"]);
       return;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "codex-endpoint-switcher",
-  "version": "1.4.0",
+  "version": "1.5.0",
   "description": "用于切换 Codex URL 和 Key 的本地网页控制台与 npm CLI",
   "main": "src/main/main.js",
   "bin": {

package/src/main/main.js

@@ -106,6 +106,14 @@ function registerHandlers() {
   ipcMain.handle("app:updateStatus", async (_event, payload) => {
     return updateService.getUpdateStatus(payload || {});
   });
+
+  ipcMain.handle("app:autoUpdate", async (_event, payload) => {
+    const result = updateService.scheduleAutoUpdate(payload || {});
+    setTimeout(() => {
+      app.quit();
+    }, 120);
+    return result;
+  });
 }
 
 app.whenReady().then(() => {

package/src/main/preload.js

@@ -58,4 +58,7 @@ contextBridge.exposeInMainWorld("codexDesktop", {
   getUpdateStatus(payload) {
     return ipcRenderer.invoke("app:updateStatus", payload);
   },
+  autoUpdate(payload) {
+    return ipcRenderer.invoke("app:autoUpdate", payload);
+  },
 });

package/src/main/update-service.js

@@ -1,12 +1,23 @@
+const os = require("node:os");
+const path = require("node:path");
+const fs = require("node:fs");
+const { spawn } = require("node:child_process");
 const packageInfo = require("../../package.json");
 
 const REGISTRY_BASE_URL = "https://registry.npmjs.org";
 const SUCCESS_CACHE_TTL_MS = 10 * 60 * 1000;
 const ERROR_CACHE_TTL_MS = 30 * 1000;
 const MAX_RETRY_TIMES = 3;
+const AUTO_UPDATE_WAIT_SECONDS = 3;
 
 let cachedStatus = null;
 
+function getAutoUpdateLogPath() {
+  const logDir = path.join(os.homedir(), ".codex");
+  fs.mkdirSync(logDir, { recursive: true });
+  return path.join(logDir, "codex-switcher-auto-update.log");
+}
+
 function sleep(timeoutMs) {
   return new Promise((resolve) => {
     setTimeout(resolve, timeoutMs);
@@ -117,6 +128,61 @@ async function getUpdateStatus(options = {}) {
   return cachedStatus;
 }
 
+function scheduleAutoUpdate(options = {}) {
+  const reopen = options.reopen !== false;
+  const logPath = getAutoUpdateLogPath();
+  const installCommand = `npm install -g ${packageInfo.name}@latest`;
+  const restartCommand = reopen ? " && codex-switcher open" : "";
+
+  if (process.platform === "win32") {
+    const command =
+      `echo [%date% %time%] 开始自动更新 ${packageInfo.name} > "${logPath}" ` +
+      `&& timeout /t ${AUTO_UPDATE_WAIT_SECONDS} /nobreak >nul ` +
+      `&& ${installCommand} >> "${logPath}" 2>&1` +
+      `${restartCommand ? `${restartCommand} >> "${logPath}" 2>&1` : ""}`;
+
+    const child = spawn("cmd.exe", ["/d", "/s", "/c", command], {
+      cwd: os.homedir(),
+      detached: true,
+      stdio: "ignore",
+      windowsHide: true,
+      env: process.env,
+    });
+
+    child.unref();
+    return {
+      scheduled: true,
+      packageName: packageInfo.name,
+      currentVersion: packageInfo.version,
+      logPath,
+      reopen,
+    };
+  }
+
+  const command =
+    `sleep ${AUTO_UPDATE_WAIT_SECONDS}; ` +
+    `echo "[auto-update] start ${packageInfo.name}" > "${logPath}"; ` +
+    `${installCommand} >> "${logPath}" 2>&1` +
+    (reopen ? ` && codex-switcher open >> "${logPath}" 2>&1` : "");
+
+  const child = spawn("sh", ["-lc", command], {
+    cwd: os.homedir(),
+    detached: true,
+    stdio: "ignore",
+    env: process.env,
+  });
+
+  child.unref();
+  return {
+    scheduled: true,
+    packageName: packageInfo.name,
+    currentVersion: packageInfo.version,
+    logPath,
+    reopen,
+  };
+}
+
 module.exports = {
   getUpdateStatus,
+  scheduleAutoUpdate,
 };

package/src/renderer/index.html

@@ -122,6 +122,9 @@
                 </div>
                 <div class="update-actions">
                   <button id="checkUpdateButton" type="button" class="ghost-button">检查更新</button>
+                  <button id="autoUpdateButton" type="button" class="primary-button" hidden>
+                    立即更新
+                  </button>
                   <button id="copyUpdateCommandButton" type="button" class="secondary-button" hidden>
                     复制升级命令
                   </button>

package/src/renderer/renderer.js

@@ -115,6 +115,12 @@ function createWebBridge() {
       const query = searchParams.toString();
       return request(`/api/app/update-status${query ? `?${query}` : ""}`);
     },
+    autoUpdate(payload = {}) {
+      return request("/api/app/auto-update", {
+        method: "POST",
+        body: JSON.stringify(payload),
+      });
+    },
   };
 }
 
@@ -306,8 +312,9 @@ function renderUpdateStatus() {
   const detail = $("#updateDetail");
   const command = $("#updateCommand");
   const copyButton = $("#copyUpdateCommandButton");
+  const autoUpdateButton = $("#autoUpdateButton");
 
-  if (!summary || !detail || !command || !copyButton) {
+  if (!summary || !detail || !command || !copyButton || !autoUpdateButton) {
     return;
   }
 
@@ -318,6 +325,7 @@ function renderUpdateStatus() {
     detail.textContent = `当前版本 ${update.currentVersion}，可以直接升级到最新版本。`;
     command.hidden = false;
     copyButton.hidden = false;
+    autoUpdateButton.hidden = false;
     return;
   }
 
@@ -326,6 +334,7 @@ function renderUpdateStatus() {
     detail.textContent = `当前版本 ${update.currentVersion}，原因：${update.lastError}`;
     command.hidden = true;
     copyButton.hidden = true;
+    autoUpdateButton.hidden = true;
     return;
   }
 
@@ -333,6 +342,7 @@ function renderUpdateStatus() {
   detail.textContent = "npm 包没有检测到更新，可以继续直接使用。";
   command.hidden = true;
   copyButton.hidden = true;
+  autoUpdateButton.hidden = true;
 }
 
 async function refreshUpdateStatus(force = false) {
@@ -760,6 +770,59 @@ async function handleCopyUpdateCommand() {
   }
 }
 
+async function handleAutoUpdate() {
+  const update = state.update;
+  if (!update?.hasUpdate) {
+    return;
+  }
+
+  const confirmed = window.confirm(
+    `确定现在自动更新到 ${update.latestVersion} 吗？程序会自动关闭，更新完成后重新打开。`,
+  );
+  if (!confirmed) {
+    return;
+  }
+
+  const summary = $("#updateSummary");
+  const detail = $("#updateDetail");
+  const checkButton = $("#checkUpdateButton");
+  const autoUpdateButton = $("#autoUpdateButton");
+  const copyButton = $("#copyUpdateCommandButton");
+
+  if (summary) {
+    summary.textContent = `正在自动更新到 ${update.latestVersion}...`;
+  }
+  if (detail) {
+    detail.textContent = "程序会先关闭，再自动安装最新 npm 包并重新打开控制台。";
+  }
+  if (checkButton) {
+    checkButton.disabled = true;
+  }
+  if (autoUpdateButton) {
+    autoUpdateButton.disabled = true;
+  }
+  if (copyButton) {
+    copyButton.disabled = true;
+  }
+
+  try {
+    const result = await bridge.autoUpdate({ reopen: true });
+    setStatus(`已开始自动更新，日志：${result.logPath}`, "success");
+  } catch (error) {
+    if (checkButton) {
+      checkButton.disabled = false;
+    }
+    if (autoUpdateButton) {
+      autoUpdateButton.disabled = false;
+    }
+    if (copyButton) {
+      copyButton.disabled = false;
+    }
+    setStatus(`自动更新失败：${error.message}`, "error");
+    await refreshUpdateStatus(true);
+  }
+}
+
 function bindEvents() {
   $("#authForm").addEventListener("submit", (event) => {
     event.preventDefault();
@@ -810,6 +873,7 @@ function bindEvents() {
   $("#checkUpdateButton").addEventListener("click", () => {
     refreshUpdateStatus(true);
   });
+  $("#autoUpdateButton").addEventListener("click", handleAutoUpdate);
   $("#copyUpdateCommandButton").addEventListener("click", handleCopyUpdateCommand);
   $("#closeAppButton").addEventListener("click", handleCloseApp);
   window.addEventListener("keydown", (event) => {

package/src/web/proxy-server.js

@@ -10,6 +10,9 @@ const httpsAgent = new https.Agent({
   keepAlive: true,
 });
 
+const UPSTREAM_REQUEST_TIMEOUT_MS = 90000;
+const WARM_UP_TIMEOUT_MS = 12000;
+
 function sleep(ms) {
   return new Promise((resolve) => setTimeout(resolve, ms));
 }
@@ -60,11 +63,13 @@ function sendUpstreamRequest({
         agent: upstreamUrl.protocol === "https:" ? httpsAgent : httpAgent,
       },
       (upstreamResponse) => {
+        upstreamResponse.socket?.setNoDelay?.(true);
         resolve(upstreamResponse);
       },
     );
 
-    upstreamRequest.setTimeout(12000, () => {
+    upstreamRequest.setNoDelay?.(true);
+    upstreamRequest.setTimeout(UPSTREAM_REQUEST_TIMEOUT_MS, () => {
       upstreamRequest.destroy(new Error("上游请求超时。"));
     });
 
@@ -135,7 +140,7 @@ async function warmUpCurrentTarget() {
       },
     );
 
-    warmRequest.setTimeout(6000, () => {
+    warmRequest.setTimeout(WARM_UP_TIMEOUT_MS, () => {
       warmRequest.destroy(new Error("预热请求超时。"));
     });
     warmRequest.on("error", reject);
@@ -144,8 +149,11 @@ async function warmUpCurrentTarget() {
 }
 
 function createProxyServer() {
-  return http.createServer(async (req, res) => {
+  const server = http.createServer(async (req, res) => {
     try {
+      req.socket?.setNoDelay?.(true);
+      res.socket?.setNoDelay?.(true);
+
       if (req.url === "/__switcher/health") {
         res.writeHead(200, { "Content-Type": "application/json; charset=utf-8" });
         res.end(JSON.stringify({ ok: true, proxyBaseUrl: profileManager.proxyBaseUrl }));
@@ -189,6 +197,12 @@ function createProxyServer() {
       );
     }
   });
+
+  server.keepAliveTimeout = 90_000;
+  server.headersTimeout = 95_000;
+  server.requestTimeout = 0;
+
+  return server;
 }
 
 module.exports = {

package/src/web/server.js

@@ -89,7 +89,23 @@ function createApp() {
     "/api/endpoints/switch",
     wrapAsync(async (req) => {
       await ensureConsoleAccess();
-      return profileManager.switchEndpoint(req.body.id);
+      const result = await profileManager.switchEndpoint(req.body.id);
+      if (result.switchedViaProxy) {
+        try {
+          await warmUpCurrentTarget();
+          return {
+            ...result,
+            proxyWarmedUp: true,
+          };
+        } catch {
+          return {
+            ...result,
+            proxyWarmedUp: false,
+          };
+        }
+      }
+
+      return result;
     }),
   );
 
@@ -189,6 +205,21 @@ function createApp() {
     }),
   );
 
+  app.post(
+    "/api/app/auto-update",
+    wrapAsync(async (req) => {
+      const result = updateService.scheduleAutoUpdate({
+        reopen: req.body?.reopen !== false,
+      });
+
+      setTimeout(() => {
+        process.emit("codex-switcher:shutdown");
+      }, 120);
+
+      return result;
+    }),
+  );
+
   app.post(
     "/api/open-path",
     wrapAsync(async (req) => {