###########################################################################
#
# Copyright (c) 2022-2024 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_flush.py
#
# @author  (last)      Micahel Garthwaite
# @date    (last)      18-May-2023
# @author  (original)  Steve Jarpe
# @date    (original)  30-Nov-2022
#
############################################################################
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
from ..common.dg_defs import DGChemDisinfectFlushStates, DGChemDisinfectFlushUIStates


class ChemicalDisinfectFlushMode(AbstractSubSystem):
    """
        Chemical Disinfect Flush Mode 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.flush_state = DGChemDisinfectFlushStates.DG_CHEM_DISINFECT_FLUSH_STATE_START.value
        self.overall_elapsed_time = 0
        self.state_elapsed_time = 0
        self.rinse_count = 0
        self.flush_UI_state = DGChemDisinfectFlushUIStates.CHEM_DISINFECT_FLUSH_UI_STATE_NOT_RUNNING.value
        self.dg_chem_disinfect_flush_timestamp = 0.0

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

    def clear_flush_info(self) -> None:
        """
        Clears public class properties that are updated by the handler.
        Specifically properties updated  by the DG broadcast message
        MSG_ID_DG_CHEM_DISINFECT_FLUSH_DATA.

        @returns : none
        """
        self.flush_state = DGChemDisinfectFlushStates.DG_CHEM_DISINFECT_FLUSH_STATE_START.value
        self.overall_elapsed_time = 0
        self.state_elapsed_time = 0

    @publish(["dg_chem_disinfect_flush_timestamp","flush_state", "overall_elapsed_time", "state_elapsed_time",
              "r1", "r2", "target_rinse_count", "rinse_count", "flush_UI_state"])
    def _handler_chem_disinfect_flush_sync(self, message: dict, timestamp = 0.0) -> None:
        """
        Handles published flush message

        @param message:  published flush 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]
        rinse_count = struct.unpack('<I', bytearray(
            message['message'][MsgFieldPositions.START_POS_FIELD_5:MsgFieldPositions.END_POS_FIELD_5]))[0]
        flush_ui_state = struct.unpack('<I', bytearray(
            message['message'][MsgFieldPositions.START_POS_FIELD_6:MsgFieldPositions.END_POS_FIELD_6]))[0]

        self.flush_state = state
        self.overall_elapsed_time = int(elapsed_time / 1000)
        self.state_elapsed_time = int(state_elapsed_time / 1000)
        self.rinse_count = rinse_count
        self.flush_UI_state = flush_ui_state
        self.dg_chem_disinfect_flush_timestamp = timestamp