import ad_toolbox.smartcondition as SmartCondition
import ad_toolbox.smartvalue as SmartValue
from ad_toolbox.smartobject import SmartObject
import time

class SmartHeatpump(SmartObject):

    def on_initialize_smart_object(self):
        super().on_initialize_smart_object()

        if 'quick_resume' in self.args:
            self.quick_resume_sensor = self.create_entity(self.args['quick_resume']['sensor']) 
            #is_on =  'on' if self.entity.get_state() != 'off' else 'off'
            #if self.get_state(self.args['quick_resume']['sensor']) != 'on': # if the sensor doesn't exist I wan't to set it as well
            #    self.set_state(self.args['quick_resume']['sensor'],state = is_on)
            self.listen_event(lambda event_name, data, kwargs: self.set_state(self.args['quick_resume']['sensor'],state = 'off'),self.args['quick_resume']['reset_event'])

        self.target_temperature = SmartValue.Evaluator(self,self.args['target_temperature'], expression_name = 'target_temperature', on_change_cb = self.on_change_target_temperature, constants = self.constants)
        self.hvac_mode = SmartValue.Evaluator(self,self.args['hvac_mode'], expression_name = 'hvac_mode', on_change_cb = self.on_change_hvac_mode, constants = self.constants)
        if 'horizontal_vane' in self.args:
            self.horizontal_vane = SmartValue.Evaluator(self,self.args['horizontal_vane'], expression_name = 'horizontal_vane', on_change_cb = self.on_change_horizontal_vane, constants = self.constants)
        else:
            self.horizontal_vane = None

        if 'fan_mode' in self.args:
            self.fan_mode = SmartValue.Evaluator(self,self.args['fan_mode'], expression_name = 'fan_mode', on_change_cb = self.on_change_fan_mode, constants = self.constants)
        else:
            self.fan_mode = None
        self.smart_conditions_evaluator = SmartCondition.Evaluator(self,self.args['smart_conditions'],condition_name = "smart_conditions",on_change_cb = self.on_smart_conditions_change,constants = self.constants, templates_library = self.templates_library)

        if 'input_buttons' in self.args:
            if 'turn_on' in self.args['input_buttons']:
                self.listen_state(lambda entity, attribute, old, new, kwargs: self.on_turn_on_button(),self.args['input_buttons']['turn_on'])
            if 'turn_off' in self.args['input_buttons']:
                self.listen_state(lambda entity, attribute, old, new, kwargs: self.on_turn_off_button(),self.args['input_buttons']['turn_off'])

    def on_turn_on_button(self): self.start_cooling()

    def on_turn_off_button(self): 
        self.stop_cooling()
        # we don't wan't quick resum if we force turn off the a/c
        if 'quick_resume' in self.args:
            self.set_state(self.args['quick_resume']['sensor'],state = 'off')

    def start_cooling(self):
        if 'quick_resume' in self.args:
            self.set_state(self.args['quick_resume']['sensor'],state = 'on')
        temperature = self.target_temperature.evaluate(False)
        hvac_mode = self.hvac_mode.evaluate(False)
        fan_mode = self.fan_mode.evaluate(False) if self.fan_mode else None
        self.log_info(f"Starting Heatpump in {hvac_mode} mode, with target_temperature of {temperature}° and {fan_mode} fan speed")
        
        #self.turn_on(self.entity_id)
        #time.sleep(2)

        self.call_service("climate/set_hvac_mode", entity_id = self.entity_id,hvac_mode = hvac_mode)
        time.sleep(1)
        self.call_service("climate/set_temperature", entity_id = self.entity_id,temperature = temperature,hvac_mode = hvac_mode)
        if self.fan_mode:
            time.sleep(1)
            self.call_service("climate/set_fan_mode", entity_id = self.entity_id,fan_mode = fan_mode)
        if self.horizontal_vane:
            time.sleep(1)
            horizontal_vane = self.horizontal_vane.evaluate(False)
            self.set_horizontal_vane(horizontal_vane)

    def stop_cooling(self):
        self.log_info("Setting Heatpump hvac_mode to off")
        self.call_service("climate/set_hvac_mode", entity_id = self.entity_id,hvac_mode = "off")


    def on_smart_conditions_change(self,prev_result,result):
        if result == SmartCondition.Result.Succeeded:
            self.start_cooling()
        else:
            self.log_info("Turning Off Heatpump")
            self.stop_cooling()

    def on_change_target_temperature(self,prev_result,result):
        if self.entity.get_state() != 'off':
            self.log_info(f"Setting target temperature to {result}°")
            self.call_service("climate/set_temperature", entity_id = self.entity_id,temperature = result)

    def on_change_hvac_mode(self,prev_result,result):
        if self.entity.get_state() != 'off':
            self.log_info(f"Setting hvac_mode to {result}")
            self.call_service("climate/set_hvac_mode", entity_id = self.entity_id,hvac_mode = result)

    def on_change_fan_mode(self,prev_result,result):
        if self.entity.get_state() != 'off':
            self.log_info(f"Setting fan_mode to {result}")
            self.call_service("climate/set_fan_mode", entity_id = self.entity_id,fan_mode = result)

    def set_horizontal_vane(self,value):
        device, entity = self.split_entity(self.entity_id)
        self.call_service("select/select_option", entity_id = f"select.{entity}_horizontal_vane",option = value)
        #self.set_state(f"select.{entity}_horizontal_vane",state = value)

    def on_change_horizontal_vane(self,prev_result,result):
        if self.entity.get_state() != 'off':
            self.log_info(f"Setting horizontal_vane to {result}")
            self.set_horizontal_vane(result)