@tokenaut/opentoken 1.3.6 → 1.4.1

package/src/tools/claude.js

@@ -0,0 +1,302 @@
+'use strict';
+
+const chalk = require('chalk');
+const inquirer = require('inquirer');
+const { API_NODES, DEFAULT_BASE_URL } = require('../models');
+const { fetchModelsForBaseUrl } = require('../fetch-models');
+const { getCurrentConfig, applyConfig } = require('../config');
+const { printLogo } = require('../logo');
+const { clearScreen } = require('../screen');
+
+let pendingConfig = { apiKey: null, baseUrl: null, model: null };
+
+function resetPending() {
+  const disk = getCurrentConfig();
+  pendingConfig = {
+    apiKey: null,
+    baseUrl: disk.baseUrl ? null : DEFAULT_BASE_URL,
+    model: null,
+  };
+}
+
+function maskApiKey(key) {
+  if (!key || key.length < 8) return key || '';
+  return key.slice(0, 6) + '****' + key.slice(-4);
+}
+
+async function handleConfigApiKey() {
+  const { apiKey } = await inquirer.prompt([
+    {
+      type: 'password',
+      name: 'apiKey',
+      prefix: '',
+      message: '请输入 API Key：',
+      mask: '*',
+    },
+  ]);
+
+  if (apiKey && apiKey.trim()) {
+    pendingConfig.apiKey = apiKey.trim();
+    console.log(chalk.green('  ✓ API Key 已更新'));
+  } else {
+    console.log(chalk.gray('  （留空，保持不变）'));
+  }
+  console.log('');
+}
+
+async function handleConfigBaseUrl() {
+  const { baseUrl: existing } = getCurrentConfig();
+  const current = pendingConfig.baseUrl !== null ? pendingConfig.baseUrl : existing;
+
+  const nodeChoices = API_NODES.map((n) => ({
+    ...n,
+    name: current === n.value ? chalk.green('► ' + n.name) : '  ' + n.name,
+  }));
+
+  const { selected } = await inquirer.prompt([
+    {
+      type: 'list',
+      name: 'selected',
+      prefix: '',
+      message: '请选择 Base URL：',
+      choices: nodeChoices,
+      default: current || API_NODES[0].value,
+    },
+  ]);
+
+  if (selected === '__custom__') {
+    const { customUrl } = await inquirer.prompt([
+      {
+        type: 'input',
+        name: 'customUrl',
+        prefix: '',
+        message: '请输入自定义 Base URL：',
+        default: current && current !== '__custom__' ? current : DEFAULT_BASE_URL,
+        validate: (v) => v.trim().startsWith('http') || '请输入合法的 URL（以 http 开头）',
+      },
+    ]);
+    pendingConfig.baseUrl = customUrl.trim();
+    console.log(chalk.green('  ✓ Base URL 已更新：' + pendingConfig.baseUrl));
+  } else {
+    pendingConfig.baseUrl = selected;
+    console.log(chalk.green('  ✓ Base URL 已更新：' + selected));
+  }
+  console.log('');
+}
+
+function getEffectiveClaudeEnv() {
+  const disk = getCurrentConfig();
+  return {
+    apiKey: pendingConfig.apiKey !== null ? pendingConfig.apiKey : disk.apiKey,
+    baseUrl: pendingConfig.baseUrl !== null ? pendingConfig.baseUrl : disk.baseUrl,
+    model: pendingConfig.model !== null ? pendingConfig.model : disk.model,
+  };
+}
+
+function grayParen(text) {
+  return chalk.gray('（' + text + '）');
+}
+
+function buildClaudeMenuChoices() {
+  const ev = getEffectiveClaudeEnv();
+  const keyShow = ev.apiKey ? maskApiKey(ev.apiKey) : '未配置';
+  const urlShow = ev.baseUrl ? ev.baseUrl : '未配置';
+  const modelShow = ev.model ? ev.model : '未配置';
+
+  return [
+    { name: '1. 配置 Base URL' + grayParen(urlShow), value: 'apiNode' },
+    { name: '2. 配置 API Key' + grayParen(keyShow), value: 'apiKey' },
+    { name: '3. 配置模型' + grayParen(modelShow), value: 'model' },
+    new inquirer.Separator(),
+    { name: '4. 应用配置并保存', value: 'apply' },
+    new inquirer.Separator(),
+    { name: '← 返回上级菜单', value: 'back' },
+  ];
+}
+
+async function handleConfigModel() {
+  const { model: existing, baseUrl, apiKey } = getEffectiveClaudeEnv();
+  const current = existing || '';
+
+  if (!baseUrl) {
+    console.log(chalk.yellow('  请先配置 Base URL，否则无法拉取模型列表。'));
+    const { modelId } = await inquirer.prompt([
+      {
+        type: 'input',
+        name: 'modelId',
+        prefix: '',
+        message: '或直接输入模型 ID：',
+        default: current,
+        validate: (v) => (v && v.trim() ? true : '请输入模型 ID'),
+      },
+    ]);
+    pendingConfig.model = modelId.trim();
+    applyConfig({ model: pendingConfig.model });
+    console.log(chalk.green('  ✓ 模型已更新：' + pendingConfig.model));
+    console.log(
+      chalk.gray(
+        '  （已写入 settings.json，并合并 API_TIMEOUT_MS 等默认环境变量）',
+      ),
+    );
+    console.log('');
+    return;
+  }
+
+  const fetched = await fetchModelsForBaseUrl(baseUrl, apiKey, {
+    filterAnthropicCompatible: true,
+  });
+
+  if (fetched.ok && fetched.models.length > 0) {
+    let ids = [...fetched.models];
+    if (current && !ids.includes(current)) {
+      ids = [current, ...ids];
+    }
+
+    const modelChoices = ids.map((id) => ({
+      name: current === id ? chalk.green('► ' + id) : '  ' + id,
+      value: id,
+    }));
+    modelChoices.push(new inquirer.Separator());
+    modelChoices.push({ name: '  手动输入其他模型 ID…', value: '__manual__' });
+
+    const defaultValue = current && ids.includes(current) ? current : ids[0];
+
+    const { model } = await inquirer.prompt([
+      {
+        type: 'list',
+        name: 'model',
+        prefix: '',
+        message: '请选择模型：',
+        choices: modelChoices,
+        default: defaultValue,
+        pageSize: 15,
+      },
+    ]);
+
+    if (model === '__manual__') {
+      const { modelId } = await inquirer.prompt([
+        {
+          type: 'input',
+          name: 'modelId',
+          prefix: '',
+          message: '请输入模型 ID：',
+          default: current || '',
+          validate: (v) => (v && v.trim() ? true : '请输入模型 ID'),
+        },
+      ]);
+      pendingConfig.model = modelId.trim();
+    } else {
+      pendingConfig.model = model;
+    }
+  } else {
+    console.log(chalk.yellow('  无法拉取模型列表：' + (fetched.error || '未知错误')));
+    console.log(chalk.gray('  请检查 Base URL、网络或 API Key 后重试，或直接输入模型 ID。'));
+    const { modelId } = await inquirer.prompt([
+      {
+        type: 'input',
+        name: 'modelId',
+        prefix: '',
+        message: '请输入模型 ID：',
+        default: current,
+        validate: (v) => (v && v.trim() ? true : '请输入模型 ID'),
+      },
+    ]);
+    pendingConfig.model = modelId.trim();
+  }
+
+  applyConfig({ model: pendingConfig.model });
+  console.log(chalk.green('  ✓ 模型已更新：' + pendingConfig.model));
+  console.log(
+    chalk.gray(
+      '  （已写入 settings.json，并合并 API_TIMEOUT_MS 等默认环境变量）',
+    ),
+  );
+  console.log('');
+}
+
+function doApplyConfig() {
+  const toWrite = {};
+  if (pendingConfig.apiKey  !== null) toWrite.apiKey  = pendingConfig.apiKey;
+  if (pendingConfig.baseUrl !== null) toWrite.baseUrl = pendingConfig.baseUrl;
+  if (pendingConfig.model   !== null) toWrite.model   = pendingConfig.model;
+
+  console.log('');
+  if (Object.keys(toWrite).length > 0) {
+    applyConfig(toWrite);
+    console.log(chalk.green('  ✓ 配置已成功写入'));
+  } else {
+    applyConfig({});
+    console.log(
+      chalk.gray(
+        '  （无 Key / URL / 模型变更；已同步 API_TIMEOUT_MS 等默认环境变量）',
+      ),
+    );
+  }
+  console.log('');
+}
+
+function paintClaudeScreen() {
+  clearScreen();
+  printLogo();
+}
+
+async function run() {
+  resetPending();
+
+  while (true) {
+    paintClaudeScreen();
+
+    const { action } = await inquirer.prompt([
+      {
+        type: 'list',
+        name: 'action',
+        prefix: '',
+        message: '请选择操作：',
+        choices: buildClaudeMenuChoices(),
+        pageSize: 10,
+      },
+    ]);
+
+    switch (action) {
+      case 'apiNode':
+        paintClaudeScreen();
+        await handleConfigBaseUrl();
+        break;
+      case 'apiKey':
+        paintClaudeScreen();
+        await handleConfigApiKey();
+        break;
+      case 'model':
+        paintClaudeScreen();
+        await handleConfigModel();
+        break;
+      case 'apply':
+        doApplyConfig();
+        return;
+      case 'back':
+        return;
+    }
+  }
+}
+
+function quickSet({ key, url, model }) {
+  const toWrite = {};
+  if (key)   toWrite.apiKey  = key;
+  if (url)   toWrite.baseUrl = url;
+  if (model) toWrite.model   = model;
+
+  if (Object.keys(toWrite).length === 0) return false;
+
+  applyConfig(toWrite);
+
+  console.log('');
+  console.log(chalk.bold(chalk.cyan('Claude Code 配置已更新：')));
+  if (key)   console.log(chalk.green('  ✓ API Key  : ') + maskApiKey(key));
+  if (url)   console.log(chalk.green('  ✓ Base URL : ') + url);
+  if (model) console.log(chalk.green('  ✓ 模型     : ') + model);
+  console.log('');
+
+  return true;
+}
+
+module.exports = { run, quickSet };

package/src/tools/codex.js

@@ -0,0 +1,327 @@
+'use strict';
+
+const chalk = require('chalk');
+const inquirer = require('inquirer');
+const { CODEX_OPENTOKEN_BASE_URL, CODEX_AUTH_KEY } = require('../models');
+const { fetchModelsForBaseUrl } = require('../fetch-models');
+const {
+  CONFIG_PATH,
+  AUTH_PATH,
+  getCurrentCodexConfig,
+  applyCodexConfig,
+} = require('../codex-config');
+const { printLogo } = require('../logo');
+const { clearScreen } = require('../screen');
+
+let pendingConfig = { apiKey: null, baseUrl: null, model: null, models: null };
+
+function resetPending() {
+  const disk = getCurrentCodexConfig();
+  pendingConfig = {
+    apiKey: null,
+    baseUrl: disk.baseUrl ? null : CODEX_OPENTOKEN_BASE_URL,
+    model: null,
+    models: null,
+  };
+}
+
+function maskApiKey(key) {
+  if (!key || key.length < 8) return key || '';
+  return key.slice(0, 6) + '****' + key.slice(-4);
+}
+
+function getEffectiveCodexEnv() {
+  const disk = getCurrentCodexConfig();
+  return {
+    apiKey: pendingConfig.apiKey !== null ? pendingConfig.apiKey : disk.apiKey,
+    baseUrl: pendingConfig.baseUrl !== null ? pendingConfig.baseUrl : disk.baseUrl,
+    model: pendingConfig.model !== null ? pendingConfig.model : disk.model,
+  };
+}
+
+function grayParen(text) {
+  return chalk.gray('（' + text + '）');
+}
+
+function buildCodexMenuChoices() {
+  const ev = getEffectiveCodexEnv();
+  const keyShow = ev.apiKey ? maskApiKey(ev.apiKey) : '未配置';
+  const urlShow = ev.baseUrl ? ev.baseUrl : CODEX_OPENTOKEN_BASE_URL;
+  const modelShow = ev.model ? ev.model : '未配置';
+
+  return [
+    { name: '1. 配置 Base URL' + grayParen(urlShow), value: 'baseUrl' },
+    { name: '2. 输入 API Key' + grayParen(keyShow), value: 'apiKey' },
+    { name: '3. 配置模型' + grayParen(modelShow), value: 'model' },
+    new inquirer.Separator(),
+    { name: '4. 应用配置并保存', value: 'apply' },
+    new inquirer.Separator(),
+    { name: '← 返回上级菜单', value: 'back' },
+  ];
+}
+
+async function handleConfigBaseUrl() {
+  const { baseUrl: existing } = getEffectiveCodexEnv();
+  const { baseUrl } = await inquirer.prompt([
+    {
+      type: 'input',
+      name: 'baseUrl',
+      prefix: '',
+      message: '请输入 Codex Base URL（需包含 /v1）：',
+      default: existing || CODEX_OPENTOKEN_BASE_URL,
+      validate: (v) => v.trim().startsWith('http') || '请输入合法的 URL（以 http 开头）',
+    },
+  ]);
+
+  pendingConfig.baseUrl = baseUrl.trim();
+  console.log(chalk.green('  ✓ Base URL 已更新：' + pendingConfig.baseUrl));
+  console.log('');
+}
+
+async function handleConfigApiKey() {
+  const { apiKey } = await inquirer.prompt([
+    {
+      type: 'password',
+      name: 'apiKey',
+      prefix: '',
+      message: '请输入 Opentoken API Key（将写入 ' + AUTH_PATH + '）：',
+      mask: '*',
+    },
+  ]);
+
+  if (apiKey && apiKey.trim()) {
+    pendingConfig.apiKey = apiKey.trim();
+    console.log(chalk.green('  ✓ API Key 已记录（保存时将写入 auth.json）'));
+  } else {
+    console.log(chalk.gray('  （留空，保持不变）'));
+  }
+  console.log('');
+}
+
+async function handleConfigModel() {
+  const { model: existing, baseUrl, apiKey } = getEffectiveCodexEnv();
+  const current = existing || '';
+  const listBase = baseUrl || CODEX_OPENTOKEN_BASE_URL;
+
+  if (!apiKey) {
+    console.log(chalk.yellow('  未配置 API Key，无法拉取 responses 模型列表。'));
+    const { modelId } = await inquirer.prompt([
+      {
+        type: 'input',
+        name: 'modelId',
+        prefix: '',
+        message: '或直接输入模型 ID：',
+        default: current,
+        validate: (v) => (v && v.trim() ? true : '请输入模型 ID'),
+      },
+    ]);
+    pendingConfig.model = modelId.trim();
+    pendingConfig.models = [pendingConfig.model];
+    applyCodexConfig({
+      baseUrl: listBase,
+      model: pendingConfig.model,
+      models: pendingConfig.models,
+      apiKey: apiKey || undefined,
+    });
+    console.log(chalk.green('  ✓ 模型已更新：' + pendingConfig.model));
+    console.log(chalk.gray('  （已写入 ' + CONFIG_PATH + '）'));
+    console.log('');
+    return;
+  }
+
+  const fetched = await fetchModelsForBaseUrl(listBase, apiKey, {
+    requiredEndpointType: 'responses',
+  });
+
+  if (fetched.ok && fetched.models.length > 0) {
+    let ids = [...fetched.models];
+    if (current && !ids.includes(current)) {
+      ids = [current, ...ids];
+    }
+
+    const modelChoices = ids.map((id) => ({
+      name: current === id ? chalk.green('► ' + id) : '  ' + id,
+      value: id,
+    }));
+    modelChoices.push(new inquirer.Separator());
+    modelChoices.push({ name: '  手动输入其他模型 ID…', value: '__manual__' });
+
+    const defaultValue = current && ids.includes(current) ? current : ids[0];
+
+    const { model } = await inquirer.prompt([
+      {
+        type: 'list',
+        name: 'model',
+        prefix: '',
+        message: '请选择 Codex responses 模型：',
+        choices: modelChoices,
+        default: defaultValue,
+        pageSize: 15,
+      },
+    ]);
+
+    if (model === '__manual__') {
+      const { modelId } = await inquirer.prompt([
+        {
+          type: 'input',
+          name: 'modelId',
+          prefix: '',
+          message: '请输入模型 ID：',
+          default: current || '',
+          validate: (v) => (v && v.trim() ? true : '请输入模型 ID'),
+        },
+      ]);
+      pendingConfig.model = modelId.trim();
+    } else {
+      pendingConfig.model = model;
+    }
+    pendingConfig.models = ids;
+  } else {
+    console.log(chalk.yellow('  无法拉取 responses 模型列表：' + (fetched.error || '未知错误')));
+    console.log(chalk.gray('  请检查 Base URL、网络或 API Key 后重试，或直接输入模型 ID。'));
+    const { modelId } = await inquirer.prompt([
+      {
+        type: 'input',
+        name: 'modelId',
+        prefix: '',
+        message: '请输入模型 ID：',
+        default: current,
+        validate: (v) => (v && v.trim() ? true : '请输入模型 ID'),
+      },
+    ]);
+    pendingConfig.model = modelId.trim();
+    pendingConfig.models = [pendingConfig.model];
+  }
+
+  applyCodexConfig({
+    baseUrl: listBase,
+    model: pendingConfig.model,
+    models: pendingConfig.models,
+    apiKey: apiKey || undefined,
+  });
+  console.log(chalk.green('  ✓ 模型已更新：' + pendingConfig.model));
+  console.log(chalk.gray('  （已写入 ' + CONFIG_PATH + '）'));
+  console.log('');
+}
+
+function printAuthHint(hasKey) {
+  console.log(
+    chalk.gray(
+      '  API Key 保存在 ' +
+        AUTH_PATH +
+        '（' +
+        CODEX_AUTH_KEY +
+        '）' +
+        (hasKey ? '，可直接运行 codex 或使用 IDE 插件。' : '；保存配置时请填写 API Key。'),
+    ),
+  );
+}
+
+function doApplyConfig() {
+  const ev = getEffectiveCodexEnv();
+  const toWrite = {};
+  if (pendingConfig.baseUrl !== null) toWrite.baseUrl = pendingConfig.baseUrl;
+  if (pendingConfig.model !== null) toWrite.model = pendingConfig.model;
+  if (pendingConfig.models !== null) toWrite.models = pendingConfig.models;
+  if (pendingConfig.apiKey !== null && pendingConfig.apiKey) {
+    toWrite.apiKey = pendingConfig.apiKey;
+  }
+
+  console.log('');
+  if (Object.keys(toWrite).length > 0) {
+    applyCodexConfig(toWrite);
+    console.log(chalk.green('  ✓ 配置已成功写入'));
+  } else {
+    applyCodexConfig({});
+    console.log(chalk.gray('  （无 URL / 模型变更；已确保 Opentoken responses provider 存在）'));
+  }
+  console.log(chalk.gray('  ' + CONFIG_PATH));
+  if (toWrite.apiKey) {
+    console.log(chalk.gray('  ' + AUTH_PATH));
+  }
+  printAuthHint(Boolean(ev.apiKey || toWrite.apiKey));
+  console.log('');
+}
+
+function paintScreen() {
+  clearScreen();
+  printLogo();
+}
+
+async function run() {
+  resetPending();
+
+  while (true) {
+    paintScreen();
+
+    const { action } = await inquirer.prompt([
+      {
+        type: 'list',
+        name: 'action',
+        prefix: '',
+        message: 'Codex 配置（~/.codex/config.toml）：',
+        choices: buildCodexMenuChoices(),
+        pageSize: 12,
+      },
+    ]);
+
+    switch (action) {
+      case 'baseUrl':
+        paintScreen();
+        await handleConfigBaseUrl();
+        break;
+      case 'apiKey':
+        paintScreen();
+        await handleConfigApiKey();
+        break;
+      case 'model':
+        paintScreen();
+        await handleConfigModel();
+        break;
+      case 'apply':
+        doApplyConfig();
+        return;
+      case 'back':
+        return;
+    }
+  }
+}
+
+async function quickSet({ key, url, model }) {
+  const toWrite = {};
+  if (url) toWrite.baseUrl = url;
+  if (model) toWrite.model = model;
+  if (key) toWrite.apiKey = key;
+
+  if (key) {
+    const listBase = url || getCurrentCodexConfig().baseUrl || CODEX_OPENTOKEN_BASE_URL;
+    const fetched = await fetchModelsForBaseUrl(listBase, key, {
+      requiredEndpointType: 'responses',
+    });
+    if (fetched.ok && fetched.models.length > 0) {
+      toWrite.models = fetched.models;
+      if (!toWrite.model) toWrite.model = fetched.models[0];
+    }
+  }
+
+  if (Object.keys(toWrite).length === 0) return false;
+
+  applyCodexConfig(toWrite);
+
+  console.log('');
+  console.log(chalk.bold(chalk.cyan('Codex 配置已更新：')));
+  if (key) console.log(chalk.green('  ✓ API Key  : ') + maskApiKey(key));
+  if (url) console.log(chalk.green('  ✓ Base URL : ') + url);
+  if (toWrite.model) console.log(chalk.green('  ✓ 模型     : ') + toWrite.model);
+  if (toWrite.models) {
+    console.log(chalk.gray('  （包含 ' + toWrite.models.length + ' 个 responses 模型）'));
+  }
+  console.log(chalk.gray('  ' + CONFIG_PATH));
+  if (key) console.log(chalk.gray('  ' + AUTH_PATH));
+  printAuthHint(Boolean(key || getCurrentCodexConfig().apiKey));
+  console.log('');
+
+  return true;
+}
+
+module.exports = { run, quickSet };