###########################################################################
#
# Copyright (c) 2019-2021 Diality Inc. - All Rights Reserved.
#
# THIS CODE MAY NOT BE COPIED OR REPRODUCED IN ANY FORM, IN PART OR IN
# WHOLE, WITHOUT THE EXPLICIT PERMISSION OF THE COPYRIGHT OWNER.
#
# @file    heat_disinfect.py
#
# @author  (last)      Quang Nguyen
# @date    (last)      05-Aug-2021
# @author  (original)  Dara Navaei
# @date    (original)  27-Feb-2021
#
############################################################################
import struct
from enum import unique
from logging import Logger

from ..common.msg_defs import MsgIds, MsgFieldPositions
from ..protocols.CAN import DenaliChannels
from ..utils.base import AbstractSubSystem, publish, DialinEnum


@unique
class HeatDisinfectStates(DialinEnum):
    DG_HEAT_DISINFECT_STATE_START = 0
    DG_HEAT_DISINFECT_STATE_DRAIN_R1 = 1
    DG_HEAT_DISINFECT_STATE_DRAIN_R2 = 2
    DG_HEAT_DISINFECT_STATE_FLUSH_DRAIN = 3
    DG_HEAT_DISINFECT_STATE_FLUSH_CIRCULATION = 4
    DG_HEAT_DISINFECT_STATE_FLUSH_R1_AND_R2 = 5
    DG_HEAT_DISINFECT_STATE_FLUSH_R2_AND_DRAIN_R1 = 6
    DG_HEAT_DISINFECT_STATE_FLUSH_DRAIN_R2 = 7
    DG_HEAT_DISINFECT_STATE_FLUSH_DRAIN_R1 = 8
    DG_HEAT_DISINFECT_STATE_FILL_WITH_WATER = 9
    DG_HEAT_DISINFECT_STATE_DISINFECT_R1_TO_R2 = 10
    DG_HEAT_DISINFECT_STATE_FILL_R2_WITH_HOT_WATER = 11
    DG_HEAT_DISINFECT_STATE_DISINFECT_R2_TO_R1 = 12
    DG_HEAT_DISINFECT_STATE_COOL_DOWN_HEATERS = 13
    DG_HEAT_DISINFECT_STATE_COOL_DOWN_RO_FILTER = 14
    DG_HEAT_DISINFECT_STATE_MIX_DRAIN_R1 = 15
    DG_HEAT_DISINFECT_STATE_MIX_DRAIN_R2 = 16
    DG_HEAT_DISINFECT_STATE_RINSE_R1_TO_R2 = 17
    DG_HEAT_DISINFECT_STATE_RINSE_R2_TO_R1_AND_DRAIN_R1 = 18
    DG_HEAT_DISINFECT_STATE_RINSE_CIRCULATION = 19
    DG_HEAT_DISINFECT_STATE_CANCEL_BASIC_PATH = 20
    DG_HEAT_DISINFECT_STATE_CANCEL_WATER_PATH = 21
    DG_HEAT_DISINFECT_STATE_COMPLETE = 22


@unique
class HeatCancellationModes(DialinEnum):
    CANCELLATION_MODE_NONE = 0
    CANCELLATION_MODE_BASIC = 1
    CANCELLATION_MODE_HOT = 2
    CANCELLATION_MODE_COLD = 3


class HeatDisinfect(AbstractSubSystem):
    """
        Heat Disinfection 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