dgxay-code-cli 0.0.0-stage → 1.0.0

package/src/commands/export.js

@@ -0,0 +1,102 @@
+'use strict';
+
+/**
+ * export：导出会话（桌面版没有这个能力，CLI 补上）。
+ *
+ *   dgxay export                把最近一个会话输出到 stdout（Markdown）
+ *   dgxay export -o chat.md     写入文件
+ *   dgxay export --format json  导出 JSON
+ *   dgxay export 2              指定会话序号
+ */
+
+const fs = require('fs');
+const path = require('path');
+const print = require('../cli/print');
+const { actionFirst } = require('../cli/args');
+const sessions = require('../core/sessions');
+
+const options = [
+  { name: 'format', type: 'string', valueName: 'md|json', desc: '输出格式：markdown（默认）/ json' },
+  { name: 'output', alias: 'o', type: 'string', valueName: 'file', desc: '写入文件而不是打印' }
+];
+
+function toMarkdown(chat) {
+  const lines = [];
+  const date = new Date(chat.createdAt || Date.now());
+  lines.push(`# ${chat.title || '新对话'}`);
+  lines.push('');
+  lines.push(`> ${date.toLocaleString()} · ${chat.mode === 'work' ? '工作模式' : '对话模式'}${chat.skill ? ` · 技能 ${chat.skill}` : ''}`);
+  lines.push('');
+  for (const message of chat.messages) {
+    if (message.role === 'user') {
+      lines.push('## 用户');
+      lines.push('');
+      lines.push(message.content || '');
+      lines.push('');
+    } else if (message.role === 'assistant') {
+      lines.push('## DgxayCode');
+      lines.push('');
+      lines.push(message.content || '');
+      lines.push('');
+      if (message.toolLog?.length) {
+        lines.push('<details><summary>工具调用</summary>');
+        lines.push('');
+        for (const tool of message.toolLog) {
+          lines.push(`- \`${tool.name}\` ${tool.ok ? '成功' : '失败'}${tool.review?.reason ? ` · ${tool.review.reason}` : ''}`);
+        }
+        lines.push('');
+        lines.push('</details>');
+        lines.push('');
+      }
+    } else if (message.role === 'terminal') {
+      lines.push('## 终端');
+      lines.push('');
+      lines.push('```');
+      lines.push(`$ ${message.command}`);
+      lines.push(`${message.output || ''}${message.stderr || ''}`.trimEnd());
+      lines.push('```');
+      lines.push('');
+    }
+  }
+  return lines.join('\n');
+}
+
+async function run({ argv }) {
+  const { action, rest, options: opts } = actionFirst(argv, options);
+  const ref = action || rest[0] || '';
+  const chats = sessions.listChats();
+  if (!chats.length) {
+    print.warn('还没有可以导出的会话');
+    return 1;
+  }
+  const chat = ref ? sessions.resolveChat(ref, chats) : chats[0];
+  if (!chat) {
+    print.fail(`找不到会话：${ref}`);
+    return 1;
+  }
+  const format = String(opts.format || 'md').toLowerCase();
+  const content =
+    format === 'json'
+      ? `${JSON.stringify(chat, null, 2)}\n`
+      : toMarkdown(chat);
+
+  if (opts.output) {
+    const target = path.resolve(opts.output);
+    fs.mkdirSync(path.dirname(target), { recursive: true });
+    fs.writeFileSync(target, content, 'utf8');
+    print.ok(`已导出「${chat.title}」到 ${target}`);
+    return 0;
+  }
+  process.stdout.write(content.endsWith('\n') ? content : `${content}\n`);
+  return 0;
+}
+
+module.exports = {
+  name: 'export',
+  aliases: ['dump'],
+  summary: '导出会话为 Markdown / JSON',
+  usage: 'dgxay export [序号|ID] [--format md|json] [-o 文件]',
+  options,
+  examples: ['dgxay export', 'dgxay export 2 -o chat.md', 'dgxay export -o chats.json --format json'],
+  run
+};

package/src/commands/mcp.js

@@ -0,0 +1,257 @@
+'use strict';
+
+/**
+ * mcp：MCP Server 管理（对应桌面版「设置 → MCP」）。
+ *
+ *   dgxay mcp list
+ *   dgxay mcp add --name filesystem --command npx --args "-y @modelcontextprotocol/server-filesystem ."
+ *   dgxay mcp start <名称|ID>
+ *   dgxay mcp stop <名称|ID>
+ *   dgxay mcp logs <名称|ID>
+ *   dgxay mcp rm <名称|ID>
+ *   dgxay mcp env <名称|ID> --env "KEY=value"    追加环境变量
+ *
+ * 注意：与桌面版一致，当前版本只管理进程，不把 MCP 工具暴露给模型。
+ */
+
+const print = require('../cli/print');
+const { parseArgs, actionFirst } = require('../cli/args');
+const { COLORS } = require('../cli/ansi');
+const mcp = require('../core/mcp');
+const cfg = require('../services/config');
+const { ask } = require('../services/input');
+
+const options = [
+  { name: 'name', type: 'string', valueName: 'text', desc: '显示名称' },
+  { name: 'command', type: 'string', valueName: 'cmd', desc: '启动命令（如 npx / uvx）' },
+  { name: 'args', type: 'string', valueName: 'text', desc: '命令参数' },
+  { name: 'env', type: 'list', valueName: 'K=V', desc: '环境变量，可重复' },
+  { name: 'cwd', type: 'string', valueName: 'dir', desc: '工作目录' },
+  { name: 'follow', alias: 'f', type: 'boolean', desc: 'logs：持续输出' },
+  { name: 'yes', alias: 'y', type: 'boolean', desc: '跳过确认' }
+];
+
+function statusText(status) {
+  if (status === 'running') return COLORS.green('运行中');
+  if (status === 'error') return COLORS.red('异常');
+  if (status === 'stopping') return COLORS.yellow('停止中');
+  return COLORS.faint('已停止');
+}
+
+function list() {
+  const servers = mcp.listServers();
+  print.blank();
+  if (!servers.length) {
+    print.hint(`还没有 MCP Server（配置保存在 ${cfg.MCP_PATH}）`);
+    print.hint('添加：dgxay mcp add --name filesystem --command npx --args "-y @modelcontextprotocol/server-filesystem ."');
+    print.blank();
+    return 0;
+  }
+  print.heading(`MCP Server（${servers.length}）`);
+  const rows = servers.map((server, index) => [
+    String(index + 1),
+    server.name,
+    statusText(server.status),
+    server.pid ? String(server.pid) : '',
+    `${server.command} ${(server.args || []).join(' ')}`.trim(),
+    server.cwd || ''
+  ]);
+  print.table(
+    [
+      { title: '#', max: 3 },
+      { title: '名称', max: 20 },
+      { title: '状态', max: 8 },
+      { title: 'PID', max: 7 },
+      { title: '命令', max: 50 },
+      { title: '工作目录', max: 30 }
+    ],
+    rows
+  );
+  print.blank();
+  print.hint('MCP 不会随应用启动而自动运行；每次启动后需要手动 start');
+  print.hint('当前版本只管理进程，尚未把 MCP 工具接入模型');
+  print.blank();
+  return 0;
+}
+
+function resolve(ref) {
+  const server = mcp.findServer(ref);
+  if (!server) {
+    print.fail(`找不到 MCP：${ref}`);
+    return null;
+  }
+  return server;
+}
+
+async function add(opts) {
+  let name = opts.name;
+  let command = opts.command;
+  if (!name && process.stdin.isTTY) name = String((await ask('名称: ')) || '').trim();
+  if (!command && process.stdin.isTTY) command = String((await ask('启动命令（如 npx）: ')) || '').trim();
+  if (!name || !command) {
+    throw Object.assign(new Error('用法：dgxay mcp add --name <名称> --command <命令> [--args "..."]'), {
+      userError: true
+    });
+  }
+  const server = mcp.addServer({
+    name,
+    command,
+    args: opts.args || '',
+    env: (opts.env || []).join('\n'),
+    cwd: opts.cwd || ''
+  });
+  print.ok(`已保存 MCP「${server.name}」（${server.id}）`);
+  print.hint(`启动：dgxay mcp start ${server.name}`);
+  return 0;
+}
+
+function start(ref) {
+  const server = resolve(ref);
+  if (!server) return 1;
+  try {
+    const result = mcp.startServer(server.id);
+    if (result.status === 'running' && result.pid) print.ok(`已启动 ${server.name}（pid ${result.pid}）`);
+    else print.ok(`${server.name} 已在运行`);
+    print.hint(`查看日志：dgxay mcp logs ${server.name}`);
+    return 0;
+  } catch (err) {
+    print.fail(err.message);
+    return 1;
+  }
+}
+
+function stop(ref) {
+  const server = resolve(ref);
+  if (!server) return 1;
+  const ok = mcp.stopServer(server.id);
+  if (ok) print.ok(`已停止 ${server.name}`);
+  else print.warn(`${server.name} 当前没有运行`);
+  return 0;
+}
+
+async function logs(ref, opts) {
+  const server = resolve(ref);
+  if (!server) return 1;
+  const shown = new Set();
+  const printLogs = () => {
+    const entries = mcp.statusOf(server.id).logs || [];
+    for (const entry of entries) {
+      const key = `${entry.at}:${entry.stream}:${entry.text}`;
+      if (shown.has(key)) continue;
+      shown.add(key);
+      print.write(entry.stream === 'stderr' ? COLORS.red(entry.text) : entry.text);
+    }
+  };
+  printLogs();
+  if (!opts.follow) {
+    if (!shown.size) print.hint('暂无日志');
+    return 0;
+  }
+  print.hint('持续输出中，Ctrl+C 退出');
+  mcp.setNotifier((event) => {
+    if (event.type === 'log' && event.id === server.id) {
+      print.write(event.stream === 'stderr' ? COLORS.red(event.text) : event.text);
+    }
+  });
+  await new Promise((resolve) => {
+    process.once('SIGINT', () => resolve());
+  });
+  return 0;
+}
+
+async function remove(ref, opts) {
+  const server = resolve(ref);
+  if (!server) return 1;
+  if (!opts.yes && process.stdin.isTTY) {
+    const answer = await ask(`删除 MCP「${server.name}」并停止进程？[y/N] `);
+    if (!/^y/i.test(String(answer || '').trim())) {
+      print.hint('已取消');
+      return 1;
+    }
+  }
+  mcp.removeServer(server.id);
+  print.ok(`已删除 ${server.name}`);
+  return 0;
+}
+
+function updateEnv(ref, opts) {
+  const server = resolve(ref);
+  if (!server) return 1;
+  const env = { ...(server.env || {}) };
+  for (const pair of opts.env || []) {
+    const index = String(pair).indexOf('=');
+    if (index > 0) env[String(pair).slice(0, index).trim()] = String(pair).slice(index + 1).trim();
+  }
+  mcp.addServer({
+    name: server.name,
+    command: server.command,
+    args: server.args,
+    env,
+    cwd: server.cwd
+  });
+  print.ok(`已更新 ${server.name} 的环境变量（${Object.keys(env).length} 项）`);
+  print.warn('重启后生效：dgxay mcp stop ' + server.name + ' && dgxay mcp start ' + server.name);
+  return 0;
+}
+
+async function run({ argv }) {
+  const { action: rawAction, rest: restArgs, options: opts } = actionFirst(argv, options);
+  const action = rawAction || 'list';
+  const ref = restArgs.join(' ');
+
+  // 兼容：dgxay mcp <名称> 视为 start
+  if (!['list', 'ls', 'add', 'start', 'run', 'stop', 'logs', 'log', 'rm', 'remove', 'delete', 'env'].includes(action)) {
+    return start(restArgs.join(' '));
+  }
+
+  switch (action) {
+    case 'list':
+    case 'ls':
+      return list();
+    case 'add':
+      return add(opts);
+    case 'start':
+    case 'run':
+      if (!ref) throw Object.assign(new Error('用法：dgxay mcp start <名称|ID>'), { userError: true });
+      return start(ref);
+    case 'stop':
+      if (!ref) throw Object.assign(new Error('用法：dgxay mcp stop <名称|ID>'), { userError: true });
+      return stop(ref);
+    case 'logs':
+    case 'log':
+      if (!ref) throw Object.assign(new Error('用法：dgxay mcp logs <名称|ID> [-f]'), { userError: true });
+      return logs(ref, opts);
+    case 'rm':
+    case 'remove':
+    case 'delete':
+      if (!ref) throw Object.assign(new Error('用法：dgxay mcp rm <名称|ID>'), { userError: true });
+      return remove(ref, opts);
+    case 'env':
+      if (!ref) throw Object.assign(new Error('用法：dgxay mcp env <名称|ID> --env "KEY=value"'), { userError: true });
+      return updateEnv(ref, opts);
+    default:
+      return 1;
+  }
+}
+
+module.exports = {
+  name: 'mcp',
+  summary: 'MCP Server 管理',
+  usage: 'dgxay mcp <list|add|start|stop|logs|rm|env>',
+  options,
+  subcommands: [
+    ['list', '列出 MCP Server'],
+    ['add', '新增配置'],
+    ['start <名称>', '启动进程'],
+    ['stop <名称>', '停止进程'],
+    ['logs <名称> [-f]', '查看日志'],
+    ['rm <名称>', '删除'],
+    ['env <名称> --env K=V', '更新环境变量']
+  ],
+  examples: [
+    'dgxay mcp add --name filesystem --command npx --args "-y @modelcontextprotocol/server-filesystem ."',
+    'dgxay mcp start filesystem',
+    'dgxay mcp logs filesystem -f'
+  ],
+  run
+};

package/src/commands/model.js

@@ -0,0 +1,223 @@
+'use strict';
+
+/**
+ * model：模型管理。
+ *
+ *   dgxay model list                 列出当前供应商的模型与上下文窗口
+ *   dgxay model use <id|序号>         设为当前模型
+ *   dgxay model add <id> [--name X] [--context-window N] [--provider P]
+ *   dgxay model rm <id> [--provider P]
+ *   dgxay model fetch [--provider P] [--base-url ...] [--api-key ...]
+ */
+
+const print = require('../cli/print');
+const { parseArgs, actionFirst } = require('../cli/args');
+const { COLORS } = require('../cli/ansi');
+const settings = require('../services/settings');
+const sessions = require('../core/sessions');
+const { askSecret } = require('../services/input');
+
+const options = [
+  { name: 'provider', alias: 'p', type: 'string', valueName: 'name', desc: '指定供应商（默认当前供应商）' },
+  { name: 'name', type: 'string', valueName: 'text', desc: '显示名称' },
+  { name: 'context-window', type: 'number', valueName: 'n', desc: '上下文窗口' },
+  { name: 'base-url', type: 'string', valueName: 'url', desc: '拉取列表时使用该地址' },
+  { name: 'api-key', type: 'string', valueName: 'key', desc: '拉取列表时使用该密钥' },
+  { name: 'add', type: 'list', valueName: 'id', desc: '拉取后把指定模型加入供应商' }
+];
+
+function list(providerRef) {
+  const config = settings.readConfig();
+  const provider = providerRef ? settings.findProvider(providerRef) : settings.activeProvider(config);
+  if (!provider) {
+    print.warn('还没有供应商，运行 dgxay config init');
+    return 1;
+  }
+  const models = provider.models || [];
+  print.blank();
+  print.heading(`${provider.name} 的模型（${models.length}）`);
+  if (!models.length) {
+    print.hint('  还没有模型，运行 dgxay model add <id>');
+    print.blank();
+    return 0;
+  }
+  const active = provider.id === config.activeProviderId ? config.activeModelId : '';
+  const rows = models.map((model, index) => [
+    String(index + 1),
+    model.id === active ? COLORS.accent('❯') : ' ',
+    model.name || model.id,
+    model.id,
+    model.contextWindow ? sessions.formatTokens(model.contextWindow) : COLORS.faint('按协议默认'),
+    provider.reviewModelId === model.id ? '审查' : ''
+  ]);
+  print.table(
+    [
+      { title: '#', max: 3 },
+      { title: '', max: 1 },
+      { title: '名称', max: 26 },
+      { title: 'ID', max: 32 },
+      { title: '上下文', max: 10 },
+      { title: '', max: 4 }
+    ],
+    rows
+  );
+  print.blank();
+  return 0;
+}
+
+async function fetchList(opts) {
+  const config = settings.readConfig();
+  const provider = opts.provider ? settings.findProvider(opts.provider) : settings.activeProvider(config);
+  let apiKey = opts.apiKey;
+  if (apiKey === undefined && provider && !(provider.authMode === 'oauth') && !provider.apiKey) {
+    const secret = await askSecret('API Key: ');
+    apiKey = secret || '';
+  }
+  process.stdout.write(`${COLORS.faint('正在拉取模型列表…')}`);
+  try {
+    let models;
+    if (opts.baseUrl) {
+      models = await settings.fetchModels({
+        baseUrl: opts.baseUrl,
+        apiKey: apiKey ?? '',
+        protocol: opts.protocol || provider?.protocol || 'openai'
+      });
+    } else {
+      if (!provider) throw new Error('没有可用的供应商');
+      models = await settings.fetchModels({ providerRef: provider.id, apiKey });
+    }
+    process.stdout.write('\r\x1b[2K');
+    print.ok(`拉到 ${models.length} 个模型`);
+    const existing = new Set((provider?.models || []).map((m) => m.id));
+    for (const id of models) {
+      print.write(`  ${existing.has(id) ? COLORS.green('✓') : ' '} ${id}`);
+    }
+    print.blank();
+    print.hint('用 dgxay model add <id> 加入供应商；拉取失败时也可以直接手动填写模型 ID');
+    return 0;
+  } catch (err) {
+    process.stdout.write('\r\x1b[2K');
+    print.fail(`拉取失败：${err.message}`);
+    print.hint('部分网关不支持 /models 接口，可以直接用 dgxay model add <id> 手动添加');
+    return 1;
+  }
+}
+
+function addModel(id, opts) {
+  const config = settings.readConfig();
+  const provider = opts.provider ? settings.findProvider(opts.provider) : settings.activeProvider(config);
+  if (!provider) {
+    print.warn('还没有供应商，运行 dgxay config init');
+    return 1;
+  }
+  const models = (provider.models || []).slice();
+  if (models.some((m) => m.id === id)) {
+    print.warn(`模型 ${id} 已存在`);
+    return 1;
+  }
+  models.push({
+    id,
+    name: opts.name || id,
+    contextWindow: Number(opts.contextWindow) || 0
+  });
+  const next = (config.providers || []).map((p) => (p.id === provider.id ? { ...p, models } : p));
+  const patch = { providers: next };
+  if (provider.id === config.activeProviderId && !config.activeModelId) patch.activeModelId = id;
+  settings.saveConfig(patch);
+  print.ok(`已添加模型 ${id}`);
+  return 0;
+}
+
+function removeModel(id, opts) {
+  const config = settings.readConfig();
+  const provider = opts.provider ? settings.findProvider(opts.provider) : settings.activeProvider(config);
+  if (!provider) {
+    print.warn('还没有供应商');
+    return 1;
+  }
+  const models = (provider.models || []).filter((m) => m.id !== id);
+  if (models.length === (provider.models || []).length) {
+    print.fail(`供应商 ${provider.name} 下没有模型 ${id}`);
+    return 1;
+  }
+  if (!models.length) {
+    print.fail('不能删除最后一个模型，请先添加其他模型');
+    return 1;
+  }
+  const reviewModelId = models.some((m) => m.id === provider.reviewModelId)
+    ? provider.reviewModelId
+    : models[0].id;
+  const next = (config.providers || []).map((p) =>
+    p.id === provider.id ? { ...p, models, reviewModelId } : p
+  );
+  const patch = { providers: next };
+  if (config.activeModelId === id) patch.activeModelId = models[0].id;
+  settings.saveConfig(patch);
+  print.ok(`已删除模型 ${id}`);
+  return 0;
+}
+
+function useModel(ref, opts) {
+  const config = settings.readConfig();
+  const provider = opts.provider ? settings.findProvider(opts.provider) : settings.activeProvider(config);
+  if (!provider) {
+    print.warn('还没有供应商');
+    return 1;
+  }
+  const models = provider.models || [];
+  const index = Number(ref);
+  const target =
+    (Number.isInteger(index) && index >= 1 && index <= models.length ? models[index - 1] : null) ||
+    models.find((m) => m.id === ref) ||
+    models.find((m) => m.id.includes(ref));
+  if (!target) {
+    print.fail(`找不到模型：${ref}`);
+    return 1;
+  }
+  settings.setActiveModel(provider.id, target.id);
+  print.ok(`当前模型：${target.name || target.id}（${provider.name}）`);
+  return 0;
+}
+
+async function run({ argv }) {
+  const { action: rawAction, rest, options: opts } = actionFirst(argv, options);
+  const action = rawAction || 'list';
+
+  // model add deepseek-chat gpt-4o 支持多个
+  if (action === 'add') {
+    if (!rest.length) throw Object.assign(new Error('用法：dgxay model add <id> [<id>…]'), { userError: true });
+    let code = 0;
+    for (const id of rest) code = Math.max(code, addModel(id, opts));
+    return code;
+  }
+  if (action === 'rm' || action === 'remove' || action === 'delete') {
+    if (!rest[0]) throw Object.assign(new Error('用法：dgxay model rm <id>'), { userError: true });
+    return removeModel(rest[0], opts);
+  }
+  if (action === 'use' || action === 'set') {
+    if (!rest[0]) throw Object.assign(new Error('用法：dgxay model use <id|序号>'), { userError: true });
+    return useModel(rest[0], opts);
+  }
+  if (action === 'fetch' || action === 'ls-remote') return fetchList(opts);
+  if (action === 'list' || action === 'ls') return list(opts.provider);
+
+  // 不是已知子命令：当作模型 ID 使用
+  return useModel(action, opts);
+}
+
+module.exports = {
+  name: 'model',
+  aliases: ['models'],
+  summary: '模型管理：列表、切换、增删、拉取',
+  usage: 'dgxay model <list|use|add|rm|fetch>',
+  options,
+  subcommands: [
+    ['list', '列出模型'],
+    ['use <id|序号>', '切换当前模型'],
+    ['add <id>', '添加模型'],
+    ['rm <id>', '删除模型'],
+    ['fetch', '从供应商接口拉取模型列表']
+  ],
+  examples: ['dgxay model list', 'dgxay model use 2', 'dgxay model add gpt-4o --name GPT-4o', 'dgxay model fetch'],
+  run
+};