@tokenaut/opentoken 1.3.5 → 1.4.0

package/src/tools/opencode.js

@@ -0,0 +1,258 @@
+'use strict';
+
+const chalk = require('chalk');
+const inquirer = require('inquirer');
+const { DEFAULT_BASE_URL, OPENCODE_PROVIDER_KEY } = require('../models');
+const { fetchModelsForBaseUrl } = require('../fetch-models');
+const {
+  OPENCODE_JSON_PATH,
+  getCurrentOpencodeOpentokenConfig,
+  applyOpencodeConfig,
+} = require('../opencode-config');
+const { printLogo } = require('../logo');
+const { clearScreen } = require('../screen');
+
+let pendingConfig = { apiKey: null, model: null };
+
+function resetPending() {
+  pendingConfig = { apiKey: null, model: null };
+}
+
+function maskApiKey(key) {
+  if (!key || key.length < 8) return key || '';
+  return key.slice(0, 6) + '****' + key.slice(-4);
+}
+
+function getEffectiveOpencodeEnv() {
+  const disk = getCurrentOpencodeOpentokenConfig();
+  return {
+    apiKey: pendingConfig.apiKey !== null ? pendingConfig.apiKey : disk.apiKey,
+    model: pendingConfig.model !== null ? pendingConfig.model : disk.model,
+  };
+}
+
+function grayParen(text) {
+  return chalk.gray('（' + text + '）');
+}
+
+function buildOpencodeMenuChoices() {
+  const ev = getEffectiveOpencodeEnv();
+  const keyShow = ev.apiKey ? maskApiKey(ev.apiKey) : '未配置';
+  const modelShow = ev.model ? ev.model : '未配置';
+
+  return [
+    new inquirer.Separator(chalk.gray('  ' + OPENCODE_JSON_PATH)),
+    { name: '1. 配置 API Key' + grayParen(keyShow), value: 'apiKey' },
+    { name: '2. 配置模型' + grayParen(modelShow), value: 'model' },
+    new inquirer.Separator(),
+    { name: '3. 应用配置并保存', value: 'apply' },
+    new inquirer.Separator(),
+    { name: '← 返回上级菜单', value: 'back' },
+  ];
+}
+
+async function handleConfigApiKey() {
+  const { apiKey } = await inquirer.prompt([
+    {
+      type: 'password',
+      name: 'apiKey',
+      prefix: '',
+      message: '请输入 Opentoken API KEY：',
+      mask: '*',
+    },
+  ]);
+
+  if (apiKey && apiKey.trim()) {
+    pendingConfig.apiKey = apiKey.trim();
+    console.log(chalk.green('  ✓ Opentoken API KEY 已更新（待应用后写入）'));
+  } else {
+    console.log(chalk.gray('  （留空，保持不变）'));
+  }
+  console.log('');
+}
+
+async function handleConfigModel() {
+  const { model: existing, apiKey } = getEffectiveOpencodeEnv();
+  const current = existing || '';
+  const listBase = DEFAULT_BASE_URL;
+
+  if (!apiKey) {
+    console.log(chalk.yellow('  请先配置 API Key，否则无法拉取模型列表。'));
+    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();
+    applyOpencodeConfig({ model: pendingConfig.model });
+    console.log(chalk.green('  ✓ 模型已更新：' + pendingConfig.model));
+    console.log(chalk.gray('  （已写入 ' + OPENCODE_JSON_PATH + '）'));
+    console.log('');
+    return;
+  }
+
+  const fetched = await fetchModelsForBaseUrl(listBase, apiKey);
+
+  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;
+    }
+
+    applyOpencodeConfig({ model: pendingConfig.model, models: ids });
+    console.log(chalk.green('  ✓ 模型已更新：' + pendingConfig.model));
+    console.log(chalk.gray('  （已写入 ' + OPENCODE_JSON_PATH + '，包含 ' + ids.length + ' 个可用模型）'));
+    console.log('');
+  } else {
+    console.log(chalk.yellow('  无法拉取模型列表：' + (fetched.error || '未知错误')));
+    console.log(chalk.gray('  请检查网络或 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();
+    applyOpencodeConfig({ model: pendingConfig.model });
+    console.log(chalk.green('  ✓ 模型已更新：' + pendingConfig.model));
+    console.log(chalk.gray('  （已写入 ' + OPENCODE_JSON_PATH + '）'));
+    console.log('');
+  }
+}
+
+function doApplyConfig() {
+  const toWrite = {};
+  if (pendingConfig.apiKey !== null) toWrite.apiKey = pendingConfig.apiKey;
+  if (pendingConfig.model !== null) toWrite.model = pendingConfig.model;
+
+  console.log('');
+  if (Object.keys(toWrite).length > 0) {
+    applyOpencodeConfig(toWrite);
+    console.log(chalk.green('  ✓ 配置已成功写入'));
+    console.log(chalk.gray('  ' + OPENCODE_JSON_PATH));
+  } else {
+    applyOpencodeConfig({});
+    console.log(chalk.gray('  （无 Key / 模型变更）'));
+    console.log(chalk.gray('  ' + OPENCODE_JSON_PATH));
+  }
+  console.log('');
+}
+
+function paintScreen() {
+  clearScreen();
+  printLogo();
+}
+
+async function run() {
+  resetPending();
+
+  while (true) {
+    paintScreen();
+
+    const { action } = await inquirer.prompt([
+      {
+        type: 'list',
+        name: 'action',
+        prefix: '',
+        message: 'OpenCode 配置（' + OPENCODE_JSON_PATH + '）：',
+        choices: buildOpencodeMenuChoices(),
+        pageSize: 14,
+      },
+    ]);
+
+    switch (action) {
+      case 'apiKey':
+        paintScreen();
+        await handleConfigApiKey();
+        break;
+      case 'model':
+        paintScreen();
+        await handleConfigModel();
+        break;
+      case 'apply':
+        doApplyConfig();
+        return;
+      case 'back':
+        return;
+    }
+  }
+}
+
+async function quickSet({ key, model }) {
+  const toWrite = {};
+  if (key) toWrite.apiKey = key;
+  if (model) toWrite.model = model;
+
+  if (Object.keys(toWrite).length === 0) return false;
+
+  if (key) {
+    const fetched = await fetchModelsForBaseUrl(DEFAULT_BASE_URL, key);
+    if (fetched.ok && fetched.models.length > 0) {
+      toWrite.models = fetched.models;
+    }
+  }
+
+  applyOpencodeConfig(toWrite);
+
+  console.log('');
+  console.log(chalk.bold(chalk.cyan('OpenCode 配置已更新')));
+  if (key) console.log(chalk.green('  ✓ API Key: ') + maskApiKey(key));
+  if (model) {
+    console.log(chalk.green('  ✓ 模型: ') + model);
+  }
+  if (toWrite.models) {
+    console.log(chalk.gray('  （包含 ' + toWrite.models.length + ' 个可用模型）'));
+  }
+  console.log(chalk.gray('  ' + OPENCODE_JSON_PATH));
+  console.log('');
+
+  return true;
+}
+
+module.exports = { run, quickSet };