@tokenaut/opentoken 1.3.5 → 1.4.0

package/src/openclaw-config.js

@@ -0,0 +1,174 @@
+'use strict';
+
+const fs = require('fs');
+const path = require('path');
+const os = require('os');
+const {
+  OPENCLAW_OPENTOKEN_BASE_URL,
+  OPENCLAW_PROVIDER_KEY,
+  OPENCLAW_DEFAULT_MODEL,
+} = require('./models');
+
+const OPENCLAW_DIR = path.join(os.homedir(), '.openclaw');
+const OPENCLAW_JSON_PATH = path.join(OPENCLAW_DIR, 'openclaw.json');
+
+function opentokenModelIdFromData(data) {
+  let model = '';
+  const primary =
+    data.agents &&
+    data.agents.defaults &&
+    data.agents.defaults.model &&
+    data.agents.defaults.model.primary != null
+      ? String(data.agents.defaults.model.primary)
+      : '';
+  const prefix = OPENCLAW_PROVIDER_KEY + '/';
+  if (primary.startsWith(prefix)) {
+    model = primary.slice(prefix.length);
+  }
+  const p =
+    data.models &&
+    data.models.providers &&
+    data.models.providers[OPENCLAW_PROVIDER_KEY];
+  if (!model && p && Array.isArray(p.models) && p.models[0] && p.models[0].id != null) {
+    model = String(p.models[0].id);
+  }
+  return model;
+}
+
+function readOpenclawRoot() {
+  if (!fs.existsSync(OPENCLAW_JSON_PATH)) {
+    return {};
+  }
+  const raw = fs.readFileSync(OPENCLAW_JSON_PATH, 'utf-8');
+  if (!String(raw).trim()) {
+    return {};
+  }
+  const data = JSON.parse(raw);
+  if (data === null || typeof data !== 'object' || Array.isArray(data)) {
+    throw new Error('openclaw.json 根节点必须是 JSON 对象');
+  }
+  return data;
+}
+
+function getCurrentOpenclawOpentokenConfig() {
+  try {
+    const data = readOpenclawRoot();
+    const p = data.models && data.models.providers && data.models.providers[OPENCLAW_PROVIDER_KEY];
+    const apiKey =
+      p && p.apiKey != null && String(p.apiKey).trim() !== ''
+        ? String(p.apiKey)
+        : '';
+
+    const model = opentokenModelIdFromData(data);
+    return { apiKey, model };
+  } catch (_) {
+    return { apiKey: '', model: '' };
+  }
+}
+
+function applyOpenclawConfig(patch) {
+  const { apiKey, model } = patch || {};
+
+  let data;
+  try {
+    data = readOpenclawRoot();
+  } catch (e) {
+    throw new Error(
+      '现有 openclaw.json 无法解析，请先备份后手动修复：' + e.message,
+    );
+  }
+
+  if (!data.models || typeof data.models !== 'object' || Array.isArray(data.models)) {
+    data.models = {};
+  }
+  if (data.models.mode == null || data.models.mode === '') {
+    data.models.mode = 'merge';
+  }
+  if (
+    !data.models.providers ||
+    typeof data.models.providers !== 'object' ||
+    Array.isArray(data.models.providers)
+  ) {
+    data.models.providers = {};
+  }
+
+  const prev =
+    data.models.providers[OPENCLAW_PROVIDER_KEY] &&
+    typeof data.models.providers[OPENCLAW_PROVIDER_KEY] === 'object' &&
+    !Array.isArray(data.models.providers[OPENCLAW_PROVIDER_KEY])
+      ? { ...data.models.providers[OPENCLAW_PROVIDER_KEY] }
+      : {};
+
+  const diskModel = opentokenModelIdFromData(data);
+  const modelId =
+    model !== undefined && model !== null && String(model).trim() !== ''
+      ? String(model).trim()
+      : diskModel || OPENCLAW_DEFAULT_MODEL;
+
+  const mergedProvider = {
+    ...prev,
+    baseUrl: OPENCLAW_OPENTOKEN_BASE_URL,
+    api: 'openai-completions',
+    models: [
+      {
+        id: modelId,
+        name: modelId,
+      },
+    ],
+  };
+  if (apiKey !== undefined) {
+    mergedProvider.apiKey = apiKey;
+  } else if (prev.apiKey != null) {
+    mergedProvider.apiKey = prev.apiKey;
+  }
+
+  data.models.providers[OPENCLAW_PROVIDER_KEY] = mergedProvider;
+
+  if (!data.agents || typeof data.agents !== 'object' || Array.isArray(data.agents)) {
+    data.agents = {};
+  }
+  if (
+    !data.agents.defaults ||
+    typeof data.agents.defaults !== 'object' ||
+    Array.isArray(data.agents.defaults)
+  ) {
+    data.agents.defaults = {};
+  }
+  if (
+    !data.agents.defaults.model ||
+    typeof data.agents.defaults.model !== 'object' ||
+    Array.isArray(data.agents.defaults.model)
+  ) {
+    data.agents.defaults.model = {};
+  }
+
+  const primaryRef = OPENCLAW_PROVIDER_KEY + '/' + modelId;
+  data.agents.defaults.model.primary = primaryRef;
+
+  if (
+    !data.agents.defaults.models ||
+    typeof data.agents.defaults.models !== 'object' ||
+    Array.isArray(data.agents.defaults.models)
+  ) {
+    data.agents.defaults.models = {};
+  }
+  if (!Object.prototype.hasOwnProperty.call(data.agents.defaults.models, primaryRef)) {
+    data.agents.defaults.models[primaryRef] = {};
+  }
+
+  if (!fs.existsSync(OPENCLAW_DIR)) {
+    fs.mkdirSync(OPENCLAW_DIR, { recursive: true });
+  }
+  fs.writeFileSync(
+    OPENCLAW_JSON_PATH,
+    JSON.stringify(data, null, 2),
+    'utf-8',
+  );
+}
+
+module.exports = {
+  OPENCLAW_DIR,
+  OPENCLAW_JSON_PATH,
+  getCurrentOpenclawOpentokenConfig,
+  applyOpenclawConfig,
+};

package/src/opencode-config.js

@@ -0,0 +1,166 @@
+'use strict';
+
+const fs = require('fs');
+const path = require('path');
+const os = require('os');
+const {
+  OPENCODE_OPENTOKEN_BASE_URL,
+  OPENCODE_PROVIDER_KEY,
+  OPENCODE_DEFAULT_MODEL,
+} = require('./models');
+
+const OPENCODE_DIR =
+  process.platform === 'win32'
+    ? path.join(os.homedir(), '.config', 'opencode')
+    : path.join(os.homedir(), '.config', 'opencode');
+
+const OPENCODE_JSON_PATH = path.join(OPENCODE_DIR, 'opencode.json');
+
+function opentokenModelIdFromData(data) {
+  const provider =
+    data.provider &&
+    data.provider[OPENCODE_PROVIDER_KEY];
+
+  if (!provider) return '';
+
+  const models = provider.models;
+  if (!models || typeof models !== 'object') return '';
+
+  const ids = Object.keys(models);
+  return ids.length > 0 ? ids[0] : '';
+}
+
+function readOpencodeRoot() {
+  if (!fs.existsSync(OPENCODE_JSON_PATH)) {
+    return {};
+  }
+  const raw = fs.readFileSync(OPENCODE_JSON_PATH, 'utf-8');
+  if (!String(raw).trim()) {
+    return {};
+  }
+  const data = JSON.parse(raw);
+  if (data === null || typeof data !== 'object' || Array.isArray(data)) {
+    throw new Error('opencode.json 根节点必须是 JSON 对象');
+  }
+  return data;
+}
+
+function getCurrentOpencodeOpentokenConfig() {
+  try {
+    const data = readOpencodeRoot();
+    const provider =
+      data.provider &&
+      data.provider[OPENCODE_PROVIDER_KEY];
+
+    const apiKey =
+      provider &&
+      provider.options &&
+      provider.options.apiKey != null &&
+      String(provider.options.apiKey).trim() !== ''
+        ? String(provider.options.apiKey)
+        : '';
+
+    const model = opentokenModelIdFromData(data);
+    return { apiKey, model };
+  } catch (_) {
+    return { apiKey: '', model: '' };
+  }
+}
+
+function applyOpencodeConfig(patch) {
+  const { apiKey, model, models } = patch || {};
+
+  let data;
+  try {
+    data = readOpencodeRoot();
+  } catch (e) {
+    throw new Error(
+      '现有 opencode.json 无法解析，请先备份后手动修复：' + e.message,
+    );
+  }
+
+  if (!data.provider || typeof data.provider !== 'object' || Array.isArray(data.provider)) {
+    data.provider = {};
+  }
+
+  if (!data.$schema) {
+    data.$schema = 'https://opencode.ai/config.json';
+  }
+
+  const prev =
+    data.provider[OPENCODE_PROVIDER_KEY] &&
+    typeof data.provider[OPENCODE_PROVIDER_KEY] === 'object' &&
+    !Array.isArray(data.provider[OPENCODE_PROVIDER_KEY])
+      ? { ...data.provider[OPENCODE_PROVIDER_KEY] }
+      : {};
+
+  const diskModel = opentokenModelIdFromData(data);
+  const modelId =
+    model !== undefined && model !== null && String(model).trim() !== ''
+      ? String(model).trim()
+      : diskModel || OPENCODE_DEFAULT_MODEL;
+
+  const modelsConfig = {};
+  
+  if (models && Array.isArray(models) && models.length > 0) {
+    models.forEach((m) => {
+      modelsConfig[m] = {
+        name: m,
+        modalities: {
+          input: ['text', 'image'],
+          output: ['text'],
+        },
+        limit: {
+          context: 120000,
+          output: 8192,
+        },
+      };
+    });
+  } else {
+    modelsConfig[modelId] = {
+      name: modelId,
+      modalities: {
+        input: ['text', 'image'],
+        output: ['text'],
+      },
+      limit: {
+        context: 120000,
+        output: 8192,
+      },
+    };
+  }
+
+  const mergedProvider = {
+    npm: '@ai-sdk/anthropic',
+    name: 'Opentoken',
+    options: {
+      baseURL: OPENCODE_OPENTOKEN_BASE_URL,
+    },
+    models: modelsConfig,
+  };
+
+  if (apiKey !== undefined) {
+    mergedProvider.options.apiKey = apiKey;
+  } else if (prev.options && prev.options.apiKey != null) {
+    mergedProvider.options.apiKey = prev.options.apiKey;
+  }
+
+  data.provider[OPENCODE_PROVIDER_KEY] = mergedProvider;
+
+  if (!fs.existsSync(OPENCODE_DIR)) {
+    fs.mkdirSync(OPENCODE_DIR, { recursive: true });
+  }
+
+  fs.writeFileSync(
+    OPENCODE_JSON_PATH,
+    JSON.stringify(data, null, 2),
+    'utf-8',
+  );
+}
+
+module.exports = {
+  OPENCODE_DIR,
+  OPENCODE_JSON_PATH,
+  getCurrentOpencodeOpentokenConfig,
+  applyOpencodeConfig,
+};

package/src/screen.js

@@ -0,0 +1,9 @@
+'use strict';
+
+function clearScreen() {
+  if (process.stdout.isTTY) {
+    console.clear();
+  }
+}
+
+module.exports = { clearScreen };

package/src/tools/atomcode.js

@@ -0,0 +1,253 @@
+'use strict';
+
+const chalk = require('chalk');
+const inquirer = require('inquirer');
+const { DEFAULT_BASE_URL, ATOMCODE_OPENTOKEN_BASE_URL } = require('../models');
+const { fetchModelsForBaseUrl } = require('../fetch-models');
+const {
+  CONFIG_PATH,
+  getCurrentAtomcodeConfig,
+  applyAtomcodeConfig,
+} = require('../atomcode-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 getEffectiveAtomcodeEnv() {
+  const disk = getCurrentAtomcodeConfig();
+  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 buildAtomcodeMenuChoices() {
+  const ev = getEffectiveAtomcodeEnv();
+  const keyShow = ev.apiKey ? maskApiKey(ev.apiKey) : '未配置';
+  const modelShow = ev.model ? ev.model : '未配置';
+
+  return [
+    new inquirer.Separator(
+      chalk.gray('  Opentoken Base URL（固定）: ' + ATOMCODE_OPENTOKEN_BASE_URL),
+    ),
+    { 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: '请输入 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 handleConfigModel() {
+  const { model: existing, apiKey } = getEffectiveAtomcodeEnv();
+  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();
+    applyAtomcodeConfig({ model: pendingConfig.model });
+    console.log(chalk.green('  ✓ 模型已更新：' + pendingConfig.model));
+    console.log(
+      chalk.gray('  （已写入 ' + CONFIG_PATH + '，base_url 为 Opentoken 固定地址）'),
+    );
+    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();
+  }
+
+  applyAtomcodeConfig({ model: pendingConfig.model });
+  console.log(chalk.green('  ✓ 模型已更新：' + pendingConfig.model));
+  console.log(
+    chalk.gray('  （已写入 ' + CONFIG_PATH + '，base_url 为 Opentoken 固定地址）'),
+  );
+  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) {
+    applyAtomcodeConfig(toWrite);
+    console.log(chalk.green('  ✓ 配置已成功写入'));
+    console.log(chalk.gray('  ' + CONFIG_PATH));
+  } else {
+    applyAtomcodeConfig({});
+    console.log(
+      chalk.gray(
+        '  （无 Key / 模型变更；已确保 default_provider 与 Opentoken 固定 base_url）',
+      ),
+    );
+    console.log(chalk.gray('  ' + CONFIG_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: 'Atomcode 配置（~/.atomcode/config.toml）：',
+        choices: buildAtomcodeMenuChoices(),
+        pageSize: 12,
+      },
+    ]);
+
+    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;
+
+  applyAtomcodeConfig(toWrite);
+
+  console.log('');
+  console.log(chalk.bold(chalk.cyan('Atomcode 配置已更新：')));
+  if (key) console.log(chalk.green('  ✓ API Key  : ') + maskApiKey(key));
+  if (model) console.log(chalk.green('  ✓ 模型     : ') + model);
+  console.log(chalk.green('  ✓ Base URL : ') + ATOMCODE_OPENTOKEN_BASE_URL + chalk.gray('（固定）'));
+  console.log(chalk.gray('  ' + CONFIG_PATH));
+  console.log('');
+
+  return true;
+}
+
+module.exports = { run, quickSet };