from .protocol import (
    make_status_request,
    make_set_relay,
    is_status_response,
    is_relay_response,
    is_error,
)


class CommunicationService:
    def __init__(self, uart_client, app_state):
        self.uart_client = uart_client
        self.app_state = app_state
        self.last_messages = []

    def request_status(self):
        self.uart_client.send_message(make_status_request())

    def set_relay(self, relay, enabled):
        self.uart_client.send_message(make_set_relay(relay, enabled))

    def poll(self):
        messages = self.uart_client.read_available_messages()
        self.last_messages = messages

        for message in messages:
            self.handle_message(message)

        return messages

    def handle_message(self, message):
        if is_status_response(message):
            self.app_state.update_from_status(message)
            return

        if is_relay_response(message):
            self.app_state.update_from_relay_response(message)
            return

        if is_error(message):
            self.app_state.set_error(message)
            return

        self.app_state.set_error({
            "type": "error",
            "message": "unknown_message_type",
            "raw": message,
        })