dgxay-code-cli 0.0.0-stage → 1.1.0

package/src/services/session-bus.js

@@ -0,0 +1,104 @@
+'use strict';
+
+/**
+ * 会话状态桥：让另一个终端里的子命令可以看到并使用当前交互会话的
+ * 工作空间、模型、权限、会话 id。
+ *
+ * 状态文件：~/.dcb/repl-state.json（0600，原子写入，每 5 秒刷新心跳）。
+ * 这样即使两个进程同时运行，也只会各自读一份最新状态，不互相覆盖配置。
+ */
+
+const fs = require('fs');
+const path = require('path');
+const { DCB_DIR } = require('./config');
+
+const STATE_PATH = path.join(DCB_DIR, 'repl-state.json');
+const HEARTBEAT_MS = 5000;
+const STALE_MS = 60000;
+
+function read() {
+  try {
+    const raw = fs.readFileSync(STATE_PATH, 'utf8');
+    const parsed = JSON.parse(raw);
+    return parsed && typeof parsed === 'object' ? parsed : null;
+  } catch {
+    return null;
+  }
+}
+
+function write(state) {
+  try {
+    fs.mkdirSync(DCB_DIR, { recursive: true, mode: 0o700 });
+    const tmp = `${STATE_PATH}.${process.pid}.tmp`;
+    fs.writeFileSync(tmp, `${JSON.stringify({ ...state, updatedAt: Date.now() }, null, 2)}\n`, {
+      encoding: 'utf8',
+      mode: 0o600
+    });
+    fs.renameSync(tmp, STATE_PATH);
+    return true;
+  } catch {
+    return false;
+  }
+}
+
+function clear(state = null) {
+  const current = state || read();
+  if (current && current.pid && current.pid !== process.pid) return false;
+  try {
+    if (fs.existsSync(STATE_PATH)) fs.unlinkSync(STATE_PATH);
+    return true;
+  } catch {
+    return false;
+  }
+}
+
+function pidAlive(pid) {
+  if (!Number.isInteger(pid) || pid <= 0) return false;
+  try {
+    process.kill(pid, 0);
+    return true;
+  } catch (err) {
+    return err.code === 'EPERM';
+  }
+}
+
+/** 读取仍然「活着」的会话状态 */
+function live() {
+  const state = read();
+  if (!state) return null;
+  if (!pidAlive(state.pid)) return null;
+  if (state.updatedAt && Date.now() - Number(state.updatedAt) > STALE_MS) return null;
+  return state;
+}
+
+/** 由 REPL 持有：周期性刷新心跳 */
+class Heartbeat {
+  constructor(getState) {
+    this.getState = getState;
+    this.timer = null;
+  }
+
+  start() {
+    this.stop();
+    const beat = () => {
+      try {
+        const state = this.getState();
+        if (state) write({ ...state, pid: process.pid });
+      } catch {
+        /* ignore */
+      }
+    };
+    beat();
+    this.timer = setInterval(beat, HEARTBEAT_MS);
+    this.timer.unref?.();
+  }
+
+  stop() {
+    if (this.timer) {
+      clearInterval(this.timer);
+      this.timer = null;
+    }
+  }
+}
+
+module.exports = { STATE_PATH, read, write, clear, live, Heartbeat, pidAlive, STALE_MS };

package/src/services/settings.js

@@ -0,0 +1,340 @@
+'use strict';
+
+/**
+ * 配置操作层：供应商 / 模型 / 权限 / 首选项的读写，供子命令与 REPL 共用。
+ */
+
+const fs = require('fs');
+const os = require('os');
+const path = require('path');
+const cfg = require('../services/config');
+const oauth = require('../core/oauth');
+const { PROTOCOLS } = require('../core/llm');
+const { resolveReviewer, resolveProviderAuth } = require('../core/prompt');
+const { listModels } = require('../core/llm');
+
+const PERMISSIONS = {
+  none: {
+    label: 'No permission',
+    short: 'read-only',
+    desc: 'Read-only access inside the workspace'
+  },
+  normal: {
+    label: 'Normal',
+    short: 'normal',
+    desc: 'All capabilities, but every operation asks you'
+  },
+  auto: {
+    label: 'Auto review',
+    short: 'auto',
+    desc: 'All capabilities; a reviewer model decides, suspicious ones ask you'
+  },
+  full: {
+    label: 'Full access',
+    short: 'full',
+    desc: 'Run commands, touch any file, unrestricted network'
+  }
+};
+
+const PROTOCOL_META = {
+  openai: {
+    short: 'ChatCompletions',
+    full: 'OpenAI ChatCompletions',
+    path: '/v1/chat/completions',
+    basePlaceholder: 'https://api.openai.com/v1',
+    modelPlaceholder: 'e.g. gpt-4o, deepseek-chat'
+  },
+  anthropic: {
+    short: 'Messages',
+    full: 'Anthropic Messages',
+    path: '/v1/messages',
+    basePlaceholder: 'https://api.anthropic.com',
+    modelPlaceholder: 'e.g. claude-sonnet-4-5'
+  },
+  responses: {
+    short: 'Responses',
+    full: 'OpenAI Responses API',
+    path: '/v1/responses',
+    basePlaceholder: 'https://api.openai.com/v1',
+    modelPlaceholder: 'e.g. gpt-5, o4-mini'
+  }
+};
+
+function newId(prefix) {
+  return `${prefix}_${Date.now().toString(36)}_${Math.random().toString(36).slice(2, 6)}`;
+}
+
+function readConfig() {
+  return cfg.readConfig();
+}
+
+function saveConfig(patch) {
+  return cfg.saveConfig(patch || {});
+}
+
+function providers() {
+  return cfg.readConfig().providers || [];
+}
+
+function findProvider(ref) {
+  if (!ref) return null;
+  const list = providers();
+  const text = String(ref).trim();
+  return (
+    list.find((p) => p.id === text) ||
+    list.find((p) => String(p.name).toLowerCase() === text.toLowerCase()) ||
+    null
+  );
+}
+
+function activeProvider(config = readConfig()) {
+  const list = config.providers || [];
+  return list.find((p) => p.id === config.activeProviderId) || list[0] || null;
+}
+
+function activeModel(config = readConfig()) {
+  const provider = activeProvider(config);
+  if (!provider) return null;
+  const models = provider.models || [];
+  const model = models.find((m) => m.id === config.activeModelId) || models[0];
+  if (!model) return null;
+  return {
+    id: model.id,
+    name: model.name || model.id,
+    providerId: provider.id,
+    providerName: provider.name,
+    protocol: provider.protocol || 'openai',
+    contextWindow: Number(model.contextWindow) || 0
+  };
+}
+
+function availablePermissions(config = readConfig()) {
+  const list = ['none', 'normal'];
+  if (resolveReviewer(config, config.activeProviderId)) list.push('auto');
+  list.push('full');
+  return list;
+}
+
+function hasReviewModel(config = readConfig()) {
+  return Boolean(resolveReviewer(config, config.activeProviderId));
+}
+
+function currentPermission(config = readConfig()) {
+  const value = config.ui?.permission || 'normal';
+  if (value === 'auto' && !hasReviewModel(config)) return 'normal';
+  return PERMISSIONS[value] ? value : 'normal';
+}
+
+function currentMode(config = readConfig()) {
+  return config.ui?.mode === 'work' ? 'work' : 'chat';
+}
+
+/**
+ * 新增或修改供应商。
+ * @param {object} input { ref, name, protocol, baseUrl, apiKey, authMode, oauthProvider, reviewModelId, models }
+ */
+function upsertProvider(input = {}) {
+  const config = readConfig();
+  const list = config.providers || [];
+  const existing = input.ref ? findProvider(input.ref) : null;
+  if (input.ref && !existing) throw new Error(`Provider not found: ${input.ref}`);
+
+  const protocol = PROTOCOLS.includes(input.protocol) ? input.protocol : existing?.protocol || 'openai';
+  const baseUrl = String(input.baseUrl || existing?.baseUrl || '').trim().replace(/\/+$/, '');
+  if (!/^https?:\/\//i.test(baseUrl)) throw new Error('Base URL must start with http:// or https://');
+
+  const models = normalizeModels(input.models, existing?.models);
+  if (!models.length) throw new Error('Provide at least one model id (--model)');
+
+  const authMode = input.authMode === 'oauth' ? 'oauth' : 'apiKey';
+  const oauthProvider = authMode === 'oauth' ? String(input.oauthProvider || '').trim() : '';
+  if (authMode === 'oauth') {
+    if (!oauthProvider) throw new Error('Pass --oauth-provider (openai/anthropic/xai/google) to use an OAuth account');
+    const account = oauth.listProviders().find((item) => item.id === oauthProvider);
+    if (!account?.connection || account.connection.expired) {
+      throw new Error(`${oauthProvider} is not connected or the login expired. Run dgxay oauth connect ${oauthProvider}`);
+    }
+  }
+
+  const record = {
+    id: existing?.id || newId('prov'),
+    name: String(input.name || existing?.name || 'My provider').trim(),
+    protocol,
+    baseUrl,
+    apiKey: input.apiKey !== undefined ? String(input.apiKey) : existing?.apiKey || '',
+    authMode,
+    oauthProvider,
+    reviewModelId:
+      input.reviewModelId !== undefined
+        ? String(input.reviewModelId || '')
+        : existing?.reviewModelId || models[0].id,
+    models
+  };
+  if (!models.some((m) => m.id === record.reviewModelId)) record.reviewModelId = models[0].id;
+
+  const next = list.map((p) => (p.id === record.id ? { ...p, ...record } : p));
+  if (!existing) next.push(record);
+
+  const patch = { providers: next };
+  if (!config.activeProviderId || existing?.id === config.activeProviderId) {
+    patch.activeProviderId = record.id;
+    const stillValid = models.find((m) => m.id === config.activeModelId);
+    if (existing?.id !== config.activeProviderId || !stillValid) patch.activeModelId = record.reviewModelId;
+  }
+  return { provider: record, config: saveConfig(patch), created: !existing };
+}
+
+function normalizeModels(input, fallback = []) {
+  let list = input;
+  if (typeof list === 'string') list = list.split(',').map((item) => item.trim()).filter(Boolean);
+  if (!Array.isArray(list) || !list.length) {
+    list = (fallback || []).map((m) => ({ id: m.id, name: m.name, contextWindow: m.contextWindow }));
+  }
+  const out = [];
+  for (const item of list) {
+    const model =
+      typeof item === 'string' ? { id: item } : { id: item?.id, name: item?.name, contextWindow: item?.contextWindow };
+    const id = String(model.id || '').trim();
+    if (!id) continue;
+    if (out.find((m) => m.id === id)) continue;
+    out.push({ id, name: String(model.name || id).trim(), contextWindow: Number(model.contextWindow) || 0 });
+  }
+  return out;
+}
+
+function removeProvider(ref) {
+  const config = readConfig();
+  const target = findProvider(ref);
+  if (!target) throw new Error(`Provider not found: ${ref}`);
+  const next = (config.providers || []).filter((p) => p.id !== target.id);
+  const patch = { providers: next };
+  if (config.activeProviderId === target.id) {
+    patch.activeProviderId = next[0]?.id || '';
+    patch.activeModelId = next[0]?.models?.[0]?.id || '';
+  }
+  return saveConfig(patch);
+}
+
+function moveProvider(ref, direction) {
+  const config = readConfig();
+  const list = (config.providers || []).slice();
+  const target = findProvider(ref);
+  if (!target) throw new Error(`Provider not found: ${ref}`);
+  const index = list.findIndex((p) => p.id === target.id);
+  const nextIndex = direction < 0 ? index - 1 : index + 1;
+  if (nextIndex < 0 || nextIndex >= list.length) return config;
+  const [item] = list.splice(index, 1);
+  list.splice(nextIndex, 0, item);
+  return saveConfig({ providers: list });
+}
+
+function setActiveModel(providerRef, modelRef) {
+  const config = readConfig();
+  const provider = providerRef ? findProvider(providerRef) : activeProvider(config);
+  if (!provider) throw new Error('No provider available');
+  const models = provider.models || [];
+  const model = models.find((m) => m.id === modelRef) || models.find((m) => m.id.includes(String(modelRef)));
+  if (!model) throw new Error(`Provider ${provider.name} has no model ${modelRef}`);
+  return saveConfig({ activeProviderId: provider.id, activeModelId: model.id });
+}
+
+function setPermission(permission) {
+  if (!PERMISSIONS[permission]) throw new Error(`Unknown permission: ${permission}`);
+  return saveConfig({ ui: { permission } });
+}
+
+function setMode(mode) {
+  return saveConfig({ ui: { mode: mode === 'work' ? 'work' : 'chat' } });
+}
+
+function setChatOptions(patch) {
+  return saveConfig({ chat: patch });
+}
+
+function setWorkspace(dir) {
+  const config = readConfig();
+  const target = dir ? path.resolve(dir) : '';
+  if (target) {
+    const stat = fs.statSync(target);
+    if (!stat.isDirectory()) throw new Error(`${target} is not a directory`);
+  }
+  const recents = (config.work?.recentWorkspaces || []).filter((item) => item !== target);
+  if (target) recents.unshift(target);
+  return saveConfig({
+    work: {
+      workspace: target,
+      recentWorkspaces: recents.slice(0, 12)
+    }
+  });
+}
+
+/** 拉取模型列表（失败不阻断，允许手动填写） */
+async function fetchModels({ providerRef, baseUrl, apiKey, protocol }) {
+  const config = readConfig();
+  const provider = providerRef ? findProvider(providerRef) : activeProvider(config);
+  const targetBaseUrl = baseUrl || provider?.baseUrl;
+  const targetProtocol = protocol || provider?.protocol || 'openai';
+  if (!targetBaseUrl) throw new Error('Missing Base URL');
+  let token = apiKey ?? provider?.apiKey ?? '';
+  let authType = 'apiKey';
+  if (provider && provider.authMode === 'oauth' && provider.oauthProvider && apiKey === undefined) {
+    token = await oauth.getAccessToken(provider.oauthProvider);
+    authType = 'oauth';
+  }
+  const controller = new AbortController();
+  const timer = setTimeout(() => controller.abort(), 20000);
+  try {
+    return await listModels({ baseUrl: targetBaseUrl, apiKey: token, protocol: targetProtocol, authType, signal: controller.signal });
+  } finally {
+    clearTimeout(timer);
+  }
+}
+
+async function authOf(provider) {
+  return resolveProviderAuth(provider);
+}
+
+function paths() {
+  return {
+    dataDir: cfg.DCB_DIR,
+    config: cfg.CONFIG_PATH,
+    chats: cfg.CHATS_PATH,
+    mcp: cfg.MCP_PATH,
+    skills: cfg.SKILLS_DIR,
+    oauth: cfg.OAUTH_PATH
+  };
+}
+
+function homeRelative(target) {
+  const home = os.homedir();
+  return target && String(target).startsWith(home) ? `~${String(target).slice(home.length)}` : target;
+}
+
+module.exports = {
+  PERMISSIONS,
+  PROTOCOL_META,
+  readConfig,
+  saveConfig,
+  providers,
+  findProvider,
+  activeProvider,
+  activeModel,
+  availablePermissions,
+  hasReviewModel,
+  currentPermission,
+  currentMode,
+  upsertProvider,
+  normalizeModels,
+  removeProvider,
+  moveProvider,
+  setActiveModel,
+  setPermission,
+  setMode,
+  setChatOptions,
+  setWorkspace,
+  fetchModels,
+  authOf,
+  paths,
+  homeRelative,
+  newId
+};

package/src/services/skills.js

@@ -0,0 +1,125 @@
+'use strict';
+
+/**
+ * 技能扫描：~/.dcb/skills
+ *  - 含 SKILL.md 的文件夹
+ *  - 单个 name.md
+ * 支持 name / description frontmatter；单次注入上限 40000 字符。
+ */
+
+const fs = require('fs');
+const path = require('path');
+const { SKILLS_DIR } = require('./config');
+
+const MAX_SKILL_CONTENT = 40000;
+
+function ensureSkillsDir() {
+  try {
+    fs.mkdirSync(SKILLS_DIR, { recursive: true, mode: 0o700 });
+  } catch {
+    /* ignore */
+  }
+}
+
+function parseFrontmatter(text) {
+  const normalized = String(text || '').replace(/\r\n/g, '\n');
+  if (!normalized.startsWith('---\n')) {
+    return { meta: {}, body: normalized };
+  }
+  const end = normalized.indexOf('\n---', 4);
+  if (end === -1) return { meta: {}, body: normalized };
+  const raw = normalized.slice(4, end);
+  const meta = {};
+  for (const line of raw.split('\n')) {
+    const idx = line.indexOf(':');
+    if (idx === -1) continue;
+    const key = line.slice(0, idx).trim();
+    let value = line.slice(idx + 1).trim();
+    if ((value.startsWith('"') && value.endsWith('"')) || (value.startsWith("'") && value.endsWith("'"))) {
+      value = value.slice(1, -1);
+    }
+    if (key) meta[key] = value;
+  }
+  return { meta, body: normalized.slice(end + 4).replace(/^\n+/, '') };
+}
+
+function describe(body) {
+  const lines = String(body || '')
+    .split('\n')
+    .map((line) => line.trim())
+    .filter(Boolean)
+    .filter((line) => !line.startsWith('#') && !line.startsWith('```') && !line.startsWith('---'));
+  const text = lines.join(' ');
+  return text.length > 160 ? `${text.slice(0, 160)}…` : text;
+}
+
+function candidateFiles() {
+  ensureSkillsDir();
+  let entries = [];
+  try {
+    entries = fs.readdirSync(SKILLS_DIR, { withFileTypes: true });
+  } catch {
+    return [];
+  }
+  const files = [];
+  for (const entry of entries) {
+    if (entry.name.startsWith('.')) continue;
+    const full = path.join(SKILLS_DIR, entry.name);
+    if (entry.isDirectory()) {
+      const skillFile = path.join(full, 'SKILL.md');
+      if (fs.existsSync(skillFile)) files.push({ file: skillFile, fallbackName: entry.name });
+    } else if (entry.isFile() && /\.md$/i.test(entry.name)) {
+      files.push({ file: full, fallbackName: entry.name.replace(/\.md$/i, '') });
+    }
+  }
+  return files;
+}
+
+function listSkills() {
+  const skills = [];
+  for (const { file, fallbackName } of candidateFiles()) {
+    let text = '';
+    try {
+      text = fs.readFileSync(file, 'utf8');
+    } catch {
+      continue;
+    }
+    const { meta, body } = parseFrontmatter(text);
+    const name = String(meta.name || fallbackName).trim();
+    if (!name) continue;
+    skills.push({
+      name,
+      description: String(meta.description || describe(body) || '').trim(),
+      path: file
+    });
+  }
+  skills.sort((a, b) => a.name.localeCompare(b.name));
+  return skills;
+}
+
+function readSkill(name) {
+  const target = String(name || '').trim().toLowerCase();
+  if (!target) return null;
+  for (const { file, fallbackName } of candidateFiles()) {
+    let text = '';
+    try {
+      text = fs.readFileSync(file, 'utf8');
+    } catch {
+      continue;
+    }
+    const { meta, body } = parseFrontmatter(text);
+    const skillName = String(meta.name || fallbackName).trim();
+    if (skillName.toLowerCase() !== target) continue;
+    const content =
+      body.length > MAX_SKILL_CONTENT ? `${body.slice(0, MAX_SKILL_CONTENT)}\n\n… (skill content truncated)` : body;
+    return {
+      name: skillName,
+      description: String(meta.description || describe(body) || '').trim(),
+      content,
+      path: file
+    };
+  }
+  return null;
+}
+
+module.exports = { SKILLS_DIR, listSkills, readSkill, ensureSkillsDir };

package/src/services/terminal.js

@@ -0,0 +1,124 @@
+'use strict';
+
+/**
+ * 终端命令执行（对应桌面版的 `!` 命令）。
+ *
+ * 与工具 run_command 的区别：这是用户自己发起的命令，输出会展示并进入对话上下文。
+ * 权限与工具一致：none 拒绝；normal 每次询问；auto 先审查；full 直接执行。
+ * 同一时间只允许一个命令；超过 5 分钟自动终止；累计输出超过 300000 字符截断。
+ */
+
+const { spawn, execFile } = require('child_process');
+const os = require('os');
+const fs = require('fs');
+
+const MAX_TOTAL = 300000;
+const TIMEOUT_MS = 300000;
+
+function killProcessTree(child) {
+  if (!child || !child.pid) return;
+  if (process.platform === 'win32') {
+    execFile('taskkill', ['/pid', String(child.pid), '/T', '/F'], () => {});
+    return;
+  }
+  try {
+    child.kill('SIGTERM');
+  } catch {
+    /* ignore */
+  }
+  setTimeout(() => {
+    try {
+      if (!child.killed) child.kill('SIGKILL');
+    } catch {
+      /* ignore */
+    }
+  }, 2000).unref?.();
+}
+
+function resolveCwd(workspace) {
+  const target = workspace || os.homedir();
+  try {
+    return fs.statSync(target).isDirectory() ? target : os.homedir();
+  } catch {
+    return os.homedir();
+  }
+}
+
+/**
+ * 运行命令。
+ * @param {object} options
+ * @param {string} options.command
+ * @param {string} [options.workspace]
+ * @param {(chunk: string, stream: 'stdout'|'stderr') => void} [options.onOutput]
+ * @returns {Promise<{ ok:boolean, code:number|null, signal:string|null, output:string, stderr:string, aborted:boolean, error?:string }>}
+ */
+function runShellCommand({ command, workspace, onOutput }) {
+  return new Promise((resolve) => {
+    const isWindows = process.platform === 'win32';
+    const shellCommand = isWindows ? `chcp 65001>nul & ${command}` : command;
+    const cwd = resolveCwd(workspace);
+    let child;
+    try {
+      child = spawn(shellCommand, {
+        cwd,
+        shell: true,
+        windowsHide: true,
+        env: { ...process.env },
+        stdio: ['ignore', 'pipe', 'pipe']
+      });
+    } catch (err) {
+      resolve({ ok: false, code: null, signal: null, output: '', stderr: '', aborted: false, error: err.message });
+      return;
+    }
+
+    let output = '';
+    let stderr = '';
+    let total = 0;
+    let truncated = false;
+    let finished = false;
+
+    const push = (stream, chunk) => {
+      const text = chunk.toString('utf8').replace(/\x1b\[[0-9;?]*[ -/]*[@-~]/g, '');
+      total += text.length;
+      if (total > MAX_TOTAL) {
+        if (!truncated) {
+          truncated = true;
+          const note = '\n… (output too long, truncated)\n';
+          stderr += note;
+          onOutput?.(note, 'stderr');
+        }
+        return;
+      }
+      if (stream === 'stderr') stderr += text;
+      else output += text;
+      onOutput?.(text, stream);
+    };
+
+    child.stdout?.on('data', (chunk) => push('stdout', chunk));
+    child.stderr?.on('data', (chunk) => push('stderr', chunk));
+
+    const finish = (code, signal) => {
+      if (finished) return;
+      finished = true;
+      clearTimeout(timer);
+      resolve({ ok: true, code, signal, output, stderr, aborted: signal === 'SIGTERM' || signal === 'SIGKILL' });
+    };
+
+    child.on('error', (err) => {
+      onOutput?.(`\n[failed to start] ${err.message}\n`, 'stderr');
+      finish(-1, null);
+    });
+    child.on('exit', (code, signal) => finish(code, signal));
+
+    const timer = setTimeout(() => {
+      onOutput?.('\n… (command ran longer than 5 minutes, terminated)\n', 'stderr');
+      killProcessTree(child);
+    }, TIMEOUT_MS);
+    child.on('exit', () => clearTimeout(timer));
+    if (child.unref) {
+      /* 保持引用，等待退出 */
+    }
+  });
+}
+
+module.exports = { runShellCommand, killProcessTree, MAX_TOTAL, TIMEOUT_MS };