Method _handle_cli_message

relay/server.py:448–492

Handle a CLI message

(
        self,
        writer: asyncio.StreamWriter,
        msg: dict[str, Any],
    )

    # ===== CLI Message Handling =====

    async def _handle_cli_message(
        self,
        writer: asyncio.StreamWriter,
        msg: dict[str, Any],
    ) -> None:
        """Handle a CLI message"""
        msg_type = msg.get("type")
        request_id = msg.get("id", "")

        if msg_type == MessageType.LIST_INSTANCES.value:
            response = InstancesMessage(
                id=request_id,
                success=True,
                data={"instances": self.registry.list_all()},
            )
            await write_frame(writer, response.to_dict())

        elif msg_type == MessageType.SET_DEFAULT.value:
            instance_id = msg.get("instance", "")
            success = self.registry.set_default(instance_id)
            if success:
                response = ResponseMessage(
                    id=request_id,
                    success=True,
                    data={"message": f"Default instance set to {instance_id}"},
                )
            else:
                response = ErrorMessage.from_code(
                    request_id,
                    ErrorCode.INSTANCE_NOT_FOUND,
                    f"Instance not found: {instance_id}. Check available instances with 'u instances'.",
                )
            await write_frame(writer, response.to_dict())

        elif msg_type == MessageType.REQUEST.value:
            response = await self._handle_request(msg)
            await write_frame(writer, response)

        else:
            response = ErrorMessage.from_code(
                request_id,
                ErrorCode.PROTOCOL_ERROR,
                f"Unknown message type: {msg_type}",
            )
            await write_frame(writer, response.to_dict())

    async def _handle_request(self, msg: dict[str, Any]) -> dict[str, Any]:
        """Handle a REQUEST message from CLI"""