from DialIn.CoreCANProtocol import (DenaliMessage,
                                    DenaliChannels)

from .utils import integer2ByteArray
from .HemodialysisDevice import HD


class HDButtons:
    """
    \class HD_Buttons

    \brief Hemodialysis Device (HD) Dialin API sub-class for button related commands.
    """

    # Buttons message IDs
    MSG_ID_HD_OFF_BUTTON_OVERRIDE = 0x8002
    MSG_ID_HD_STOP_BUTTON_OVERRIDE = 0x8003

    def __init__(self, can_interface):
        """
        HD_Buttons constructor

        \param can_interface: the denali can interface object
        """
        self.can_interface = can_interface

    def cmd_off_button_override(self, reset, state):
        """
        Constructs and sends the Off button override command

        \param reset:  integer - 1 to reset a previous override, 0 to override
        \param state:  integer - 1 to override off button to "pressed", 0 to "released"
        \returns 1 if successful, zero otherwise
        """

        rst = integer2ByteArray(reset)
        sta = integer2ByteArray(state)
        payload = rst + sta

        message = DenaliMessage.build_message(channel_id=DenaliChannels.dialin_to_hd_ch_id,
                                              message_id=self.MSG_ID_HD_OFF_BUTTON_OVERRIDE,
                                              payload=payload)

        print("override off button")

        # Send message
        received_message = self.can_interface.send(message)

        # If there is content...
        if received_message is not None:
            if reset == HD.RESET:
                str_res = "reset back to normal"
            else:
                str_res = ("pressed" if state != 0 else "released")

            print("Off button overridden to " + str_res + ":" +
                  str(received_message['message'][DenaliMessage.PAYLOAD_START_INDEX]))
            # response payload is OK or not OK
            return received_message['message'][DenaliMessage.PAYLOAD_START_INDEX]
        else:
            print("Timeout!!!!")
            return False

    def cmd_stop_button_override(self, reset, state):
        """
        Constructs and sends the Stop button override command

        \param reset:  integer - 1 to reset a previous override, 0 to override
        \param state:  integer - 1 to override stop button to "pressed", 0 to "released"
        \returns 1 if successful, zero otherwise
        """

        rst = integer2ByteArray(reset)
        sta = integer2ByteArray(state)
        payload = rst + sta

        message = DenaliMessage.build_message(channel_id=DenaliChannels.dialin_to_hd_ch_id,
                                              message_id=self.MSG_ID_HD_STOP_BUTTON_OVERRIDE,
                                              payload=payload)

        print("override stop button")

        # Send message
        received_message = self.can_interface.send(message)

        # If there is content...
        if received_message is not None:
            if reset == HD.RESET:
                str_res = "reset back to normal"
            else:
                str_res = ("pressed" if state != 0 else "released")
            print("Stop button overridden to " + str_res + ":" +
                  str(received_message['message'][DenaliMessage.PAYLOAD_START_INDEX]))

            # response payload is OK or not OK
            return received_message["message"][DenaliMessage.PAYLOAD_START_INDEX]
        else:
            print("Timeout!!!!")
            return False