###########################################################################
#
# Copyright (c) 2020-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    buttons.py
#
# @author  (last)      Micahel Garthwaite
# @date    (last)      18-Aug-2023
# @author  (original)  Peter Lucia
# @date    (original)  02-Apr-2020
#
############################################################################
import struct
from logging import Logger

from leahi_dialin.common.msg_defs import MsgIds, MsgFieldPositions
from leahi_dialin.protocols.CAN import DenaliMessage, DenaliCanMessenger, DenaliChannels
from leahi_dialin.utils.base import AbstractSubSystem, publish
from leahi_dialin.utils.conversions import integer_to_bytearray, byte_to_bytearray, float_to_bytearray


class TDProxy(AbstractSubSystem):
    """
    Dialysate Delivery (DD) Dialin API sub-class for TD proxy ( injection ) related commands.
    """

    def __init__(self, can_interface: DenaliCanMessenger, logger: Logger):
        """
        HD_Buttons constructor

        @param can_interface: the Denali CAN interface object
        """

        super().__init__()
        self.can_interface = can_interface
        self.logger = logger

        if self.can_interface is not None:
            channel_id = DenaliChannels.td_to_dd_ch_id
            msg_id = MsgIds.MSG_ID_DD_GEN_DIALYSATE_REQUEST_DATA.value
            self.can_interface.register_receiving_publication_function(channel_id, msg_id,
                                                                       self._handler_dialysate_delivery_request_response)

        self.dialysate_delivery_request_start = 0
        self.dialysate_delivery_request_dial_rate = 0.0
        self.dialysate_delivery_request_uf_rate = 0.0
        self.dialysate_delivery_request_dial_temp = 0.0
        self.dialysate_delivery_request_bypass = 0
        self.dialysate_delivery_request_acid = 0
        self.dialysate_delivery_request_bicarb = 0
        self.dd_td_to_dd_request_response_timestamp = 0.0

    @publish(["dd_td_to_dd_request_response_timestamp",
              "dialysate_delivery_request_start","dialysate_delivery_request_dial_rate",
              "dialysate_delivery_request_uf_rate","dialysate_delivery_request_dial_temp",
              "dialysate_delivery_request_bypass","dialysate_delivery_request_acid",
              "dialysate_delivery_request_bicarb",
              ])
    def _handler_dialysate_delivery_request_response(self, message, timestamp=0.0):
        self.dialysate_delivery_request_start = struct.unpack('i', bytearray(
            message['message'][MsgFieldPositions.START_POS_FIELD_1:MsgFieldPositions.END_POS_FIELD_1]))[0]
        self.dialysate_delivery_request_dial_rate = struct.unpack('f', bytearray(
            message['message'][MsgFieldPositions.START_POS_FIELD_2:MsgFieldPositions.END_POS_FIELD_2]))[0]
        self.dialysate_delivery_request_uf_rate = struct.unpack('f', bytearray(
            message['message'][MsgFieldPositions.START_POS_FIELD_3:MsgFieldPositions.END_POS_FIELD_3]))[0]
        self.dialysate_delivery_request_dial_temp = struct.unpack('f', bytearray(
            message['message'][MsgFieldPositions.START_POS_FIELD_4:MsgFieldPositions.END_POS_FIELD_4]))[0]
        self.dialysate_delivery_request_bypass = struct.unpack('i', bytearray(
            message['message'][MsgFieldPositions.START_POS_FIELD_5:MsgFieldPositions.END_POS_FIELD_5]))[0]
        self.dialysate_delivery_request_acid = struct.unpack('i', bytearray(
            message['message'][MsgFieldPositions.START_POS_FIELD_6:MsgFieldPositions.END_POS_FIELD_6]))[0]
        self.dialysate_delivery_request_bicarb = struct.unpack('i', bytearray(
            message['message'][MsgFieldPositions.START_POS_FIELD_7:MsgFieldPositions.END_POS_FIELD_7]))[0]

        self.dd_td_to_dd_request_response_timestamp = timestamp