229 lines
9.3 KiB
JavaScript
229 lines
9.3 KiB
JavaScript
import { exec, spawn } from "child_process";
|
||
import { promisify } from "util";
|
||
import { existsSync, readFileSync } from "fs";
|
||
import { join } from "path";
|
||
const execAsync = promisify(exec);
|
||
// 进程管理器 - 跟踪所有启动的开发服务器
|
||
const managedProcesses = new Map();
|
||
export const devServerTool = {
|
||
name: "dev_server",
|
||
description: "开发服务器管理。启动/停止/重启开发服务器，查看运行状态和实时日志。支持自动检测项目的 dev 命令。",
|
||
inputSchema: {
|
||
type: "object",
|
||
properties: {
|
||
project_path: {
|
||
type: "string",
|
||
description: "项目根目录（绝对路径）",
|
||
},
|
||
action: {
|
||
type: "string",
|
||
description: "操作类型",
|
||
enum: ["start", "stop", "restart", "status", "logs", "list"],
|
||
},
|
||
command: {
|
||
type: "string",
|
||
description: "自定义启动命令（可选，默认自动检测 npm run dev 等）",
|
||
},
|
||
name: {
|
||
type: "string",
|
||
description: "服务器名称/标识（可选，默认使用目录名）",
|
||
},
|
||
tail: {
|
||
type: "number",
|
||
description: "显示最近几行日志（默认 30）",
|
||
},
|
||
},
|
||
required: ["action"],
|
||
},
|
||
};
|
||
function getServerName(projectPath, name) {
|
||
if (name)
|
||
return name;
|
||
if (projectPath)
|
||
return projectPath.split(/[/\\]/).pop() || "server";
|
||
return "default";
|
||
}
|
||
function detectDevCommand(projectPath) {
|
||
const pkgPath = join(projectPath, "package.json");
|
||
if (existsSync(pkgPath)) {
|
||
try {
|
||
const pkg = JSON.parse(readFileSync(pkgPath, "utf-8"));
|
||
const scripts = pkg.scripts || {};
|
||
// 按优先级检测
|
||
for (const name of ["dev", "start:dev", "serve", "start"]) {
|
||
if (scripts[name])
|
||
return `npm run ${name}`;
|
||
}
|
||
}
|
||
catch { }
|
||
}
|
||
if (existsSync(join(projectPath, "manage.py"))) {
|
||
return "python manage.py runserver";
|
||
}
|
||
if (existsSync(join(projectPath, "main.py"))) {
|
||
return "python main.py";
|
||
}
|
||
if (existsSync(join(projectPath, "app.py"))) {
|
||
return "python app.py";
|
||
}
|
||
return null;
|
||
}
|
||
export async function executeDevServer(args) {
|
||
const { project_path, action, command, name, tail = 30 } = args;
|
||
switch (action) {
|
||
case "list": {
|
||
if (managedProcesses.size === 0) {
|
||
return "# 开发服务器\n\n_没有正在运行的服务器_";
|
||
}
|
||
const output = ["# 运行中的开发服务器", ""];
|
||
for (const [key, info] of managedProcesses) {
|
||
const running = !info.process.killed && info.process.exitCode === null;
|
||
const uptime = Math.round((Date.now() - info.startTime) / 1000);
|
||
output.push(`## ${running ? "🟢" : "🔴"} ${key}`, `- 命令: \`${info.command}\``, `- 目录: ${info.cwd}`, `- PID: ${info.process.pid}`, `- 运行时间: ${uptime}s`, `- 状态: ${running ? "运行中" : "已停止"}`, ``);
|
||
}
|
||
return output.join("\n");
|
||
}
|
||
case "start": {
|
||
if (!project_path)
|
||
return "❌ start 需要 project_path 参数";
|
||
const serverName = getServerName(project_path, name);
|
||
const existing = managedProcesses.get(serverName);
|
||
if (existing && !existing.process.killed && existing.process.exitCode === null) {
|
||
return `⚠️ 服务器 "${serverName}" 已在运行中 (PID: ${existing.process.pid})\n\n使用 action=restart 重启，或 action=stop 先停止`;
|
||
}
|
||
const startCmd = command || detectDevCommand(project_path);
|
||
if (!startCmd) {
|
||
return `❌ 未检测到启动命令，请手动指定 command 参数\n\n常见命令:\n- npm run dev\n- npm start\n- python app.py`;
|
||
}
|
||
// 解析命令
|
||
const isWin = process.platform === "win32";
|
||
const child = spawn(isWin ? "cmd" : "sh", [isWin ? "/c" : "-c", startCmd], {
|
||
cwd: project_path,
|
||
stdio: ["ignore", "pipe", "pipe"],
|
||
detached: false,
|
||
});
|
||
const logs = [];
|
||
const maxLogs = 200;
|
||
const addLog = (data, stream) => {
|
||
const lines = data.toString().split("\n").filter(Boolean);
|
||
for (const line of lines) {
|
||
logs.push(`[${stream}] ${line}`);
|
||
if (logs.length > maxLogs)
|
||
logs.shift();
|
||
}
|
||
};
|
||
child.stdout?.on("data", (data) => addLog(data, "out"));
|
||
child.stderr?.on("data", (data) => addLog(data, "err"));
|
||
managedProcesses.set(serverName, {
|
||
process: child,
|
||
command: startCmd,
|
||
cwd: project_path,
|
||
startTime: Date.now(),
|
||
logs,
|
||
});
|
||
// 等待一会检查是否立即崩溃
|
||
await new Promise((r) => setTimeout(r, 2000));
|
||
const crashed = child.exitCode !== null;
|
||
if (crashed) {
|
||
const output = logs.join("\n");
|
||
managedProcesses.delete(serverName);
|
||
return `# ❌ 服务器启动失败\n\n命令: \`${startCmd}\`\n退出码: ${child.exitCode}\n\n\`\`\`\n${output.slice(0, 3000)}\n\`\`\``;
|
||
}
|
||
return [
|
||
`# ✅ 服务器已启动`,
|
||
``,
|
||
`- 名称: ${serverName}`,
|
||
`- 命令: \`${startCmd}\``,
|
||
`- PID: ${child.pid}`,
|
||
`- 目录: ${project_path}`,
|
||
``,
|
||
`最近日志:`,
|
||
"```",
|
||
logs.slice(-10).join("\n") || "(等待输出...)",
|
||
"```",
|
||
``,
|
||
`💡 使用 \`dev_server action=logs\` 查看实时日志`,
|
||
].join("\n");
|
||
}
|
||
case "stop": {
|
||
const serverName = getServerName(project_path, name);
|
||
const info = managedProcesses.get(serverName);
|
||
if (!info) {
|
||
return `❌ 未找到服务器 "${serverName}"\n\n使用 action=list 查看所有运行中的服务器`;
|
||
}
|
||
const pid = info.process.pid;
|
||
try {
|
||
// Windows 需要 taskkill 杀进程树
|
||
if (process.platform === "win32" && pid) {
|
||
await execAsync(`taskkill /PID ${pid} /T /F`).catch(() => { });
|
||
}
|
||
else {
|
||
info.process.kill("SIGTERM");
|
||
}
|
||
}
|
||
catch { }
|
||
managedProcesses.delete(serverName);
|
||
return `# ✅ 服务器已停止\n\n- 名称: ${serverName}\n- PID: ${pid}`;
|
||
}
|
||
case "restart": {
|
||
const serverName = getServerName(project_path, name);
|
||
const info = managedProcesses.get(serverName);
|
||
if (info) {
|
||
try {
|
||
if (process.platform === "win32" && info.process.pid) {
|
||
await execAsync(`taskkill /PID ${info.process.pid} /T /F`).catch(() => { });
|
||
}
|
||
else {
|
||
info.process.kill("SIGTERM");
|
||
}
|
||
}
|
||
catch { }
|
||
managedProcesses.delete(serverName);
|
||
await new Promise((r) => setTimeout(r, 1000));
|
||
}
|
||
// 重新启动
|
||
return executeDevServer({
|
||
project_path: info?.cwd || project_path,
|
||
action: "start",
|
||
command: command || info?.command,
|
||
name: serverName,
|
||
});
|
||
}
|
||
case "status": {
|
||
const serverName = getServerName(project_path, name);
|
||
const info = managedProcesses.get(serverName);
|
||
if (!info) {
|
||
return `❌ 未找到服务器 "${serverName}"`;
|
||
}
|
||
const running = !info.process.killed && info.process.exitCode === null;
|
||
const uptime = Math.round((Date.now() - info.startTime) / 1000);
|
||
return [
|
||
`# ${running ? "🟢" : "🔴"} ${serverName}`,
|
||
``,
|
||
`- 状态: ${running ? "运行中" : `已停止 (退出码: ${info.process.exitCode})`}`,
|
||
`- 命令: \`${info.command}\``,
|
||
`- PID: ${info.process.pid}`,
|
||
`- 运行时间: ${uptime}s`,
|
||
`- 目录: ${info.cwd}`,
|
||
].join("\n");
|
||
}
|
||
case "logs": {
|
||
const serverName = getServerName(project_path, name);
|
||
const info = managedProcesses.get(serverName);
|
||
if (!info) {
|
||
return `❌ 未找到服务器 "${serverName}"`;
|
||
}
|
||
const recentLogs = info.logs.slice(-tail);
|
||
return [
|
||
`# 📋 ${serverName} 日志（最近 ${tail} 行）`,
|
||
``,
|
||
"```",
|
||
recentLogs.join("\n") || "(无日志)",
|
||
"```",
|
||
].join("\n");
|
||
}
|
||
default:
|
||
return `❌ 未知操作: ${action}`;
|
||
}
|
||
}
|
||
//# sourceMappingURL=devServer.js.map
|