@pikecode/api-key-manager 1.1.0 → 1.1.2

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@pikecode/api-key-manager",
-  "version": "1.1.00",
+  "version": "1.1.2",
   "description": "A CLI tool for managing and switching multiple API provider configurations",
   "main": "src/index.js",
   "bin": {

package/src/commands/add.js

@@ -390,7 +390,7 @@ class ProviderAdder extends BaseCommand {
 
   async importCodexConfig() {
     try {
-      const { readCodexFiles } = require('../utils/codex-files');
+      const { readCodexFiles, extractBaseUrlFromConfigToml } = require('../utils/codex-files');
       const codexFiles = await readCodexFiles();
 
       if (!codexFiles.authJson) {
@@ -405,19 +405,15 @@ class ProviderAdder extends BaseCommand {
         return null;
       }
 
-      // 尝试从 config.toml 获取 base URL
-      // 支持 api_base_url（akm 写入的格式）和 api_base（某些旧配置可能使用）
+      // 从 config.toml 中读取当前激活 provider 的 base_url
+      // 优先从 [model_providers.<key>] section 读取，兼容旧的顶层 api_base_url 格式
       let baseUrl = null;
       if (codexFiles.configToml) {
-        const baseUrlMatch = codexFiles.configToml.match(/api_base_url\s*=\s*["']([^"']+)["']/);
-        if (baseUrlMatch) {
-          baseUrl = baseUrlMatch[1];
-        } else {
-          // 兼容旧格式
-          const legacyMatch = codexFiles.configToml.match(/api_base\s*=\s*["']([^"']+)["']/);
-          if (legacyMatch) {
-            baseUrl = legacyMatch[1];
-          }
+        baseUrl = extractBaseUrlFromConfigToml(codexFiles.configToml);
+        if (!baseUrl) {
+          // 兼容旧格式（akm 之前错误写入的顶层字段）
+          const legacyMatch = codexFiles.configToml.match(/^api_base_url\s*=\s*["']([^"']+)["']/m);
+          if (legacyMatch) baseUrl = legacyMatch[1];
         }
       }
 

package/src/utils/codex-files.js

@@ -87,106 +87,83 @@ async function backupCodexFiles(codexHome = resolveCodexHome()) {
   return backupDir;
 }
 
-async function applyCodexProfile(profile, options = {}) {
-  if (!profile || (profile.configToml == null && profile.authJson == null)) {
-    throw new Error('Codex 配置为空，无法切换');
-  }
-
-  const codexHome = await ensureCodexHome(profile.codexHome || options.codexHome);
-  const { configTomlPath, authJsonPath } = buildCodexPaths(codexHome);
-
-  const backupDir = await backupCodexFiles(codexHome);
-
-  if (profile.configToml != null) {
-    await fs.writeFile(configTomlPath, profile.configToml, 'utf8');
-    await setSecurePermissions(configTomlPath);
-  }
-
-  if (profile.authJson != null) {
-    await fs.writeFile(authJsonPath, profile.authJson, 'utf8');
-    await setSecurePermissions(authJsonPath);
-  }
-
-  return { codexHome, backupDir };
-}
-
 /**
- * 确保 config.toml 中设置了 preferred_auth_method = "apikey"
+ * 移除 config.toml 顶层（第一个 [section] 之前）的 api_base_url 字段
+ * 只清理顶层，不影响 [model_providers.xxx] section 内的 base_url
  * @param {string} configToml - 现有的 config.toml 内容
  * @returns {string} 更新后的 config.toml 内容
  */
-function ensureApiKeyAuthMethod(configToml) {
-  if (!configToml) {
-    // 如果没有 config.toml，创建一个最小配置
-    return 'preferred_auth_method = "apikey"\n';
-  }
-
-  // 移除所有已存在的 preferred_auth_method 行（防止重复）
-  const authMethodRegex = /^preferred_auth_method\s*=\s*["']?[^"'\n]*["']?\s*\n?/gm;
-  let cleanedConfig = configToml.replace(authMethodRegex, '');
-
-  // 清理可能存在的损坏字符（如孤立的 { 或空行堆积）
-  cleanedConfig = cleanedConfig.replace(/^[{}]+\s*\n?/gm, '');
-  cleanedConfig = cleanedConfig.replace(/^\n{3,}/gm, '\n\n');
+function removeTopLevelApiBaseUrl(configToml) {
+  if (!configToml) return configToml;
 
-  // 在文件开头添加 preferred_auth_method
-  const newLine = 'preferred_auth_method = "apikey"\n';
+  const sectionStart = configToml.search(/^\[/m);
+  const topLevel = sectionStart === -1 ? configToml : configToml.slice(0, sectionStart);
+  const rest = sectionStart === -1 ? '' : configToml.slice(sectionStart);
 
-  // 如果清理后为空，直接返回新配置
-  if (!cleanedConfig.trim()) {
-    return newLine;
-  }
+  const cleaned = topLevel.replace(/^api_base_url\s*=\s*["']?[^"'\n]*["']?\s*\n?/m, '');
+  return cleaned + rest;
+}
 
-  return newLine + cleanedConfig;
+/**
+ * 从 config.toml 中提取当前激活的 model_provider 的 base_url
+ * 读取 model_provider 字段，再从对应的 [model_providers.<key>] section 中取 base_url
+ * @param {string} configToml - config.toml 内容
+ * @returns {string|null} base_url 或 null
+ */
+function extractBaseUrlFromConfigToml(configToml) {
+  if (!configToml) return null;
+
+  // 读取 model_provider = "xxx"
+  const providerMatch = configToml.match(/^model_provider\s*=\s*["']([^"']+)["']/m);
+  if (!providerMatch) return null;
+
+  const providerKey = providerMatch[1];
+
+  // 在对应 section 中找 base_url
+  const sectionRegex = new RegExp(
+    `\\[model_providers\\.${providerKey}\\][^\\[]*base_url\\s*=\\s*["']([^"']+)["']`,
+    's'
+  );
+  const urlMatch = configToml.match(sectionRegex);
+  return urlMatch ? urlMatch[1] : null;
 }
 
 /**
- * 更新 config.toml 中的 api_base_url
+ * 更新 config.toml 中的 model_provider 和对应的 [model_providers.akm] section
+ * 使用固定 key "akm" 管理 akm 切换的供应商，不影响用户其他自定义配置
  * @param {string} configToml - 现有的 config.toml 内容
- * @param {string|null} baseUrl - API base URL，null 时移除该配置
+ * @param {string} baseUrl - 新供应商的 base_url
  * @returns {string} 更新后的 config.toml 内容
  */
-function updateApiBaseUrl(configToml, baseUrl) {
-  if (!configToml) {
-    configToml = '';
-  }
-
-  // 匹配 api_base_url 配置行
-  const baseUrlRegex = /^api_base_url\s*=\s*["']?[^"'\n]*["']?\s*\n?/m;
-
-  if (baseUrl) {
-    // 需要设置 api_base_url
-    const newLine = `api_base_url = "${baseUrl}"`;
-
-    if (configToml.match(baseUrlRegex)) {
-      // 替换现有的
-      return configToml.replace(baseUrlRegex, newLine + '\n');
-    }
-    // 空配置，直接返回新行
-    if (configToml.length === 0) {
-      return newLine + '\n';
-    }
-    // 找到第一个 section [xxx]，在它之前插入
-    const sectionMatch = configToml.match(/^\[/m);
-    if (sectionMatch) {
-      const insertPos = configToml.indexOf(sectionMatch[0]);
-      const before = configToml.slice(0, insertPos);
-      const after = configToml.slice(insertPos);
-      // 确保前面有换行分隔
-      const separator = before.endsWith('\n') ? '' : '\n';
-      return before + separator + newLine + '\n\n' + after;
-    }
-    // 没有 section，在文件末尾添加
-    const separator = configToml.endsWith('\n') ? '' : '\n';
-    return configToml + separator + newLine + '\n';
+function updateModelProvider(configToml, baseUrl) {
+  if (!configToml) configToml = '';
+
+  const providerKey = 'akm';
+  const newSection = [
+    `[model_providers.${providerKey}]`,
+    `name = "${providerKey}"`,
+    `base_url = "${baseUrl}"`,
+    'wire_api = "responses"',
+    'requires_openai_auth = true'
+  ].join('\n');
+
+  // 更新顶层 model_provider 字段
+  let result = configToml.match(/^model_provider\s*=/m)
+    ? configToml.replace(/^model_provider\s*=\s*["'][^"'\n]*["']/m, `model_provider = "${providerKey}"`)
+    : `model_provider = "${providerKey}"\n` + configToml;
+
+  // 替换或追加 [model_providers.akm] section
+  const sectionRegex = /\[model_providers\.akm\](?:\n(?!\[)[^\n]*)*\n?/;
+  if (result.match(sectionRegex)) {
+    result = result.replace(sectionRegex, newSection + '\n');
   } else {
-    // 移除 api_base_url（使用官方 API）
-    return configToml.replace(baseUrlRegex, '');
+    result = result.trimEnd() + '\n\n' + newSection + '\n';
   }
+
+  return result;
 }
 
 /**
- * 构建 auth.json 内容
  * @param {string} apiKey - API Key
  * @returns {string} auth.json 内容
  */
@@ -195,11 +172,14 @@ function buildAuthJson(apiKey) {
 }
 
 /**
- * 应用 Codex 配置（写入 config.toml 和 auth.json）
- * 用于切换供应商时确保配置文件正确
+ * 应用 Codex 配置（写入 auth.json，更新 config.toml 中的 provider 路由）
+ * akm 负责：
+ *   1. 写入 auth.json（API Key）
+ *   2. 更新 config.toml 中的 model_provider 和 [model_providers.akm] section
+ *   3. 清理之前错误写入的顶层 api_base_url 字段
  * @param {object} config - 供应商配置
  * @param {object} options - 选项
- * @returns {Promise<{codexHome: string, backupDir: string|null}>}
+ * @returns {Promise<{codexHome: string}>}
  */
 async function applyCodexConfig(config, options = {}) {
   if (!config || !config.authToken) {
@@ -209,41 +189,41 @@ async function applyCodexConfig(config, options = {}) {
   const codexHome = await ensureCodexHome(options.codexHome);
   const { configTomlPath, authJsonPath } = buildCodexPaths(codexHome);
 
-  // 备份现有配置
-  const backupDir = await backupCodexFiles(codexHome);
-
-  // 读取现有 config.toml
-  let existingConfigToml = null;
-  if (await fs.pathExists(configTomlPath)) {
-    existingConfigToml = await fs.readFile(configTomlPath, 'utf8');
-  }
-
-  // 确保设置了 preferred_auth_method = "apikey"
-  let updatedConfigToml = ensureApiKeyAuthMethod(existingConfigToml);
-
-  // 更新 api_base_url（如果有则设置，没有则移除）
-  updatedConfigToml = updateApiBaseUrl(updatedConfigToml, config.baseUrl || null);
-
-  await fs.writeFile(configTomlPath, updatedConfigToml, 'utf8');
-  await setSecurePermissions(configTomlPath);
-
-  // 写入 auth.json
+  // 写入 auth.json（API Key）
   const authJsonContent = buildAuthJson(config.authToken);
   await fs.writeFile(authJsonPath, authJsonContent, 'utf8');
   await setSecurePermissions(authJsonPath);
 
-  return { codexHome, backupDir };
+  // 更新 config.toml：设置 model_provider + [model_providers.akm]，清理旧的顶层 api_base_url
+  if (config.baseUrl) {
+    const existingToml = await fs.pathExists(configTomlPath)
+      ? await fs.readFile(configTomlPath, 'utf8')
+      : '';
+    let updatedToml = removeTopLevelApiBaseUrl(existingToml);
+    updatedToml = updateModelProvider(updatedToml, config.baseUrl);
+    await fs.writeFile(configTomlPath, updatedToml, 'utf8');
+    await setSecurePermissions(configTomlPath);
+  } else if (await fs.pathExists(configTomlPath)) {
+    // 没有 baseUrl，只清理旧的顶层 api_base_url
+    const existingToml = await fs.readFile(configTomlPath, 'utf8');
+    const cleanedToml = removeTopLevelApiBaseUrl(existingToml);
+    if (cleanedToml !== existingToml) {
+      await fs.writeFile(configTomlPath, cleanedToml, 'utf8');
+      await setSecurePermissions(configTomlPath);
+    }
+  }
+
+  return { codexHome };
 }
 
 module.exports = {
   resolveCodexHome,
   buildCodexPaths,
   readCodexFiles,
-  applyCodexProfile,
   applyCodexConfig,
   backupCodexFiles,
-  ensureApiKeyAuthMethod,
-  updateApiBaseUrl,
+  removeTopLevelApiBaseUrl,
+  extractBaseUrlFromConfigToml,
+  updateModelProvider,
   buildAuthJson
 };
-