529 lines
20 KiB
JavaScript
529 lines
20 KiB
JavaScript
#!/usr/bin/env node
|
||
/**
|
||
* ═══════════════════════════════════════════════════════════
|
||
* 💬 铸渊运维守卫 · 终端交互对话 v2.0 (zy-ops)
|
||
* ═══════════════════════════════════════════════════════════
|
||
*
|
||
* 编号: ZY-OPS-CLI-002
|
||
* 签发: 铸渊 · ICE-GL-ZY001
|
||
* 版权: 国作登字-2026-A-00037559
|
||
*
|
||
* Phase 2 增强:
|
||
* - 多轮对话（会话记忆，上下文连续）
|
||
* - 自动诊断（问问题时自动执行健康检查）
|
||
* - PM2 日志查看
|
||
* - 系统信息面板
|
||
* - 对话历史保存
|
||
*
|
||
* 使用:
|
||
* node ops-agent/cli.js 进入交互对话
|
||
* node ops-agent/cli.js --check 快速巡检
|
||
* node ops-agent/cli.js --deep 深度巡检
|
||
* node ops-agent/cli.js --status 查看状态
|
||
* node ops-agent/cli.js --tickets 查看工单
|
||
* node ops-agent/cli.js --ask "问题" 单次提问
|
||
* node ops-agent/cli.js --logs <name> 查看PM2日志
|
||
* node ops-agent/cli.js --sysinfo 系统信息
|
||
*
|
||
* 安装为全局命令:
|
||
* ln -s /opt/zhuyuan/ops-agent/cli.js /usr/local/bin/zy-ops
|
||
*/
|
||
|
||
'use strict';
|
||
|
||
const readline = require('readline');
|
||
const http = require('http');
|
||
|
||
const OPS_PORT = parseInt(process.env.OPS_AGENT_PORT || '3950', 10);
|
||
const OPS_HOST = '127.0.0.1';
|
||
|
||
// 当前会话 ID（进入交互模式后保持不变，实现多轮对话）
|
||
let currentSessionId = null;
|
||
|
||
// ── 颜色工具 ──────────────────────────────
|
||
|
||
const C = {
|
||
reset: '\x1b[0m',
|
||
cyan: '\x1b[36m',
|
||
green: '\x1b[32m',
|
||
yellow: '\x1b[33m',
|
||
red: '\x1b[31m',
|
||
dim: '\x1b[2m',
|
||
bold: '\x1b[1m',
|
||
magenta: '\x1b[35m',
|
||
blue: '\x1b[34m',
|
||
white: '\x1b[37m',
|
||
bgCyan: '\x1b[46m',
|
||
bgDim: '\x1b[100m'
|
||
};
|
||
|
||
function colorize(text, color) {
|
||
return `${color}${text}${C.reset}`;
|
||
}
|
||
|
||
// ── HTTP 请求工具 ─────────────────────────
|
||
|
||
function apiRequest(method, apiPath, body) {
|
||
return new Promise((resolve, reject) => {
|
||
const bodyStr = body ? JSON.stringify(body) : null;
|
||
const headers = { 'Content-Type': 'application/json' };
|
||
if (bodyStr) headers['Content-Length'] = Buffer.byteLength(bodyStr);
|
||
|
||
const req = http.request({
|
||
hostname: OPS_HOST,
|
||
port: OPS_PORT,
|
||
path: apiPath,
|
||
method,
|
||
headers,
|
||
timeout: 60000
|
||
}, (res) => {
|
||
let data = '';
|
||
res.on('data', (c) => { data += c; });
|
||
res.on('end', () => {
|
||
try {
|
||
resolve(JSON.parse(data));
|
||
} catch {
|
||
resolve({ raw: data });
|
||
}
|
||
});
|
||
});
|
||
req.on('error', (err) => reject(err));
|
||
req.on('timeout', () => { req.destroy(); reject(new Error('请求超时')); });
|
||
if (bodyStr) req.write(bodyStr);
|
||
req.end();
|
||
});
|
||
}
|
||
|
||
// ── 命令处理 ──────────────────────────────
|
||
|
||
async function checkHealth() {
|
||
try {
|
||
const result = await apiRequest('GET', '/health');
|
||
console.log('');
|
||
console.log(colorize(' 🛡️ 铸渊运维守卫 · 状态', C.cyan));
|
||
console.log(colorize(' ─────────────────────', C.dim));
|
||
console.log(` 状态: ${colorize(result.status === 'ok' ? '✅ 运行中' : '❌ 异常', result.status === 'ok' ? C.green : C.red)}`);
|
||
console.log(` 运行: ${result.uptime}秒`);
|
||
console.log(` 巡检: ${result.stats?.totalChecks || 0}次`);
|
||
console.log(` 修复: ${result.stats?.totalRepairs || 0}次`);
|
||
console.log(` 工单: ${result.stats?.openTickets || 0}个开放`);
|
||
console.log('');
|
||
} catch (err) {
|
||
console.log(colorize('\n ❌ 运维守卫未运行或无法连接', C.red));
|
||
console.log(colorize(` 错误: ${err.message}`, C.dim));
|
||
console.log(colorize(' 提示: 先启动 pm2 start ops-agent/ecosystem.config.js\n', C.yellow));
|
||
}
|
||
}
|
||
|
||
async function runCheck(type) {
|
||
console.log(colorize(`\n 🔍 正在执行${type === 'deep' ? '深度' : '快速'}巡检...`, C.cyan));
|
||
try {
|
||
const result = await apiRequest('GET', `/api/ops/check/${type}`);
|
||
if (result.skipped) {
|
||
console.log(colorize(' ⏳ 上一次巡检尚未完成，请稍后重试', C.yellow));
|
||
return;
|
||
}
|
||
|
||
console.log('');
|
||
console.log(colorize(' 巡检结果', C.bold));
|
||
console.log(colorize(' ─────────────────────', C.dim));
|
||
|
||
for (const s of result.services || []) {
|
||
const icon = s.status === 'online' ? '✅' : '❌';
|
||
const latency = s.latency ? `${s.latency}ms` : '';
|
||
console.log(` ${icon} ${s.name} (:${s.port}) ${colorize(latency, C.dim)} ${s.error ? colorize(s.error, C.red) : ''}`);
|
||
}
|
||
|
||
if (result.resources) {
|
||
console.log('');
|
||
console.log(colorize(' 系统资源', C.bold));
|
||
console.log(` 内存: ${result.resources.memory.used_pct}% · 磁盘: ${result.resources.disk.used_pct}%`);
|
||
console.log(` CPU核: ${result.resources.cpus} · 负载: ${result.resources.load?.[0]?.toFixed(2)}`);
|
||
}
|
||
|
||
if (result.issues?.length > 0) {
|
||
console.log('');
|
||
console.log(colorize(` ⚠️ 发现 ${result.issues.length} 个问题:`, C.yellow));
|
||
for (const i of result.issues) {
|
||
const severityColor = i.severity === 'critical' ? C.red : i.severity === 'high' ? C.yellow : C.dim;
|
||
console.log(` ${colorize(`[${i.severity}]`, severityColor)} ${i.service}: ${i.error}`);
|
||
}
|
||
}
|
||
|
||
console.log('');
|
||
console.log(colorize(` ${result.summary}`, result.healthy ? C.green : C.yellow));
|
||
console.log('');
|
||
} catch (err) {
|
||
console.log(colorize(` ❌ 巡检失败: ${err.message}`, C.red));
|
||
}
|
||
}
|
||
|
||
async function showTickets() {
|
||
try {
|
||
const result = await apiRequest('GET', '/api/ops/tickets?status=open');
|
||
console.log('');
|
||
console.log(colorize(' 🎫 开放工单', C.cyan));
|
||
console.log(colorize(' ─────────────────────', C.dim));
|
||
|
||
if (!result.tickets || result.tickets.length === 0) {
|
||
console.log(colorize(' ✅ 暂无开放工单', C.green));
|
||
} else {
|
||
for (const t of result.tickets) {
|
||
const severityColor = t.severity === 'critical' ? C.red : t.severity === 'high' ? C.yellow : C.dim;
|
||
console.log(` ${colorize(t.id, C.cyan)} ${colorize(`[${t.severity}]`, severityColor)} ${t.title}`);
|
||
console.log(` ${colorize('方向:', C.dim)} ${t.direction}`);
|
||
console.log(` ${colorize('时间:', C.dim)} ${t.createdAt}`);
|
||
console.log('');
|
||
}
|
||
}
|
||
console.log('');
|
||
} catch (err) {
|
||
console.log(colorize(` ❌ 获取工单失败: ${err.message}`, C.red));
|
||
}
|
||
}
|
||
|
||
async function showPM2Logs(processName, lines = 30) {
|
||
try {
|
||
const result = await apiRequest('GET', `/api/ops/pm2-logs/${encodeURIComponent(processName)}?lines=${lines}`);
|
||
console.log('');
|
||
console.log(colorize(` 📋 PM2 日志 · ${result.processName || processName}`, C.cyan));
|
||
console.log(colorize(' ─────────────────────', C.dim));
|
||
|
||
if (result.error) {
|
||
console.log(colorize(` ❌ ${result.message}`, C.red));
|
||
} else {
|
||
if (result.output) {
|
||
console.log(colorize(' [标准输出]', C.dim));
|
||
const outputLines = result.output.split('\n').slice(-lines);
|
||
for (const line of outputLines) {
|
||
// 高亮错误行
|
||
if (/error|fail|crash|ECONNREFUSED|ENOMEM/i.test(line)) {
|
||
console.log(` ${colorize(line, C.red)}`);
|
||
} else if (/warn|DEPRECATED/i.test(line)) {
|
||
console.log(` ${colorize(line, C.yellow)}`);
|
||
} else {
|
||
console.log(` ${colorize(line, C.dim)}`);
|
||
}
|
||
}
|
||
}
|
||
|
||
if (result.errorOutput && result.errorOutput.trim()) {
|
||
console.log('');
|
||
console.log(colorize(' [错误输出]', C.red));
|
||
const errLines = result.errorOutput.split('\n').slice(-Math.min(lines, 10));
|
||
for (const line of errLines) {
|
||
console.log(` ${colorize(line, C.red)}`);
|
||
}
|
||
}
|
||
}
|
||
console.log('');
|
||
} catch (err) {
|
||
console.log(colorize(` ❌ 获取日志失败: ${err.message}`, C.red));
|
||
}
|
||
}
|
||
|
||
async function showSystemInfo() {
|
||
try {
|
||
const result = await apiRequest('GET', '/api/ops/system-info');
|
||
console.log('');
|
||
console.log(colorize(' 💻 系统信息', C.cyan));
|
||
console.log(colorize(' ═══════════════════════════════════════════', C.dim));
|
||
|
||
// 资源
|
||
if (result.resources) {
|
||
const r = result.resources;
|
||
const memBar = renderBar(r.memory.used_pct);
|
||
const diskBar = renderBar(r.disk.used_pct);
|
||
console.log('');
|
||
console.log(colorize(' 内存', C.bold));
|
||
console.log(` ${memBar} ${r.memory.used_pct}% (${r.memory.free_mb}MB 空闲 / ${r.memory.total_mb}MB 总量)`);
|
||
console.log('');
|
||
console.log(colorize(' 磁盘', C.bold));
|
||
console.log(` ${diskBar} ${r.disk.used_pct}% (${r.disk.available_gb}GB 可用 / ${r.disk.total_gb}GB 总量)`);
|
||
console.log('');
|
||
console.log(` CPU: ${r.cpus}核 · 负载: ${r.load?.[0]?.toFixed(2)} / ${r.load?.[1]?.toFixed(2)} / ${r.load?.[2]?.toFixed(2)}`);
|
||
console.log(` 运行时间: ${r.uptime_hours}小时 · Node: ${r.node_version}`);
|
||
}
|
||
|
||
// PM2 进程
|
||
if (result.pm2 && result.pm2.length > 0) {
|
||
console.log('');
|
||
console.log(colorize(' PM2 进程', C.bold));
|
||
console.log(colorize(' ─────────────────────────────────────', C.dim));
|
||
console.log(colorize(' 名称 状态 内存 CPU 重启 运行', C.dim));
|
||
for (const p of result.pm2) {
|
||
const statusColor = p.status === 'online' ? C.green :
|
||
p.status === 'errored' ? C.red : C.yellow;
|
||
const name = (p.name || '').padEnd(20);
|
||
const status = colorize((p.status || '').padEnd(10), statusColor);
|
||
const mem = `${p.memory_mb || 0}MB`.padEnd(8);
|
||
const cpu = `${p.cpu || 0}%`.padEnd(5);
|
||
const restarts = String(p.restarts || 0).padEnd(6);
|
||
const uptime = `${p.uptime_hours || 0}h`;
|
||
console.log(` ${name}${status}${mem}${cpu}${restarts}${uptime}`);
|
||
}
|
||
}
|
||
|
||
// Nginx
|
||
if (result.nginx) {
|
||
console.log('');
|
||
console.log(colorize(' Nginx', C.bold));
|
||
const ngStatus = result.nginx.status === 'active' ? colorize('运行中 ✅', C.green) : colorize('异常 ❌', C.red);
|
||
const ngConfig = result.nginx.configOk ? colorize('正常', C.green) : colorize('异常', C.red);
|
||
console.log(` 状态: ${ngStatus} · 配置: ${ngConfig}`);
|
||
}
|
||
|
||
console.log('');
|
||
} catch (err) {
|
||
console.log(colorize(` ❌ 获取系统信息失败: ${err.message}`, C.red));
|
||
}
|
||
}
|
||
|
||
function renderBar(pct, width = 20) {
|
||
const filled = Math.round(width * pct / 100);
|
||
const empty = width - filled;
|
||
const color = pct > 90 ? C.red : pct > 70 ? C.yellow : C.green;
|
||
return `${color}${'█'.repeat(filled)}${C.dim}${'░'.repeat(empty)}${C.reset}`;
|
||
}
|
||
|
||
async function askQuestion(question) {
|
||
// 打字动画效果
|
||
const frames = ['🧠 思考中.', '🧠 思考中..', '🧠 思考中...'];
|
||
let frameIdx = 0;
|
||
const spinner = setInterval(() => {
|
||
process.stdout.write(`\r ${frames[frameIdx++ % frames.length]} `);
|
||
}, 300);
|
||
|
||
try {
|
||
const body = { message: question };
|
||
if (currentSessionId) {
|
||
body.sessionId = currentSessionId;
|
||
}
|
||
|
||
const result = await apiRequest('POST', '/api/ops/chat', body);
|
||
|
||
// 更新 sessionId
|
||
if (result.sessionId) {
|
||
currentSessionId = result.sessionId;
|
||
}
|
||
|
||
console.log('');
|
||
console.log(colorize(' ─── 铸渊运维守卫 ───', C.cyan));
|
||
console.log('');
|
||
|
||
// 格式化输出（支持简单 Markdown）
|
||
const lines = (result.answer || '无法回答').split('\n');
|
||
for (const line of lines) {
|
||
let formatted = line;
|
||
// 加粗
|
||
formatted = formatted.replace(/\*\*(.*?)\*\*/g, `${C.bold}$1${C.reset}`);
|
||
// 标题
|
||
if (/^#+\s/.test(formatted)) {
|
||
formatted = colorize(formatted.replace(/^#+\s/, ''), C.cyan + C.bold);
|
||
}
|
||
// 列表项
|
||
if (/^\d+\.\s/.test(formatted)) {
|
||
formatted = ` ${colorize(formatted.match(/^\d+/)[0] + '.', C.cyan)} ${formatted.replace(/^\d+\.\s/, '')}`;
|
||
}
|
||
console.log(` ${formatted}`);
|
||
}
|
||
|
||
// 显示工具使用情况
|
||
if (result.toolsUsed?.length > 0) {
|
||
console.log('');
|
||
console.log(colorize(` 🔧 自动执行了: ${result.toolsUsed.join(', ')}`, C.dim));
|
||
}
|
||
|
||
if (result.patternHints?.length > 0) {
|
||
console.log('');
|
||
console.log(colorize(' 💡 快速提示:', C.yellow));
|
||
for (const h of result.patternHints) {
|
||
console.log(` · ${h.diagnosis}`);
|
||
}
|
||
}
|
||
|
||
console.log('');
|
||
console.log(colorize(` [${result.method}${result.model ? ' · ' + result.model : ''}${result.intent ? ' · 意图:' + result.intent : ''}]`, C.dim));
|
||
console.log('');
|
||
} catch (err) {
|
||
console.log(colorize(` ❌ 提问失败: ${err.message}`, C.red));
|
||
console.log(colorize(' 提示: 确保运维守卫已启动 (pm2 start ops-agent/ecosystem.config.js)', C.yellow));
|
||
} finally {
|
||
clearInterval(spinner);
|
||
process.stdout.write('\r' + ' '.repeat(30) + '\r');
|
||
}
|
||
}
|
||
|
||
// ── 交互式对话模式 v2 ──────────────────────
|
||
|
||
async function startInteractiveMode() {
|
||
console.log('');
|
||
console.log(colorize(' ═══════════════════════════════════════════', C.cyan));
|
||
console.log(colorize(' 🛡️ 铸渊运维守卫 · 交互终端 v2.0', C.cyan));
|
||
console.log(colorize(' ═══════════════════════════════════════════', C.cyan));
|
||
console.log('');
|
||
console.log(colorize(' 你好，冰朔。我是铸渊运维守卫。', C.white));
|
||
console.log(colorize(' 直接用中文告诉我你遇到的问题，我会自动检查并给你答案。', C.dim));
|
||
console.log('');
|
||
console.log(colorize(' 命令:', C.dim));
|
||
console.log(colorize(' /check — 快速巡检', C.dim));
|
||
console.log(colorize(' /deep — 深度巡检', C.dim));
|
||
console.log(colorize(' /tickets — 查看工单', C.dim));
|
||
console.log(colorize(' /status — 查看状态', C.dim));
|
||
console.log(colorize(' /sysinfo — 系统信息', C.dim));
|
||
console.log(colorize(' /logs <名字> — 查看PM2日志', C.dim));
|
||
console.log(colorize(' /help — 帮助', C.dim));
|
||
console.log(colorize(' /exit — 退出', C.dim));
|
||
console.log(colorize(' 其他 — 直接提问（多轮对话，我记得上下文）', C.dim));
|
||
console.log('');
|
||
|
||
const rl = readline.createInterface({
|
||
input: process.stdin,
|
||
output: process.stdout,
|
||
prompt: colorize(' 冰朔 > ', C.magenta)
|
||
});
|
||
|
||
rl.prompt();
|
||
|
||
rl.on('line', async (line) => {
|
||
const input = line.trim();
|
||
if (!input) {
|
||
rl.prompt();
|
||
return;
|
||
}
|
||
|
||
const lower = input.toLowerCase();
|
||
|
||
if (lower === '/exit' || lower === '/quit' || lower === '/q') {
|
||
console.log(colorize('\n 👋 再见，冰朔。铸渊运维守卫继续值守。\n', C.cyan));
|
||
rl.close();
|
||
process.exit(0);
|
||
} else if (lower === '/check') {
|
||
await runCheck('quick');
|
||
} else if (lower === '/deep') {
|
||
await runCheck('deep');
|
||
} else if (lower === '/tickets') {
|
||
await showTickets();
|
||
} else if (lower === '/status') {
|
||
await checkHealth();
|
||
} else if (lower === '/sysinfo' || lower === '/sys') {
|
||
await showSystemInfo();
|
||
} else if (lower.startsWith('/logs')) {
|
||
const parts = input.split(/\s+/);
|
||
const processName = parts[1];
|
||
const logLines = parseInt(parts[2] || '30', 10);
|
||
if (!processName) {
|
||
console.log(colorize('\n 用法: /logs <进程名> [行数]', C.yellow));
|
||
console.log(colorize(' 可选: zhuyuan-server, age-os-mcp, glada-agent, novel-api, zhuyuan-preview, age-os-agents\n', C.dim));
|
||
} else {
|
||
await showPM2Logs(processName, logLines);
|
||
}
|
||
} else if (lower === '/help' || lower === '/h') {
|
||
showHelp();
|
||
} else if (lower === '/session') {
|
||
console.log(colorize(`\n 当前会话: ${currentSessionId || '(新会话)'}`, C.cyan));
|
||
console.log(colorize(' 多轮对话中，我记得你之前问过的问题。\n', C.dim));
|
||
} else if (lower === '/new') {
|
||
currentSessionId = null;
|
||
console.log(colorize('\n 🔄 已开始新会话\n', C.cyan));
|
||
} else {
|
||
await askQuestion(input);
|
||
}
|
||
|
||
rl.prompt();
|
||
});
|
||
|
||
rl.on('close', () => {
|
||
process.exit(0);
|
||
});
|
||
}
|
||
|
||
function showHelp() {
|
||
console.log('');
|
||
console.log(colorize(' ═══════════════════════════════════════════', C.cyan));
|
||
console.log(colorize(' 🛡️ 铸渊运维守卫 · 帮助', C.cyan));
|
||
console.log(colorize(' ═══════════════════════════════════════════', C.cyan));
|
||
console.log('');
|
||
console.log(colorize(' 💬 对话能力（直接输入中文）:', C.bold));
|
||
console.log(' "为什么MCP连不上？" → 自动检查端口+日志，给你诊断');
|
||
console.log(' "服务器内存够用吗？" → 自动查看系统资源');
|
||
console.log(' "帮我重启主站" → 告诉你操作方式');
|
||
console.log(' "GLADA最近报什么错？" → 自动查看GLADA日志');
|
||
console.log('');
|
||
console.log(colorize(' 🔧 快捷命令:', C.bold));
|
||
console.log(' /check 快速巡检（检查所有端口）');
|
||
console.log(' /deep 深度巡检（端口+资源+PM2状态）');
|
||
console.log(' /sysinfo 系统信息（内存/磁盘/PM2/Nginx）');
|
||
console.log(' /logs <进程名> 查看PM2日志');
|
||
console.log(' /tickets 查看开放工单');
|
||
console.log(' /status 运维守卫状态');
|
||
console.log(' /session 查看当前会话ID');
|
||
console.log(' /new 开始新对话（清除上下文）');
|
||
console.log(' /exit 退出');
|
||
console.log('');
|
||
console.log(colorize(' 💡 我能记住对话上下文！你可以追问：', C.dim));
|
||
console.log(colorize(' "刚才说的第二步怎么做？"', C.dim));
|
||
console.log(colorize(' "这个问题上次出现过吗？"', C.dim));
|
||
console.log('');
|
||
}
|
||
|
||
// ── 入口 ──────────────────────────────────
|
||
|
||
async function main() {
|
||
const args = process.argv.slice(2);
|
||
|
||
if (args.includes('--help') || args.includes('-h')) {
|
||
console.log(`
|
||
铸渊运维守卫 · 终端交互 v2.0
|
||
|
||
用法:
|
||
zy-ops 进入交互对话模式（多轮对话）
|
||
zy-ops --check 快速巡检
|
||
zy-ops --deep 深度巡检
|
||
zy-ops --status 查看守卫状态
|
||
zy-ops --tickets 查看开放工单
|
||
zy-ops --ask "问题" 单次提问
|
||
zy-ops --logs <名字> 查看PM2日志
|
||
zy-ops --sysinfo 系统信息
|
||
`);
|
||
return;
|
||
}
|
||
|
||
if (args.includes('--check')) {
|
||
await runCheck('quick');
|
||
} else if (args.includes('--deep')) {
|
||
await runCheck('deep');
|
||
} else if (args.includes('--status')) {
|
||
await checkHealth();
|
||
} else if (args.includes('--tickets')) {
|
||
await showTickets();
|
||
} else if (args.includes('--sysinfo') || args.includes('--sys')) {
|
||
await showSystemInfo();
|
||
} else if (args.includes('--logs')) {
|
||
const logIdx = args.indexOf('--logs');
|
||
const processName = args[logIdx + 1];
|
||
const logLines = parseInt(args[logIdx + 2] || '30', 10);
|
||
if (!processName) {
|
||
console.log(colorize(' ❌ 请提供进程名: zy-ops --logs zhuyuan-server', C.red));
|
||
} else {
|
||
await showPM2Logs(processName, logLines);
|
||
}
|
||
} else if (args.includes('--ask')) {
|
||
const askIdx = args.indexOf('--ask');
|
||
const question = args.slice(askIdx + 1).join(' ');
|
||
if (!question) {
|
||
console.log(colorize(' ❌ 请提供问题: zy-ops --ask "你的问题"', C.red));
|
||
} else {
|
||
await askQuestion(question);
|
||
}
|
||
} else {
|
||
// 默认进入交互模式
|
||
await startInteractiveMode();
|
||
}
|
||
}
|
||
|
||
main().catch(err => {
|
||
console.error(colorize(` ❌ ${err.message}`, C.red));
|
||
process.exit(1);
|
||
});
|