catlog22/Claude-Code-Workflow
1
0
Fork 0
mirror of https://github.com/catlog22/Claude-Code-Workflow.git synced 2026-02-14 02:42:04 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat(websocket): integrate A2UI message handling and answer callback registration

Browse Source
This commit is contained in:
catlog22
2026-02-03 21:18:19 +08:00
parent bb4cd0529e
commit 9bb50a13fa
4 changed files with 30 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
ccw/src/core/websocket.ts
View File

@@ -1,6 +1,8 @@
import { createHash } from 'crypto';
import type { IncomingMessage } from 'http';
import type { Duplex } from 'stream';
import { a2uiWebSocketHandler, handleA2UIMessage } from './a2ui/A2UIWebSocketHandler.js';
import { handleAnswer } from '../tools/ask-question.js';
// WebSocket clients for real-time notifications
export const wsClients = new Set<Duplex>();
@@ -174,6 +176,11 @@ export function handleWebSocketUpgrade(req: IncomingMessage, socket: Duplex, _he
case 0x1: // Text frame
if (payload) {
console.log('[WS] Received:', payload);
// Try to handle as A2UI message
const handledAsA2UI = handleA2UIMessage(payload, a2uiWebSocketHandler, handleAnswer);
if (handledAsA2UI) {
console.log('[WS] Handled as A2UI message');
}
}
break;
case 0x8: // Close frame