dgxay-code-cli 0.0.0-stage → 1.1.0

package/src/cli/approval.js

@@ -0,0 +1,186 @@
+'use strict';
+
+/**
+ * 授权确认组件（对应桌面版 permission:request 弹窗）。
+ *
+ * - 普通权限：每次操作都必须确认；
+ * - 自动审查：模型放行则直接执行，可疑操作转到这里；
+ * - 非交互模式（管道输入）下默认拒绝，避免无人值守时被 AI 静默执行；
+ * - 按 Esc 或回答 n 视为拒绝；等待超时 180 秒同样按拒绝处理。
+ */
+
+const readline = require('readline');
+const { COLORS, displayWidth, truncate, termWidth, wrapText } = require('./ansi');
+
+const APPROVAL_TIMEOUT = 180000;
+
+function frameText(lines, options = {}) {
+  const width = Math.min(options.width || 70, Math.max(24, termWidth() - 2));
+  const title = options.title || '';
+  const accentText = (text) => COLORS.accent(text);
+  const body = [];
+  if (title) {
+    const head = ` ${title} `;
+    const fill = Math.max(0, width - displayWidth(head) - 2);
+    body.push(`${accentText('┌')}${COLORS.bold(head)}${COLORS.accent(`${'─'.repeat(fill)}┐`)}`);
+  } else {
+    body.push(COLORS.accent(`┌${'─'.repeat(width)}┐`));
+  }
+  for (const line of lines) {
+    const text = typeof line === 'string' ? line : line.text;
+    const raw = typeof line === 'string' ? text : line.raw || text;
+    const visible = displayWidth(raw);
+    const pad = Math.max(0, width - visible - 2);
+    body.push(`${COLORS.accent('│')} ${text}${' '.repeat(pad)} ${COLORS.accent('│')}`);
+  }
+  body.push(COLORS.accent(`└${'─'.repeat(width)}┘`));
+  return body.join('\n');
+}
+
+function question(prompt) {
+  return new Promise((resolve) => {
+    const rl = readline.createInterface({ input: process.stdin, output: process.stdout });
+    rl.question(prompt, (answer) => {
+      rl.close();
+      resolve(answer);
+    });
+    rl.on('SIGINT', () => {
+      rl.close();
+      resolve('');
+    });
+  });
+}
+
+/** 单键读取（y / n / Esc / Enter），非 TTY 时回退到整行输入 */
+function readKey({ timeout = APPROVAL_TIMEOUT } = {}) {
+  return new Promise((resolve) => {
+    if (!process.stdin.isTTY) {
+      question('').then((answer) => {
+        const text = String(answer || '').trim().toLowerCase();
+        resolve({ key: text, line: text });
+      });
+      return;
+    }
+    let done = false;
+    readline.emitKeypressEvents(process.stdin);
+    if (process.stdin.isTTY) {
+      try {
+        process.stdin.setRawMode(true);
+      } catch {
+        /* ignore */
+      }
+    }
+    const finish = (result) => {
+      if (done) return;
+      done = true;
+      clearTimeout(timer);
+      process.stdin.removeListener('keypress', onKey);
+      if (process.stdin.isTTY) {
+        try {
+          process.stdin.setRawMode(false);
+        } catch {
+          /* ignore */
+        }
+      }
+      resolve(result);
+    };
+    const onKey = (str, key = {}) => {
+      if (key.ctrl && key.name === 'c') {
+        finish({ key: 'escape', line: '' });
+        return;
+      }
+      if (key.name === 'escape') {
+        finish({ key: 'escape', line: '' });
+        return;
+      }
+      if (key.name === 'return' || key.name === 'enter' || str === '\r' || str === '\n') {
+        finish({ key: 'submit', line: '' });
+        return;
+      }
+      if (!key.ctrl && !key.meta && str) {
+        const text = String(str).toLowerCase();
+        if (text === 'y' || text === 'n') {
+          finish({ key: text, line: text });
+          return;
+        }
+        if (text === 'q') {
+          finish({ key: 'escape', line: '' });
+          return;
+        }
+        // 其他字符先忽略，避免误触
+        process.stdout.write('\x07');
+      }
+    };
+    const timer = setTimeout(() => finish({ key: 'timeout', line: '' }), timeout);
+    process.stdin.on('keypress', onKey);
+    process.stdin.resume();
+  });
+}
+
+/**
+ * 请求用户授权一次操作。
+ * @returns {Promise<boolean>}
+ */
+async function askApproval(request, options = {}) {
+  const interactive = Boolean(process.stdin.isTTY);
+  if (!interactive) {
+    // 非交互：一律拒绝，并提示如何绕过
+    process.stderr.write(
+      `${COLORS.yellow('[permission]')} ${request.title}：${request.operation}\n` +
+        `${COLORS.dim('  Not an interactive terminal — denied. Use --permission full or run interactively.')}\n`
+    );
+    return false;
+  }
+
+  const lines = [];
+  lines.push({ text: `${COLORS.faint('op')}  ${COLORS.bold(request.operation)}`, raw: `op  ${request.operation}` });
+  for (const line of String(request.detail || '').split('\n')) {
+    if (!line.trim()) continue;
+    const idx = line.indexOf('：');
+    if (idx > 0) {
+      const label = line.slice(0, idx);
+      const value = line.slice(idx + 1);
+      for (const piece of wrapText(value, 56)) {
+        lines.push({
+          text: `${COLORS.faint(truncate(label, 6))}  ${piece}`,
+          raw: `${label}  ${piece}`
+        });
+      }
+    } else {
+      for (const piece of wrapText(line, 60)) lines.push({ text: piece, raw: piece });
+    }
+  }
+  if (request.workspace) {
+    lines.push({ text: '', raw: '' });
+    lines.push({ text: COLORS.faint(`workspace ${request.workspace}`), raw: `workspace ${request.workspace}` });
+  }
+  if (request.review) {
+    lines.push({ text: '', raw: '' });
+    const who = request.reviewer ? `reviewer ${request.reviewer}` : 'auto-review';
+    lines.push({ text: COLORS.red(`${who} flagged this`), raw: `${who} flagged this` });
+    for (const piece of wrapText(request.review, 58)) {
+      lines.push({ text: COLORS.dim(`  ${piece}`), raw: `  ${piece}` });
+    }
+  }
+
+  process.stdout.write(`${frameText(lines, { title: request.title })}\n`);
+
+  const promptText = `${COLORS.accent('Allow this?')} ${COLORS.faint('[y] yes  [n] no')} `;
+  process.stdout.write(promptText);
+  const result = await readKey({ timeout: options.timeout || APPROVAL_TIMEOUT });
+
+  if (result.key === 'y') {
+    process.stdout.write(`${COLORS.green('allowed')}\n`);
+    return true;
+  }
+  const label =
+    result.key === 'timeout'
+      ? COLORS.yellow('timed out — denied')
+      : result.key === 'escape'
+        ? COLORS.dim('denied')
+        : COLORS.red('denied');
+  process.stdout.write(`${label}\n`);
+  return false;
+}
+
+module.exports = { askApproval, readKey, question, frameText, APPROVAL_TIMEOUT };

package/src/cli/args.js

@@ -0,0 +1,162 @@
+'use strict';
+
+/**
+ * 轻量参数解析：支持 --flag、--flag value、--flag=value、-abc 组合、-- 终止符。
+ *
+ * 每个命令声明自己的选项：{ name, alias, type: 'string'|'boolean'|'number'|'list', desc, values }
+ */
+
+function camel(name) {
+  return String(name).replace(/-([a-z])/g, (m, c) => c.toUpperCase());
+}
+
+function parseArgs(argv, spec = [], parseOptions = {}) {
+  const allowUnknown = Boolean(parseOptions.allowUnknown);
+  const options = {};
+  const positionals = [];
+  const lookup = new Map();
+  for (const item of spec) {
+    lookup.set(`--${item.name}`, item);
+    if (item.alias) {
+      for (const alias of [].concat(item.alias)) lookup.set(alias.length === 1 ? `-${alias}` : alias, item);
+    }
+    if (item.type === 'list') options[camel(item.name)] = [];
+  }
+
+  const rest = argv.slice();
+  while (rest.length) {
+    const token = rest.shift();
+    if (token === '--') {
+      positionals.push(...rest);
+      break;
+    }
+    // -h / -v 是约定俗成的通用开关；但命令自己声明了同名短选项时以命令为准
+    const takesHelp = !lookup.has('-h');
+    const takesVersion = !lookup.has('-v');
+    if ((token === '-h' || token === '--help') && takesHelp) {
+      options.help = true;
+      continue;
+    }
+    if ((token === '-v' || token === '--version') && (takesVersion || token === '--version')) {
+      options.version = true;
+      continue;
+    }
+
+    if (token.startsWith('--')) {
+      const eq = token.indexOf('=');
+      const name = eq === -1 ? token : token.slice(0, eq);
+      const inline = eq === -1 ? undefined : token.slice(eq + 1);
+      const item = lookup.get(name);
+      if (!item) {
+        if (allowUnknown) {
+          // 全局解析阶段：不认识的选项原样保留，交给子命令自己解析
+          positionals.push(token);
+          continue;
+        }
+        const error = new Error(`未知选项：${name}`);
+        error.userError = true;
+        throw error;
+      }
+      const key = camel(item.name);
+      if (item.type === 'boolean') {
+        options[key] = inline === undefined ? true : !/^(false|0|no)$/i.test(inline);
+        continue;
+      }
+      let value = inline;
+      if (value === undefined) {
+        value = rest.shift();
+        if (value === undefined) {
+          const error = new Error(`选项 ${name} 需要一个值`);
+          error.userError = true;
+          throw error;
+        }
+      }
+      if (item.type === 'list') options[key] = options[key].concat([value]);
+      else if (item.type === 'number') options[key] = Number(value);
+      else options[key] = value;
+      continue;
+    }
+
+    // 任意 -x / -abc 组合短选项（未知的仍然报错，避免把 -5 这类输入吞掉）
+    if (/^-[A-Za-z]/.test(token)) {
+      // 组合短选项：-abc，只有最后一位可以带值
+      const chars = token.slice(1).split('');
+      for (let i = 0; i < chars.length; i += 1) {
+        const item = lookup.get(`-${chars[i]}`);
+        if (!item) {
+          if (allowUnknown) {
+            positionals.push(token);
+            break;
+          }
+          const error = new Error(`未知选项：-${chars[i]}`);
+          error.userError = true;
+          throw error;
+        }
+        const key = camel(item.name);
+        if (item.type === 'boolean') {
+          options[key] = true;
+          continue;
+        }
+        const value = i === chars.length - 1 ? rest.shift() : chars.slice(i + 1).join('');
+        if (value === undefined) {
+          const error = new Error(`选项 -${chars[i]} 需要一个值`);
+          error.userError = true;
+          throw error;
+        }
+        if (item.type === 'list') options[key] = options[key].concat([value]);
+        else if (item.type === 'number') options[key] = Number(value);
+        else options[key] = value;
+        break;
+      }
+      continue;
+    }
+
+    positionals.push(token);
+  }
+
+  return { options, positionals };
+}
+
+/**
+ * 多级子命令解析：第一个位置参数是动作名，剩余参数才按 spec 解析选项。
+ * 返回 { action, rest, options, positionals }
+ */
+function actionFirst(argv, spec = []) {
+  const list = argv.slice();
+  const actionSpec = {};
+  for (const item of Array.isArray(spec) ? spec : []) {
+    if (item && item.name) actionSpec[item.name] = item;
+  }
+  const defaultAction = spec && spec.action && typeof spec.action === 'string' ? spec.action : '';
+  const actionIndex = list.findIndex((token) => !token.startsWith('-'));
+  const action = actionIndex === -1 ? defaultAction : list[actionIndex];
+  if (actionIndex !== -1) list.splice(actionIndex, 1);
+  const parsed = parseArgs(list, Object.values(actionSpec));
+  return { action, rest: parsed.positionals, options: parsed.options, positionals: parsed.positionals };
+}
+
+/**
+ * 简单选择器：把用户输入按名称或序号解析成列表中的一项。
+ * 序号 1 表示第一项（与 /chat list 的输出保持一致）。
+ */
+function pick(list, ref, label = '项目') {
+  if (!list.length) return null;
+  if (!ref) return list[0];
+  const text = String(ref).trim();
+  const index = Number(text);
+  if (Number.isInteger(index) && index >= 1 && index <= list.length) return list[index - 1];
+  const exact = list.find((item) => item.id === text || String(item.name || '').toLowerCase() === text.toLowerCase());
+  if (exact) return exact;
+  const partial = list.filter(
+    (item) => String(item.id).startsWith(text) || String(item.name || '').toLowerCase().startsWith(text.toLowerCase())
+  );
+  if (partial.length === 1) return partial[0];
+  if (partial.length > 1) {
+    const error = new Error(`${label}名称不唯一：「${text}」匹配到 ${partial.length} 项`);
+    error.userError = true;
+    throw error;
+  }
+  return null;
+}
+
+module.exports = { parseArgs, actionFirst, pick, camel };

package/src/cli/command-table.js

@@ -0,0 +1,52 @@
+'use strict';
+
+/**
+ * 子命令注册表。每个命令：{ name, aliases, summary, usage, options, run(ctx) }
+ * ctx = { argv, io, config, print, error, exitCode }
+ */
+
+const chat = require('../commands/chat');
+const config = require('../commands/config');
+const provider = require('../commands/provider');
+const model = require('../commands/model');
+const oauth = require('../commands/oauth');
+const mcp = require('../commands/mcp');
+const skill = require('../commands/skill');
+const workspace = require('../commands/workspace');
+const permission = require('../commands/permission');
+const context = require('../commands/context');
+const run = require('../commands/run');
+const exportCmd = require('../commands/export');
+const doctor = require('../commands/doctor');
+
+const COMMANDS = [
+  run,
+  chat,
+  config,
+  provider,
+  model,
+  oauth,
+  mcp,
+  skill,
+  workspace,
+  permission,
+  context,
+  exportCmd,
+  doctor
+];
+
+function findCommand(name) {
+  const target = String(name || '').toLowerCase();
+  return (
+    COMMANDS.find((cmd) => cmd.name === target) ||
+    COMMANDS.find((cmd) => (cmd.aliases || []).includes(target)) ||
+    null
+  );
+}
+
+/** 汇总所有子命令的选项（用于全局解析后分发给命令自身） */
+function specFor(command) {
+  return command?.options || [];
+}
+
+module.exports = { COMMANDS, findCommand, specFor };