@seandong/seno 0.1.0

package/dist/cli/commands.js

@@ -0,0 +1,667 @@
+import * as crypto from 'crypto';
+import * as fs from 'fs';
+import * as path from 'path';
+import { execSync } from 'child_process';
+import chalk from 'chalk';
+import ora from 'ora';
+import { loadServers, saveServers } from '../config/manager.js';
+import { syncSkills } from '../skills/sync.js';
+import { CancelError, promptQuestion, promptChoice } from './prompt.js';
+import { modelInfo, modelSet, modelAuth } from '../commands/model.js';
+export const COMMAND_TREE = [
+    {
+        name: 'skills',
+        description: '管理 Skills',
+        subCommands: [
+            { name: 'list', description: '列出所有 Skills' },
+            { name: 'sync', description: '手动同步远程 Skills' },
+        ],
+    },
+    {
+        name: 'server',
+        description: '管理 SSH 服务器',
+        subCommands: [
+            { name: 'list', description: '列出所有服务器' },
+            { name: 'add', description: '添加新服务器' },
+            { name: 'remove', description: '删除指定服务器' },
+        ],
+    },
+    {
+        name: 'plugin',
+        description: 'ONES 插件开发助手',
+        subCommands: [
+            { name: 'init', description: '初始化新插件项目' },
+        ],
+    },
+    {
+        name: 'model',
+        description: '管理 LLM 模型配置',
+        subCommands: [
+            { name: 'set', description: '切换 Provider 和模型' },
+            { name: 'auth', description: '配置 Provider 授权' },
+        ],
+    },
+    { name: 'clear', description: '清空对话历史' },
+    { name: 'exit', description: '退出 SENO CLI' },
+    { name: 'help', description: '显示帮助信息' },
+];
+/**
+ * 根据当前输入计算候选列表
+ * @param input 当前输入（含 /），保留尾部空格用于触发子命令候选
+ */
+export function getCommandCandidates(input) {
+    if (!input.startsWith('/'))
+        return [];
+    const body = input.slice(1); // 去掉 /
+    const hasSpace = body.includes(' ');
+    if (!hasSpace) {
+        // 主命令阶段："/", "/he", "/server"（无空格）
+        const prefix = body.toLowerCase();
+        return COMMAND_TREE.filter((cmd) => cmd.name.startsWith(prefix)).map((cmd) => ({
+            text: `/${cmd.name}`,
+            description: cmd.description,
+        }));
+    }
+    // 子命令阶段："/server ", "/server li" 等（含空格）
+    const spaceIndex = body.indexOf(' ');
+    const mainName = body.slice(0, spaceIndex).toLowerCase();
+    const subInput = body.slice(spaceIndex + 1).trimStart().toLowerCase();
+    const mainCmd = COMMAND_TREE.find((c) => c.name === mainName);
+    if (!mainCmd?.subCommands)
+        return [];
+    return mainCmd.subCommands
+        .filter((sub) => sub.name.startsWith(subInput))
+        .map((sub) => ({
+        text: `/${mainName} ${sub.name}`,
+        description: sub.description,
+    }));
+}
+// ─── Command Execution ───────────────────────────────────────
+/**
+ * 判断输入是否为斜杠命令
+ */
+export function isSlashCommand(input) {
+    return input.startsWith('/');
+}
+/**
+ * 处理斜杠命令
+ * @returns false 表示应退出 REPL 循环
+ */
+export async function handleSlashCommand(input, context) {
+    const parts = input.trim().split(/\s+/);
+    const command = parts[0].toLowerCase();
+    switch (command) {
+        case '/help':
+            printHelp();
+            return true;
+        case '/model':
+            return handleModelCommand(parts.slice(1), context);
+        case '/skills':
+            return handleSkillsCommand(parts.slice(1), context);
+        case '/server':
+            return handleServerCommand(parts.slice(1), context);
+        case '/plugin':
+            return handlePluginCommand(parts.slice(1), context);
+        case '/clear':
+            context.agentLoop.clearHistory();
+            console.log(chalk.green('  ✓ 对话历史已清空'));
+            return true;
+        case '/exit':
+            console.log(chalk.gray('  再见 👋'));
+            return false;
+        default:
+            console.log(chalk.yellow(`  未知命令: ${command}`) +
+                chalk.gray('，输入 /help 查看可用命令'));
+            return true;
+    }
+}
+// ─── /help ────────────────────────────────────────────────────
+function printHelp() {
+    console.log();
+    console.log(chalk.bold('  可用命令:'));
+    console.log();
+    for (const cmd of COMMAND_TREE) {
+        if (cmd.subCommands) {
+            for (const sub of cmd.subCommands) {
+                const full = `/${cmd.name} ${sub.name}`;
+                console.log(`  ${chalk.cyan(full.padEnd(22))}${sub.description}`);
+            }
+        }
+        else {
+            console.log(`  ${chalk.cyan(`/${cmd.name}`.padEnd(22))}${cmd.description}`);
+        }
+    }
+    console.log();
+    console.log(chalk.gray('  直接输入文本与 AI Agent 对话'));
+    console.log();
+}
+// ─── /skills ──────────────────────────────────────────────────
+function printSkills(skills) {
+    if (skills.length === 0) {
+        console.log(chalk.yellow('\n  没有可用的 Skills\n'));
+        return;
+    }
+    console.log();
+    console.log(chalk.bold('  Available Skills:'));
+    console.log();
+    // 计算列宽
+    const nameWidth = Math.max(6, ...skills.map((s) => s.name.length));
+    const versionWidth = Math.max(7, ...skills.map((s) => (s.version || '-').length));
+    const domainWidth = Math.max(8, ...skills.map((s) => s.domain.length));
+    // 表头
+    console.log(`  ${chalk.gray('Name'.padEnd(nameWidth + 2))}` +
+        `${chalk.gray('Version'.padEnd(versionWidth + 2))}` +
+        `${chalk.gray('Domain'.padEnd(domainWidth + 2))}` +
+        `${chalk.gray('Description')}`);
+    console.log(chalk.gray(`  ${'─'.repeat(nameWidth + versionWidth + domainWidth + 46)}`));
+    // 行
+    for (const skill of skills) {
+        const desc = skill.description.length > 40
+            ? skill.description.slice(0, 40) + '...'
+            : skill.description;
+        const ver = skill.version || '-';
+        console.log(`  ${skill.name.padEnd(nameWidth + 2)}` +
+            `${chalk.green(ver.padEnd(versionWidth + 2))}` +
+            `${chalk.cyan(skill.domain.padEnd(domainWidth + 2))}` +
+            `${chalk.gray(desc)}`);
+    }
+    console.log();
+}
+// ─── /skills ─────────────────────────────────────────────────
+async function handleSkillsCommand(args, context) {
+    const subCommand = args[0]?.toLowerCase();
+    switch (subCommand) {
+        case 'sync':
+            return skillsSync(context);
+        case 'list':
+        case undefined:
+            printSkills(context.skills);
+            return true;
+        default:
+            console.log(chalk.yellow(`  未知子命令: ${subCommand}`) +
+                chalk.gray('，可用: list, sync'));
+            return true;
+    }
+}
+async function skillsSync(context) {
+    const spinner = ora({ text: '同步 Skills...', indent: 2, discardStdin: false });
+    spinner.start();
+    try {
+        const result = await syncSkills(context.config);
+        if (result.success) {
+            spinner.succeed(`Skills 已同步 (${result.skillCount} 个技能)`);
+            // 重新加载 skills
+            context.skills = await context.reloadSkills();
+        }
+        else {
+            spinner.fail(`Skills 同步失败: ${result.error}`);
+        }
+    }
+    catch (error) {
+        spinner.fail(`Skills 同步失败: ${error instanceof Error ? error.message : String(error)}`);
+    }
+    return true;
+}
+// ─── /model ──────────────────────────────────────────────────
+async function handleModelCommand(args, context) {
+    const subCommand = args[0]?.toLowerCase();
+    try {
+        switch (subCommand) {
+            case 'set':
+                await modelSet(context.rl, context.config, context.switchLLMProvider);
+                return true;
+            case 'auth':
+                await modelAuth(context.rl, context.config, context.switchLLMProvider);
+                return true;
+            case undefined:
+                await modelInfo(context.config);
+                return true;
+            default:
+                console.log(chalk.yellow(`  未知子命令: ${subCommand}`) +
+                    chalk.gray('，可用: set, auth'));
+                return true;
+        }
+    }
+    catch (error) {
+        if (error instanceof CancelError) {
+            console.log();
+            return true;
+        }
+        throw error;
+    }
+}
+// ─── /server ──────────────────────────────────────────────────
+async function handleServerCommand(args, context) {
+    const subCommand = args[0]?.toLowerCase();
+    switch (subCommand) {
+        case 'list':
+        case undefined:
+            await serverList();
+            return true;
+        case 'add':
+            return serverAdd(context.rl);
+        case 'remove': {
+            const name = args[1];
+            if (!name) {
+                console.log(chalk.yellow('  用法: /server remove <name>'));
+                return true;
+            }
+            await serverRemove(name);
+            return true;
+        }
+        default:
+            console.log(chalk.yellow(`  未知子命令: ${subCommand}`) +
+                chalk.gray('，可用: list, add, remove'));
+            return true;
+    }
+}
+async function serverList() {
+    const { servers } = await loadServers();
+    if (servers.length === 0) {
+        console.log(chalk.yellow('\n  没有已配置的服务器，使用 /server add 添加\n'));
+        return;
+    }
+    console.log();
+    console.log(chalk.bold('  SSH 服务器:'));
+    console.log();
+    const nameWidth = Math.max(6, ...servers.map((s) => s.name.length));
+    const hostWidth = Math.max(6, ...servers.map((s) => s.host.length));
+    const userWidth = Math.max(6, ...servers.map((s) => s.user.length));
+    const authWidth = 10;
+    console.log(`  ${chalk.gray('Name'.padEnd(nameWidth + 2))}` +
+        `${chalk.gray('Host'.padEnd(hostWidth + 2))}` +
+        `${chalk.gray('User'.padEnd(userWidth + 2))}` +
+        `${chalk.gray('Port')}  ` +
+        `${chalk.gray('Auth'.padEnd(authWidth))}` +
+        `${chalk.gray('Description')}`);
+    console.log(chalk.gray(`  ${'─'.repeat(nameWidth + hostWidth + userWidth + authWidth + 30)}`));
+    for (const s of servers) {
+        const auth = s.auth_type === 'password' ? 'password' : 'ssh-key';
+        console.log(`  ${s.name.padEnd(nameWidth + 2)}` +
+            `${s.host.padEnd(hostWidth + 2)}` +
+            `${s.user.padEnd(userWidth + 2)}` +
+            `${String(s.port).padEnd(6)}` +
+            `${auth.padEnd(authWidth)}` +
+            `${chalk.gray(s.description || '')}`);
+    }
+    console.log();
+}
+async function serverAdd(rl) {
+    try {
+        console.log();
+        console.log(chalk.gray('  添加服务器 [按 ESC 键取消]'));
+        console.log();
+        const name = await promptQuestion(rl, chalk.cyan('  Server name: '));
+        if (!name) {
+            console.log(chalk.yellow('  服务器名称不能为空'));
+            return true;
+        }
+        const host = await promptQuestion(rl, chalk.cyan('  Host: '));
+        if (!host) {
+            console.log(chalk.yellow('  主机地址不能为空'));
+            return true;
+        }
+        const userInput = await promptQuestion(rl, chalk.cyan('  User [root]: '));
+        const user = userInput || 'root';
+        const portInput = await promptQuestion(rl, chalk.cyan('  Port [22]: '));
+        const port = portInput ? parseInt(portInput, 10) : 22;
+        if (isNaN(port)) {
+            console.log(chalk.yellow('  端口号无效'));
+            return true;
+        }
+        // 认证方式选择
+        console.log();
+        console.log(chalk.cyan('  Auth method:'));
+        const authIdx = await promptChoice(rl, '', [
+            'SSH Key',
+            'Password',
+        ], 0);
+        const authType = authIdx === 1 ? 'password' : 'ssh-key';
+        let identity_file;
+        let password;
+        if (authType === 'ssh-key') {
+            const identityInput = await promptQuestion(rl, chalk.cyan('  Identity file [~/.ssh/id_rsa]: '));
+            identity_file = identityInput || '~/.ssh/id_rsa';
+        }
+        else {
+            password = await promptQuestion(rl, chalk.cyan('  Password: '));
+            if (!password) {
+                console.log(chalk.yellow('  密码不能为空'));
+                return true;
+            }
+        }
+        const description = await promptQuestion(rl, chalk.cyan('  Description: '));
+        // 保存
+        const config = await loadServers();
+        // 检查重名
+        if (config.servers.some((s) => s.name === name)) {
+            console.log(chalk.yellow(`\n  服务器 "${name}" 已存在，请使用其他名称\n`));
+            return true;
+        }
+        config.servers.push({
+            name,
+            host,
+            user,
+            port,
+            auth_type: authType,
+            identity_file,
+            password,
+            description,
+        });
+        await saveServers(config);
+        console.log(chalk.green(`\n  ✓ Server "${name}" added successfully\n`));
+        return true;
+    }
+    catch (error) {
+        if (error instanceof CancelError) {
+            // ESC 取消，静默返回
+            console.log();
+            return true;
+        }
+        throw error;
+    }
+}
+async function serverRemove(name) {
+    const config = await loadServers();
+    const index = config.servers.findIndex((s) => s.name === name);
+    if (index === -1) {
+        console.log(chalk.yellow(`\n  未找到服务器: ${name}\n`));
+        return;
+    }
+    config.servers.splice(index, 1);
+    await saveServers(config);
+    console.log(chalk.green(`\n  ✓ Server "${name}" removed\n`));
+}
+// ─── /plugin ─────────────────────────────────────────────────
+/**
+ * 安全执行 shell 命令，返回 stdout 或 null（失败时）
+ */
+function runCommand(cmd, timeout = 15000) {
+    try {
+        return execSync(cmd, {
+            encoding: 'utf-8',
+            timeout,
+            stdio: ['pipe', 'pipe', 'pipe'],
+        }).trim();
+    }
+    catch {
+        return null;
+    }
+}
+/** 获取 nvm 目录路径 */
+function getNvmDir() {
+    return process.env.NVM_DIR || path.join(process.env.HOME || '~', '.nvm');
+}
+/** 检查 nvm 是否已安装 */
+function isNvmInstalled() {
+    const nvmScript = path.join(getNvmDir(), 'nvm.sh');
+    try {
+        fs.accessSync(nvmScript);
+        return true;
+    }
+    catch {
+        return false;
+    }
+}
+/** 通过 nvm 执行命令（先 source nvm.sh），通过 stdin 传入脚本避免引号问题 */
+function runWithNvm(cmd) {
+    const nvmDir = getNvmDir();
+    const script = `source "${nvmDir}/nvm.sh"\n${cmd}`;
+    try {
+        return execSync('bash', {
+            input: script,
+            encoding: 'utf-8',
+            timeout: 300000,
+            stdio: ['pipe', 'pipe', 'pipe'],
+        }).trim();
+    }
+    catch {
+        return null;
+    }
+}
+/** 检查 nvm 中是否已安装 Node.js 18，返回版本号或 null */
+function getNode18Version() {
+    const result = runWithNvm('nvm ls 18');
+    if (!result)
+        return null;
+    const match = result.match(/v(18\.\d+\.\d+)/);
+    return match ? match[1] : null;
+}
+/** 确保 nvm 已安装，返回是否成功 */
+async function ensureNvm() {
+    if (isNvmInstalled())
+        return true;
+    const spinner = ora({ text: '安装 nvm...', indent: 2, discardStdin: false });
+    spinner.start();
+    try {
+        execSync('curl -o- https://raw.githubusercontent.com/nvm-sh/nvm/v0.40.1/install.sh | bash', { encoding: 'utf-8', stdio: ['pipe', 'pipe', 'pipe'], timeout: 120000 });
+        if (!isNvmInstalled()) {
+            spinner.fail('nvm 安装失败');
+            return false;
+        }
+        spinner.succeed('nvm 安装成功');
+        return true;
+    }
+    catch {
+        spinner.fail('nvm 安装失败');
+        return false;
+    }
+}
+/** 确保 nvm 中存在 Node.js 18（仅安装，不切换当前版本） */
+async function ensureNode18() {
+    // 1. 确保 nvm 已安装
+    if (!(await ensureNvm()))
+        return false;
+    // 2. 检查 nvm 中是否已有 v18
+    const existing = getNode18Version();
+    if (existing) {
+        console.log(chalk.green('  ✓ ') + `${'Node.js 18'.padEnd(12)}${chalk.gray(`v${existing}`)}`);
+        return true;
+    }
+    // 3. nvm 中没有 v18，安装
+    const spinner = ora({ text: '通过 nvm 安装 Node.js 18...', indent: 2, discardStdin: false });
+    spinner.start();
+    runWithNvm('nvm install 18');
+    const installed = getNode18Version();
+    if (installed) {
+        spinner.succeed(`Node.js 18 安装成功 ${chalk.gray(`v${installed}`)}`);
+        return true;
+    }
+    spinner.fail('Node.js 18 安装失败');
+    return false;
+}
+/** 自动安装 ONES CLI */
+async function installOnesCli() {
+    const spinner = ora({ text: '安装 ONES CLI...', indent: 2, discardStdin: false });
+    spinner.start();
+    try {
+        execSync('npm install -g @ones/cli --registry=https://npm.partner.ones.cn/registry/', { encoding: 'utf-8', stdio: ['pipe', 'pipe', 'pipe'], timeout: 120000 });
+        const version = runCommand('ones --version');
+        if (version) {
+            spinner.succeed(`ONES CLI 安装成功 ${chalk.gray(version)}`);
+            return true;
+        }
+    }
+    catch { /* fall through */ }
+    spinner.fail('ONES CLI 安装失败');
+    console.log(chalk.gray('  手动安装: npm install -g @ones/cli --registry=https://npm.partner.ones.cn/registry/'));
+    return false;
+}
+/** 自动安装 npx（通过 npm） */
+async function installNpx() {
+    const spinner = ora({ text: '安装 npx...', indent: 2, discardStdin: false });
+    spinner.start();
+    try {
+        execSync('npm install -g npx', {
+            encoding: 'utf-8',
+            stdio: ['pipe', 'pipe', 'pipe'],
+            timeout: 120000,
+        });
+        const version = runCommand('npx --version');
+        if (version) {
+            spinner.succeed(`npx 安装成功 ${chalk.gray(version)}`);
+            return true;
+        }
+    }
+    catch { /* fall through */ }
+    spinner.fail('npx 安装失败');
+    console.log(chalk.gray('  手动安装: npm install -g npx'));
+    return false;
+}
+async function handlePluginCommand(args, context) {
+    const subCommand = args[0]?.toLowerCase();
+    switch (subCommand) {
+        case 'init':
+            return pluginInit(context.rl);
+        case undefined:
+            console.log(chalk.yellow('  用法: /plugin init') +
+                chalk.gray('  初始化新插件项目'));
+            return true;
+        default:
+            console.log(chalk.yellow(`  未知子命令: ${subCommand}`) +
+                chalk.gray('，可用: init'));
+            return true;
+    }
+}
+async function pluginInit(rl) {
+    try {
+        console.log();
+        console.log(chalk.bold('  ONES 插件项目初始化'));
+        console.log(chalk.gray('  [按 ESC 键取消]'));
+        console.log();
+        // ── 环境预检 ──────────────────────────────────────────
+        console.log(chalk.bold('  环境预检:'));
+        console.log();
+        // Node.js 18 通过 nvm 单独检查（不依赖当前激活版本）
+        const node18Ok = await ensureNode18();
+        const checks = [
+            { label: 'npm', cmd: 'npm --version' },
+            {
+                label: 'ONES CLI',
+                cmd: 'ones --version',
+                autoFix: installOnesCli,
+            },
+            {
+                label: 'npx',
+                cmd: 'npx --version',
+                autoFix: installNpx,
+            },
+        ];
+        let allRequiredPassed = node18Ok;
+        for (const check of checks) {
+            let version = runCommand(check.cmd);
+            if (version) {
+                console.log(chalk.green('  ✓ ') +
+                    `${check.label.padEnd(12)}${chalk.gray(version)}`);
+                continue;
+            }
+            // 未通过 - 尝试自动修复
+            if (check.autoFix) {
+                const fixed = await check.autoFix();
+                if (fixed) {
+                    version = runCommand(check.cmd);
+                    if (version)
+                        continue;
+                }
+            }
+            // 无法自动修复或修复失败
+            allRequiredPassed = false;
+            console.log(chalk.red('  ✗ ') + `${check.label} 未安装`);
+        }
+        console.log();
+        if (!allRequiredPassed) {
+            console.log(chalk.yellow('  请先安装缺失的依赖，然后重新执行 /plugin init\n'));
+            return true;
+        }
+        console.log(chalk.green('  环境预检通过'));
+        console.log();
+        // ── 收集插件信息 ────────────────────────────────────────
+        const pluginName = await promptQuestion(rl, chalk.cyan('  插件名称: '));
+        if (!pluginName) {
+            console.log(chalk.yellow('  插件名称不能为空'));
+            return true;
+        }
+        // 校验名称格式（仅允许字母、数字、连字符、下划线）
+        if (!/^[a-zA-Z0-9_-]+$/.test(pluginName)) {
+            console.log(chalk.yellow('  插件名称只能包含字母、数字、连字符(-)和下划线(_)'));
+            return true;
+        }
+        const pluginDir = await promptQuestion(rl, chalk.cyan('  项目目录: ') + chalk.gray('(留空为当前目录) '));
+        // ── 创建项目 ──────────────────────────────────────────
+        // ones create -d <name>           → 在当前目录创建项目
+        // ones create -d <name> -s <dir>  → 在 <dir>/ 创建项目
+        const projectPath = path.resolve(pluginDir || '.');
+        // 确保 -s 目标目录存在（ones create 要求该路径已存在）
+        if (pluginDir) {
+            try {
+                fs.mkdirSync(path.resolve(pluginDir), { recursive: true });
+            }
+            catch {
+                console.log(chalk.red(`\n  目录创建失败: ${pluginDir}\n`));
+                return true;
+            }
+        }
+        console.log();
+        console.log(chalk.gray(`  正在创建插件项目: ${pluginName}`) +
+            (pluginDir ? chalk.gray(` → ${pluginDir}/`) : '') +
+            chalk.gray(' ...'));
+        console.log();
+        let createCmd = `ones create -d ${pluginName}`;
+        if (pluginDir) {
+            createCmd += ` -s ${pluginDir}`;
+        }
+        const appId = crypto.randomBytes(4).toString('hex');
+        try {
+            execSync(createCmd, {
+                input: appId + '\n',
+                stdio: ['pipe', 'inherit', 'inherit'],
+                timeout: 300000,
+            });
+        }
+        catch (error) {
+            const err = error;
+            if (err.status) {
+                console.log(chalk.red(`\n  创建失败 (exit code: ${err.status})\n`));
+            }
+            else {
+                console.log(chalk.red(`\n  创建失败: ${err.message ?? error}\n`));
+            }
+            return true;
+        }
+        // ones create 可能 exit 0 但实际失败，检查项目目录是否存在
+        if (!fs.existsSync(projectPath)) {
+            console.log(chalk.red(`\n  创建失败: 项目目录不存在 ${projectPath}\n`));
+            return true;
+        }
+        console.log(chalk.green(`  ✓ 插件项目 "${pluginName}" 创建成功`));
+        // ── 写入 .nvmrc ─────────────────────────────────────────
+        try {
+            fs.writeFileSync(path.join(projectPath, '.nvmrc'), '18\n', 'utf-8');
+            console.log(chalk.green('  ✓ ') + '.nvmrc 已写入');
+        }
+        catch {
+            console.log(chalk.yellow('  ⚠ .nvmrc 写入失败'));
+        }
+        console.log();
+        // ── 后续指引 ──────────────────────────────────────────
+        console.log(chalk.bold('  本地调试步骤:'));
+        console.log();
+        console.log(chalk.cyan('  1. ') + `cd ${projectPath}`);
+        console.log(chalk.cyan('  2. ') + 'npx op login');
+        console.log(chalk.cyan('  3. ') + 'npx op pickteam local');
+        console.log(chalk.cyan('  4. ') + 'npx op invoke run');
+        console.log();
+        console.log(chalk.bold('  开发文档: ') +
+            chalk.underline('https://developer.ones.cn/zh-CN/docs/guide/getting-started/development'));
+        console.log();
+        return true;
+    }
+    catch (error) {
+        if (error instanceof CancelError) {
+            console.log();
+            return true;
+        }
+        throw error;
+    }
+}
+//# sourceMappingURL=commands.js.map