import struct
from ..common.msg_defs import MsgIds, MsgFieldPositions
from ..protocols.CAN import DenaliMessage, DenaliChannels
from ..utils.base import _AbstractSubSystem, _publish, DialinEnum
from logging import Logger
from enum import unique


@unique
class HeatCancellationModes(DialinEnum):

    CANCELLATION_MODE_NONE = 0
    CANCELLATION_MODE_BASIC = 1
    CANCELLATION_MODE_HOT = 2
    CANCELLATION_MODE_COLD = 3


class HeatDisinfect(_AbstractSubSystem):
    """
    \class HeatDisinfect

    \brief Heat disinfect class with APIs to set the timing of each of the stages.
    """

    def __init__(self, can_interface, logger: Logger):

        super().__init__()

        self.can_interface = can_interface
        self.logger = logger

        self.heat_disinfect_state = 0
        self.heat_disinfect_ui_state = 0
        self.overall_elapsed_time = 0
        self.state_elapsed_time = 0
        self.heat_disinfect_target_time = 0
        self.heat_disinfect_count_down_time = 0
        self.cancellation_mode = 0
        self.r1_level = 0
        self.r2_level = 0

        if self.can_interface is not None:
            channel_id = DenaliChannels.dg_sync_broadcast_ch_id
            msg_id = MsgIds.MSG_ID_DG_HEAT_DISINFECT_DATA.value
            self.can_interface.register_receiving_publication_function(channel_id, msg_id,
                                                                       self._handler_heat_disinfect_sync)

            channel_id = DenaliChannels.dg_to_ui_ch_id
            msg_id = MsgIds.MSG_ID_DG_HEAT_DISINFECT_TIME_DATA.value
            self.can_interface.register_receiving_publication_function(channel_id, msg_id,
                                                                       self._handler_heat_disinfect_to_ui_sync)

    @_publish(["heat_disinfect_target_time", "heat_disinfect_count_down_time"])
    def _handler_heat_disinfect_to_ui_sync(self, message):
        """
        Handles published heat disinfect message

        @param message:  published heat disinfect UI data message
        @returns none
        """
        disinfect_target_time = struct.unpack('i', bytearray(
            message['message'][MsgFieldPositions.START_POS_FIELD_1:MsgFieldPositions.END_POS_FIELD_1]))[0]
        disinfect_count_down_time = struct.unpack('i', bytearray(
            message['message'][MsgFieldPositions.START_POS_FIELD_2:MsgFieldPositions.END_POS_FIELD_2]))[0]

        self.heat_disinfect_target_time = int(disinfect_target_time / 1000)
        self.heat_disinfect_count_down_time = disinfect_count_down_time

    @_publish(["heat_disinfect_state", "overall_elapsed_time", "state_elapsed_time", "cancellation_mode", "r1_level",
               "r2_level", "heat_disinfect_ui_state"])
    def _handler_heat_disinfect_sync(self, message):
        """
        Handles published heat disinfect message

        @param message:  published heat disinfect data message
        @returns none
        """
        state = struct.unpack('i', bytearray(
            message['message'][MsgFieldPositions.START_POS_FIELD_1:MsgFieldPositions.END_POS_FIELD_1]))[0]
        elapsed_time = struct.unpack('i', bytearray(
            message['message'][MsgFieldPositions.START_POS_FIELD_2:MsgFieldPositions.END_POS_FIELD_2]))[0]
        state_elapsed_time = struct.unpack('i', bytearray(
            message['message'][MsgFieldPositions.START_POS_FIELD_3:MsgFieldPositions.END_POS_FIELD_3]))[0]
        cancellation_mode = struct.unpack('i', bytearray(
            message['message'][MsgFieldPositions.START_POS_FIELD_4:MsgFieldPositions.END_POS_FIELD_4]))[0]
        r1 = struct.unpack('f', bytearray(
            message['message'][MsgFieldPositions.START_POS_FIELD_5:MsgFieldPositions.END_POS_FIELD_5]))[0]
        r2 = struct.unpack('f', bytearray(
            message['message'][MsgFieldPositions.START_POS_FIELD_6:MsgFieldPositions.END_POS_FIELD_6]))[0]
        ui = struct.unpack('i', bytearray(
            message['message'][MsgFieldPositions.START_POS_FIELD_7:MsgFieldPositions.END_POS_FIELD_7]))[0]

        self.heat_disinfect_state = state
        self.overall_elapsed_time = int(elapsed_time / 1000)
        self.state_elapsed_time = int(state_elapsed_time / 1000)
        self.cancellation_mode = cancellation_mode
        self.r1_level = r1
        self.r2_level = r2
        self.heat_disinfect_ui_state = ui