bigwo/dev-assistant-mcp/dist/tools/envCheck.js

import { exec } from "child_process";
import { promisify } from "util";
import { platform, totalmem, freemem, cpus, hostname, userInfo } from "os";
const execAsync = promisify(exec);
export const envCheckTool = {
    name: "env_check",
    description: "环境检测。检查 Node.js、Python、Git 等工具版本，检测端口占用，查看系统资源（CPU/内存/磁盘），验证开发环境是否就绪。",
    inputSchema: {
        type: "object",
        properties: {
            checks: {
                type: "string",
                description: "要检查的项目，逗号分隔（可选）：all, node, python, git, ports, system, docker。默认 all。",
            },
            ports: {
                type: "string",
                description: "要检测的端口号，逗号分隔（如 3000,8080,5173）",
            },
        },
        required: [],
    },
};
async function getVersion(cmd) {
    try {
        const { stdout } = await execAsync(cmd, { timeout: 5000 });
        return stdout.trim().split("\n")[0];
    }
    catch {
        return "❌ 未安装";
    }
}
async function checkPort(port) {
    try {
        const cmd = process.platform === "win32"
            ? `netstat -ano | findstr :${port}`
            : `lsof -i :${port} 2>/dev/null || ss -tlnp | grep :${port}`;
        const { stdout } = await execAsync(cmd, { timeout: 5000 });
        return stdout.trim().length > 0;
    }
    catch {
        return false;
    }
}
async function getDiskSpace() {
    try {
        const cmd = process.platform === "win32"
            ? 'powershell -command "Get-PSDrive -PSProvider FileSystem | Select-Object Name, @{N=\'Used(GB)\';E={[math]::Round($_.Used/1GB,1)}}, @{N=\'Free(GB)\';E={[math]::Round($_.Free/1GB,1)}} | Format-Table -AutoSize"'
            : "df -h / /home 2>/dev/null";
        const { stdout } = await execAsync(cmd, { timeout: 10000 });
        return stdout.trim();
    }
    catch {
        return "无法获取";
    }
}
export async function executeEnvCheck(args) {
    const { checks = "all", ports } = args;
    const checkList = checks.split(",").map((c) => c.trim().toLowerCase());
    const doAll = checkList.includes("all");
    const output = [
        `# 环境检测报告`,
        ``,
        `🖥️ ${hostname()} | ${platform()} | ${userInfo().username}`,
        `📅 ${new Date().toLocaleString("zh-CN")}`,
        ``,
    ];
    // 系统资源
    if (doAll || checkList.includes("system")) {
        const totalMem = (totalmem() / 1024 / 1024 / 1024).toFixed(1);
        const freeMem = (freemem() / 1024 / 1024 / 1024).toFixed(1);
        const usedMem = ((totalmem() - freemem()) / 1024 / 1024 / 1024).toFixed(1);
        const memPercent = ((1 - freemem() / totalmem()) * 100).toFixed(0);
        const cpuInfo = cpus();
        const cpuModel = cpuInfo[0]?.model || "未知";
        output.push(`## 💻 系统资源`, ``, `| 项目 | 值 |`, `|------|------|`, `| CPU | ${cpuModel} |`, `| 核心数 | ${cpuInfo.length} |`, `| 内存 | ${usedMem} / ${totalMem} GB (${memPercent}%) |`, `| 空闲内存 | ${freeMem} GB |`, ``);
        const disk = await getDiskSpace();
        if (disk !== "无法获取") {
            output.push(`### 磁盘空间`, "```", disk, "```", ``);
        }
    }
    // Node.js
    if (doAll || checkList.includes("node")) {
        const [nodeVer, npmVer, npxVer, yarnVer, pnpmVer] = await Promise.all([
            getVersion("node --version"),
            getVersion("npm --version"),
            getVersion("npx --version"),
            getVersion("yarn --version"),
            getVersion("pnpm --version"),
        ]);
        output.push(`## 📦 Node.js 生态`, ``, `| 工具 | 版本 |`, `|------|------|`, `| Node.js | ${nodeVer} |`, `| npm | ${npmVer} |`, `| npx | ${npxVer} |`, `| yarn | ${yarnVer} |`, `| pnpm | ${pnpmVer} |`, ``);
    }
    // Python
    if (doAll || checkList.includes("python")) {
        const [pyVer, py3Ver, pipVer] = await Promise.all([
            getVersion("python --version"),
            getVersion("python3 --version"),
            getVersion("pip --version"),
        ]);
        output.push(`## 🐍 Python 生态`, ``, `| 工具 | 版本 |`, `|------|------|`, `| Python | ${pyVer} |`, `| Python3 | ${py3Ver} |`, `| pip | ${pipVer} |`, ``);
    }
    // Git
    if (doAll || checkList.includes("git")) {
        const gitVer = await getVersion("git --version");
        const gitUser = await getVersion("git config --global user.name");
        const gitEmail = await getVersion("git config --global user.email");
        output.push(`## 🔧 Git`, ``, `| 项目 | 值 |`, `|------|------|`, `| 版本 | ${gitVer} |`, `| 用户名 | ${gitUser} |`, `| 邮箱 | ${gitEmail} |`, ``);
    }
    // Docker
    if (doAll || checkList.includes("docker")) {
        const [dockerVer, composeVer] = await Promise.all([
            getVersion("docker --version"),
            getVersion("docker compose version 2>&1 || docker-compose --version 2>&1"),
        ]);
        output.push(`## 🐳 Docker`, ``, `| 工具 | 版本 |`, `|------|------|`, `| Docker | ${dockerVer} |`, `| Compose | ${composeVer} |`, ``);
    }
    // 端口检测
    if (doAll || checkList.includes("ports") || ports) {
        const portsToCheck = ports
            ? ports.split(",").map((p) => parseInt(p.trim())).filter(Boolean)
            : [3000, 3001, 5173, 5174, 8080, 8081, 4000, 4173, 5000, 5500];
        const portResults = await Promise.all(portsToCheck.map(async (port) => ({
            port,
            occupied: await checkPort(port),
        })));
        output.push(`## 🔌 端口状态`, ``, `| 端口 | 状态 |`, `|------|------|`, ...portResults.map((r) => `| ${r.port} | ${r.occupied ? "🔴 已占用" : "🟢 空闲"} |`), ``);
    }
    // 总结
    output.push(`---`, `_检测完成_`);
    return output.join("\n");
}
//# sourceMappingURL=envCheck.js.map