Validate state values in WebSocket message handler
This commit is contained in:
parent
5df3392ad9
commit
d30a092df5
1 changed files with 3 additions and 2 deletions
|
|
@ -447,8 +447,9 @@ const server = Bun.serve<SessionData>({
|
|||
// Handle state message
|
||||
if (msg.type === "state") {
|
||||
const sessionState = sessionStates.get(ws.data.sessionId);
|
||||
if (sessionState && msg.state) {
|
||||
sessionState.state = msg.state;
|
||||
const validStates = ["ready", "thinking", "permission", "question", "complete", "interrupted"];
|
||||
if (sessionState && msg.state && validStates.includes(msg.state)) {
|
||||
sessionState.state = msg.state as SessionState["state"];
|
||||
sessionState.dirty = true;
|
||||
// Broadcast SSE state event
|
||||
broadcastSSE({
|
||||
|
|
|
|||
Loading…
Reference in a new issue