@tokenaut/opentoken 1.3.5 → 1.4.0

package/src/cli.js

@@ -1,88 +1,105 @@
 'use strict';
 
-const { applyConfig, SETTINGS_PATH } = require('./config');
-const { applyCodexConfig, CONFIG_PATH: CODEX_CONFIG_PATH, AUTH_PATH: CODEX_AUTH_PATH, getCurrentCodexConfig } = require('./codex-config');
+const chalk = require('chalk');
+const inquirer = require('inquirer');
+const { printLogo } = require('./logo');
+const { clearScreen } = require('./screen');
+const claude = require('./tools/claude');
+const atomcode = require('./tools/atomcode');
+const hermes = require('./tools/hermes');
+const openclaw = require('./tools/openclaw');
+const opencode = require('./tools/opencode');
+const codex = require('./tools/codex');
+const { DEFAULT_BASE_URL } = require('./models');
 
-const DEFAULT_BASE_URL = 'https://gw.opentoken.io/';
-const CODEX_DEFAULT_BASE_URL = 'https://gw.opentoken.io/v1';
-
-const CC_DEFAULT_MODEL_ALIAS = 'opus';
-const CX_DEFAULT_MODEL_ALIAS = '5.4';
-
-const CC_MODEL_MAP = {
-  opus: 'claude-opus-4-8',
-  sonnet: 'claude-sonnet-4-6',
-  haiku: 'claude-haiku-4-5',
-  'claude-opus-4-8': 'claude-opus-4-8',
-  'claude-sonnet-4-6': 'claude-sonnet-4-6',
-  'claude-haiku-4-5': 'claude-haiku-4-5',
-};
-
-const CC_SONNET_MODEL = 'claude-sonnet-4-6';
-const CC_HAIKU_MODEL = 'claude-haiku-4-5';
+function isCcCommand(tool) {
+  return tool === 'cc' || tool === 'claude';
+}
 
-const CX_MODEL_MAP = {
-  '5.5': 'gpt-5.5',
-  '5.4': 'gpt-5.4',
-  'gpt5.5': 'gpt-5.5',
-  'gpt5.4': 'gpt-5.4',
-  'gpt-5.5': 'gpt-5.5',
-  'gpt-5.4': 'gpt-5.4',
-};
+function isAtomcodeCommand(tool) {
+  return tool === 'atomcode' || tool === 'ac';
+}
 
-function resolveModel(input, modelMap, defaultAlias) {
-  if (!input) {
-    return modelMap[defaultAlias];
-  }
+function isHermesCommand(tool) {
+  return tool === 'hermes' || tool === 'hm';
+}
 
-  const trimmed = String(input).trim().toLowerCase();
-  if (modelMap[trimmed]) {
-    return modelMap[trimmed];
-  }
+function isOpenclawCommand(tool) {
+  return tool === 'openclaw' || tool === 'oc';
+}
 
-  return String(input).trim();
+function isOpencodeCommand(tool) {
+  return tool === 'opencode' || tool === 'od';
 }
 
-function maskApiKey(key) {
-  if (!key || key.length < 8) {
-    return key || '';
-  }
+function isCodexCommand(tool) {
+  return tool === 'codex' || tool === 'cx';
+}
 
-  return `${key.slice(0, 6)}****${key.slice(-4)}`;
+function printVersion() {
+  const pkg = require('../package.json');
+  console.log(`${pkg.name} ${pkg.version}`);
 }
 
 function printHelp() {
   console.log('');
-  console.log('OpenToken 一键配置 CLI');
+  console.log(chalk.bold('常用：'));
+  console.log('  opentoken help | --help | -h           显示本帮助');
+  console.log('  opentoken version | --version | -v | -V  显示版本号');
   console.log('');
-  console.log('用法：');
-  console.log('  opentoken <tool> [--key <api_key>] [--model <model>] [--url <base_url>]');
+  console.log(chalk.bold('用法：'));
+  console.log('  opentoken                              进入主菜单');
+  console.log('  opentoken cc                           进入 Claude Code 配置菜单');
+  console.log('  opentoken cc --key   <apikey>          设置 API Key');
+  console.log('  opentoken cc --url   <baseurl>         设置 Base URL');
+  console.log('  opentoken cc --model <model>           设置模型');
+  console.log('  opentoken cc -k <k> -u <u> -m <m>      一键全部设置');
   console.log('');
-  console.log('工具：');
-  console.log('  cc       Claude Code 配置（~/.claude/settings.json）');
-  console.log('  codex    Codex 配置（~/.codex/config.toml + ~/.codex/auth.json）');
-  console.log('  cx       同 codex（缩写）');
+  console.log('  opentoken atomcode                     进入 Atomcode 配置（~/.atomcode/config.toml）');
+  console.log('  opentoken ac                           同上（缩写）');
+  console.log('  opentoken ac --key <apikey>            仅设置 API Key（base_url 固定为 Opentoken /v1）');
+  console.log('  opentoken ac --model <model>           仅设置模型');
+  console.log('  opentoken ac -k <k> -m <m>             一键设置 Key 与模型');
   console.log('');
-  console.log('Claude Code 模型别名：');
-  console.log(`  opus   → ${CC_MODEL_MAP.opus} (默认)`);
-  console.log(`  sonnet → ${CC_MODEL_MAP.sonnet}`);
-  console.log(`  haiku  → ${CC_MODEL_MAP.haiku}`);
+  console.log('  opentoken openclaw                   进入 OpenClaw 配置（~/.openclaw/openclaw.json）');
+  console.log('  opentoken oc                         同上（缩写）');
+  console.log('  opentoken oc --key <apikey>          写入 models.providers.opentoken.apiKey');
+  console.log('  opentoken oc --model <model>         设置 agents.defaults.model.primary（opentoken/<id>）');
+  console.log('  opentoken oc -k <k> -m <m>           一键设置 Key 与主模型');
   console.log('');
-  console.log('Codex 模型别名：');
-  console.log(`  5.5 / gpt5.5 / gpt-5.5 → gpt-5.5`);
-  console.log(`  5.4 / gpt5.4 / gpt-5.4 → gpt-5.4 (默认)`);
+  console.log('  opentoken hermes                       进入 Hermes 配置（~/.hermes）');
+  console.log('  opentoken hm                           同上（缩写）');
+  console.log('  opentoken hm --key <apikey>            设置 Opentoken API KEY（写入 ~/.hermes/.env）');
+  console.log('  opentoken hm --model <model>           设置 config.yaml 内 model.default');
+  console.log('  opentoken hm -k <k> -m <m>             一键设置 Key 与默认模型');
   console.log('');
-  console.log('示例：');
-  console.log('  opentoken cc --key sk-xxxxx');
-  console.log('  opentoken cc --model sonnet');
-  console.log('  opentoken codex --key sk-xxxxx');
-  console.log('  opentoken cx --model 5.5');
-  console.log('  opentoken cx --url https://gw.opentoken.io/v1');
+  console.log('  opentoken opencode                     进入 OpenCode 配置（~/.config/opencode/opencode.json）');
+  console.log('  opentoken od                           同上（缩写）');
+  console.log('  opentoken od --key <apikey>            设置 Opentoken API KEY');
+  console.log('  opentoken od --model <model>           设置默认模型');
+  console.log('  opentoken od -k <k> -m <m>             一键设置 Key 与模型');
   console.log('');
-  console.log('说明：');
-  console.log('  各参数均可独立使用，只修改指定的配置项，未指定的保持不变');
-  console.log(`  Claude Code 默认 Base URL: ${DEFAULT_BASE_URL}`);
-  console.log(`  Codex 默认 Base URL: ${CODEX_DEFAULT_BASE_URL}`);
+  console.log('  opentoken codex                        进入 Codex 配置（~/.codex/config.toml）');
+  console.log('  opentoken cx                           同上（缩写）');
+  console.log('  opentoken cx --key <apikey>            拉取 responses 模型并写入配置（Key 需 export 生效）');
+  console.log('  opentoken cx --url <baseurl>           设置 Codex Base URL（需包含 /v1）');
+  console.log('  opentoken cx --model <model>           设置默认模型');
+  console.log('  opentoken cx -k <k> -u <u> -m <m>      一键设置 URL 与模型');
+  console.log('');
+  console.log(chalk.bold('快捷别名（--key / -k，--url / -u，--model / -m）'));
+  console.log('');
+  console.log(chalk.bold('示例：'));
+  console.log('  opentoken cc -k sk-xxx');
+  console.log(`  opentoken cc -k sk-xxx -u ${DEFAULT_BASE_URL} -m claude-opus-4-5`);
+  console.log('');
+  console.log(chalk.bold('主菜单：'));
+  console.log('  1. 配置 Claude Code');
+  console.log('  2. 配置 Atomcode');
+  console.log('  3. 配置 OpenClaw');
+  console.log('  4. 配置 Hermes');
+  console.log('  5. 配置 OpenCode');
+  console.log('  6. 配置 Codex');
+  console.log('  7. 退出');
   console.log('');
 }
 
@@ -91,138 +108,181 @@ function parseArgs(argv) {
   const result = {
     tool: null,
     key: null,
-    model: null,
     url: null,
+    model: null,
     help: false,
+    version: false,
   };
 
   let i = 0;
+
   if (args[0] && !args[0].startsWith('-')) {
     result.tool = args[0].toLowerCase();
     i = 1;
   }
 
-  for (; i < args.length; i += 1) {
-    const current = args[i];
+  for (; i < args.length; i++) {
+    const a = args[i];
     const next = args[i + 1];
-
-    if (current === '--help' || current === '-h') {
-      result.help = true;
-    } else if ((current === '--key' || current === '-k') && next) {
+    if (a === '--help' || a === '-h') result.help = true;
+    else if (a === '--version' || a === '-v' || a === '-V') result.version = true;
+    else if ((a === '--key' || a === '-k') && next) {
       result.key = next;
-      i += 1;
-    } else if ((current === '--model' || current === '-m') && next) {
-      result.model = next;
-      i += 1;
-    } else if ((current === '--url' || current === '-u') && next) {
+      i++;
+    } else if ((a === '--url' || a === '-u') && next) {
       result.url = next;
-      i += 1;
+      i++;
+    } else if ((a === '--model' || a === '-m') && next) {
+      result.model = next;
+      i++;
     }
   }
 
   return result;
 }
 
-function isCcCommand(tool) {
-  return tool === 'cc';
-}
+async function mainMenu() {
+  while (true) {
+    clearScreen();
+    printLogo();
+
+    const { action } = await inquirer.prompt([
+      {
+        type: 'list',
+        name: 'action',
+        prefix: '',
+        message: '请选择：',
+        choices: [
+          { name: '1. 配置 Claude Code', value: 'claude' },
+          { name: '2. 配置 Atomcode', value: 'atomcode' },
+          { name: '3. 配置 OpenClaw', value: 'openclaw' },
+          { name: '4. 配置 Hermes', value: 'hermes' },
+          { name: '5. 配置 OpenCode', value: 'opencode' },
+          { name: '6. 配置 Codex', value: 'codex' },
+          { name: '7. 退出', value: 'exit' },
+        ],
+        pageSize: 8,
+      },
+    ]);
+
+    if (action === 'exit') {
+      clearScreen();
+      console.log(chalk.cyan('再见！'));
+      return;
+    }
 
-function isCodexCommand(tool) {
-  return tool === 'codex' || tool === 'cx';
-}
+    if (action === 'claude') {
+      await claude.run();
+    }
+
+    if (action === 'atomcode') {
+      await atomcode.run();
+    }
+
+    if (action === 'openclaw') {
+      await openclaw.run();
+    }
+
+    if (action === 'hermes') {
+      await hermes.run();
+    }
 
-function validateArgs(args) {
-  if (!isCcCommand(args.tool) && !isCodexCommand(args.tool)) {
-    throw new Error('未知工具，请使用: opentoken cc 或 opentoken codex');
+    if (action === 'opencode') {
+      await opencode.run();
+    }
+
+    if (action === 'codex') {
+      await codex.run();
+    }
   }
 }
 
-async function runCc(args) {
-  const patch = {};
+async function run() {
+  const args = parseArgs(process.argv);
 
-  if (args.key && String(args.key).trim()) {
-    patch.apiKey = String(args.key).trim();
+  if (args.help || args.tool === 'help') {
+    printHelp();
+    return;
   }
 
-  if (args.model !== null) {
-    patch.model = resolveModel(args.model, CC_MODEL_MAP, CC_DEFAULT_MODEL_ALIAS);
-    patch.sonnetModel = resolveModel('sonnet', CC_MODEL_MAP, CC_DEFAULT_MODEL_ALIAS);
-    patch.haikuModel = resolveModel('haiku', CC_MODEL_MAP, CC_DEFAULT_MODEL_ALIAS);
+  if (args.version || args.tool === 'version') {
+    printVersion();
+    return;
   }
 
-  if (args.url && String(args.url).trim()) {
-    patch.baseUrl = String(args.url).trim();
+  if (isCcCommand(args.tool) && (args.key || args.url || args.model)) {
+    clearScreen();
+    printLogo();
+    claude.quickSet({ key: args.key, url: args.url, model: args.model });
+    return;
   }
 
-  if (Object.keys(patch).length === 0) {
-    console.log('');
-    console.log('未指定任何配置项，请使用 --key / --model / --url');
-    console.log('');
+  if (isAtomcodeCommand(args.tool) && (args.key || args.model)) {
+    clearScreen();
+    printLogo();
+    atomcode.quickSet({ key: args.key, model: args.model });
     return;
   }
 
-  applyConfig(patch);
-
-  console.log('');
-  console.log('Claude Code 配置已更新：');
-  if (patch.apiKey) console.log(`  API Key: ${maskApiKey(patch.apiKey)}`);
-  if (patch.baseUrl) console.log(`  Base URL: ${patch.baseUrl}`);
-  if (patch.model) console.log(`  Model: ${patch.model}`);
-  if (patch.sonnetModel) console.log(`  Sonnet: ${patch.sonnetModel}`);
-  if (patch.haikuModel) console.log(`  Haiku: ${patch.haikuModel}`);
-  console.log(`  File: ${SETTINGS_PATH}`);
-  console.log('');
-}
-
-async function runCodex(args) {
-  const patch = {};
-
-  if (args.key && String(args.key).trim()) {
-    patch.apiKey = String(args.key).trim();
+  if (isHermesCommand(args.tool) && (args.key || args.model)) {
+    clearScreen();
+    printLogo();
+    hermes.quickSet({ key: args.key, model: args.model });
+    return;
   }
 
-  if (args.model !== null) {
-    patch.model = resolveModel(args.model, CX_MODEL_MAP, CX_DEFAULT_MODEL_ALIAS);
+  if (isOpenclawCommand(args.tool) && (args.key || args.model)) {
+    clearScreen();
+    printLogo();
+    openclaw.quickSet({ key: args.key, model: args.model });
+    return;
   }
 
-  if (args.url && String(args.url).trim()) {
-    patch.baseUrl = String(args.url).trim();
+  if (isOpencodeCommand(args.tool) && (args.key || args.model)) {
+    clearScreen();
+    printLogo();
+    await opencode.quickSet({ key: args.key, model: args.model });
+    return;
   }
 
-  if (Object.keys(patch).length === 0) {
-    console.log('');
-    console.log('未指定任何配置项，请使用 --key / --model / --url');
-    console.log('');
+  if (isCodexCommand(args.tool) && (args.key || args.url || args.model)) {
+    clearScreen();
+    printLogo();
+    await codex.quickSet({ key: args.key, url: args.url, model: args.model });
     return;
   }
 
-  applyCodexConfig(patch);
+  if (isCcCommand(args.tool)) {
+    await claude.run();
+    return;
+  }
 
-  console.log('');
-  console.log('Codex 配置已更新：');
-  if (patch.apiKey) console.log(`  API Key: ${maskApiKey(patch.apiKey)}`);
-  if (patch.baseUrl) console.log(`  Base URL: ${patch.baseUrl}`);
-  if (patch.model) console.log(`  Model: ${patch.model}`);
-  console.log(`  Config: ${CODEX_CONFIG_PATH}`);
-  console.log(`  Auth:   ${CODEX_AUTH_PATH}`);
-  console.log('');
-}
+  if (isAtomcodeCommand(args.tool)) {
+    await atomcode.run();
+    return;
+  }
 
-async function run() {
-  const args = parseArgs(process.argv);
+  if (isHermesCommand(args.tool)) {
+    await hermes.run();
+    return;
+  }
 
-  if (args.help || args.tool === null) {
-    printHelp();
+  if (isOpenclawCommand(args.tool)) {
+    await openclaw.run();
     return;
   }
 
-  validateArgs(args);
+  if (isOpencodeCommand(args.tool)) {
+    await opencode.run();
+    return;
+  }
 
-  if (isCcCommand(args.tool)) {
-    await runCc(args);
-  } else if (isCodexCommand(args.tool)) {
-    await runCodex(args);
+  if (isCodexCommand(args.tool)) {
+    await codex.run();
+    return;
   }
+
+  await mainMenu();
 }
 
-module.exports = run;
+module.exports = { run };

package/src/codex-config.js

@@ -1,18 +1,26 @@
 'use strict';
 
 const fs = require('fs');
-const os = require('os');
 const path = require('path');
+const os = require('os');
 const TOML = require('@iarna/toml');
+const {
+  CODEX_OPENTOKEN_BASE_URL,
+  CODEX_PROVIDER_KEY,
+  CODEX_AUTH_KEY,
+  CODEX_DEFAULT_MODEL,
+} = require('./models');
 
 const CONFIG_DIR = path.join(os.homedir(), '.codex');
 const CONFIG_PATH = path.join(CONFIG_DIR, 'config.toml');
 const AUTH_PATH = path.join(CONFIG_DIR, 'auth.json');
-
-const PROVIDER_KEY = 'opentoken';
-const AUTH_KEY = 'OPENAI_API_KEY';
 const MODEL_AVAILABILITY_LEVEL = 4;
 
+function readRaw() {
+  if (!fs.existsSync(CONFIG_PATH)) return '';
+  return fs.readFileSync(CONFIG_PATH, 'utf-8');
+}
+
 function readAuthFile() {
   if (!fs.existsSync(AUTH_PATH)) return {};
   try {
@@ -26,6 +34,21 @@ function readAuthFile() {
   }
 }
 
+function readApiKey() {
+  const auth = readAuthFile();
+  const fromAuth = auth[CODEX_AUTH_KEY];
+  if (fromAuth != null && String(fromAuth).trim()) {
+    return String(fromAuth).trim();
+  }
+  if (process.env[CODEX_AUTH_KEY] && String(process.env[CODEX_AUTH_KEY]).trim()) {
+    return String(process.env[CODEX_AUTH_KEY]).trim();
+  }
+  if (process.env.OPENTOKEN_API_KEY && String(process.env.OPENTOKEN_API_KEY).trim()) {
+    return String(process.env.OPENTOKEN_API_KEY).trim();
+  }
+  return '';
+}
+
 function applyCodexAuth(apiKey) {
   const key = apiKey == null ? '' : String(apiKey).trim();
   if (!key) return;
@@ -34,7 +57,7 @@ function applyCodexAuth(apiKey) {
   const data = {
     ...prev,
     auth_mode: 'apikey',
-    [AUTH_KEY]: key,
+    [CODEX_AUTH_KEY]: key,
   };
 
   if (!fs.existsSync(CONFIG_DIR)) {
@@ -51,8 +74,7 @@ function ensurePlainObject(parent, key) {
 }
 
 function parseConfig() {
-  if (!fs.existsSync(CONFIG_PATH)) return {};
-  const raw = fs.readFileSync(CONFIG_PATH, 'utf-8');
+  const raw = readRaw();
   if (!raw || !String(raw).trim()) return {};
   try {
     const data = TOML.parse(raw);
@@ -72,15 +94,14 @@ function getCurrentCodexConfig() {
       ? data.model_providers
       : {};
   const provider =
-    providers[PROVIDER_KEY] && typeof providers[PROVIDER_KEY] === 'object' && !Array.isArray(providers[PROVIDER_KEY])
-      ? providers[PROVIDER_KEY]
+    providers[CODEX_PROVIDER_KEY] &&
+    typeof providers[CODEX_PROVIDER_KEY] === 'object' &&
+    !Array.isArray(providers[CODEX_PROVIDER_KEY])
+      ? providers[CODEX_PROVIDER_KEY]
       : {};
 
-  const auth = readAuthFile();
-  const apiKey = auth[AUTH_KEY] || '';
-
   return {
-    apiKey,
+    apiKey: readApiKey(),
     baseUrl: provider.base_url != null ? String(provider.base_url) : '',
     model: data.model != null ? String(data.model) : '',
   };
@@ -105,8 +126,10 @@ function applyCodexConfig(patch) {
 
   const providers = ensurePlainObject(data, 'model_providers');
   const prevProvider =
-    providers[PROVIDER_KEY] && typeof providers[PROVIDER_KEY] === 'object' && !Array.isArray(providers[PROVIDER_KEY])
-      ? { ...providers[PROVIDER_KEY] }
+    providers[CODEX_PROVIDER_KEY] &&
+    typeof providers[CODEX_PROVIDER_KEY] === 'object' &&
+    !Array.isArray(providers[CODEX_PROVIDER_KEY])
+      ? { ...providers[CODEX_PROVIDER_KEY] }
       : {};
 
   const nextModel =
@@ -114,24 +137,24 @@ function applyCodexConfig(patch) {
       ? String(model).trim()
       : data.model != null && String(data.model).trim()
         ? String(data.model).trim()
-        : 'gpt-5.4';
+        : CODEX_DEFAULT_MODEL;
   const nextBaseUrl =
     baseUrl !== undefined && String(baseUrl).trim()
       ? String(baseUrl).trim()
       : prevProvider.base_url != null && String(prevProvider.base_url).trim()
         ? String(prevProvider.base_url).trim()
-        : 'https://gw.opentoken.io/v1';
+        : CODEX_OPENTOKEN_BASE_URL;
 
   data.model = nextModel;
-  data.model_provider = PROVIDER_KEY;
+  data.model_provider = CODEX_PROVIDER_KEY;
   const nextProvider = {
     ...prevProvider,
-    name: 'OpenToken',
+    name: 'Opentoken',
     base_url: nextBaseUrl,
     wire_api: 'responses',
   };
   delete nextProvider.env_key;
-  providers[PROVIDER_KEY] = nextProvider;
+  providers[CODEX_PROVIDER_KEY] = nextProvider;
 
   const profileModels = uniqueModels(models, nextModel);
   const profiles = ensurePlainObject(data, 'profiles');
@@ -142,7 +165,7 @@ function applyCodexConfig(patch) {
         : {};
     profiles[id] = {
       ...prevProfile,
-      model_provider: PROVIDER_KEY,
+      model_provider: CODEX_PROVIDER_KEY,
       model: id,
     };
   });
@@ -166,7 +189,6 @@ function applyCodexConfig(patch) {
 module.exports = {
   CONFIG_PATH,
   AUTH_PATH,
-  PROVIDER_KEY,
   getCurrentCodexConfig,
   applyCodexConfig,
   applyCodexAuth,