aws-runtime-bridge 1.9.79 → 1.9.81

package/dist/index.js

@@ -1,574 +1,3 @@
 #!/usr/bin/env node
-/**
- * Runtime Bridge 服务主入口
- *
- * 提供 Agent 运行时与调度器之间的桥接功能
- */
-import express from "express";
-import cors from "cors";
-import { existsSync, readFileSync } from "node:fs";
-import path from "node:path";
-import os from "node:os";
-import { allowedCorsOrigins, port, validateProductionToken } from "./config.js";
-import { propertiesRouter } from "./routes/properties.js";
-import { ymlRouter } from "./routes/yml.js";
-import { instanceRouter, setGracefulShutdownFn } from "./routes/instance.js";
-import { skillsRouter } from "./routes/skills.js";
-import { mcpRouter } from "./routes/mcp.js";
-import { terminalRouter, sdkSessions } from "./routes/terminal.js";
-import { sessionsRouter } from "./routes/sessions.js";
-import { gitRouter } from "./routes/git.js";
-import { fileSnapshotRouter } from "./routes/file-snapshot.js";
-import { acodeTodosRouter } from "./routes/acode-todos.js";
-import { subAgentsRouter } from "./routes/sub-agents.js";
-import { fileBrowserRouter } from "./routes/file-browser.js";
-import { aiSourcesRouter } from "./routes/ai-sources.js";
-import { processesRouter } from "./routes/processes.js";
-import { systemMetricsRouter } from "./routes/system-metrics.js";
-import { taskContextRouter } from "./routes/task-context.js";
-import { fileDiffRollbackRouter } from "./routes/file-diff-rollback.js";
-import { attachPtyWebSocketServer, closeAllPtySessions, ptyRouter, ptySessions, } from "./routes/pty.js";
-import { runtimeBindingRouter, logRuntimeBindingStartupState, } from "./routes/runtime-binding.js";
-import { dashboardRouter } from "./routes/dashboard.js";
-import { eventsRouter } from "./routes/events.js";
-import { loadPersistedSessions, savePersistedSessions, } from "./services/terminal-persistence.js";
-import { detectOrphanProcesses } from "./services/process-detector.js";
-import { startOrphanMonitor, stopOrphanMonitor, } from "./services/orphan-monitor.js";
-import { getAgentProcessManager } from "./services/agent-process-manager.js";
-import { adapterRegistry } from "./adapter/index.js";
-import { logger } from "./utils/logger.js";
-import { autoRegister, unregister, bridgeRestartCleanup, cancelBackgroundRegisterRetries, } from "./services/auto-register.js";
-import { stopTunnelClient } from "./services/tunnel-client.js";
-import { ensureAwsClientAgentMcpReleased } from "./services/aws-client-agent-mcp.js";
-import { ensureAcodePackageReleased } from "./services/acode-package.js";
-import { ensureStartupConfig } from "./services/startup-config-wizard.js";
-import { handleCliCommand } from "./services/cli-commands.js";
-import { migrateOldPathsOnStartup } from "./services/legacy-path-migration.js";
-import { subAgentStore } from "./services/sub-agent-manager-instance.js";
-import { buildGracefulShutdownPlan, shouldCleanupPersistedSessionsOnStartup, } from "./services/runtime-lifecycle-policy.js";
-import { deliverPendingBridgeLifecycleNotifications } from "./services/lifecycle-state.js";
-const cliCommandResult = await handleCliCommand();
-if (cliCommandResult.handled) {
-    process.exit(cliCommandResult.exitCode);
-}
-// 验证生产环境配置
-await ensureStartupConfig();
-validateProductionToken();
-ensureAwsClientAgentMcpReleased();
-ensureAcodePackageReleased();
-migrateOldPathsOnStartup();
-logRuntimeBindingStartupState();
-// ★ 启动时自动注册
-async function performAutoRegister() {
-    try {
-        const success = await autoRegister();
-        if (success) {
-            logger.info("[runtime-bridge] 自动注册成功");
-        }
-        else {
-            // 检查注册状态
-            const { loadConfig } = await import("./services/auto-register.js");
-            const config = loadConfig();
-            // 判断是否有配置（userKey 为必填项；tenantId 可自动推导，不参与判断）
-            const hasConfig = Boolean(config.userKey);
-            if (hasConfig && config.enabled) {
-                // 有配置但暂时注册失败（如 server 尚未启动）：
-                // autoRegisterSingle 已调度后台持续重试，连上后自动完成注册与隧道建立，
-                // bridge 继续正常提供本地服务
-                logger.warn("[runtime-bridge] 自动注册暂时失败，将在后台持续重试直到与调度中心建立连接");
-            }
-            else {
-                // 没有配置，正常启动服务
-                logger.info("[runtime-bridge] 自动注册未配置，继续启动服务");
-                logger.info("[runtime-bridge] 要启用自动注册，请创建配置文件 ~/.aws-bridge/config.json");
-            }
-        }
-    }
-    catch (error) {
-        const err = error;
-        logger.warn("[runtime-bridge] 自动注册出错:", err.message);
-    }
-}
-performAutoRegister()
-    .then(() => {
-    // Bridge 启动后只做非破坏性同步：调度中心会通过 /runtime/sessions 恢复运行态。
-    // 不再默认把该 bridge 下的 Agent 标记为 offline/stopped，避免 server/bridge
-    // 短暂重启导致正在运行的 Agent 被误停。
-    return performBridgeRestartCleanup();
-})
-    .then(() => deliverPendingBridgeLifecycleNotifications())
-    .then(() => {
-    logger.info("[runtime-bridge] Bridge 生命周期通知同步完成");
-})
-    .catch((err) => {
-    logger.warn("[runtime-bridge] 自动注册或清理出错:", err);
-});
-// Bridge 启动状态同步。
-// 默认不清理 Agent 状态；只有显式设置 AWS_BRIDGE_RESTART_CLEANUP=true 时才保留旧行为。
-async function performBridgeRestartCleanup() {
-    try {
-        if (process.env.AWS_BRIDGE_RESTART_CLEANUP !== "true") {
-            logger.info("[runtime-bridge] 跳过 Bridge 重启清理，保留 Agent 状态供调度中心恢复");
-            return;
-        }
-        // 只有在已注册的情况下才执行清理
-        const { getRegistrationState } = await import("./services/auto-register.js");
-        const state = getRegistrationState();
-        if (!state.registered || !state.instanceId) {
-            logger.info("[runtime-bridge] 实例未注册，跳过 Bridge 重启清理");
-            return;
-        }
-        logger.info("[runtime-bridge] ★ Bridge 重启清理已显式启用，正在通知调度中心清理 Agent 状态...");
-        const result = await bridgeRestartCleanup();
-        if (result.success) {
-            logger.info(`[runtime-bridge] ★ Bridge 重启清理成功，清理了 ${result.agentCount || 0} 个 Agent`);
-        }
-        else {
-            logger.warn(`[runtime-bridge] ★ Bridge 重启清理失败: ${result.error}`);
-        }
-    }
-    catch (error) {
-        const err = error;
-        logger.warn("[runtime-bridge] Bridge 重启清理出错:", err.message);
-    }
-}
-// ★ 启动时重建进程注册表
-async function rebuildProcessRegistry() {
-    try {
-        const persistedSessions = await loadPersistedSessions();
-        if (persistedSessions.length === 0) {
-            logger.info("[runtime-bridge] 无持久化会话，跳过进程注册表重建");
-            return;
-        }
-        const manager = getAgentProcessManager();
-        await manager.rebuildRegistry(persistedSessions.map((s) => ({
-            agentId: s.agentId,
-            sessionId: s.sessionId,
-            pid: s.pid,
-            workspacePath: s.workspacePath,
-            command: s.command,
-            mode: "sdk",
-        })));
-        logger.info(`[runtime-bridge] 进程注册表重建完成: ${persistedSessions.length} 条记录`);
-    }
-    catch (error) {
-        const err = error;
-        logger.warn("[runtime-bridge] 进程注册表重建失败:", err.message);
-    }
-}
-rebuildProcessRegistry().catch((err) => {
-    logger.warn("[runtime-bridge] 进程注册表重建出错:", err);
-});
-// ★ 启动时检测孤儿进程。
-// 默认保留持久化 running 会话并重建进程注册表，避免 runtime-bridge/server
-// 重启误杀仍在运行的 SDK 进程。只有显式设置
-// AWS_BRIDGE_CLEAN_ORPHANS_ON_STARTUP=true 时才执行历史清理逻辑。
-async function detectOrphansOnStartup() {
-    try {
-        // ★★★ 检查是否存在保留会话标记文件
-        const fs = await import("node:fs/promises");
-        const path = await import("node:path");
-        const os = await import("node:os");
-        const restartFlagFile = path.join(os.tmpdir(), "agentswork-runtime-bridge", ".preserve-sessions");
-        let preserveSessions = false;
-        try {
-            const flagContent = await fs.readFile(restartFlagFile, "utf-8");
-            const flag = JSON.parse(flagContent);
-            preserveSessions = flag.preserveSessions === true;
-            // 读取后删除标记文件
-            await fs.unlink(restartFlagFile);
-            logger.info(`[runtime-bridge] ★ 检测到保留会话标记，preserveSessions=${preserveSessions}`);
-        }
-        catch {
-            // 标记文件不存在，正常启动
-        }
-        const persistedSessions = await loadPersistedSessions();
-        if (persistedSessions.length === 0) {
-            logger.info("[runtime-bridge] 启动时无持久化会话，无需检测孤儿进程");
-            return;
-        }
-        const shouldCleanup = shouldCleanupPersistedSessionsOnStartup(preserveSessions);
-        if (!shouldCleanup) {
-            logger.info(`[runtime-bridge] ★ 保留 ${persistedSessions.length} 个持久化会话并重建进程注册表`);
-            // 重建进程注册表
-            const manager = getAgentProcessManager();
-            await manager.rebuildRegistry(persistedSessions.map((s) => ({
-                agentId: s.agentId,
-                sessionId: s.sessionId,
-                pid: s.pid,
-                workspacePath: s.workspacePath,
-                command: s.command,
-                mode: "sdk",
-            })));
-            return;
-        }
-        logger.info(`[runtime-bridge] ★ 已显式启用启动清理，检测 ${persistedSessions.length} 个持久化会话的孤儿进程...`);
-        const orphans = await detectOrphanProcesses(persistedSessions.map((s) => ({
-            agentId: s.agentId,
-            pid: s.pid,
-            workspacePath: s.workspacePath,
-            command: s.command,
-        })));
-        if (orphans.length > 0) {
-            logger.warn(`[runtime-bridge] ★ 检测到 ${orphans.length} 个孤儿进程仍在运行：`);
-            for (const orphan of orphans) {
-                logger.warn(`  - agentId=${orphan.agentId}, pid=${orphan.process.pid}, command=${orphan.process.command?.slice(0, 50)}`);
-            }
-            // ★ 自动清理所有孤儿进程
-            logger.info("[runtime-bridge] 正在自动清理所有孤儿进程...");
-            for (const orphan of orphans) {
-                try {
-                    if (orphan.process.pid) {
-                        // 使用跨平台方式终止进程
-                        if (process.platform === "win32") {
-                            const { execSync } = await import("node:child_process");
-                            execSync(`taskkill /PID ${orphan.process.pid} /T /F`, {
-                                encoding: "utf8",
-                                timeout: 3000,
-                            });
-                            logger.info(`[runtime-bridge] 已自动终止孤儿进程: agentId=${orphan.agentId}, PID=${orphan.process.pid}`);
-                        }
-                        else {
-                            process.kill(orphan.process.pid, "SIGKILL");
-                            logger.info(`[runtime-bridge] 已自动终止孤儿进程: agentId=${orphan.agentId}, PID=${orphan.process.pid}`);
-                        }
-                    }
-                }
-                catch (killError) {
-                    const err = killError;
-                    logger.warn(`[runtime-bridge] 自动终止孤儿进程失败: agentId=${orphan.agentId}, PID=${orphan.process.pid}, error=${err.message}`);
-                }
-            }
-            // 等待进程完全终止
-            await new Promise((resolve) => setTimeout(resolve, 1000));
-        }
-        await savePersistedSessions([]);
-        logger.info(`[runtime-bridge] ★ 启动清理完成，已清空所有持久化会话（清理了 ${orphans.length} 个孤儿进程）`);
-    }
-    catch (error) {
-        const err = error;
-        logger.warn("[runtime-bridge] 启动时孤儿进程检测失败:", err.message);
-    }
-}
-// 执行孤儿进程检测
-detectOrphansOnStartup().catch((err) => {
-    logger.warn("[runtime-bridge] 孤儿进程检测出错:", err);
-});
-// ★ 启动孤儿进程定期监控
-startOrphanMonitor().catch((err) => {
-    logger.warn("[runtime-bridge] 启动孤儿进程监控失败:", err);
-});
-// 创建 Express 应用
-const app = express();
-// 中间件配置
-app.use(cors({
-    origin(origin, callback) {
-        // 允许无来源请求（非浏览器客户端、curl 等）
-        if (!origin) {
-            callback(null, true);
-            return;
-        }
-        // AWS_RUNTIME_CORS_ORIGINS=* 时放行所有来源
-        if (allowedCorsOrigins.includes("*")) {
-            callback(null, true);
-            return;
-        }
-        if (allowedCorsOrigins.includes(origin)) {
-            callback(null, true);
-            return;
-        }
-        // 允许 bridge 自身端口的来源（dashboard/登录页面同源请求）
-        // 无论主机名是 localhost、127.0.0.1 还是外部 IP，只要端口匹配就放行
-        try {
-            const url = new URL(origin);
-            if (url.port === String(port)) {
-                callback(null, true);
-                return;
-            }
-        }
-        catch {
-            // URL 解析失败，继续拒绝
-        }
-        callback(new Error(`[runtime-bridge] CORS origin is not allowed: ${origin}`));
-    },
-}));
-app.use(express.json({ limit: "1mb" }));
-// 注册路由
-app.use("/runtime", runtimeBindingRouter);
-app.use("/runtime", propertiesRouter);
-app.use("/runtime", ymlRouter);
-app.use("/runtime", instanceRouter);
-app.use("/runtime", skillsRouter);
-app.use("/runtime", mcpRouter);
-app.use("/runtime", terminalRouter);
-app.use("/runtime", sessionsRouter);
-app.use("/runtime", processesRouter);
-app.use("/runtime", systemMetricsRouter);
-app.use("/runtime", eventsRouter);
-app.use("/runtime", gitRouter);
-app.use("/runtime", fileSnapshotRouter);
-app.use("/runtime", acodeTodosRouter);
-app.use("/runtime", subAgentsRouter);
-app.use("/runtime", aiSourcesRouter);
-app.use("/runtime", taskContextRouter);
-app.use("/runtime", fileDiffRollbackRouter);
-app.use("/pty", ptyRouter);
-app.use("/", dashboardRouter);
-app.use("/api/file-browser", fileBrowserRouter);
-/**
- * 启动 HTTP 服务并对端口占用给出可诊断错误，避免未捕获异常直接崩溃。
- */
-function startServer() {
-    const server = app.listen(port, () => {
-        logger.info(`[runtime-bridge] listening on ${port}`);
-    });
-    attachPtyWebSocketServer(server);
-    server.on("error", (error) => {
-        if (error.code === "EADDRINUSE") {
-            logger.error(`[runtime-bridge] port ${port} is already in use. ` +
-                "Stop the existing runtime-bridge process or set AWS_RUNTIME_BRIDGE_PORT to another port before starting.");
-            process.exitCode = 1;
-            return;
-        }
-        logger.error("[runtime-bridge] failed to start server:", error);
-        process.exitCode = 1;
-    });
-    return server;
-}
-/**
- * 优雅关闭处理
- *
- * 在收到终止信号时：
- * 1. 停止所有 SDK 会话（终止 Agent 进程）
- * 2. 验证所有进程已终止
- * 3. 清空持久化状态（避免恢复残留会话）
- * 4. 关闭 HTTP 服务器
- *
- * ★★★ 修复：添加 preserveSessions 参数
- * - preserveSessions=true: aws-mcp-server 触发的关闭，保留会话状态
- * - preserveSessions=false: bridge 自己关闭，清理所有会话
- */
-let isShuttingDown = false;
-async function gracefulShutdown(signal, server, preserveSessions = false) {
-    if (isShuttingDown) {
-        logger.info(`[runtime-bridge] 已在关闭中，忽略 ${signal}`);
-        return;
-    }
-    isShuttingDown = true;
-    // ★★★ 防御性检测：如果 preserveSessions 未显式设为 true，
-    // 检查 .preserve-sessions 标记文件是否存在。
-    // 这是为了覆盖所有调用路径（包括非信号处理器路径）都能正确识别 aws-mcp-server 重启场景。
-    if (!preserveSessions) {
-        const flagExists = checkPreserveFlagSync();
-        if (flagExists) {
-            preserveSessions = true;
-            logger.info(`[runtime-bridge] ★ 检测到 .preserve-sessions 标记文件，自动升级为保留会话模式`);
-        }
-    }
-    logger.info(`[runtime-bridge] 收到 ${signal} 信号，开始优雅关闭... (preserveSessions=${preserveSessions})`);
-    const startTime = Date.now();
-    const shutdownPlan = buildGracefulShutdownPlan(preserveSessions);
-    try {
-        // 1. 停止所有 SDK 会话
-        const sdkCount = sdkSessions.size;
-        if (shutdownPlan.terminateSdkSessions) {
-            logger.info(`[runtime-bridge] 正在停止 ${sdkCount} 个 SDK 会话...`);
-            for (const [sessionId, entry] of sdkSessions.entries()) {
-                try {
-                    const providerId = entry.providerId || "claude-code";
-                    const adapter = adapterRegistry.get(providerId);
-                    // 尝试获取 PID 用于日志
-                    let pidInfo = "";
-                    try {
-                        const pid = adapter.getSessionPid?.(sessionId);
-                        if (pid) {
-                            pidInfo = ` (PID: ${pid})`;
-                        }
-                    }
-                    catch {
-                        // 忽略 PID 获取错误
-                    }
-                    await adapter.terminateSession(sessionId);
-                    logger.info(`[runtime-bridge] SDK 会话 ${sessionId} 已终止${pidInfo}`);
-                    sdkSessions.delete(sessionId);
-                }
-                catch (e) {
-                    const err = e;
-                    logger.error(`[runtime-bridge] SDK 会话 ${sessionId} 终止失败:`, err.message);
-                    // 即使失败也从内存中移除
-                    sdkSessions.delete(sessionId);
-                }
-            }
-        }
-        else {
-            logger.info(`[runtime-bridge] ★ 保留 ${sdkCount} 个 SDK 会话，不在重启恢复场景中终止 Agent 进程`);
-        }
-        const ptyCount = ptySessions.size;
-        if (shutdownPlan.closePtySessions) {
-            logger.info(`[runtime-bridge] 正在停止 ${ptyCount} 个 PTY 会话...`);
-            closeAllPtySessions("shutdown");
-        }
-        else {
-            logger.info(`[runtime-bridge] ★ 保留 ${ptyCount} 个 PTY 会话，不在重启恢复场景中关闭终端进程`);
-        }
-        // 3. 停止孤儿进程监控
-        logger.info("[runtime-bridge] 正在停止孤儿进程监控...");
-        stopOrphanMonitor();
-        // 4. ★ 验证所有进程已终止（等待一小段时间让进程完全退出）
-        if (shutdownPlan.terminateResidualProcesses) {
-            logger.info("[runtime-bridge] 等待进程完全终止...");
-            await new Promise((resolve) => setTimeout(resolve, 1000));
-        }
-        // 5. 检查是否还有残留进程，如果有则强制终止
-        if (shutdownPlan.terminateResidualProcesses) {
-            try {
-                const { detectOrphanProcesses, terminateProcessTree } = await import("./services/process-detector.js");
-                const { loadPersistedSessions } = await import("./services/terminal-persistence.js");
-                const persistedSessions = await loadPersistedSessions();
-                if (persistedSessions.length > 0) {
-                    const orphans = await detectOrphanProcesses(persistedSessions.map((s) => ({
-                        agentId: s.agentId,
-                        pid: s.pid,
-                        workspacePath: s.workspacePath,
-                        command: s.command,
-                    })));
-                    if (orphans.length > 0) {
-                        logger.warn(`[runtime-bridge] 检测到 ${orphans.length} 个残留进程，正在强制终止...`);
-                        for (const orphan of orphans) {
-                            if (orphan.process.pid) {
-                                // 使用进程树终止（跨平台）
-                                const result = await terminateProcessTree(orphan.process.pid);
-                                logger.info(`[runtime-bridge] 已强制终止残留进程树：agentId=${orphan.agentId}, PID=${orphan.process.pid}, 终止了 ${result.terminated} 个进程`);
-                            }
-                        }
-                    }
-                }
-            }
-            catch (checkError) {
-                const err = checkError;
-                logger.warn("[runtime-bridge] 残留进程检查失败:", err.message);
-            }
-        }
-        else {
-            logger.info("[runtime-bridge] ★ 保留模式跳过残留进程强制终止，等待调度中心恢复会话");
-        }
-        // 6. 注销实例
-        if (shutdownPlan.unregisterInstance) {
-            logger.info("[runtime-bridge] 正在注销实例...");
-            try {
-                await unregister();
-            }
-            catch (e) {
-                const err = e;
-                logger.error("[runtime-bridge] 注销实例失败:", err.message);
-            }
-        }
-        else {
-            logger.info("[runtime-bridge] ★ 保留模式跳过实例注销，避免调度中心将 Agent 标记为 stopped");
-        }
-        // 6.5 停止后台注册重试 + 关闭隧道客户端（若运行中）
-        cancelBackgroundRegisterRetries();
-        if (shutdownPlan.stopTunnelClient) {
-            logger.info("[runtime-bridge] 正在关闭隧道客户端...");
-            stopTunnelClient();
-        }
-        // 7. ★★★ 修复：根据 preserveSessions 决定是否清空持久化状态
-        // - preserveSessions=true (aws-mcp-server 触发): 保留会话状态，aws-mcp-server 重启后可恢复
-        // - preserveSessions=false (bridge 自己关闭): 清空所有会话，agent 应被清理
-        if (shutdownPlan.clearPersistedSessions) {
-            logger.info("[runtime-bridge] 正在清空持久化会话状态...");
-            try {
-                await savePersistedSessions([]);
-            }
-            catch (e) {
-                const err = e;
-                logger.error("[runtime-bridge] 清空持久化状态失败:", err.message);
-            }
-        }
-        else {
-            logger.info("[runtime-bridge] ★ 保留持久化会话状态（aws-mcp-server 重启恢复场景）");
-        }
-        // 8. 关闭 HTTP 服务器
-        const elapsed = Date.now() - startTime;
-        logger.info(`[runtime-bridge] 优雅关闭完成 (耗时 ${elapsed}ms)，停止 ${sdkCount} SDK 会话`);
-        server.close(() => {
-            logger.info("[runtime-bridge] HTTP 服务器已关闭");
-            process.exit(process.exitCode || 0);
-        });
-        // 强制退出超时（防止卡住）
-        setTimeout(() => {
-            logger.warn("[runtime-bridge] 优雅关闭超时，强制退出");
-            process.exit(1);
-        }, 15000);
-    }
-    catch (e) {
-        const err = e;
-        logger.error("[runtime-bridge] 优雅关闭过程中发生错误:", err);
-        process.exit(1);
-    }
-}
-// 启动前清理僵尸 sub-agent 记录（bridge 重启前未完成的 running/pending 记录）
-subAgentStore.reconcileZombies().then((fixed) => {
-    if (fixed > 0) {
-        logger.info(`[runtime-bridge] 已清理 ${fixed} 个僵尸 sub-agent 记录`);
-    }
-}).catch((err) => {
-    logger.warn("[runtime-bridge] 清理僵尸 sub-agent 记录失败:", err.message);
-});
-// 启动服务器
-const server = startServer();
-/**
- * ★★★ 修复：同步检测 .preserve-sessions 标记文件
- *
- * 当 aws-mcp-server 调用 /runtime/prepare-restart 后，
- * 会在临时目录创建该标记文件。桥接器收到 SIGTERM/SIGINT 时，
- * 应检查此文件来决定是否保留会话。
- *
- * 信号处理器必须使用同步 I/O，因为异步操作在信号上下文中不可靠。
- */
-const RESTART_FLAG_FILE = path.join(os.tmpdir(), "agentswork-runtime-bridge", ".preserve-sessions");
-function checkPreserveFlagSync() {
-    try {
-        if (existsSync(RESTART_FLAG_FILE)) {
-            const content = readFileSync(RESTART_FLAG_FILE, "utf-8");
-            const flag = JSON.parse(content);
-            return flag.preserveSessions === true;
-        }
-    }
-    catch {
-        // 文件不存在或无法解析，按不保留处理
-    }
-    return false;
-}
-// ★★★ 设置 gracefulShutdown 函数引用，供 instance router 使用
-setGracefulShutdownFn(async (signal, preserveSessions) => {
-    await gracefulShutdown(signal, server, preserveSessions);
-});
-// 注册终止信号处理器
-// ★★★ 修复：收到终止信号时先检查 aws-mcp-server 是否已调用 prepare-restart
-// 如果标记文件存在，说明是 aws-mcp-server 重启场景，需要保留会话状态
-process.on("SIGTERM", () => {
-    const preserve = checkPreserveFlagSync();
-    gracefulShutdown("SIGTERM", server, preserve);
-});
-process.on("SIGINT", () => {
-    const preserve = checkPreserveFlagSync();
-    gracefulShutdown("SIGINT", server, preserve);
-});
-// Windows 支持（Ctrl+C 在 Windows 上触发 SIGBREAK）
-if (process.platform === "win32") {
-    process.on("SIGBREAK", () => {
-        const preserve = checkPreserveFlagSync();
-        gracefulShutdown("SIGBREAK", server, preserve);
-    });
-}
-// 未捕获异常处理
-process.on("uncaughtException", (error) => {
-    logger.error("[runtime-bridge] 未捕获异常:", error);
-    // ★ 未捕获异常时不检查标记文件 — 异常重启需要完整清理
-    gracefulShutdown("uncaughtException", server);
-});
-process.on("unhandledRejection", (reason) => {
-    logger.error("[runtime-bridge] 未处理的 Promise 拒绝:", reason);
-    // 不立即退出，只记录错误
-});
+import I from"express";import E from"cors";import{existsSync as D,readFileSync as B}from"node:fs";import x from"node:path";import O from"node:os";import{allowedCorsOrigins as P,port as w,validateProductionToken as G}from"./config.js";import{propertiesRouter as F}from"./routes/properties.js";import{ymlRouter as K}from"./routes/yml.js";import{instanceRouter as M,setGracefulShutdownFn as N}from"./routes/instance.js";import{skillsRouter as _}from"./routes/skills.js";import{mcpRouter as L}from"./routes/mcp.js";import{terminalRouter as j,sdkSessions as S}from"./routes/terminal.js";import{sessionsRouter as U}from"./routes/sessions.js";import{gitRouter as W}from"./routes/git.js";import{fileSnapshotRouter as z}from"./routes/file-snapshot.js";import{acodeTodosRouter as J}from"./routes/acode-todos.js";import{subAgentsRouter as Y}from"./routes/sub-agents.js";import{fileBrowserRouter as H}from"./routes/file-browser.js";import{aiSourcesRouter as Z}from"./routes/ai-sources.js";import{processesRouter as q}from"./routes/processes.js";import{systemMetricsRouter as Q}from"./routes/system-metrics.js";import{taskContextRouter as V}from"./routes/task-context.js";import{fileDiffRollbackRouter as X}from"./routes/file-diff-rollback.js";import{attachPtyWebSocketServer as ee,closeAllPtySessions as re,ptyRouter as te,ptySessions as ie}from"./routes/pty.js";import{runtimeBindingRouter as ne,logRuntimeBindingStartupState as oe}from"./routes/runtime-binding.js";import{dashboardRouter as se}from"./routes/dashboard.js";import{eventsRouter as ae}from"./routes/events.js";import{loadPersistedSessions as y,savePersistedSessions as $}from"./services/terminal-persistence.js";import{detectOrphanProcesses as me}from"./services/process-detector.js";import{startOrphanMonitor as ue,stopOrphanMonitor as ce}from"./services/orphan-monitor.js";import{getAgentProcessManager as C}from"./services/agent-process-manager.js";import{adapterRegistry as de}from"./adapter/index.js";import{logger as e}from"./utils/logger.js";import{autoRegister as pe,unregister as ge,bridgeRestartCleanup as fe,cancelBackgroundRegisterRetries as le}from"./services/auto-register.js";import{stopTunnelClient as be}from"./services/tunnel-client.js";import{ensureAwsClientAgentMcpReleased as he}from"./services/aws-client-agent-mcp.js";import{ensureAcodePackageReleased as we}from"./services/acode-package.js";import{ensureStartupConfig as Se}from"./services/startup-config-wizard.js";import{handleCliCommand as Re}from"./services/cli-commands.js";import{migrateOldPathsOnStartup as Ie}from"./services/legacy-path-migration.js";import{subAgentStore as Pe}from"./services/sub-agent-manager-instance.js";import{buildGracefulShutdownPlan as ye,shouldCleanupPersistedSessionsOnStartup as $e}from"./services/runtime-lifecycle-policy.js";import{deliverPendingBridgeLifecycleNotifications as Ce}from"./services/lifecycle-state.js";const k=await Re();k.handled&&process.exit(k.exitCode),await Se(),G(),he(),we(),Ie(),oe();async function ke(){try{if(await pe())e.info("[runtime-bridge] \u81EA\u52A8\u6CE8\u518C\u6210\u529F");else{const{loadConfig:i}=await import("./services/auto-register.js"),o=i();!!o.userKey&&o.enabled?e.warn("[runtime-bridge] \u81EA\u52A8\u6CE8\u518C\u6682\u65F6\u5931\u8D25\uFF0C\u5C06\u5728\u540E\u53F0\u6301\u7EED\u91CD\u8BD5\u76F4\u5230\u4E0E\u8C03\u5EA6\u4E2D\u5FC3\u5EFA\u7ACB\u8FDE\u63A5"):(e.info("[runtime-bridge] \u81EA\u52A8\u6CE8\u518C\u672A\u914D\u7F6E\uFF0C\u7EE7\u7EED\u542F\u52A8\u670D\u52A1"),e.info("[runtime-bridge] \u8981\u542F\u7528\u81EA\u52A8\u6CE8\u518C\uFF0C\u8BF7\u521B\u5EFA\u914D\u7F6E\u6587\u4EF6 ~/.aws-bridge/config.json"))}}catch(r){const i=r;e.warn("[runtime-bridge] \u81EA\u52A8\u6CE8\u518C\u51FA\u9519:",i.message)}}ke().then(()=>Te()).then(()=>Ce()).then(()=>{e.info("[runtime-bridge] Bridge \u751F\u547D\u5468\u671F\u901A\u77E5\u540C\u6B65\u5B8C\u6210")}).catch(r=>{e.warn("[runtime-bridge] \u81EA\u52A8\u6CE8\u518C\u6216\u6E05\u7406\u51FA\u9519:",r)});async function Te(){try{if(process.env.AWS_BRIDGE_RESTART_CLEANUP!=="true"){e.info("[runtime-bridge] \u8DF3\u8FC7 Bridge \u91CD\u542F\u6E05\u7406\uFF0C\u4FDD\u7559 Agent \u72B6\u6001\u4F9B\u8C03\u5EA6\u4E2D\u5FC3\u6062\u590D");return}const{getRegistrationState:r}=await import("./services/auto-register.js"),i=r();if(!i.registered||!i.instanceId){e.info("[runtime-bridge] \u5B9E\u4F8B\u672A\u6CE8\u518C\uFF0C\u8DF3\u8FC7 Bridge \u91CD\u542F\u6E05\u7406");return}e.info("[runtime-bridge] \u2605 Bridge \u91CD\u542F\u6E05\u7406\u5DF2\u663E\u5F0F\u542F\u7528\uFF0C\u6B63\u5728\u901A\u77E5\u8C03\u5EA6\u4E2D\u5FC3\u6E05\u7406 Agent \u72B6\u6001...");const o=await fe();o.success?e.info(`[runtime-bridge] \u2605 Bridge \u91CD\u542F\u6E05\u7406\u6210\u529F\uFF0C\u6E05\u7406\u4E86 ${o.agentCount||0} \u4E2A Agent`):e.warn(`[runtime-bridge] \u2605 Bridge \u91CD\u542F\u6E05\u7406\u5931\u8D25: ${o.error}`)}catch(r){const i=r;e.warn("[runtime-bridge] Bridge \u91CD\u542F\u6E05\u7406\u51FA\u9519:",i.message)}}async function ve(){try{const r=await y();if(r.length===0){e.info("[runtime-bridge] \u65E0\u6301\u4E45\u5316\u4F1A\u8BDD\uFF0C\u8DF3\u8FC7\u8FDB\u7A0B\u6CE8\u518C\u8868\u91CD\u5EFA");return}await C().rebuildRegistry(r.map(o=>({agentId:o.agentId,sessionId:o.sessionId,pid:o.pid,workspacePath:o.workspacePath,command:o.command,mode:"sdk"}))),e.info(`[runtime-bridge] \u8FDB\u7A0B\u6CE8\u518C\u8868\u91CD\u5EFA\u5B8C\u6210: ${r.length} \u6761\u8BB0\u5F55`)}catch(r){const i=r;e.warn("[runtime-bridge] \u8FDB\u7A0B\u6CE8\u518C\u8868\u91CD\u5EFA\u5931\u8D25:",i.message)}}ve().catch(r=>{e.warn("[runtime-bridge] \u8FDB\u7A0B\u6CE8\u518C\u8868\u91CD\u5EFA\u51FA\u9519:",r)});async function Ae(){try{const r=await import("node:fs/promises"),i=await import("node:path"),o=await import("node:os"),f=i.join(o.tmpdir(),"agentswork-runtime-bridge",".preserve-sessions");let u=!1;try{const t=await r.readFile(f,"utf-8");u=JSON.parse(t).preserveSessions===!0,await r.unlink(f),e.info(`[runtime-bridge] \u2605 \u68C0\u6D4B\u5230\u4FDD\u7559\u4F1A\u8BDD\u6807\u8BB0\uFF0CpreserveSessions=${u}`)}catch{}const a=await y();if(a.length===0){e.info("[runtime-bridge] \u542F\u52A8\u65F6\u65E0\u6301\u4E45\u5316\u4F1A\u8BDD\uFF0C\u65E0\u9700\u68C0\u6D4B\u5B64\u513F\u8FDB\u7A0B");return}if(!$e(u)){e.info(`[runtime-bridge] \u2605 \u4FDD\u7559 ${a.length} \u4E2A\u6301\u4E45\u5316\u4F1A\u8BDD\u5E76\u91CD\u5EFA\u8FDB\u7A0B\u6CE8\u518C\u8868`),await C().rebuildRegistry(a.map(s=>({agentId:s.agentId,sessionId:s.sessionId,pid:s.pid,workspacePath:s.workspacePath,command:s.command,mode:"sdk"})));return}e.info(`[runtime-bridge] \u2605 \u5DF2\u663E\u5F0F\u542F\u7528\u542F\u52A8\u6E05\u7406\uFF0C\u68C0\u6D4B ${a.length} \u4E2A\u6301\u4E45\u5316\u4F1A\u8BDD\u7684\u5B64\u513F\u8FDB\u7A0B...`);const c=await me(a.map(t=>({agentId:t.agentId,pid:t.pid,workspacePath:t.workspacePath,command:t.command})));if(c.length>0){e.warn(`[runtime-bridge] \u2605 \u68C0\u6D4B\u5230 ${c.length} \u4E2A\u5B64\u513F\u8FDB\u7A0B\u4ECD\u5728\u8FD0\u884C\uFF1A`);for(const t of c)e.warn(`  - agentId=${t.agentId}, pid=${t.process.pid}, command=${t.process.command?.slice(0,50)}`);e.info("[runtime-bridge] \u6B63\u5728\u81EA\u52A8\u6E05\u7406\u6240\u6709\u5B64\u513F\u8FDB\u7A0B...");for(const t of c)try{if(t.process.pid)if(process.platform==="win32"){const{execSync:s}=await import("node:child_process");s(`taskkill /PID ${t.process.pid} /T /F`,{encoding:"utf8",timeout:3e3}),e.info(`[runtime-bridge] \u5DF2\u81EA\u52A8\u7EC8\u6B62\u5B64\u513F\u8FDB\u7A0B: agentId=${t.agentId}, PID=${t.process.pid}`)}else process.kill(t.process.pid,"SIGKILL"),e.info(`[runtime-bridge] \u5DF2\u81EA\u52A8\u7EC8\u6B62\u5B64\u513F\u8FDB\u7A0B: agentId=${t.agentId}, PID=${t.process.pid}`)}catch(s){const d=s;e.warn(`[runtime-bridge] \u81EA\u52A8\u7EC8\u6B62\u5B64\u513F\u8FDB\u7A0B\u5931\u8D25: agentId=${t.agentId}, PID=${t.process.pid}, error=${d.message}`)}await new Promise(t=>setTimeout(t,1e3))}await $([]),e.info(`[runtime-bridge] \u2605 \u542F\u52A8\u6E05\u7406\u5B8C\u6210\uFF0C\u5DF2\u6E05\u7A7A\u6240\u6709\u6301\u4E45\u5316\u4F1A\u8BDD\uFF08\u6E05\u7406\u4E86 ${c.length} \u4E2A\u5B64\u513F\u8FDB\u7A0B\uFF09`)}catch(r){const i=r;e.warn("[runtime-bridge] \u542F\u52A8\u65F6\u5B64\u513F\u8FDB\u7A0B\u68C0\u6D4B\u5931\u8D25:",i.message)}}Ae().catch(r=>{e.warn("[runtime-bridge] \u5B64\u513F\u8FDB\u7A0B\u68C0\u6D4B\u51FA\u9519:",r)}),ue().catch(r=>{e.warn("[runtime-bridge] \u542F\u52A8\u5B64\u513F\u8FDB\u7A0B\u76D1\u63A7\u5931\u8D25:",r)});const n=I();n.use(E({origin(r,i){if(!r){i(null,!0);return}if(P.includes("*")){i(null,!0);return}if(P.includes(r)){i(null,!0);return}try{if(new URL(r).port===String(w)){i(null,!0);return}}catch{}i(new Error(`[runtime-bridge] CORS origin is not allowed: ${r}`))}})),n.use(I.json({limit:"1mb"})),n.use("/runtime",ne),n.use("/runtime",F),n.use("/runtime",K),n.use("/runtime",M),n.use("/runtime",_),n.use("/runtime",L),n.use("/runtime",j),n.use("/runtime",U),n.use("/runtime",q),n.use("/runtime",Q),n.use("/runtime",ae),n.use("/runtime",W),n.use("/runtime",z),n.use("/runtime",J),n.use("/runtime",Y),n.use("/runtime",Z),n.use("/runtime",V),n.use("/runtime",X),n.use("/pty",te),n.use("/",se),n.use("/api/file-browser",H);function Ee(){const r=n.listen(w,()=>{e.info(`[runtime-bridge] listening on ${w}`)});return ee(r),r.on("error",i=>{if(i.code==="EADDRINUSE"){e.error(`[runtime-bridge] port ${w} is already in use. Stop the existing runtime-bridge process or set AWS_RUNTIME_BRIDGE_PORT to another port before starting.`),process.exitCode=1;return}e.error("[runtime-bridge] failed to start server:",i),process.exitCode=1}),r}let T=!1;async function b(r,i,o=!1){if(T){e.info(`[runtime-bridge] \u5DF2\u5728\u5173\u95ED\u4E2D\uFF0C\u5FFD\u7565 ${r}`);return}T=!0,o||R()&&(o=!0,e.info("[runtime-bridge] \u2605 \u68C0\u6D4B\u5230 .preserve-sessions \u6807\u8BB0\u6587\u4EF6\uFF0C\u81EA\u52A8\u5347\u7EA7\u4E3A\u4FDD\u7559\u4F1A\u8BDD\u6A21\u5F0F")),e.info(`[runtime-bridge] \u6536\u5230 ${r} \u4FE1\u53F7\uFF0C\u5F00\u59CB\u4F18\u96C5\u5173\u95ED... (preserveSessions=${o})`);const f=Date.now(),u=ye(o);try{const a=S.size;if(u.terminateSdkSessions){e.info(`[runtime-bridge] \u6B63\u5728\u505C\u6B62 ${a} \u4E2A SDK \u4F1A\u8BDD...`);for(const[t,s]of S.entries())try{const d=s.providerId||"claude-code",p=de.get(d);let g="";try{const m=p.getSessionPid?.(t);m&&(g=` (PID: ${m})`)}catch{}await p.terminateSession(t),e.info(`[runtime-bridge] SDK \u4F1A\u8BDD ${t} \u5DF2\u7EC8\u6B62${g}`),S.delete(t)}catch(d){const p=d;e.error(`[runtime-bridge] SDK \u4F1A\u8BDD ${t} \u7EC8\u6B62\u5931\u8D25:`,p.message),S.delete(t)}}else e.info(`[runtime-bridge] \u2605 \u4FDD\u7559 ${a} \u4E2A SDK \u4F1A\u8BDD\uFF0C\u4E0D\u5728\u91CD\u542F\u6062\u590D\u573A\u666F\u4E2D\u7EC8\u6B62 Agent \u8FDB\u7A0B`);const l=ie.size;if(u.closePtySessions?(e.info(`[runtime-bridge] \u6B63\u5728\u505C\u6B62 ${l} \u4E2A PTY \u4F1A\u8BDD...`),re("shutdown")):e.info(`[runtime-bridge] \u2605 \u4FDD\u7559 ${l} \u4E2A PTY \u4F1A\u8BDD\uFF0C\u4E0D\u5728\u91CD\u542F\u6062\u590D\u573A\u666F\u4E2D\u5173\u95ED\u7EC8\u7AEF\u8FDB\u7A0B`),e.info("[runtime-bridge] \u6B63\u5728\u505C\u6B62\u5B64\u513F\u8FDB\u7A0B\u76D1\u63A7..."),ce(),u.terminateResidualProcesses&&(e.info("[runtime-bridge] \u7B49\u5F85\u8FDB\u7A0B\u5B8C\u5168\u7EC8\u6B62..."),await new Promise(t=>setTimeout(t,1e3))),u.terminateResidualProcesses)try{const{detectOrphanProcesses:t,terminateProcessTree:s}=await import("./services/process-detector.js"),{loadPersistedSessions:d}=await import("./services/terminal-persistence.js"),p=await d();if(p.length>0){const g=await t(p.map(m=>({agentId:m.agentId,pid:m.pid,workspacePath:m.workspacePath,command:m.command})));if(g.length>0){e.warn(`[runtime-bridge] \u68C0\u6D4B\u5230 ${g.length} \u4E2A\u6B8B\u7559\u8FDB\u7A0B\uFF0C\u6B63\u5728\u5F3A\u5236\u7EC8\u6B62...`);for(const m of g)if(m.process.pid){const A=await s(m.process.pid);e.info(`[runtime-bridge] \u5DF2\u5F3A\u5236\u7EC8\u6B62\u6B8B\u7559\u8FDB\u7A0B\u6811\uFF1AagentId=${m.agentId}, PID=${m.process.pid}, \u7EC8\u6B62\u4E86 ${A.terminated} \u4E2A\u8FDB\u7A0B`)}}}}catch(t){const s=t;e.warn("[runtime-bridge] \u6B8B\u7559\u8FDB\u7A0B\u68C0\u67E5\u5931\u8D25:",s.message)}else e.info("[runtime-bridge] \u2605 \u4FDD\u7559\u6A21\u5F0F\u8DF3\u8FC7\u6B8B\u7559\u8FDB\u7A0B\u5F3A\u5236\u7EC8\u6B62\uFF0C\u7B49\u5F85\u8C03\u5EA6\u4E2D\u5FC3\u6062\u590D\u4F1A\u8BDD");if(u.unregisterInstance){e.info("[runtime-bridge] \u6B63\u5728\u6CE8\u9500\u5B9E\u4F8B...");try{await ge()}catch(t){const s=t;e.error("[runtime-bridge] \u6CE8\u9500\u5B9E\u4F8B\u5931\u8D25:",s.message)}}else e.info("[runtime-bridge] \u2605 \u4FDD\u7559\u6A21\u5F0F\u8DF3\u8FC7\u5B9E\u4F8B\u6CE8\u9500\uFF0C\u907F\u514D\u8C03\u5EA6\u4E2D\u5FC3\u5C06 Agent \u6807\u8BB0\u4E3A stopped");if(le(),u.stopTunnelClient&&(e.info("[runtime-bridge] \u6B63\u5728\u5173\u95ED\u96A7\u9053\u5BA2\u6237\u7AEF..."),be()),u.clearPersistedSessions){e.info("[runtime-bridge] \u6B63\u5728\u6E05\u7A7A\u6301\u4E45\u5316\u4F1A\u8BDD\u72B6\u6001...");try{await $([])}catch(t){const s=t;e.error("[runtime-bridge] \u6E05\u7A7A\u6301\u4E45\u5316\u72B6\u6001\u5931\u8D25:",s.message)}}else e.info("[runtime-bridge] \u2605 \u4FDD\u7559\u6301\u4E45\u5316\u4F1A\u8BDD\u72B6\u6001\uFF08aws-mcp-server \u91CD\u542F\u6062\u590D\u573A\u666F\uFF09");const c=Date.now()-f;e.info(`[runtime-bridge] \u4F18\u96C5\u5173\u95ED\u5B8C\u6210 (\u8017\u65F6 ${c}ms)\uFF0C\u505C\u6B62 ${a} SDK \u4F1A\u8BDD`),i.close(()=>{e.info("[runtime-bridge] HTTP \u670D\u52A1\u5668\u5DF2\u5173\u95ED"),process.exit(process.exitCode||0)}),setTimeout(()=>{e.warn("[runtime-bridge] \u4F18\u96C5\u5173\u95ED\u8D85\u65F6\uFF0C\u5F3A\u5236\u9000\u51FA"),process.exit(1)},15e3)}catch(a){const l=a;e.error("[runtime-bridge] \u4F18\u96C5\u5173\u95ED\u8FC7\u7A0B\u4E2D\u53D1\u751F\u9519\u8BEF:",l),process.exit(1)}}Pe.reconcileZombies().then(r=>{r>0&&e.info(`[runtime-bridge] \u5DF2\u6E05\u7406 ${r} \u4E2A\u50F5\u5C38 sub-agent \u8BB0\u5F55`)}).catch(r=>{e.warn("[runtime-bridge] \u6E05\u7406\u50F5\u5C38 sub-agent \u8BB0\u5F55\u5931\u8D25:",r.message)});const h=Ee(),v=x.join(O.tmpdir(),"agentswork-runtime-bridge",".preserve-sessions");function R(){try{if(D(v)){const r=B(v,"utf-8");return JSON.parse(r).preserveSessions===!0}}catch{}return!1}N(async(r,i)=>{await b(r,h,i)}),process.on("SIGTERM",()=>{const r=R();b("SIGTERM",h,r)}),process.on("SIGINT",()=>{const r=R();b("SIGINT",h,r)}),process.platform==="win32"&&process.on("SIGBREAK",()=>{const r=R();b("SIGBREAK",h,r)}),process.on("uncaughtException",r=>{e.error("[runtime-bridge] \u672A\u6355\u83B7\u5F02\u5E38:",r),b("uncaughtException",h)}),process.on("unhandledRejection",r=>{e.error("[runtime-bridge] \u672A\u5904\u7406\u7684 Promise \u62D2\u7EDD:",r)});
+

package/dist/middleware/auth.d.ts

@@ -24,4 +24,3 @@ export declare function validateToken(req: Request, res: Response, next: NextFun
  * 适用于可从面板 UI 或外部 API 两种途径触发的操作。
  */
 export declare function validateTokenOrPanelSession(req: Request, res: Response, next: NextFunction): void;
-//# sourceMappingURL=auth.d.ts.map