@tokenaut/opentoken 1.3.5 → 1.4.0

package/src/tools/hermes.js

@@ -0,0 +1,274 @@
+'use strict';
+
+const chalk = require('chalk');
+const inquirer = require('inquirer');
+const { DEFAULT_BASE_URL, HERMES_OPENTOKEN_BASE_URL } = require('../models');
+const { fetchModelsForBaseUrl } = require('../fetch-models');
+const {
+  CONFIG_YAML_PATH,
+  ENV_PATH,
+  getCurrentHermesConfig,
+  applyHermesConfig,
+} = require('../hermes-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 getEffectiveHermesEnv() {
+  const disk = getCurrentHermesConfig();
+  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 buildHermesMenuChoices() {
+  const ev = getEffectiveHermesEnv();
+  const keyShow = ev.apiKey ? maskApiKey(ev.apiKey) : '未配置';
+  const modelShow = ev.model ? ev.model : '未配置';
+
+  return [
+    new inquirer.Separator(
+      chalk.gray(
+        '  Opentoken（custom provider）: ' +
+          HERMES_OPENTOKEN_BASE_URL +
+          ' · Opentoken API KEY → ' +
+          ENV_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（写入 ~/.hermes/.env）：',
+      mask: '*',
+    },
+  ]);
+
+  if (apiKey && apiKey.trim()) {
+    pendingConfig.apiKey = apiKey.trim();
+    console.log(chalk.green('  ✓ Opentoken API KEY 已更新（待应用后写入 .env）'));
+  } else {
+    console.log(chalk.gray('  （留空，保持不变）'));
+  }
+  console.log('');
+}
+
+async function handleConfigModel() {
+  const { model: existing, apiKey } = getEffectiveHermesEnv();
+  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();
+    applyHermesConfig({ model: pendingConfig.model });
+    console.log(chalk.green('  ✓ 模型已更新：' + pendingConfig.model));
+    console.log(
+      chalk.gray(
+        '  （已写入 ' +
+          CONFIG_YAML_PATH +
+          '，provider=custom，base_url=' +
+          HERMES_OPENTOKEN_BASE_URL +
+          '）',
+      ),
+    );
+    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;
+    }
+  } 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();
+  }
+
+  applyHermesConfig({ model: pendingConfig.model });
+  console.log(chalk.green('  ✓ 模型已更新：' + pendingConfig.model));
+  console.log(
+    chalk.gray(
+      '  （已写入 ' +
+        CONFIG_YAML_PATH +
+        '，provider=custom，base_url=' +
+        HERMES_OPENTOKEN_BASE_URL +
+        '）',
+    ),
+  );
+  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) {
+    applyHermesConfig(toWrite);
+    console.log(chalk.green('  ✓ 配置已成功写入'));
+    console.log(chalk.gray('  ' + CONFIG_YAML_PATH));
+    console.log(chalk.gray('  ' + ENV_PATH));
+  } else {
+    applyHermesConfig({});
+    console.log(
+      chalk.gray(
+        '  （无 Key / 模型变更；已确保 provider=custom 与 Opentoken base_url）',
+      ),
+    );
+    console.log(chalk.gray('  ' + CONFIG_YAML_PATH));
+    console.log(chalk.gray('  ' + ENV_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: 'Hermes 配置（~/.hermes/config.yaml · ~/.hermes/.env）：',
+        choices: buildHermesMenuChoices(),
+        pageSize: 14,
+      },
+    ]);
+
+    switch (action) {
+      case 'apiKey':
+        paintScreen();
+        await handleConfigApiKey();
+        break;
+      case 'model':
+        paintScreen();
+        await handleConfigModel();
+        break;
+      case 'apply':
+        doApplyConfig();
+        return;
+      case 'back':
+        return;
+    }
+  }
+}
+
+function quickSet({ key, model }) {
+  const toWrite = {};
+  if (key) toWrite.apiKey = key;
+  if (model) toWrite.model = model;
+
+  if (Object.keys(toWrite).length === 0) return false;
+
+  applyHermesConfig(toWrite);
+
+  console.log('');
+  console.log(chalk.bold(chalk.cyan('Hermes 配置已更新：')));
+  if (key) console.log(chalk.green('  ✓ Opentoken API KEY（.env）: ') + maskApiKey(key));
+  if (model) console.log(chalk.green('  ✓ model.default       : ') + model);
+  console.log(chalk.green('  ✓ provider / base_url : ') + 'custom · ' + HERMES_OPENTOKEN_BASE_URL);
+  console.log(chalk.gray('  ' + CONFIG_YAML_PATH));
+  console.log(chalk.gray('  ' + ENV_PATH));
+  console.log('');
+
+  return true;
+}
+
+module.exports = { run, quickSet };

package/src/tools/openclaw.js

@@ -0,0 +1,244 @@
+'use strict';
+
+const chalk = require('chalk');
+const inquirer = require('inquirer');
+const { DEFAULT_BASE_URL, OPENCLAW_PROVIDER_KEY } = require('../models');
+const { fetchModelsForBaseUrl } = require('../fetch-models');
+const {
+  OPENCLAW_JSON_PATH,
+  getCurrentOpenclawOpentokenConfig,
+  applyOpenclawConfig,
+} = require('../openclaw-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 getEffectiveOpenclawEnv() {
+  const disk = getCurrentOpenclawOpentokenConfig();
+  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 buildOpenclawMenuChoices() {
+  const ev = getEffectiveOpenclawEnv();
+  const keyShow = ev.apiKey ? maskApiKey(ev.apiKey) : '未配置';
+  const modelShow = ev.model ? OPENCLAW_PROVIDER_KEY + '/' + ev.model : '未配置';
+
+  return [
+    new inquirer.Separator(chalk.gray('  ' + OPENCLAW_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 } = getEffectiveOpenclawEnv();
+  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();
+    applyOpenclawConfig({ model: pendingConfig.model });
+    console.log(chalk.green('  ✓ 模型已更新：' + OPENCLAW_PROVIDER_KEY + '/' + pendingConfig.model));
+    console.log(chalk.gray('  （已写入 ' + OPENCLAW_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;
+    }
+  } 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();
+  }
+
+  applyOpenclawConfig({ model: pendingConfig.model });
+  console.log(chalk.green('  ✓ 模型已更新：' + OPENCLAW_PROVIDER_KEY + '/' + pendingConfig.model));
+  console.log(chalk.gray('  （已写入 ' + OPENCLAW_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) {
+    applyOpenclawConfig(toWrite);
+    console.log(chalk.green('  ✓ 配置已成功写入'));
+    console.log(chalk.gray('  ' + OPENCLAW_JSON_PATH));
+  } else {
+    applyOpenclawConfig({});
+    console.log(chalk.gray('  （无 Key / 模型变更）'));
+    console.log(chalk.gray('  ' + OPENCLAW_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: 'OpenClaw 配置（' + OPENCLAW_JSON_PATH + '）：',
+        choices: buildOpenclawMenuChoices(),
+        pageSize: 14,
+      },
+    ]);
+
+    switch (action) {
+      case 'apiKey':
+        paintScreen();
+        await handleConfigApiKey();
+        break;
+      case 'model':
+        paintScreen();
+        await handleConfigModel();
+        break;
+      case 'apply':
+        doApplyConfig();
+        return;
+      case 'back':
+        return;
+    }
+  }
+}
+
+function quickSet({ key, model }) {
+  const toWrite = {};
+  if (key) toWrite.apiKey = key;
+  if (model) toWrite.model = model;
+
+  if (Object.keys(toWrite).length === 0) return false;
+
+  applyOpenclawConfig(toWrite);
+
+  console.log('');
+  console.log(chalk.bold(chalk.cyan('OpenClaw 配置已更新')));
+  if (key) console.log(chalk.green('  ✓ API Key: ') + maskApiKey(key));
+  if (model) {
+    console.log(chalk.green('  ✓ 模型: ') + OPENCLAW_PROVIDER_KEY + '/' + model);
+  }
+  console.log(chalk.gray('  ' + OPENCLAW_JSON_PATH));
+  console.log('');
+
+  return true;
+}
+
+module.exports = { run, quickSet };