###########################################################################
#
# 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    chemical_disinfect.py
#
# @author  (last)      Quang Nguyen
# @date    (last)      05-Aug-2021
# @author  (original)  Dara Navaei
# @date    (original)  02-May-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 ChemCancellationModes(DialinEnum):
    CANCELLATION_MODE_NONE = 0
    CANCELLATION_MODE_BASIC = 1
    CANCELLATION_MODE_WATER = 2
    CANCELLATION_MODE_CHEMICAL = 3


class ChemicalDisinfect(AbstractSubSystem):
    """
    @Brief Chemical 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.chemical_disinfect_state = 0
        self.overall_elapsed_time = 0
        self.state_elapsed_time = 0
        self.chemical_disinfect_target_time = 0
        self.chemical_disinfect_elapsed_time = 0
        self.cancellation_mode = 0
        self.r1_level = 0
        self.r2_level = 0
        self.target_post_rinse_count = 0
        self.current_post_rinse_count = 0
        self.chemical_disinfect_ui_state = 0

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

            channel_id = DenaliChannels.dg_to_ui_ch_id
            msg_id = MsgIds.MSG_ID_DG_CHEM_DISINFECT_TIME_DATA.value
            self.can_interface.register_receiving_publication_function(channel_id, msg_id,
                                                                       self._handler_chemical_disinfect_to_ui_sync)

    @publish(["chemical_disinfect_elapsed_time"])
    def _handler_chemical_disinfect_to_ui_sync(self, message):
        """
               Handles published chemical disinfect message

               @param message:  published chemical disinfect to 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_elapsed_time = struct.unpack('i', bytearray(
            message['message'][MsgFieldPositions.START_POS_FIELD_2:MsgFieldPositions.END_POS_FIELD_2]))[0]

        self.chemical_disinfect_target_time = int(disinfect_target_time / 1000)
        self.chemical_disinfect_target_time = int(disinfect_elapsed_time / 1000)

    @publish(["chemical_disinfect_state", "overall_elapsed_time", "state_elapsed_time", "cancellation_mode",
              "r1_level", "r2_level", "target_post_rinse_count", "current_post_rinse_count",
              "chemical_disinfect_ui_state"])
    def _handler_chemical_disinfect_sync(self, message):
        """
        Handles published chemical disinfect message

        @param message:  published chemical 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]
        target_post_rinse_cnt = struct.unpack('i', bytearray(
            message['message'][MsgFieldPositions.START_POS_FIELD_7:MsgFieldPositions.END_POS_FIELD_7]))[0]
        current_post_rinse_cnt = struct.unpack('i', bytearray(
            message['message'][MsgFieldPositions.START_POS_FIELD_8:MsgFieldPositions.END_POS_FIELD_8]))[0]
        chem_ui_state = struct.unpack('i', bytearray(
            message['message'][MsgFieldPositions.START_POS_FIELD_9:MsgFieldPositions.END_POS_FIELD_9]))[0]

        self.chemical_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.target_post_rinse_count = target_post_rinse_cnt
        self.current_post_rinse_count = current_post_rinse_cnt
        self.chemical_disinfect_ui_state = chem_ui_state