Cyrene_v2/internal/services/gateway/ws_route.go

package gateway

import (
	"context"
	"encoding/json"

	"git.yeij.top/AskaEth/Cyrene_v2/internal/runtime/orchestration"
)

// route dispatches one inbound frame by its type.
func (c *wsConn) route(in wsInbound) {
	switch in.Type {
	case "mode_switch":
		c.publishPublic("ws.mode_switch", rawData(in.Data))
	case "chat":
		c.publishPublic(wsChatSubject, rawData(in.Data))
	case "voice_input":
		// Voice is not wired to a live pipeline yet; surface the payload on the
		// bus so an upper layer (or a later milestone) can consume it, rather
		// than dropping it as unknown.
		c.publishPublic(wsVoiceInput, rawData(in.Data))
	case "subagent_ask":
		c.handleSubagentAsk(in)
	case "task_output_query":
		c.handleTaskOutputQuery(in)
	default:
		c.enqueueJSON(wsEvent{Type: "error", Data: map[string]any{"message": "unknown event type"}})
	}
}

// subagentAskRequest is the decoded payload of a subagent_ask frame. Mode selects
// the audit window (""/full/tail/keyword); n bounds a tail query; query is the
// keyword for a keyword query.
type subagentAskRequest struct {
	ID    string `json:"id"`
	Mode  string `json:"mode"`
	N     int    `json:"n"`
	Query string `json:"query"`
}

// taskOutputRequest is the decoded payload of a task_output_query frame.
type taskOutputRequest struct {
	TaskID string `json:"taskId"`
}

// handleSubagentAsk audits a sub-agent's context and replies with a
// ws.subagent_event frame carrying the real context window. When no auditor is
// injected (gateway not wired with a scheduler) it replies with a clear error
// rather than a fabricated "pending" snapshot.
func (c *wsConn) handleSubagentAsk(in wsInbound) {
	if c.h.auditor == nil {
		c.enqueueJSON(wsEvent{Type: "error", Data: map[string]any{"message": "subagent audit unavailable"}})
		return
	}
	var req subagentAskRequest
	_ = json.Unmarshal(in.Data, &req)
	if req.ID == "" {
		c.enqueueJSON(wsEvent{Type: "error", Data: map[string]any{"message": "subagent id required"}})
		return
	}
	mode := orchestration.AuditMode(req.Mode)
	if mode == "" {
		mode = orchestration.AuditFull
	}
	res, err := c.h.auditor.AuditContext(context.Background(), req.ID, mode, req.N, req.Query)
	if err != nil {
		c.h.log.Debug("subagent audit", "id", req.ID, "err", err)
		c.enqueueJSON(wsEvent{Type: "error", Data: map[string]any{"message": err.Error()}})
		return
	}
	c.enqueueJSON(wsEvent{Type: "subagent_event", Data: res})
}

// handleTaskOutputQuery fetches the real task output for taskId and replies with
// a ws.task_output frame. When no task store is injected it replies with a clear
// error rather than an empty {}. A task that is still running returns the output
// captured so far.
func (c *wsConn) handleTaskOutputQuery(in wsInbound) {
	if c.h.taskOutputs == nil {
		c.enqueueJSON(wsEvent{Type: "error", Data: map[string]any{"message": "task output unavailable"}})
		return
	}
	var req taskOutputRequest
	_ = json.Unmarshal(in.Data, &req)
	if req.TaskID == "" {
		c.enqueueJSON(wsEvent{Type: "error", Data: map[string]any{"message": "task id required"}})
		return
	}
	ev, err := c.h.taskOutputs.TaskOutput(context.Background(), req.TaskID)
	if err != nil {
		c.h.log.Debug("task output query", "id", req.TaskID, "err", err)
		c.enqueueJSON(wsEvent{Type: "error", Data: map[string]any{"message": err.Error()}})
		return
	}
	text := ev.Output.Stdout
	if ev.Output.Stderr != "" {
		if text != "" {
			text += "\n"
		}
		text += ev.Output.Stderr
	}
	c.enqueueJSON(wsEvent{Type: "task_output", Data: map[string]any{
		"taskId": ev.ID,
		"output": text,
		"status": string(ev.Status),
	}})
}