codex-endpoint-switcher 1.0.0

package/src/main/profile-manager.js

@@ -0,0 +1,439 @@
+const fs = require("node:fs/promises");
+const path = require("node:path");
+const os = require("node:os");
+const { randomUUID } = require("node:crypto");
+
+const codexRoot = path.join(os.homedir(), ".codex");
+const backupDir = path.join(codexRoot, "backup");
+const mainConfigPath = path.join(codexRoot, "config.toml");
+const authConfigPath = path.join(codexRoot, "auth.json");
+const endpointStorePath = path.join(codexRoot, "endpoint-presets.json");
+
+function escapeRegExp(value) {
+  return value.replace(/[.*+?^${}()|[\]\\]/g, "\\$&");
+}
+
+function escapeTomlString(value) {
+  return String(value).replace(/\\/g, "\\\\").replace(/"/g, '\\"');
+}
+
+function stripQuotedValue(value) {
+  if (typeof value !== "string") {
+    return value;
+  }
+
+  const trimmed = value.trim();
+  if (trimmed.startsWith('"') && trimmed.endsWith('"')) {
+    return trimmed.slice(1, -1);
+  }
+
+  return trimmed;
+}
+
+function maskKey(value) {
+  const key = String(value || "").trim();
+  if (!key) {
+    return "";
+  }
+
+  if (key.length <= 10) {
+    return `${key.slice(0, 3)}***${key.slice(-2)}`;
+  }
+
+  return `${key.slice(0, 6)}***${key.slice(-4)}`;
+}
+
+function normalizeUrl(url) {
+  const trimmed = String(url || "").trim();
+  let parsedUrl;
+
+  try {
+    parsedUrl = new URL(trimmed);
+  } catch {
+    throw new Error("URL 格式不正确。");
+  }
+
+  if (!["http:", "https:"].includes(parsedUrl.protocol)) {
+    throw new Error("URL 仅支持 http 或 https。");
+  }
+
+  return parsedUrl.toString();
+}
+
+function validateEndpointPayload(payload) {
+  const note = String(payload.note || "").trim();
+  const url = normalizeUrl(payload.url);
+  const key = String(payload.key || "").trim();
+
+  if (!note) {
+    throw new Error("备注不能为空。");
+  }
+
+  if (!key) {
+    throw new Error("Key 不能为空。");
+  }
+
+  return { note, url, key };
+}
+
+async function ensureDirectory(targetPath) {
+  await fs.mkdir(targetPath, { recursive: true });
+}
+
+async function ensureFileExists(targetPath, errorMessage) {
+  try {
+    await fs.access(targetPath);
+  } catch {
+    throw new Error(errorMessage);
+  }
+}
+
+async function readTextFile(targetPath) {
+  return fs.readFile(targetPath, "utf8");
+}
+
+async function writeTextFile(targetPath, content) {
+  await fs.writeFile(targetPath, content, "utf8");
+}
+
+async function readJsonFile(targetPath, fallbackValue) {
+  try {
+    const content = await readTextFile(targetPath);
+    return JSON.parse(content);
+  } catch (error) {
+    if (error.code === "ENOENT") {
+      return fallbackValue;
+    }
+
+    if (fallbackValue !== undefined) {
+      return fallbackValue;
+    }
+
+    throw error;
+  }
+}
+
+async function writeJsonFile(targetPath, value) {
+  await writeTextFile(targetPath, `${JSON.stringify(value, null, 2)}\n`);
+}
+
+function readTomlScalar(content, key) {
+  const pattern = new RegExp(`^\\s*${escapeRegExp(key)}\\s*=\\s*(.+?)\\s*$`, "m");
+  const matched = content.match(pattern);
+  if (!matched) {
+    return "";
+  }
+
+  return stripQuotedValue(matched[1]);
+}
+
+function readProviderBaseUrl(content, provider) {
+  if (!provider) {
+    return "";
+  }
+
+  const lines = content.split(/\r?\n/);
+  let insideProviderSection = false;
+
+  for (const line of lines) {
+    const trimmed = line.trim();
+    if (/^\[.+\]$/.test(trimmed)) {
+      insideProviderSection = trimmed === `[model_providers.${provider}]`;
+      continue;
+    }
+
+    if (!insideProviderSection) {
+      continue;
+    }
+
+    const matched = line.match(/^\s*base_url\s*=\s*"(.+?)"\s*$/);
+    if (matched) {
+      return matched[1].trim();
+    }
+  }
+
+  return "";
+}
+
+function updateProviderBaseUrl(content, provider, nextUrl) {
+  const newline = content.includes("\r\n") ? "\r\n" : "\n";
+  const lines = content.split(/\r?\n/);
+  const output = [];
+  const replacementLine = `base_url = "${escapeTomlString(nextUrl)}"`;
+  let foundSection = false;
+  let insideProviderSection = false;
+  let replaced = false;
+
+  for (const line of lines) {
+    const trimmed = line.trim();
+    if (/^\[.+\]$/.test(trimmed)) {
+      if (insideProviderSection && !replaced) {
+        output.push(replacementLine);
+        replaced = true;
+      }
+
+      insideProviderSection = trimmed === `[model_providers.${provider}]`;
+      if (insideProviderSection) {
+        foundSection = true;
+      }
+
+      output.push(line);
+      continue;
+    }
+
+    if (insideProviderSection && /^\s*base_url\s*=/.test(line)) {
+      output.push(replacementLine);
+      replaced = true;
+      continue;
+    }
+
+    output.push(line);
+  }
+
+  if (insideProviderSection && !replaced) {
+    output.push(replacementLine);
+    replaced = true;
+  }
+
+  if (!foundSection) {
+    throw new Error(`未找到模型提供商配置段：${provider}`);
+  }
+
+  if (!replaced) {
+    throw new Error("未能更新 base_url。");
+  }
+
+  return output.join(newline);
+}
+
+function buildEndpointResponse(endpoint, activeId) {
+  return {
+    id: endpoint.id,
+    note: endpoint.note,
+    url: endpoint.url,
+    key: endpoint.key,
+    maskedKey: maskKey(endpoint.key),
+    createdAt: endpoint.createdAt,
+    updatedAt: endpoint.updatedAt,
+    isActive: endpoint.id === activeId,
+  };
+}
+
+async function ensureEndpointStore() {
+  await ensureFileExists(mainConfigPath, `未找到 Codex 主配置文件：${mainConfigPath}`);
+  await ensureDirectory(backupDir);
+
+  try {
+    await fs.access(endpointStorePath);
+  } catch {
+    await writeJsonFile(endpointStorePath, []);
+  }
+}
+
+async function readEndpointStore() {
+  await ensureEndpointStore();
+  const data = await readJsonFile(endpointStorePath, []);
+  return Array.isArray(data) ? data : [];
+}
+
+async function writeEndpointStore(endpoints) {
+  await writeJsonFile(endpointStorePath, endpoints);
+}
+
+async function readCurrentConfigAndAuth() {
+  await ensureEndpointStore();
+  const configContent = await readTextFile(mainConfigPath);
+  const authConfig = await readJsonFile(authConfigPath, {});
+  const provider = readTomlScalar(configContent, "model_provider");
+
+  return {
+    configContent,
+    authConfig,
+    provider,
+    model: readTomlScalar(configContent, "model"),
+    baseUrl: readProviderBaseUrl(configContent, provider),
+    apiKey: String(authConfig.OPENAI_API_KEY || "").trim(),
+  };
+}
+
+async function backupCurrentState() {
+  const timestamp = new Date().toISOString().replace(/[:.]/g, "-");
+  const configBackupPath = path.join(backupDir, `config.toml.switch-${timestamp}.bak`);
+  const authBackupPath = path.join(backupDir, `auth.json.switch-${timestamp}.bak`);
+
+  await fs.copyFile(mainConfigPath, configBackupPath);
+  try {
+    await fs.copyFile(authConfigPath, authBackupPath);
+  } catch (error) {
+    if (error.code !== "ENOENT") {
+      throw error;
+    }
+  }
+
+  return {
+    configBackupPath,
+    authBackupPath,
+  };
+}
+
+async function getCurrentEndpointSummary() {
+  const [currentState, endpoints] = await Promise.all([
+    readCurrentConfigAndAuth(),
+    readEndpointStore(),
+  ]);
+
+  const activeEndpoint = endpoints.find(
+    (item) => item.url === currentState.baseUrl && item.key === currentState.apiKey,
+  );
+
+  return {
+    model: currentState.model,
+    provider: currentState.provider,
+    currentUrl: currentState.baseUrl,
+    currentKeyMasked: maskKey(currentState.apiKey),
+    currentNote: activeEndpoint ? activeEndpoint.note : "未命名连接",
+    activeEndpointId: activeEndpoint ? activeEndpoint.id : "",
+    endpointStorePath,
+    authConfigPath,
+    mainConfigPath,
+    totalEndpoints: endpoints.length,
+  };
+}
+
+async function listEndpoints() {
+  const [currentState, endpoints] = await Promise.all([
+    readCurrentConfigAndAuth(),
+    readEndpointStore(),
+  ]);
+
+  const activeEndpoint = endpoints.find(
+    (item) => item.url === currentState.baseUrl && item.key === currentState.apiKey,
+  );
+  const activeId = activeEndpoint ? activeEndpoint.id : "";
+
+  return endpoints
+    .slice()
+    .sort((left, right) => String(right.updatedAt).localeCompare(String(left.updatedAt)))
+    .map((endpoint) => buildEndpointResponse(endpoint, activeId));
+}
+
+async function createEndpoint(payload) {
+  const endpoint = validateEndpointPayload(payload);
+  const endpoints = await readEndpointStore();
+  const now = new Date().toISOString();
+
+  const nextItem = {
+    id: randomUUID(),
+    note: endpoint.note,
+    url: endpoint.url,
+    key: endpoint.key,
+    createdAt: now,
+    updatedAt: now,
+  };
+
+  endpoints.push(nextItem);
+  await writeEndpointStore(endpoints);
+  return buildEndpointResponse(nextItem, "");
+}
+
+async function updateEndpoint(id, payload) {
+  const safeId = String(id || "").trim();
+  if (!safeId) {
+    throw new Error("缺少要更新的端点 ID。");
+  }
+
+  const endpoint = validateEndpointPayload(payload);
+  const endpoints = await readEndpointStore();
+  const index = endpoints.findIndex((item) => item.id === safeId);
+
+  if (index === -1) {
+    throw new Error("未找到要更新的端点。");
+  }
+
+  endpoints[index] = {
+    ...endpoints[index],
+    note: endpoint.note,
+    url: endpoint.url,
+    key: endpoint.key,
+    updatedAt: new Date().toISOString(),
+  };
+
+  await writeEndpointStore(endpoints);
+  return buildEndpointResponse(endpoints[index], "");
+}
+
+async function deleteEndpoint(id) {
+  const safeId = String(id || "").trim();
+  if (!safeId) {
+    throw new Error("缺少要删除的端点 ID。");
+  }
+
+  const endpoints = await readEndpointStore();
+  const index = endpoints.findIndex((item) => item.id === safeId);
+
+  if (index === -1) {
+    throw new Error("未找到要删除的端点。");
+  }
+
+  const [removed] = endpoints.splice(index, 1);
+  await writeEndpointStore(endpoints);
+
+  return {
+    id: removed.id,
+    note: removed.note,
+  };
+}
+
+async function switchEndpoint(id) {
+  const safeId = String(id || "").trim();
+  if (!safeId) {
+    throw new Error("缺少要切换的端点 ID。");
+  }
+
+  const endpoints = await readEndpointStore();
+  const target = endpoints.find((item) => item.id === safeId);
+
+  if (!target) {
+    throw new Error("未找到要切换的端点。");
+  }
+
+  const { configContent, authConfig, provider } = await readCurrentConfigAndAuth();
+  if (!provider) {
+    throw new Error("当前 config.toml 未找到 model_provider。");
+  }
+
+  const nextConfigContent = updateProviderBaseUrl(configContent, provider, target.url);
+  const nextAuthConfig = {
+    ...authConfig,
+    OPENAI_API_KEY: target.key,
+  };
+
+  const backupPaths = await backupCurrentState();
+  await writeTextFile(mainConfigPath, nextConfigContent);
+  await writeJsonFile(authConfigPath, nextAuthConfig);
+
+  return {
+    ...backupPaths,
+    ...(await getCurrentEndpointSummary()),
+  };
+}
+
+function getManagedPaths() {
+  return {
+    codexRoot,
+    backupDir,
+    mainConfigPath,
+    authConfigPath,
+    endpointStorePath,
+  };
+}
+
+module.exports = {
+  createEndpoint,
+  deleteEndpoint,
+  getCurrentEndpointSummary,
+  getManagedPaths,
+  listEndpoints,
+  switchEndpoint,
+  updateEndpoint,
+};

package/src/renderer/index.html

@@ -0,0 +1,140 @@
+<!doctype html>
+<html lang="zh-CN">
+  <head>
+    <meta charset="UTF-8" />
+    <meta
+      http-equiv="Content-Security-Policy"
+      content="default-src 'self'; style-src 'self' 'unsafe-inline'; script-src 'self';"
+    />
+    <meta name="viewport" content="width=device-width, initial-scale=1.0" />
+    <title>Codex 连接切换器</title>
+    <link rel="stylesheet" href="./styles.css" />
+  </head>
+  <body>
+    <div class="page-shell">
+      <header class="hero-card">
+        <div class="hero-copy">
+          <p class="eyebrow">CODEX ENDPOINT SWITCHER</p>
+          <h1>只切换 URL 和 Key</h1>
+          <p class="hero-text">
+            你只需要维护 <code>备注</code>、<code>URL</code>、<code>Key</code>。
+            切换时应用只会更新 <code>config.toml</code> 的 <code>base_url</code> 和
+            <code>auth.json</code> 的 <code>OPENAI_API_KEY</code>，其它配置保持不动。
+          </p>
+          <div class="hero-actions">
+            <button id="refreshButton" class="ghost-button">刷新状态</button>
+            <button id="openCodexRootButton" class="ghost-button">打开 Codex 目录</button>
+            <button id="openEndpointStoreButton" class="ghost-button">打开连接数据文件</button>
+          </div>
+        </div>
+        <div class="hero-side">
+          <div class="orb orb-a"></div>
+          <div class="orb orb-b"></div>
+          <div class="hero-badge" id="heroBadge">读取中</div>
+        </div>
+      </header>
+
+      <main class="content-grid">
+        <section class="panel current-panel">
+          <div class="panel-header">
+            <div>
+              <p class="panel-kicker">当前状态</p>
+              <h2>当前生效连接</h2>
+            </div>
+            <span class="panel-tag" id="activeEndpointTag">未识别</span>
+          </div>
+          <div class="current-grid">
+            <article class="metric-card">
+              <span class="metric-label">当前备注</span>
+              <strong id="currentNote">-</strong>
+            </article>
+            <article class="metric-card">
+              <span class="metric-label">提供商</span>
+              <strong id="currentProvider">-</strong>
+            </article>
+            <article class="metric-card metric-card-wide">
+              <span class="metric-label">当前 URL</span>
+              <strong id="currentBaseUrl">-</strong>
+            </article>
+            <article class="metric-card">
+              <span class="metric-label">当前 Key</span>
+              <strong id="currentKeyMasked">-</strong>
+            </article>
+            <article class="metric-card">
+              <span class="metric-label">模型</span>
+              <strong id="currentModel">-</strong>
+            </article>
+          </div>
+        </section>
+
+        <section class="panel action-panel">
+          <div class="panel-header">
+            <div>
+              <p class="panel-kicker">编辑区</p>
+              <h2>新增或编辑连接</h2>
+            </div>
+          </div>
+
+          <form id="endpointForm" class="stack-form">
+            <input id="endpointId" type="hidden" />
+            <label class="field">
+              <span>备注</span>
+              <input
+                id="endpointNote"
+                type="text"
+                placeholder="例如：公司 / 家里 / 备用"
+                autocomplete="off"
+              />
+            </label>
+            <label class="field">
+              <span>URL</span>
+              <input
+                id="endpointUrl"
+                type="text"
+                placeholder="例如：https://sub.jlypx.de/"
+                autocomplete="off"
+              />
+            </label>
+            <label class="field">
+              <span>Key</span>
+              <input
+                id="endpointKey"
+                type="password"
+                placeholder="输入 OPENAI_API_KEY"
+                autocomplete="off"
+              />
+            </label>
+            <p class="field-hint">连接数据仅保存在本机 `~/.codex/endpoint-presets.json`。</p>
+            <div class="action-row">
+              <button id="saveEndpointButton" type="submit" class="primary-button">新增连接</button>
+              <button id="clearFormButton" type="button" class="ghost-button">清空表单</button>
+            </div>
+          </form>
+        </section>
+
+        <section class="panel profiles-panel">
+          <div class="panel-header">
+            <div>
+              <p class="panel-kicker">Endpoints</p>
+              <h2>连接列表</h2>
+            </div>
+            <span class="profile-count" id="endpointCount">0 条</span>
+          </div>
+          <div id="endpointsList" class="profiles-list"></div>
+        </section>
+
+        <section class="panel footer-panel">
+          <div class="panel-header">
+            <div>
+              <p class="panel-kicker">输出</p>
+              <h2>执行结果</h2>
+            </div>
+          </div>
+          <div id="statusBox" class="status-box info">应用已启动，正在读取当前连接。</div>
+        </section>
+      </main>
+    </div>
+
+    <script src="./renderer.js"></script>
+  </body>
+</html>