@jingyi0605/codingns 0.1.0

package/dist/server/ws/terminal-ws-hub.js

@@ -0,0 +1,181 @@
+import { AppError } from "../shared/errors/app-error.js";
+export class TerminalWsHub {
+    terminalService;
+    clientSubscriptions = new WeakMap();
+    constructor(terminalService) {
+        this.terminalService = terminalService;
+    }
+    handleMessage(client, payload, authContext) {
+        if (isTerminalSubscribeMessage(payload)) {
+            void this.subscribeTerminal(client, payload, authContext);
+            return true;
+        }
+        if (isTerminalInputMessage(payload)) {
+            void this.handleTerminalInput(client, payload);
+            return true;
+        }
+        if (isTerminalResizeMessage(payload)) {
+            void this.handleTerminalResize(client, payload);
+            return true;
+        }
+        return false;
+    }
+    cleanupClient(client) {
+        const subscriptions = this.clientSubscriptions.get(client);
+        if (!subscriptions) {
+            return;
+        }
+        for (const subscription of subscriptions.values()) {
+            subscription.close();
+        }
+        subscriptions.clear();
+    }
+    async subscribeTerminal(client, payload, authContext) {
+        const subscriptions = this.getClientSubscriptions(client);
+        subscriptions.get(payload.terminalId)?.close();
+        subscriptions.delete(payload.terminalId);
+        try {
+            const subscription = this.terminalService.subscribeTerminal(payload.terminalId, payload.lastCursor ?? null, {
+                onBackfill: async (backfill) => {
+                    client.send(JSON.stringify({
+                        type: "terminal.backfill",
+                        terminalId: payload.terminalId,
+                        truncated: backfill.truncated,
+                        cursorReset: backfill.cursorReset,
+                        latestCursor: backfill.latestCursor,
+                        chunks: backfill.chunks
+                    }));
+                },
+                onOutput: async (chunk) => {
+                    client.send(JSON.stringify({
+                        type: "terminal.output",
+                        terminalId: payload.terminalId,
+                        chunk
+                    }));
+                },
+                onStatus: async (terminal) => {
+                    client.send(JSON.stringify({
+                        type: "terminal.status",
+                        terminal
+                    }));
+                },
+                onExit: async (event) => {
+                    client.send(JSON.stringify({
+                        type: "terminal.exit",
+                        terminalId: payload.terminalId,
+                        requestedClose: event.requestedClose,
+                        terminal: event.terminal
+                    }));
+                }
+            });
+            client.send(JSON.stringify({
+                type: "terminal.subscribed",
+                terminalId: payload.terminalId,
+                userId: authContext.user.userId
+            }));
+            subscriptions.set(payload.terminalId, subscription);
+        }
+        catch (error) {
+            subscriptions.delete(payload.terminalId);
+            const appError = error instanceof AppError
+                ? error
+                : new AppError({
+                    statusCode: 500,
+                    errorCode: "INTERNAL_ERROR",
+                    detail: "订阅终端失败"
+                });
+            client.send(JSON.stringify({
+                type: "terminal.error",
+                terminalId: payload.terminalId,
+                error_code: appError.errorCode,
+                detail: appError.message,
+                timestamp: new Date().toISOString()
+            }));
+        }
+    }
+    async handleTerminalInput(client, payload) {
+        try {
+            this.terminalService.writeInput(payload.terminalId, payload.content);
+            client.send(JSON.stringify({
+                type: "terminal.input.accepted",
+                terminalId: payload.terminalId
+            }));
+        }
+        catch (error) {
+            const appError = error instanceof AppError
+                ? error
+                : new AppError({
+                    statusCode: 500,
+                    errorCode: "INTERNAL_ERROR",
+                    detail: "终端输入失败"
+                });
+            client.send(JSON.stringify({
+                type: "terminal.error",
+                terminalId: payload.terminalId,
+                error_code: appError.errorCode,
+                detail: appError.message,
+                timestamp: new Date().toISOString()
+            }));
+        }
+    }
+    async handleTerminalResize(client, payload) {
+        try {
+            this.terminalService.resizeTerminal(payload.terminalId, payload.cols, payload.rows);
+            client.send(JSON.stringify({
+                type: "terminal.resize.accepted",
+                terminalId: payload.terminalId,
+                cols: payload.cols,
+                rows: payload.rows
+            }));
+        }
+        catch (error) {
+            const appError = error instanceof AppError
+                ? error
+                : new AppError({
+                    statusCode: 500,
+                    errorCode: "INTERNAL_ERROR",
+                    detail: "终端尺寸调整失败"
+                });
+            client.send(JSON.stringify({
+                type: "terminal.error",
+                terminalId: payload.terminalId,
+                error_code: appError.errorCode,
+                detail: appError.message,
+                timestamp: new Date().toISOString()
+            }));
+        }
+    }
+    getClientSubscriptions(client) {
+        let subscriptions = this.clientSubscriptions.get(client);
+        if (!subscriptions) {
+            subscriptions = new Map();
+            this.clientSubscriptions.set(client, subscriptions);
+        }
+        return subscriptions;
+    }
+}
+function isTerminalSubscribeMessage(payload) {
+    const candidate = payload;
+    return (typeof payload === "object" &&
+        payload !== null &&
+        candidate?.type === "terminal.subscribe" &&
+        typeof candidate?.terminalId === "string");
+}
+function isTerminalInputMessage(payload) {
+    const candidate = payload;
+    return (typeof payload === "object" &&
+        payload !== null &&
+        candidate?.type === "terminal.input" &&
+        typeof candidate?.terminalId === "string" &&
+        typeof candidate?.content === "string");
+}
+function isTerminalResizeMessage(payload) {
+    const candidate = payload;
+    return (typeof payload === "object" &&
+        payload !== null &&
+        candidate?.type === "terminal.resize" &&
+        typeof candidate?.terminalId === "string" &&
+        typeof candidate?.cols === "number" &&
+        typeof candidate?.rows === "number");
+}
+//# sourceMappingURL=terminal-ws-hub.js.map

package/dist/server/ws/terminal-ws-hub.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"terminal-ws-hub.js","sourceRoot":"","sources":["../../../src/ws/terminal-ws-hub.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,QAAQ,EAAE,MAAM,+BAA+B,CAAC;AAyBzD,MAAM,OAAO,aAAa;IAGK;IAFZ,mBAAmB,GAAG,IAAI,OAAO,EAA6C,CAAC;IAEhG,YAA6B,eAAgC;QAAhC,oBAAe,GAAf,eAAe,CAAiB;IAAG,CAAC;IAEjE,aAAa,CACX,MAAiB,EACjB,OAAgB,EAChB,WAAwB;QAExB,IAAI,0BAA0B,CAAC,OAAO,CAAC,EAAE,CAAC;YACxC,KAAK,IAAI,CAAC,iBAAiB,CAAC,MAAM,EAAE,OAAO,EAAE,WAAW,CAAC,CAAC;YAC1D,OAAO,IAAI,CAAC;QACd,CAAC;QAED,IAAI,sBAAsB,CAAC,OAAO,CAAC,EAAE,CAAC;YACpC,KAAK,IAAI,CAAC,mBAAmB,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;YAC/C,OAAO,IAAI,CAAC;QACd,CAAC;QAED,IAAI,uBAAuB,CAAC,OAAO,CAAC,EAAE,CAAC;YACrC,KAAK,IAAI,CAAC,oBAAoB,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;YAChD,OAAO,IAAI,CAAC;QACd,CAAC;QAED,OAAO,KAAK,CAAC;IACf,CAAC;IAED,aAAa,CAAC,MAAiB;QAC7B,MAAM,aAAa,GAAG,IAAI,CAAC,mBAAmB,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QAE3D,IAAI,CAAC,aAAa,EAAE,CAAC;YACnB,OAAO;QACT,CAAC;QAED,KAAK,MAAM,YAAY,IAAI,aAAa,CAAC,MAAM,EAAE,EAAE,CAAC;YAClD,YAAY,CAAC,KAAK,EAAE,CAAC;QACvB,CAAC;QAED,aAAa,CAAC,KAAK,EAAE,CAAC;IACxB,CAAC;IAEO,KAAK,CAAC,iBAAiB,CAC7B,MAAiB,EACjB,OAAiC,EACjC,WAAwB;QAExB,MAAM,aAAa,GAAG,IAAI,CAAC,sBAAsB,CAAC,MAAM,CAAC,CAAC;QAC1D,aAAa,CAAC,GAAG,CAAC,OAAO,CAAC,UAAU,CAAC,EAAE,KAAK,EAAE,CAAC;QAC/C,aAAa,CAAC,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;QAEzC,IAAI,CAAC;YACH,MAAM,YAAY,GAAG,IAAI,CAAC,eAAe,CAAC,iBAAiB,CACzD,OAAO,CAAC,UAAU,EAClB,OAAO,CAAC,UAAU,IAAI,IAAI,EAC1B;gBACE,UAAU,EAAE,KAAK,EAAE,QAAQ,EAAE,EAAE;oBAC7B,MAAM,CAAC,IAAI,CACT,IAAI,CAAC,SAAS,CAAC;wBACb,IAAI,EAAE,mBAAmB;wBACzB,UAAU,EAAE,OAAO,CAAC,UAAU;wBAC9B,SAAS,EAAE,QAAQ,CAAC,SAAS;wBAC7B,WAAW,EAAE,QAAQ,CAAC,WAAW;wBACjC,YAAY,EAAE,QAAQ,CAAC,YAAY;wBACnC,MAAM,EAAE,QAAQ,CAAC,MAAM;qBACxB,CAAC,CACH,CAAC;gBACJ,CAAC;gBACD,QAAQ,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE;oBACxB,MAAM,CAAC,IAAI,CACT,IAAI,CAAC,SAAS,CAAC;wBACb,IAAI,EAAE,iBAAiB;wBACvB,UAAU,EAAE,OAAO,CAAC,UAAU;wBAC9B,KAAK;qBACN,CAAC,CACH,CAAC;gBACJ,CAAC;gBACD,QAAQ,EAAE,KAAK,EAAE,QAAQ,EAAE,EAAE;oBAC3B,MAAM,CAAC,IAAI,CACT,IAAI,CAAC,SAAS,CAAC;wBACb,IAAI,EAAE,iBAAiB;wBACvB,QAAQ;qBACT,CAAC,CACH,CAAC;gBACJ,CAAC;gBACD,MAAM,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE;oBACtB,MAAM,CAAC,IAAI,CACT,IAAI,CAAC,SAAS,CAAC;wBACb,IAAI,EAAE,eAAe;wBACrB,UAAU,EAAE,OAAO,CAAC,UAAU;wBAC9B,cAAc,EAAE,KAAK,CAAC,cAAc;wBACpC,QAAQ,EAAE,KAAK,CAAC,QAAQ;qBACzB,CAAC,CACH,CAAC;gBACJ,CAAC;aACF,CACF,CAAC;YAEF,MAAM,CAAC,IAAI,CACT,IAAI,CAAC,SAAS,CAAC;gBACb,IAAI,EAAE,qBAAqB;gBAC3B,UAAU,EAAE,OAAO,CAAC,UAAU;gBAC9B,MAAM,EAAE,WAAW,CAAC,IAAI,CAAC,MAAM;aAChC,CAAC,CACH,CAAC;YAEF,aAAa,CAAC,GAAG,CAAC,OAAO,CAAC,UAAU,EAAE,YAAY,CAAC,CAAC;QACtD,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,aAAa,CAAC,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;YACzC,MAAM,QAAQ,GACZ,KAAK,YAAY,QAAQ;gBACvB,CAAC,CAAC,KAAK;gBACP,CAAC,CAAC,IAAI,QAAQ,CAAC;oBACX,UAAU,EAAE,GAAG;oBACf,SAAS,EAAE,gBAAgB;oBAC3B,MAAM,EAAE,QAAQ;iBACjB,CAAC,CAAC;YAET,MAAM,CAAC,IAAI,CACT,IAAI,CAAC,SAAS,CAAC;gBACb,IAAI,EAAE,gBAAgB;gBACtB,UAAU,EAAE,OAAO,CAAC,UAAU;gBAC9B,UAAU,EAAE,QAAQ,CAAC,SAAS;gBAC9B,MAAM,EAAE,QAAQ,CAAC,OAAO;gBACxB,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;aACpC,CAAC,CACH,CAAC;QACJ,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,mBAAmB,CAC/B,MAAiB,EACjB,OAA6B;QAE7B,IAAI,CAAC;YACH,IAAI,CAAC,eAAe,CAAC,UAAU,CAAC,OAAO,CAAC,UAAU,EAAE,OAAO,CAAC,OAAO,CAAC,CAAC;YACrE,MAAM,CAAC,IAAI,CACT,IAAI,CAAC,SAAS,CAAC;gBACb,IAAI,EAAE,yBAAyB;gBAC/B,UAAU,EAAE,OAAO,CAAC,UAAU;aAC/B,CAAC,CACH,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,QAAQ,GACZ,KAAK,YAAY,QAAQ;gBACvB,CAAC,CAAC,KAAK;gBACP,CAAC,CAAC,IAAI,QAAQ,CAAC;oBACX,UAAU,EAAE,GAAG;oBACf,SAAS,EAAE,gBAAgB;oBAC3B,MAAM,EAAE,QAAQ;iBACjB,CAAC,CAAC;YAET,MAAM,CAAC,IAAI,CACT,IAAI,CAAC,SAAS,CAAC;gBACb,IAAI,EAAE,gBAAgB;gBACtB,UAAU,EAAE,OAAO,CAAC,UAAU;gBAC9B,UAAU,EAAE,QAAQ,CAAC,SAAS;gBAC9B,MAAM,EAAE,QAAQ,CAAC,OAAO;gBACxB,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;aACpC,CAAC,CACH,CAAC;QACJ,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,oBAAoB,CAChC,MAAiB,EACjB,OAA8B;QAE9B,IAAI,CAAC;YACH,IAAI,CAAC,eAAe,CAAC,cAAc,CAAC,OAAO,CAAC,UAAU,EAAE,OAAO,CAAC,IAAI,EAAE,OAAO,CAAC,IAAI,CAAC,CAAC;YACpF,MAAM,CAAC,IAAI,CACT,IAAI,CAAC,SAAS,CAAC;gBACb,IAAI,EAAE,0BAA0B;gBAChC,UAAU,EAAE,OAAO,CAAC,UAAU;gBAC9B,IAAI,EAAE,OAAO,CAAC,IAAI;gBAClB,IAAI,EAAE,OAAO,CAAC,IAAI;aACnB,CAAC,CACH,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,QAAQ,GACZ,KAAK,YAAY,QAAQ;gBACvB,CAAC,CAAC,KAAK;gBACP,CAAC,CAAC,IAAI,QAAQ,CAAC;oBACX,UAAU,EAAE,GAAG;oBACf,SAAS,EAAE,gBAAgB;oBAC3B,MAAM,EAAE,UAAU;iBACnB,CAAC,CAAC;YAET,MAAM,CAAC,IAAI,CACT,IAAI,CAAC,SAAS,CAAC;gBACb,IAAI,EAAE,gBAAgB;gBACtB,UAAU,EAAE,OAAO,CAAC,UAAU;gBAC9B,UAAU,EAAE,QAAQ,CAAC,SAAS;gBAC9B,MAAM,EAAE,QAAQ,CAAC,OAAO;gBACxB,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;aACpC,CAAC,CACH,CAAC;QACJ,CAAC;IACH,CAAC;IAEO,sBAAsB,CAAC,MAAiB;QAC9C,IAAI,aAAa,GAAG,IAAI,CAAC,mBAAmB,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QAEzD,IAAI,CAAC,aAAa,EAAE,CAAC;YACnB,aAAa,GAAG,IAAI,GAAG,EAA6B,CAAC;YACrD,IAAI,CAAC,mBAAmB,CAAC,GAAG,CAAC,MAAM,EAAE,aAAa,CAAC,CAAC;QACtD,CAAC;QAED,OAAO,aAAa,CAAC;IACvB,CAAC;CACF;AAED,SAAS,0BAA0B,CAAC,OAAgB;IAClD,MAAM,SAAS,GAAG,OAAyC,CAAC;IAE5D,OAAO,CACL,OAAO,OAAO,KAAK,QAAQ;QAC3B,OAAO,KAAK,IAAI;QAChB,SAAS,EAAE,IAAI,KAAK,oBAAoB;QACxC,OAAO,SAAS,EAAE,UAAU,KAAK,QAAQ,CAC1C,CAAC;AACJ,CAAC;AAED,SAAS,sBAAsB,CAAC,OAAgB;IAC9C,MAAM,SAAS,GAAG,OAAyC,CAAC;IAE5D,OAAO,CACL,OAAO,OAAO,KAAK,QAAQ;QAC3B,OAAO,KAAK,IAAI;QAChB,SAAS,EAAE,IAAI,KAAK,gBAAgB;QACpC,OAAO,SAAS,EAAE,UAAU,KAAK,QAAQ;QACvC,OAAO,SAAS,EAAE,OAAO,KAAK,QAAQ,CACzC,CAAC;AACJ,CAAC;AAED,SAAS,uBAAuB,CAAC,OAAgB;IAC/C,MAAM,SAAS,GAAG,OAAyC,CAAC;IAE5D,OAAO,CACL,OAAO,OAAO,KAAK,QAAQ;QAC3B,OAAO,KAAK,IAAI;QAChB,SAAS,EAAE,IAAI,KAAK,iBAAiB;QACrC,OAAO,SAAS,EAAE,UAAU,KAAK,QAAQ;QACzC,OAAO,SAAS,EAAE,IAAI,KAAK,QAAQ;QACnC,OAAO,SAAS,EAAE,IAAI,KAAK,QAAQ,CACpC,CAAC;AACJ,CAAC"}

package/dist/server/ws/workbench-ws-hub.d.ts

@@ -0,0 +1,30 @@
+import type { WebSocket } from "ws";
+import type { AuthContext } from "../modules/auth/auth-service.js";
+import type { WorkbenchService } from "../modules/workbench/workbench-service.js";
+import type { WorkspacePanelSnapshotService } from "../modules/workbench/workspace-panel-snapshot-service.js";
+export declare class WorkbenchWsHub {
+    private readonly workbenchService;
+    private readonly workspacePanelSnapshotService;
+    private readonly clientUsers;
+    private readonly userChannels;
+    private readonly clientFileTreeSubscriptions;
+    private readonly clientGitSubscriptions;
+    private readonly clientTerminalManagerSubscriptions;
+    private readonly clientWorkspaceManagementSubscriptions;
+    constructor(workbenchService: WorkbenchService, workspacePanelSnapshotService: WorkspacePanelSnapshotService);
+    handleMessage(client: WebSocket, payload: unknown, authContext: AuthContext): boolean;
+    cleanupClient(client: WebSocket): void;
+    broadcastSnapshot(userId: string): Promise<void>;
+    private attachClient;
+    private getOrCreateChannel;
+    private syncTitlesAndBroadcast;
+    private sendWorkbenchSnapshotToClient;
+    private refreshAndBroadcast;
+    private refreshSidebarSubscriptions;
+    private ensureFileTreeSubscription;
+    private refreshFileTreeSubscriptions;
+    private refreshGitSubscription;
+    private refreshTerminalManagerSubscription;
+    private refreshWorkspaceManagementSubscription;
+    private reportAsyncError;
+}