Phase3

apps/node-agent/internal/ws/client.go

@@ -274,6 +274,46 @@ func (c *Client) dispatch(ctx context.Context, env *protocol.Envelope) {
 		}
 		c.runtime.HandleDelete(ctx, correlationID, meta)
 
+	case protocol.TypeServerLogsSubscribe:
+		var payload protocol.ServerLogsSubscribePayload
+		if err := json.Unmarshal(env.Payload, &payload); err != nil {
+			c.log.Warn("decode server.logs.subscribe", "error", err)
+			return
+		}
+		c.runtime.HandleLogsSubscribe(ctx, env.MessageID, payload)
+
+	case protocol.TypeServerCommand:
+		var payload protocol.ServerCommandPayload
+		if err := json.Unmarshal(env.Payload, &payload); err != nil {
+			c.log.Warn("decode server.command", "error", err)
+			return
+		}
+		c.runtime.HandleCommand(ctx, env.MessageID, payload)
+
+	case protocol.TypeServerFilesList:
+		var payload protocol.ServerFilesListPayload
+		if err := json.Unmarshal(env.Payload, &payload); err != nil {
+			c.log.Warn("decode server.files.list", "error", err)
+			return
+		}
+		c.runtime.HandleFilesList(ctx, env.MessageID, payload)
+
+	case protocol.TypeServerFilesRead:
+		var payload protocol.ServerFilesReadPayload
+		if err := json.Unmarshal(env.Payload, &payload); err != nil {
+			c.log.Warn("decode server.files.read", "error", err)
+			return
+		}
+		c.runtime.HandleFilesRead(ctx, env.MessageID, payload)
+
+	case protocol.TypeServerFilesWrite:
+		var payload protocol.ServerFilesWritePayload
+		if err := json.Unmarshal(env.Payload, &payload); err != nil {
+			c.log.Warn("decode server.files.write", "error", err)
+			return
+		}
+		c.runtime.HandleFilesWrite(ctx, env.MessageID, payload)
+
 	default:
 		c.log.Debug("ignored control message", "type", env.Type)
 	}
@@ -333,6 +373,11 @@ func (c *Client) SendState(ctx context.Context, correlationID string, payload pr
 	return c.send(ctx, protocol.TypeServerState, correlationID, payload)
 }
 
+// SendLog implements runtime.Responder.
+func (c *Client) SendLog(ctx context.Context, correlationID string, payload protocol.ServerLogPayload) error {
+	return c.send(ctx, protocol.TypeServerLog, correlationID, payload)
+}
+
 // SendOperationComplete implements runtime.Responder.
 func (c *Client) SendOperationComplete(ctx context.Context, correlationID string, payload protocol.ServerOperationResultPayload) error {
 	return c.send(ctx, protocol.TypeServerOperationComplete, correlationID, payload)