agent-tower 0.1.0

package/dist/services/workspace.service.js

@@ -0,0 +1,351 @@
+import { prisma } from '../utils/index.js';
+import { WorkspaceStatus, TaskStatus, SessionStatus, SessionPurpose } from '../types/index.js';
+import { WorktreeManager } from '../git/worktree.manager.js';
+import { execGit } from '../git/git-cli.js';
+import { NotFoundError, ServiceError } from '../errors.js';
+import { getSessionManager, getEventBus } from '../core/container.js';
+/** 过滤条件：只返回用户可见的 CHAT session */
+const visibleSessionsFilter = { where: { purpose: { not: SessionPurpose.COMMIT_MSG } } };
+export class WorkspaceService {
+    sessionService = getSessionManager();
+    eventBus = getEventBus();
+    // ── Queries ──────────────────────────────────────────────────────────────────
+    async findById(id) {
+        return prisma.workspace.findUnique({
+            where: { id },
+            include: { sessions: visibleSessionsFilter, task: { include: { project: true } } },
+        });
+    }
+    /**
+     * 获取 Task 下所有 Workspace
+     */
+    async findByTaskId(taskId) {
+        const task = await prisma.task.findUnique({ where: { id: taskId } });
+        if (!task) {
+            throw new NotFoundError('Task', taskId);
+        }
+        return prisma.workspace.findMany({
+            where: { taskId },
+            include: { sessions: visibleSessionsFilter },
+            orderBy: { createdAt: 'desc' },
+        });
+    }
+    // ── Create ───────────────────────────────────────────────────────────────────
+    /**
+     * 创建 Workspace
+     *
+     * - 默认分支名: at/{workspace-short-id}（ID 前 8 位）
+     * - 支持用户自定义分支名
+     * - 创建前校验分支名合法性 & 是否已存在
+     * - 创建后自动将关联 Task 状态改为 IN_PROGRESS
+     * - 失败时回滚已创建的数据库记录
+     */
+    async create(taskId, branchName) {
+        const task = await prisma.task.findUnique({
+            where: { id: taskId },
+            include: { project: true },
+        });
+        if (!task) {
+            throw new NotFoundError('Task', taskId);
+        }
+        const worktreeManager = new WorktreeManager(task.project.repoPath);
+        // 查找可复用的 MERGED workspace（branch 已通过 update-ref 保留）
+        if (!branchName) {
+            const mergedWorkspace = await prisma.workspace.findFirst({
+                where: { taskId, status: WorkspaceStatus.MERGED },
+                orderBy: { updatedAt: 'desc' },
+            });
+            if (mergedWorkspace) {
+                const worktreePath = await worktreeManager.ensureWorktreeExists(mergedWorkspace.branchName);
+                const updated = await prisma.workspace.update({
+                    where: { id: mergedWorkspace.id },
+                    data: {
+                        status: WorkspaceStatus.ACTIVE,
+                        worktreePath,
+                    },
+                    include: { sessions: true, task: { include: { project: true } } },
+                });
+                // Task 状态回退到 IN_PROGRESS
+                if (task.status !== TaskStatus.IN_PROGRESS && task.status !== TaskStatus.TODO) {
+                    await prisma.task.update({
+                        where: { id: taskId },
+                        data: { status: TaskStatus.IN_PROGRESS },
+                    });
+                    this.eventBus.emit('task:updated', {
+                        taskId,
+                        projectId: task.projectId,
+                        status: TaskStatus.IN_PROGRESS,
+                    });
+                }
+                return updated;
+            }
+        }
+        // 先在数据库创建记录以获取 ID（用于生成默认分支名）
+        const workspace = await prisma.workspace.create({
+            data: {
+                taskId,
+                branchName: '', // 占位，稍后更新
+                worktreePath: '', // 占位，稍后更新
+                status: WorkspaceStatus.ACTIVE,
+            },
+        });
+        try {
+            // 生成分支名：用户指定 or 自动生成 at/{shortId}
+            const branch = branchName || `at/${workspace.id.slice(0, 8)}`;
+            // WorktreeManager.create 内部已做分支名合法性校验和重复检查
+            const worktreePath = await worktreeManager.create(branch);
+            // 更新数据库记录：填入真正的 branchName 和 worktreePath
+            const updated = await prisma.workspace.update({
+                where: { id: workspace.id },
+                data: { branchName: branch, worktreePath },
+                include: { sessions: true, task: { include: { project: true } } },
+            });
+            // 将关联 Task 状态改为 IN_PROGRESS（仅当当前为 TODO 时）
+            if (task.status === TaskStatus.TODO) {
+                await prisma.task.update({
+                    where: { id: taskId },
+                    data: { status: TaskStatus.IN_PROGRESS },
+                });
+                this.eventBus.emit('task:updated', {
+                    taskId,
+                    projectId: task.projectId,
+                    status: TaskStatus.IN_PROGRESS,
+                });
+            }
+            return updated;
+        }
+        catch (err) {
+            // 回滚：删除已创建的数据库记录
+            await prisma.workspace.delete({ where: { id: workspace.id } }).catch(() => {
+                // 忽略回滚失败
+            });
+            throw err;
+        }
+    }
+    // ── Delete ───────────────────────────────────────────────────────────────────
+    /**
+     * 删除 Workspace
+     *
+     * - 删除前检查是否有 RUNNING 状态的 Session，有则拒绝删除
+     * - 删除时先停止所有关联 Session 的进程
+     * - 删除时调用 WorktreeManager.remove 清理 worktree
+     * - worktree 清理失败时仍然删除数据库记录（记录警告日志）
+     */
+    async delete(id) {
+        const workspace = await prisma.workspace.findUnique({
+            where: { id },
+            include: {
+                sessions: true,
+                task: { include: { project: true } },
+            },
+        });
+        if (!workspace) {
+            throw new NotFoundError('Workspace', id);
+        }
+        // 停止所有活跃的 Session（RUNNING 和 PENDING 状态）
+        const activeSessions = workspace.sessions.filter((s) => s.status === SessionStatus.PENDING || s.status === SessionStatus.RUNNING);
+        for (const session of activeSessions) {
+            try {
+                await this.sessionService.stop(session.id);
+            }
+            catch {
+                // 忽略停止失败
+            }
+        }
+        // 清理 worktree
+        try {
+            const worktreeManager = new WorktreeManager(workspace.task.project.repoPath);
+            await worktreeManager.remove(workspace.worktreePath);
+        }
+        catch (err) {
+            // worktree 清理失败时记录警告但不阻断删除
+            console.warn(`[WorkspaceService] Failed to remove worktree for workspace ${id}: ${err instanceof Error ? err.message : err}`);
+        }
+        // 删除数据库记录（级联删除 sessions）
+        await prisma.workspace.delete({ where: { id } });
+        return true;
+    }
+    // ── Diff ─────────────────────────────────────────────────────────────────────
+    async getDiff(id) {
+        const workspace = await prisma.workspace.findUnique({
+            where: { id },
+            include: { task: { include: { project: true } } },
+        });
+        if (!workspace) {
+            throw new NotFoundError('Workspace', id);
+        }
+        const worktreeManager = new WorktreeManager(workspace.task.project.repoPath);
+        return worktreeManager.getDiff(workspace.worktreePath, workspace.task.project.mainBranch);
+    }
+    // ── Merge ────────────────────────────────────────────────────────────────────
+    // ── Git Operations ──────────────────────────────────────────────────────────
+    /**
+     * Rebase 工作空间分支到最新的基础分支
+     */
+    async rebase(id) {
+        const workspace = await prisma.workspace.findUnique({
+            where: { id },
+            include: { task: { include: { project: true } } },
+        });
+        if (!workspace) {
+            throw new NotFoundError('Workspace', id);
+        }
+        const worktreeManager = new WorktreeManager(workspace.task.project.repoPath);
+        await worktreeManager.rebase(workspace.worktreePath, workspace.task.project.mainBranch);
+    }
+    /**
+     * 获取工作空间的 Git 操作状态
+     */
+    async getGitStatus(id) {
+        const workspace = await prisma.workspace.findUnique({
+            where: { id },
+            include: { task: { include: { project: true } } },
+        });
+        if (!workspace) {
+            throw new NotFoundError('Workspace', id);
+        }
+        const worktreeManager = new WorktreeManager(workspace.task.project.repoPath);
+        return worktreeManager.getGitOperationStatus(workspace.worktreePath, workspace.task.project.mainBranch);
+    }
+    /**
+     * 中止工作空间当前进行中的 Git 操作
+     */
+    async abortOperation(id) {
+        const workspace = await prisma.workspace.findUnique({
+            where: { id },
+            include: { task: { include: { project: true } } },
+        });
+        if (!workspace) {
+            throw new NotFoundError('Workspace', id);
+        }
+        const worktreeManager = new WorktreeManager(workspace.task.project.repoPath);
+        await worktreeManager.abortOperation(workspace.worktreePath);
+    }
+    // ── Merge (squash) ──────────────────────────────────────────────────────────
+    /**
+     * 合并 Workspace 到主分支（squash merge）
+     *
+     * @param commitMessage - 可选的自定义 commit message
+     * @returns squash commit 的 SHA
+     */
+    async merge(id, commitMessage) {
+        const workspace = await prisma.workspace.findUnique({
+            where: { id },
+            include: { task: { include: { project: true } } },
+        });
+        if (!workspace) {
+            throw new NotFoundError('Workspace', id);
+        }
+        // 优先使用传入的 commitMessage，其次使用 AI 生成的缓存
+        const message = commitMessage || workspace.commitMessage || undefined;
+        const worktreeManager = new WorktreeManager(workspace.task.project.repoPath);
+        const { sha } = await worktreeManager.merge(workspace.worktreePath, workspace.task.project.mainBranch, message ? { commitMessage: message } : undefined);
+        // 更新 workspace：标记 MERGED，清空 worktreePath（物理目录已删除）
+        await prisma.workspace.update({
+            where: { id },
+            data: { status: WorkspaceStatus.MERGED, worktreePath: '' },
+        });
+        // Task 推进到 DONE
+        const advanceableStatuses = [TaskStatus.IN_PROGRESS, TaskStatus.IN_REVIEW];
+        if (advanceableStatuses.includes(workspace.task.status)) {
+            await prisma.task.update({
+                where: { id: workspace.task.id },
+                data: { status: TaskStatus.DONE },
+            });
+            this.eventBus.emit('task:updated', {
+                taskId: workspace.task.id,
+                projectId: workspace.task.projectId,
+                status: TaskStatus.DONE,
+            });
+        }
+        return sha;
+    }
+    // ── Archive ──────────────────────────────────────────────────────────────────
+    /**
+     * 归档 Workspace（标记状态为 ABANDONED）
+     */
+    async archive(id) {
+        const workspace = await prisma.workspace.findUnique({
+            where: { id },
+            include: { sessions: true },
+        });
+        if (!workspace) {
+            throw new NotFoundError('Workspace', id);
+        }
+        if (workspace.status !== WorkspaceStatus.ACTIVE) {
+            throw new ServiceError(`Cannot archive workspace in ${workspace.status} status`, 'INVALID_WORKSPACE_STATE', 400);
+        }
+        // 停止所有活跃的 Session
+        const activeSessions = workspace.sessions.filter((s) => s.status === SessionStatus.PENDING || s.status === SessionStatus.RUNNING);
+        for (const session of activeSessions) {
+            try {
+                await this.sessionService.stop(session.id);
+            }
+            catch {
+                // 忽略停止失败
+            }
+        }
+        return prisma.workspace.update({
+            where: { id },
+            data: { status: WorkspaceStatus.ABANDONED },
+            include: { sessions: true, task: { include: { project: true } } },
+        });
+    }
+    // ── Cleanup ──────────────────────────────────────────────────────────────────
+    /**
+     * 清理所有 ABANDONED/MERGED 状态且关联 Task 为 DONE 的 Workspace
+     *
+     * @returns 被清理的 Workspace 数量
+     */
+    async cleanup() {
+        const workspaces = await prisma.workspace.findMany({
+            where: {
+                status: { in: [WorkspaceStatus.ABANDONED, WorkspaceStatus.MERGED] },
+                task: { status: TaskStatus.DONE },
+            },
+            include: { task: { include: { project: true } } },
+        });
+        let cleaned = 0;
+        for (const workspace of workspaces) {
+            try {
+                const worktreeManager = new WorktreeManager(workspace.task.project.repoPath);
+                // 清理残留 worktree（如果还存在）
+                if (workspace.worktreePath) {
+                    await worktreeManager.remove(workspace.worktreePath);
+                }
+                // Task 已 DONE，branch 不再需要，删除
+                if (workspace.branchName) {
+                    try {
+                        await execGit(workspace.task.project.repoPath, ['branch', '-D', workspace.branchName]);
+                    }
+                    catch {
+                        // branch 可能已不存在，忽略
+                    }
+                }
+            }
+            catch (err) {
+                console.warn(`[WorkspaceService] cleanup: failed for workspace ${workspace.id}: ${err instanceof Error ? err.message : err}`);
+            }
+            await prisma.workspace.delete({ where: { id: workspace.id } });
+            cleaned++;
+        }
+        return cleaned;
+    }
+    // ── Startup Prune ────────────────────────────────────────────────────────────
+    /**
+     * 服务启动时调用：对所有项目执行 git worktree prune
+     */
+    static async pruneAllWorktrees() {
+        const projects = await prisma.project.findMany();
+        for (const project of projects) {
+            try {
+                const worktreeManager = new WorktreeManager(project.repoPath);
+                await worktreeManager.prune();
+            }
+            catch (err) {
+                console.warn(`[WorkspaceService] prune failed for project ${project.id}: ${err instanceof Error ? err.message : err}`);
+            }
+        }
+    }
+}
+//# sourceMappingURL=workspace.service.js.map

package/dist/services/workspace.service.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"workspace.service.js","sourceRoot":"","sources":["../../src/services/workspace.service.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,mBAAmB,CAAC;AAC3C,OAAO,EAAE,eAAe,EAAE,UAAU,EAAE,aAAa,EAAE,cAAc,EAAE,MAAM,mBAAmB,CAAC;AAC/F,OAAO,EAAE,eAAe,EAAE,MAAM,4BAA4B,CAAC;AAC7D,OAAO,EAAE,OAAO,EAAE,MAAM,mBAAmB,CAAC;AAC5C,OAAO,EAAE,aAAa,EAAE,YAAY,EAAE,MAAM,cAAc,CAAC;AAC3D,OAAO,EAAE,iBAAiB,EAAE,WAAW,EAAE,MAAM,sBAAsB,CAAC;AAItE,iCAAiC;AACjC,MAAM,qBAAqB,GAAG,EAAE,KAAK,EAAE,EAAE,OAAO,EAAE,EAAE,GAAG,EAAE,cAAc,CAAC,UAAU,EAAE,EAAE,EAAE,CAAC;AAEzF,MAAM,OAAO,gBAAgB;IACnB,cAAc,GAAG,iBAAiB,EAAE,CAAC;IACrC,QAAQ,GAAa,WAAW,EAAE,CAAC;IAE3C,gFAAgF;IAEhF,KAAK,CAAC,QAAQ,CAAC,EAAU;QACvB,OAAO,MAAM,CAAC,SAAS,CAAC,UAAU,CAAC;YACjC,KAAK,EAAE,EAAE,EAAE,EAAE;YACb,OAAO,EAAE,EAAE,QAAQ,EAAE,qBAAqB,EAAE,IAAI,EAAE,EAAE,OAAO,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,EAAE,EAAE;SACnF,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,YAAY,CAAC,MAAc;QAC/B,MAAM,IAAI,GAAG,MAAM,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,CAAC,CAAC;QACrE,IAAI,CAAC,IAAI,EAAE,CAAC;YACV,MAAM,IAAI,aAAa,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QAC1C,CAAC;QAED,OAAO,MAAM,CAAC,SAAS,CAAC,QAAQ,CAAC;YAC/B,KAAK,EAAE,EAAE,MAAM,EAAE;YACjB,OAAO,EAAE,EAAE,QAAQ,EAAE,qBAAqB,EAAE;YAC5C,OAAO,EAAE,EAAE,SAAS,EAAE,MAAM,EAAE;SAC/B,CAAC,CAAC;IACL,CAAC;IAED,gFAAgF;IAEhF;;;;;;;;OAQG;IACH,KAAK,CAAC,MAAM,CAAC,MAAc,EAAE,UAAmB;QAC9C,MAAM,IAAI,GAAG,MAAM,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC;YACxC,KAAK,EAAE,EAAE,EAAE,EAAE,MAAM,EAAE;YACrB,OAAO,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE;SAC3B,CAAC,CAAC;QAEH,IAAI,CAAC,IAAI,EAAE,CAAC;YACV,MAAM,IAAI,aAAa,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QAC1C,CAAC;QAED,MAAM,eAAe,GAAG,IAAI,eAAe,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QAEnE,oDAAoD;QACpD,IAAI,CAAC,UAAU,EAAE,CAAC;YAChB,MAAM,eAAe,GAAG,MAAM,MAAM,CAAC,SAAS,CAAC,SAAS,CAAC;gBACvD,KAAK,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE,eAAe,CAAC,MAAM,EAAE;gBACjD,OAAO,EAAE,EAAE,SAAS,EAAE,MAAM,EAAE;aAC/B,CAAC,CAAC;YAEH,IAAI,eAAe,EAAE,CAAC;gBACpB,MAAM,YAAY,GAAG,MAAM,eAAe,CAAC,oBAAoB,CAAC,eAAe,CAAC,UAAU,CAAC,CAAC;gBAE5F,MAAM,OAAO,GAAG,MAAM,MAAM,CAAC,SAAS,CAAC,MAAM,CAAC;oBAC5C,KAAK,EAAE,EAAE,EAAE,EAAE,eAAe,CAAC,EAAE,EAAE;oBACjC,IAAI,EAAE;wBACJ,MAAM,EAAE,eAAe,CAAC,MAAM;wBAC9B,YAAY;qBACb;oBACD,OAAO,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,IAAI,EAAE,EAAE,OAAO,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,EAAE,EAAE;iBAClE,CAAC,CAAC;gBAEH,yBAAyB;gBACzB,IAAI,IAAI,CAAC,MAAM,KAAK,UAAU,CAAC,WAAW,IAAI,IAAI,CAAC,MAAM,KAAK,UAAU,CAAC,IAAI,EAAE,CAAC;oBAC9E,MAAM,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC;wBACvB,KAAK,EAAE,EAAE,EAAE,EAAE,MAAM,EAAE;wBACrB,IAAI,EAAE,EAAE,MAAM,EAAE,UAAU,CAAC,WAAW,EAAE;qBACzC,CAAC,CAAC;oBACH,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,cAAc,EAAE;wBACjC,MAAM;wBACN,SAAS,EAAE,IAAI,CAAC,SAAS;wBACzB,MAAM,EAAE,UAAU,CAAC,WAAW;qBAC/B,CAAC,CAAC;gBACL,CAAC;gBAED,OAAO,OAAO,CAAC;YACjB,CAAC;QACH,CAAC;QAED,6BAA6B;QAC7B,MAAM,SAAS,GAAG,MAAM,MAAM,CAAC,SAAS,CAAC,MAAM,CAAC;YAC9C,IAAI,EAAE;gBACJ,MAAM;gBACN,UAAU,EAAE,EAAE,EAAE,UAAU;gBAC1B,YAAY,EAAE,EAAE,EAAE,UAAU;gBAC5B,MAAM,EAAE,eAAe,CAAC,MAAM;aAC/B;SACF,CAAC,CAAC;QAEH,IAAI,CAAC;YACH,kCAAkC;YAClC,MAAM,MAAM,GAAG,UAAU,IAAI,MAAM,SAAS,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC;YAE9D,2CAA2C;YAC3C,MAAM,YAAY,GAAG,MAAM,eAAe,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;YAE1D,0CAA0C;YAC1C,MAAM,OAAO,GAAG,MAAM,MAAM,CAAC,SAAS,CAAC,MAAM,CAAC;gBAC5C,KAAK,EAAE,EAAE,EAAE,EAAE,SAAS,CAAC,EAAE,EAAE;gBAC3B,IAAI,EAAE,EAAE,UAAU,EAAE,MAAM,EAAE,YAAY,EAAE;gBAC1C,OAAO,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,IAAI,EAAE,EAAE,OAAO,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,EAAE,EAAE;aAClE,CAAC,CAAC;YAEH,0CAA0C;YAC1C,IAAI,IAAI,CAAC,MAAM,KAAK,UAAU,CAAC,IAAI,EAAE,CAAC;gBACpC,MAAM,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC;oBACvB,KAAK,EAAE,EAAE,EAAE,EAAE,MAAM,EAAE;oBACrB,IAAI,EAAE,EAAE,MAAM,EAAE,UAAU,CAAC,WAAW,EAAE;iBACzC,CAAC,CAAC;gBACH,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,cAAc,EAAE;oBACjC,MAAM;oBACN,SAAS,EAAE,IAAI,CAAC,SAAS;oBACzB,MAAM,EAAE,UAAU,CAAC,WAAW;iBAC/B,CAAC,CAAC;YACL,CAAC;YAED,OAAO,OAAO,CAAC;QACjB,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,iBAAiB;YACjB,MAAM,MAAM,CAAC,SAAS,CAAC,MAAM,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,EAAE,SAAS,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE;gBACxE,SAAS;YACX,CAAC,CAAC,CAAC;YACH,MAAM,GAAG,CAAC;QACZ,CAAC;IACH,CAAC;IAED,gFAAgF;IAEhF;;;;;;;OAOG;IACH,KAAK,CAAC,MAAM,CAAC,EAAU;QACrB,MAAM,SAAS,GAAG,MAAM,MAAM,CAAC,SAAS,CAAC,UAAU,CAAC;YAClD,KAAK,EAAE,EAAE,EAAE,EAAE;YACb,OAAO,EAAE;gBACP,QAAQ,EAAE,IAAI;gBACd,IAAI,EAAE,EAAE,OAAO,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,EAAE;aACrC;SACF,CAAC,CAAC;QAEH,IAAI,CAAC,SAAS,EAAE,CAAC;YACf,MAAM,IAAI,aAAa,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC;QAC3C,CAAC;QAED,wCAAwC;QACxC,MAAM,cAAc,GAAG,SAAS,CAAC,QAAQ,CAAC,MAAM,CAC9C,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,KAAK,aAAa,CAAC,OAAO,IAAI,CAAC,CAAC,MAAM,KAAK,aAAa,CAAC,OAAO,CAChF,CAAC;QACF,KAAK,MAAM,OAAO,IAAI,cAAc,EAAE,CAAC;YACrC,IAAI,CAAC;gBACH,MAAM,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;YAC7C,CAAC;YAAC,MAAM,CAAC;gBACP,SAAS;YACX,CAAC;QACH,CAAC;QAED,cAAc;QACd,IAAI,CAAC;YACH,MAAM,eAAe,GAAG,IAAI,eAAe,CAAC,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;YAC7E,MAAM,eAAe,CAAC,MAAM,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC;QACvD,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,2BAA2B;YAC3B,OAAO,CAAC,IAAI,CACV,8DAA8D,EAAE,KAAK,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,EAAE,CAChH,CAAC;QACJ,CAAC;QAED,yBAAyB;QACzB,MAAM,MAAM,CAAC,SAAS,CAAC,MAAM,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,CAAC;QACjD,OAAO,IAAI,CAAC;IACd,CAAC;IAED,gFAAgF;IAEhF,KAAK,CAAC,OAAO,CAAC,EAAU;QACtB,MAAM,SAAS,GAAG,MAAM,MAAM,CAAC,SAAS,CAAC,UAAU,CAAC;YAClD,KAAK,EAAE,EAAE,EAAE,EAAE;YACb,OAAO,EAAE,EAAE,IAAI,EAAE,EAAE,OAAO,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,EAAE,EAAE;SAClD,CAAC,CAAC;QAEH,IAAI,CAAC,SAAS,EAAE,CAAC;YACf,MAAM,IAAI,aAAa,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC;QAC3C,CAAC;QAED,MAAM,eAAe,GAAG,IAAI,eAAe,CAAC,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QAC7E,OAAO,eAAe,CAAC,OAAO,CAC5B,SAAS,CAAC,YAAY,EACtB,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,UAAU,CAClC,CAAC;IACJ,CAAC;IAED,gFAAgF;IAEhF,+EAA+E;IAE/E;;OAEG;IACH,KAAK,CAAC,MAAM,CAAC,EAAU;QACrB,MAAM,SAAS,GAAG,MAAM,MAAM,CAAC,SAAS,CAAC,UAAU,CAAC;YAClD,KAAK,EAAE,EAAE,EAAE,EAAE;YACb,OAAO,EAAE,EAAE,IAAI,EAAE,EAAE,OAAO,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,EAAE,EAAE;SAClD,CAAC,CAAC;QAEH,IAAI,CAAC,SAAS,EAAE,CAAC;YACf,MAAM,IAAI,aAAa,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC;QAC3C,CAAC;QAED,MAAM,eAAe,GAAG,IAAI,eAAe,CAAC,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QAC7E,MAAM,eAAe,CAAC,MAAM,CAC1B,SAAS,CAAC,YAAY,EACtB,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,UAAU,CAClC,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,YAAY,CAAC,EAAU;QAC3B,MAAM,SAAS,GAAG,MAAM,MAAM,CAAC,SAAS,CAAC,UAAU,CAAC;YAClD,KAAK,EAAE,EAAE,EAAE,EAAE;YACb,OAAO,EAAE,EAAE,IAAI,EAAE,EAAE,OAAO,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,EAAE,EAAE;SAClD,CAAC,CAAC;QAEH,IAAI,CAAC,SAAS,EAAE,CAAC;YACf,MAAM,IAAI,aAAa,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC;QAC3C,CAAC;QAED,MAAM,eAAe,GAAG,IAAI,eAAe,CAAC,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QAC7E,OAAO,eAAe,CAAC,qBAAqB,CAC1C,SAAS,CAAC,YAAY,EACtB,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,UAAU,CAClC,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,cAAc,CAAC,EAAU;QAC7B,MAAM,SAAS,GAAG,MAAM,MAAM,CAAC,SAAS,CAAC,UAAU,CAAC;YAClD,KAAK,EAAE,EAAE,EAAE,EAAE;YACb,OAAO,EAAE,EAAE,IAAI,EAAE,EAAE,OAAO,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,EAAE,EAAE;SAClD,CAAC,CAAC;QAEH,IAAI,CAAC,SAAS,EAAE,CAAC;YACf,MAAM,IAAI,aAAa,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC;QAC3C,CAAC;QAED,MAAM,eAAe,GAAG,IAAI,eAAe,CAAC,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QAC7E,MAAM,eAAe,CAAC,cAAc,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC;IAC/D,CAAC;IAED,+EAA+E;IAE/E;;;;;OAKG;IACH,KAAK,CAAC,KAAK,CAAC,EAAU,EAAE,aAAsB;QAC5C,MAAM,SAAS,GAAG,MAAM,MAAM,CAAC,SAAS,CAAC,UAAU,CAAC;YAClD,KAAK,EAAE,EAAE,EAAE,EAAE;YACb,OAAO,EAAE,EAAE,IAAI,EAAE,EAAE,OAAO,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,EAAE,EAAE;SAClD,CAAC,CAAC;QAEH,IAAI,CAAC,SAAS,EAAE,CAAC;YACf,MAAM,IAAI,aAAa,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC;QAC3C,CAAC;QAED,sCAAsC;QACtC,MAAM,OAAO,GAAG,aAAa,IAAI,SAAS,CAAC,aAAa,IAAI,SAAS,CAAC;QAEtE,MAAM,eAAe,GAAG,IAAI,eAAe,CAAC,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QAC7E,MAAM,EAAE,GAAG,EAAE,GAAG,MAAM,eAAe,CAAC,KAAK,CACzC,SAAS,CAAC,YAAY,EACtB,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,UAAU,EACjC,OAAO,CAAC,CAAC,CAAC,EAAE,aAAa,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC,SAAS,CACjD,CAAC;QAEF,kDAAkD;QAClD,MAAM,MAAM,CAAC,SAAS,CAAC,MAAM,CAAC;YAC5B,KAAK,EAAE,EAAE,EAAE,EAAE;YACb,IAAI,EAAE,EAAE,MAAM,EAAE,eAAe,CAAC,MAAM,EAAE,YAAY,EAAE,EAAE,EAAE;SAC3D,CAAC,CAAC;QAEH,gBAAgB;QAChB,MAAM,mBAAmB,GAAG,CAAC,UAAU,CAAC,WAAW,EAAE,UAAU,CAAC,SAAS,CAAC,CAAC;QAC3E,IAAI,mBAAmB,CAAC,QAAQ,CAAC,SAAS,CAAC,IAAI,CAAC,MAAoB,CAAC,EAAE,CAAC;YACtE,MAAM,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC;gBACvB,KAAK,EAAE,EAAE,EAAE,EAAE,SAAS,CAAC,IAAI,CAAC,EAAE,EAAE;gBAChC,IAAI,EAAE,EAAE,MAAM,EAAE,UAAU,CAAC,IAAI,EAAE;aAClC,CAAC,CAAC;YACH,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,cAAc,EAAE;gBACjC,MAAM,EAAE,SAAS,CAAC,IAAI,CAAC,EAAE;gBACzB,SAAS,EAAE,SAAS,CAAC,IAAI,CAAC,SAAS;gBACnC,MAAM,EAAE,UAAU,CAAC,IAAI;aACxB,CAAC,CAAC;QACL,CAAC;QAED,OAAO,GAAG,CAAC;IACb,CAAC;IAED,gFAAgF;IAEhF;;OAEG;IACH,KAAK,CAAC,OAAO,CAAC,EAAU;QACtB,MAAM,SAAS,GAAG,MAAM,MAAM,CAAC,SAAS,CAAC,UAAU,CAAC;YAClD,KAAK,EAAE,EAAE,EAAE,EAAE;YACb,OAAO,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE;SAC5B,CAAC,CAAC;QAEH,IAAI,CAAC,SAAS,EAAE,CAAC;YACf,MAAM,IAAI,aAAa,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC;QAC3C,CAAC;QAED,IAAI,SAAS,CAAC,MAAM,KAAK,eAAe,CAAC,MAAM,EAAE,CAAC;YAChD,MAAM,IAAI,YAAY,CACpB,+BAA+B,SAAS,CAAC,MAAM,SAAS,EACxD,yBAAyB,EACzB,GAAG,CACJ,CAAC;QACJ,CAAC;QAED,kBAAkB;QAClB,MAAM,cAAc,GAAG,SAAS,CAAC,QAAQ,CAAC,MAAM,CAC9C,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,KAAK,aAAa,CAAC,OAAO,IAAI,CAAC,CAAC,MAAM,KAAK,aAAa,CAAC,OAAO,CAChF,CAAC;QACF,KAAK,MAAM,OAAO,IAAI,cAAc,EAAE,CAAC;YACrC,IAAI,CAAC;gBACH,MAAM,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;YAC7C,CAAC;YAAC,MAAM,CAAC;gBACP,SAAS;YACX,CAAC;QACH,CAAC;QAED,OAAO,MAAM,CAAC,SAAS,CAAC,MAAM,CAAC;YAC7B,KAAK,EAAE,EAAE,EAAE,EAAE;YACb,IAAI,EAAE,EAAE,MAAM,EAAE,eAAe,CAAC,SAAS,EAAE;YAC3C,OAAO,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,IAAI,EAAE,EAAE,OAAO,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,EAAE,EAAE;SAClE,CAAC,CAAC;IACL,CAAC;IAED,gFAAgF;IAEhF;;;;OAIG;IACH,KAAK,CAAC,OAAO;QACX,MAAM,UAAU,GAAG,MAAM,MAAM,CAAC,SAAS,CAAC,QAAQ,CAAC;YACjD,KAAK,EAAE;gBACL,MAAM,EAAE,EAAE,EAAE,EAAE,CAAC,eAAe,CAAC,SAAS,EAAE,eAAe,CAAC,MAAM,CAAC,EAAE;gBACnE,IAAI,EAAE,EAAE,MAAM,EAAE,UAAU,CAAC,IAAI,EAAE;aAClC;YACD,OAAO,EAAE,EAAE,IAAI,EAAE,EAAE,OAAO,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,EAAE,EAAE;SAClD,CAAC,CAAC;QAEH,IAAI,OAAO,GAAG,CAAC,CAAC;QAEhB,KAAK,MAAM,SAAS,IAAI,UAAU,EAAE,CAAC;YACnC,IAAI,CAAC;gBACH,MAAM,eAAe,GAAG,IAAI,eAAe,CAAC,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;gBAE7E,uBAAuB;gBACvB,IAAI,SAAS,CAAC,YAAY,EAAE,CAAC;oBAC3B,MAAM,eAAe,CAAC,MAAM,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC;gBACvD,CAAC;gBAED,6BAA6B;gBAC7B,IAAI,SAAS,CAAC,UAAU,EAAE,CAAC;oBACzB,IAAI,CAAC;wBACH,MAAM,OAAO,CAAC,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,EAAE,CAAC,QAAQ,EAAE,IAAI,EAAE,SAAS,CAAC,UAAU,CAAC,CAAC,CAAC;oBACzF,CAAC;oBAAC,MAAM,CAAC;wBACP,mBAAmB;oBACrB,CAAC;gBACH,CAAC;YACH,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,OAAO,CAAC,IAAI,CACV,oDAAoD,SAAS,CAAC,EAAE,KAAK,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,EAAE,CAChH,CAAC;YACJ,CAAC;YAED,MAAM,MAAM,CAAC,SAAS,CAAC,MAAM,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,EAAE,SAAS,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;YAC/D,OAAO,EAAE,CAAC;QACZ,CAAC;QAED,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,gFAAgF;IAEhF;;OAEG;IACH,MAAM,CAAC,KAAK,CAAC,iBAAiB;QAC5B,MAAM,QAAQ,GAAG,MAAM,MAAM,CAAC,OAAO,CAAC,QAAQ,EAAE,CAAC;QAEjD,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;YAC/B,IAAI,CAAC;gBACH,MAAM,eAAe,GAAG,IAAI,eAAe,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;gBAC9D,MAAM,eAAe,CAAC,KAAK,EAAE,CAAC;YAChC,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,OAAO,CAAC,IAAI,CACV,+CAA+C,OAAO,CAAC,EAAE,KAAK,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,EAAE,CACzG,CAAC;YACJ,CAAC;QACH,CAAC;IACH,CAAC;CACF"}

package/dist/socket/events.d.ts

@@ -0,0 +1,2 @@
+export { NAMESPACE, ClientEvents, ServerEvents, type SubscribePayload, type UnsubscribePayload, type SessionInputPayload, type SessionResizePayload, type SessionStdoutPayload, type SessionPatchPayload, type SessionExitPayload, type SessionErrorPayload, type SessionIdPayload, type SessionSubscribedPayload, type SessionUnsubscribedPayload, type TaskUpdatedPayload, type TaskDeletedPayload, type TerminalInputPayload, type TerminalResizePayload, type TerminalStdoutPayload, type TerminalExitPayload, type TerminalSubscribedPayload, type TerminalUnsubscribedPayload, type AgentStatus, type AgentStatusPayload, type AckResponse, type ClientEventType, type ServerEventType, } from '@agent-tower/shared/socket';
+//# sourceMappingURL=events.d.ts.map

package/dist/socket/events.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"events.d.ts","sourceRoot":"","sources":["../../src/socket/events.ts"],"names":[],"mappings":"AACA,OAAO,EACL,SAAS,EACT,YAAY,EACZ,YAAY,EACZ,KAAK,gBAAgB,EACrB,KAAK,kBAAkB,EACvB,KAAK,mBAAmB,EACxB,KAAK,oBAAoB,EACzB,KAAK,oBAAoB,EACzB,KAAK,mBAAmB,EACxB,KAAK,kBAAkB,EACvB,KAAK,mBAAmB,EACxB,KAAK,gBAAgB,EACrB,KAAK,wBAAwB,EAC7B,KAAK,0BAA0B,EAC/B,KAAK,kBAAkB,EACvB,KAAK,kBAAkB,EACvB,KAAK,oBAAoB,EACzB,KAAK,qBAAqB,EAC1B,KAAK,qBAAqB,EAC1B,KAAK,mBAAmB,EACxB,KAAK,yBAAyB,EAC9B,KAAK,2BAA2B,EAChC,KAAK,WAAW,EAChB,KAAK,kBAAkB,EACvB,KAAK,WAAW,EAChB,KAAK,eAAe,EACpB,KAAK,eAAe,GACrB,MAAM,4BAA4B,CAAA"}

package/dist/socket/events.js

@@ -0,0 +1,3 @@
+// 从 shared 包重新导出事件类型
+export { NAMESPACE, ClientEvents, ServerEvents, } from '@agent-tower/shared/socket';
+//# sourceMappingURL=events.js.map

package/dist/socket/events.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"events.js","sourceRoot":"","sources":["../../src/socket/events.ts"],"names":[],"mappings":"AAAA,qBAAqB;AACrB,OAAO,EACL,SAAS,EACT,YAAY,EACZ,YAAY,GAyBb,MAAM,4BAA4B,CAAA"}

package/dist/socket/index.d.ts

@@ -0,0 +1,20 @@
+import { Server } from 'socket.io';
+import type { FastifyInstance } from 'fastify';
+import { type AgentStatusPayload } from './events.js';
+/**
+ * 获取 Socket.IO 实例
+ */
+export declare function getIO(): Server;
+/**
+ * 初始化 Socket.IO 服务
+ */
+export declare function initializeSocket(fastify: FastifyInstance): Promise<Server>;
+/**
+ * 关闭 Socket.IO 服务
+ */
+export declare function closeSocket(): Promise<void>;
+export * from './events.js';
+export * from './rooms.js';
+export { type AuthenticatedSocket } from './middleware/index.js';
+export declare function broadcastAgentStatus(payload: AgentStatusPayload): void;
+//# sourceMappingURL=index.d.ts.map

package/dist/socket/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/socket/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,WAAW,CAAA;AAClC,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,SAAS,CAAA;AAG9C,OAAO,EAAa,KAAK,kBAAkB,EAAE,MAAM,aAAa,CAAA;AAMhE;;GAEG;AACH,wBAAgB,KAAK,IAAI,MAAM,CAK9B;AAED;;GAEG;AACH,wBAAsB,gBAAgB,CAAC,OAAO,EAAE,eAAe,GAAG,OAAO,CAAC,MAAM,CAAC,CA6BhF;AAED;;GAEG;AACH,wBAAsB,WAAW,IAAI,OAAO,CAAC,IAAI,CAAC,CAqBjD;AAGD,cAAc,aAAa,CAAA;AAC3B,cAAc,YAAY,CAAA;AAC1B,OAAO,EAAE,KAAK,mBAAmB,EAAE,MAAM,uBAAuB,CAAA;AAEhE,wBAAgB,oBAAoB,CAAC,OAAO,EAAE,kBAAkB,GAAG,IAAI,CAEtE"}

package/dist/socket/index.js

@@ -0,0 +1,77 @@
+import { Server } from 'socket.io';
+import { authMiddleware } from './middleware/index.js';
+import { SocketGateway } from './socket-gateway.js';
+import { NAMESPACE } from './events.js';
+import { getEventBus, getSessionManager, getTerminalManager, getNotificationService } from '../core/container.js';
+let io = null;
+let socketGateway = null;
+/**
+ * 获取 Socket.IO 实例
+ */
+export function getIO() {
+    if (!io) {
+        throw new Error('Socket.IO not initialized');
+    }
+    return io;
+}
+/**
+ * 初始化 Socket.IO 服务
+ */
+export async function initializeSocket(fastify) {
+    // Clean up previous instance to prevent listener accumulation during dev hot-reload
+    socketGateway?.destroy();
+    socketGateway = null;
+    io = new Server(fastify.server, {
+        cors: {
+            origin: process.env.CORS_ORIGIN || '*',
+            methods: ['GET', 'POST'],
+            credentials: true,
+        },
+        pingTimeout: 60000,
+        pingInterval: 25000,
+    });
+    const nsp = io.of(NAMESPACE);
+    nsp.use(authMiddleware);
+    const tm = await getTerminalManager();
+    getNotificationService();
+    socketGateway = new SocketGateway(nsp, getEventBus(), getSessionManager(), tm);
+    nsp.on('connection', (socket) => {
+        console.log(`[Socket] Connected: ${socket.id}`);
+        socketGateway?.register(socket);
+    });
+    console.log('[Socket.IO] Initialized namespace:', NAMESPACE);
+    return io;
+}
+/**
+ * 关闭 Socket.IO 服务
+ */
+export async function closeSocket() {
+    if (io) {
+        socketGateway?.destroy();
+        // Kill all active agent session pipelines on shutdown
+        getSessionManager().destroyAll();
+        // Kill all standalone terminals on shutdown
+        try {
+            const tm = await getTerminalManager();
+            tm.destroyAll();
+        }
+        catch {
+            // TerminalManager may not have been initialized; safe to ignore
+        }
+        await new Promise((resolve) => {
+            io.close(() => {
+                console.log('[Socket.IO] Closed');
+                resolve();
+            });
+        });
+        socketGateway = null;
+        io = null;
+    }
+}
+// 导出类型和工具
+export * from './events.js';
+export * from './rooms.js';
+export function broadcastAgentStatus(payload) {
+    socketGateway?.broadcastAgentStatus(payload);
+}
+//# sourceMappingURL=index.js.map

package/dist/socket/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/socket/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,WAAW,CAAA;AAElC,OAAO,EAAE,cAAc,EAAE,MAAM,uBAAuB,CAAA;AACtD,OAAO,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAA;AACnD,OAAO,EAAE,SAAS,EAA2B,MAAM,aAAa,CAAA;AAChE,OAAO,EAAE,WAAW,EAAE,iBAAiB,EAAE,kBAAkB,EAAE,sBAAsB,EAAE,MAAM,sBAAsB,CAAA;AAEjH,IAAI,EAAE,GAAkB,IAAI,CAAA;AAC5B,IAAI,aAAa,GAAyB,IAAI,CAAA;AAE9C;;GAEG;AACH,MAAM,UAAU,KAAK;IACnB,IAAI,CAAC,EAAE,EAAE,CAAC;QACR,MAAM,IAAI,KAAK,CAAC,2BAA2B,CAAC,CAAA;IAC9C,CAAC;IACD,OAAO,EAAE,CAAA;AACX,CAAC;AAED;;GAEG;AACH,MAAM,CAAC,KAAK,UAAU,gBAAgB,CAAC,OAAwB;IAC7D,oFAAoF;IACpF,aAAa,EAAE,OAAO,EAAE,CAAA;IACxB,aAAa,GAAG,IAAI,CAAA;IAEpB,EAAE,GAAG,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,EAAE;QAC9B,IAAI,EAAE;YACJ,MAAM,EAAE,OAAO,CAAC,GAAG,CAAC,WAAW,IAAI,GAAG;YACtC,OAAO,EAAE,CAAC,KAAK,EAAE,MAAM,CAAC;YACxB,WAAW,EAAE,IAAI;SAClB;QACD,WAAW,EAAE,KAAK;QAClB,YAAY,EAAE,KAAK;KACpB,CAAC,CAAA;IAEF,MAAM,GAAG,GAAG,EAAE,CAAC,EAAE,CAAC,SAAS,CAAC,CAAA;IAC5B,GAAG,CAAC,GAAG,CAAC,cAAc,CAAC,CAAA;IAEvB,MAAM,EAAE,GAAG,MAAM,kBAAkB,EAAE,CAAA;IACrC,sBAAsB,EAAE,CAAA;IACxB,aAAa,GAAG,IAAI,aAAa,CAAC,GAAG,EAAE,WAAW,EAAE,EAAE,iBAAiB,EAAE,EAAE,EAAE,CAAC,CAAA;IAC9E,GAAG,CAAC,EAAE,CAAC,YAAY,EAAE,CAAC,MAAM,EAAE,EAAE;QAC9B,OAAO,CAAC,GAAG,CAAC,uBAAuB,MAAM,CAAC,EAAE,EAAE,CAAC,CAAA;QAC/C,aAAa,EAAE,QAAQ,CAAC,MAAM,CAAC,CAAA;IACjC,CAAC,CAAC,CAAA;IAEF,OAAO,CAAC,GAAG,CAAC,oCAAoC,EAAE,SAAS,CAAC,CAAA;IAE5D,OAAO,EAAE,CAAA;AACX,CAAC;AAED;;GAEG;AACH,MAAM,CAAC,KAAK,UAAU,WAAW;IAC/B,IAAI,EAAE,EAAE,CAAC;QACP,aAAa,EAAE,OAAO,EAAE,CAAA;QACxB,sDAAsD;QACtD,iBAAiB,EAAE,CAAC,UAAU,EAAE,CAAA;QAChC,4CAA4C;QAC5C,IAAI,CAAC;YACH,MAAM,EAAE,GAAG,MAAM,kBAAkB,EAAE,CAAA;YACrC,EAAE,CAAC,UAAU,EAAE,CAAA;QACjB,CAAC;QAAC,MAAM,CAAC;YACP,gEAAgE;QAClE,CAAC;QACD,MAAM,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,EAAE;YAClC,EAAG,CAAC,KAAK,CAAC,GAAG,EAAE;gBACb,OAAO,CAAC,GAAG,CAAC,oBAAoB,CAAC,CAAA;gBACjC,OAAO,EAAE,CAAA;YACX,CAAC,CAAC,CAAA;QACJ,CAAC,CAAC,CAAA;QACF,aAAa,GAAG,IAAI,CAAA;QACpB,EAAE,GAAG,IAAI,CAAA;IACX,CAAC;AACH,CAAC;AAED,UAAU;AACV,cAAc,aAAa,CAAA;AAC3B,cAAc,YAAY,CAAA;AAG1B,MAAM,UAAU,oBAAoB,CAAC,OAA2B;IAC9D,aAAa,EAAE,oBAAoB,CAAC,OAAO,CAAC,CAAA;AAC9C,CAAC"}

package/dist/socket/middleware/auth.d.ts

@@ -0,0 +1,13 @@
+import type { Socket } from 'socket.io';
+type NextFunction = (err?: Error) => void;
+export interface AuthenticatedSocket extends Socket {
+    userId?: string;
+    username?: string;
+}
+/**
+ * Socket 认证中间件
+ * 隧道请求（带 CF 头）必须携带有效 token，本地请求放行
+ */
+export declare function authMiddleware(socket: AuthenticatedSocket, next: NextFunction): void;
+export {};
+//# sourceMappingURL=auth.d.ts.map

package/dist/socket/middleware/auth.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"auth.d.ts","sourceRoot":"","sources":["../../../src/socket/middleware/auth.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,WAAW,CAAA;AAGvC,KAAK,YAAY,GAAG,CAAC,GAAG,CAAC,EAAE,KAAK,KAAK,IAAI,CAAA;AAEzC,MAAM,WAAW,mBAAoB,SAAQ,MAAM;IACjD,MAAM,CAAC,EAAE,MAAM,CAAA;IACf,QAAQ,CAAC,EAAE,MAAM,CAAA;CAClB;AAED;;;GAGG;AACH,wBAAgB,cAAc,CAC5B,MAAM,EAAE,mBAAmB,EAC3B,IAAI,EAAE,YAAY,QA2BnB"}

package/dist/socket/middleware/auth.js

@@ -0,0 +1,29 @@
+import { TunnelService } from '../../services/tunnel.service.js';
+/**
+ * Socket 认证中间件
+ * 隧道请求（带 CF 头）必须携带有效 token，本地请求放行
+ */
+export function authMiddleware(socket, next) {
+    // 检查是否为隧道请求
+    const headers = socket.request.headers;
+    const isTunnel = !!(headers['cf-connecting-ip'] || headers['cf-ray']);
+    if (isTunnel && TunnelService.isRunning()) {
+        const token = socket.handshake.auth?.token ??
+            socket.handshake.query?.token;
+        if (!token || !TunnelService.validateToken(token)) {
+            return next(new Error('Unauthorized: valid tunnel token required'));
+        }
+    }
+    // 设置用户标识
+    const authToken = socket.handshake.auth?.token;
+    if (authToken) {
+        socket.userId = authToken;
+        socket.username = `User-${authToken.slice(0, 6)}`;
+    }
+    else {
+        socket.userId = `anonymous-${socket.id.slice(0, 8)}`;
+        socket.username = 'Anonymous';
+    }
+    next();
+}
+//# sourceMappingURL=auth.js.map

package/dist/socket/middleware/auth.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"auth.js","sourceRoot":"","sources":["../../../src/socket/middleware/auth.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,aAAa,EAAE,MAAM,kCAAkC,CAAA;AAShE;;;GAGG;AACH,MAAM,UAAU,cAAc,CAC5B,MAA2B,EAC3B,IAAkB;IAElB,YAAY;IACZ,MAAM,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC,OAAO,CAAA;IACtC,MAAM,QAAQ,GAAG,CAAC,CAAC,CAAC,OAAO,CAAC,kBAAkB,CAAC,IAAI,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAA;IAErE,IAAI,QAAQ,IAAI,aAAa,CAAC,SAAS,EAAE,EAAE,CAAC;QAC1C,MAAM,KAAK,GACT,MAAM,CAAC,SAAS,CAAC,IAAI,EAAE,KAAK;YAC3B,MAAM,CAAC,SAAS,CAAC,KAAK,EAAE,KAA4B,CAAA;QAEvD,IAAI,CAAC,KAAK,IAAI,CAAC,aAAa,CAAC,aAAa,CAAC,KAAK,CAAC,EAAE,CAAC;YAClD,OAAO,IAAI,CAAC,IAAI,KAAK,CAAC,2CAA2C,CAAC,CAAC,CAAA;QACrE,CAAC;IACH,CAAC;IAED,SAAS;IACT,MAAM,SAAS,GAAG,MAAM,CAAC,SAAS,CAAC,IAAI,EAAE,KAAK,CAAA;IAC9C,IAAI,SAAS,EAAE,CAAC;QACd,MAAM,CAAC,MAAM,GAAG,SAAS,CAAA;QACzB,MAAM,CAAC,QAAQ,GAAG,QAAQ,SAAS,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAA;IACnD,CAAC;SAAM,CAAC;QACN,MAAM,CAAC,MAAM,GAAG,aAAa,MAAM,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAA;QACpD,MAAM,CAAC,QAAQ,GAAG,WAAW,CAAA;IAC/B,CAAC;IAED,IAAI,EAAE,CAAA;AACR,CAAC"}

package/dist/socket/middleware/error-handler.d.ts

@@ -0,0 +1,11 @@
+import type { Socket } from 'socket.io';
+import type { AckResponse } from '../events.js';
+/**
+ * 包装 handler 函数，统一处理错误
+ */
+export declare function withErrorHandler<T, R>(handler: (socket: Socket, payload: T) => Promise<R>): (socket: Socket, payload: T, callback?: (response: AckResponse<R>) => void) => Promise<void>;
+/**
+ * 创建带验证的 handler
+ */
+export declare function createHandler<T, R>(validator: (payload: unknown) => T, handler: (socket: Socket, payload: T) => Promise<R>): (socket: Socket, payload: unknown, callback?: ((response: AckResponse<R>) => void) | undefined) => Promise<void>;
+//# sourceMappingURL=error-handler.d.ts.map

package/dist/socket/middleware/error-handler.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"error-handler.d.ts","sourceRoot":"","sources":["../../../src/socket/middleware/error-handler.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,WAAW,CAAA;AACvC,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,cAAc,CAAA;AAK/C;;GAEG;AACH,wBAAgB,gBAAgB,CAAC,CAAC,EAAE,CAAC,EACnC,OAAO,EAAE,CAAC,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,KAAK,OAAO,CAAC,CAAC,CAAC,IAGjD,QAAQ,MAAM,EACd,SAAS,CAAC,EACV,WAAW,CAAC,QAAQ,EAAE,WAAW,CAAC,CAAC,CAAC,KAAK,IAAI,mBAwBhD;AAED;;GAEG;AACH,wBAAgB,aAAa,CAAC,CAAC,EAAE,CAAC,EAChC,SAAS,EAAE,CAAC,OAAO,EAAE,OAAO,KAAK,CAAC,EAClC,OAAO,EAAE,CAAC,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,KAAK,OAAO,CAAC,CAAC,CAAC,YAjCzC,MAAM,8DAE2B,IAAI,gCAqChD"}

package/dist/socket/middleware/error-handler.js

@@ -0,0 +1,39 @@
+// 通用错误事件名
+const ERROR_EVENT = 'error';
+/**
+ * 包装 handler 函数，统一处理错误
+ */
+export function withErrorHandler(handler) {
+    return async (socket, payload, callback) => {
+        try {
+            const result = await handler(socket, payload);
+            callback?.({ success: true, data: result });
+        }
+        catch (error) {
+            const message = error instanceof Error ? error.message : 'Unknown error';
+            console.error(`[Socket Error] ${message}`, error);
+            callback?.({
+                success: false,
+                error: {
+                    code: 'INTERNAL_ERROR',
+                    message,
+                },
+            });
+            // 也发送错误事件给客户端
+            socket.emit(ERROR_EVENT, {
+                code: 'INTERNAL_ERROR',
+                message,
+            });
+        }
+    };
+}
+/**
+ * 创建带验证的 handler
+ */
+export function createHandler(validator, handler) {
+    return withErrorHandler(async (socket, payload) => {
+        const validated = validator(payload);
+        return handler(socket, validated);
+    });
+}
+//# sourceMappingURL=error-handler.js.map

package/dist/socket/middleware/error-handler.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"error-handler.js","sourceRoot":"","sources":["../../../src/socket/middleware/error-handler.ts"],"names":[],"mappings":"AAGA,UAAU;AACV,MAAM,WAAW,GAAG,OAAO,CAAA;AAE3B;;GAEG;AACH,MAAM,UAAU,gBAAgB,CAC9B,OAAmD;IAEnD,OAAO,KAAK,EACV,MAAc,EACd,OAAU,EACV,QAA6C,EAC7C,EAAE;QACF,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,MAAM,EAAE,OAAO,CAAC,CAAA;YAC7C,QAAQ,EAAE,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,CAAA;QAC7C,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,OAAO,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,eAAe,CAAA;YACxE,OAAO,CAAC,KAAK,CAAC,kBAAkB,OAAO,EAAE,EAAE,KAAK,CAAC,CAAA;YAEjD,QAAQ,EAAE,CAAC;gBACT,OAAO,EAAE,KAAK;gBACd,KAAK,EAAE;oBACL,IAAI,EAAE,gBAAgB;oBACtB,OAAO;iBACR;aACF,CAAC,CAAA;YAEF,cAAc;YACd,MAAM,CAAC,IAAI,CAAC,WAAW,EAAE;gBACvB,IAAI,EAAE,gBAAgB;gBACtB,OAAO;aACR,CAAC,CAAA;QACJ,CAAC;IACH,CAAC,CAAA;AACH,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,aAAa,CAC3B,SAAkC,EAClC,OAAmD;IAEnD,OAAO,gBAAgB,CAAC,KAAK,EAAE,MAAc,EAAE,OAAgB,EAAE,EAAE;QACjE,MAAM,SAAS,GAAG,SAAS,CAAC,OAAO,CAAC,CAAA;QACpC,OAAO,OAAO,CAAC,MAAM,EAAE,SAAS,CAAC,CAAA;IACnC,CAAC,CAAC,CAAA;AACJ,CAAC"}

package/dist/socket/middleware/index.d.ts

@@ -0,0 +1,3 @@
+export { authMiddleware, type AuthenticatedSocket } from './auth.js';
+export { withErrorHandler, createHandler } from './error-handler.js';
+//# sourceMappingURL=index.d.ts.map

package/dist/socket/middleware/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/socket/middleware/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAE,KAAK,mBAAmB,EAAE,MAAM,WAAW,CAAA;AACpE,OAAO,EAAE,gBAAgB,EAAE,aAAa,EAAE,MAAM,oBAAoB,CAAA"}

package/dist/socket/middleware/index.js

@@ -0,0 +1,3 @@
+export { authMiddleware } from './auth.js';
+export { withErrorHandler, createHandler } from './error-handler.js';
+//# sourceMappingURL=index.js.map

package/dist/socket/middleware/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/socket/middleware/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAA4B,MAAM,WAAW,CAAA;AACpE,OAAO,EAAE,gBAAgB,EAAE,aAAa,EAAE,MAAM,oBAAoB,CAAA"}