from dataclasses import dataclass, field, fields
from datetime import timedelta

from appdaemon.entity import Entity
from appdaemon.plugins.hass.hassapi import Hass


@dataclass(init=False)
class LivingRoomTV(Hass):
    remote_entity: str = None
    playing_entity: str = None
    # media_entity: str = None
    # hardware_entity: str = None

    def initialize(self):
        self.log(self.args)
        for f in fields(self):
            if f.name in self.args:
                entity_id = self.args[f.name]
                if not self.entity_exists(entity_id):
                    self.log(f'{entity_id} does not exist', level="WARNING")
                else:
                    setattr(self, f.name, entity_id)
            else:
                self.log(f'{f.name} field is unset', level="WARNING")
        self.log(repr(self))

        self.listen_state(self.turn_on_soundbar, self.playing_entity, new='playing')
        self.log(f'Waiting for {self.playing_entity} to go to playing')

    def turn_on_soundbar(self, entity=None, attribute=None, old=None, new=None, kwargs=None):
        if self.get_state(self.remote_entity) != 'on':
            self.log('Turning on remote')
            self.turn_on(self.remote_entity)
        # if self.remote_entity.state != 'on':
        #     self.log('Turning on remote')
        #     self.remote_entity.turn_on()

        self.log('Turning on soundbar')
        self.call_service(service='remote/send_command',
                          entity_id=self.remote_entity,
                          device='BoseTV', command='TV')

    def state_change(self, entity=None, attribute=None, old=None, new=None, kwargs=None):
        try:
            if entity.endswith('2') or entity.endswith('3'):
                name = new['attributes']['entity_id']
                app_name = new['attributes']['app_name']
                state = new['state']
                self.log(f'{name}: {app_name} {state}')
            else:
                self.log(f'{entity}: {new}')
        except KeyError as e:
            self.log(f'Failed getting attribute {e} from {entity}')

    def cancel_callbacks(self, entity=None, attribute=None, old=None, new=None, kwargs=None):
        if (callbacks := self.get_scheduler_entries().get(self.name)):
            for handle, info in callbacks.items():
                self.cancel_timer(handle)
                self.log(f'Cancelled {handle}')

    def hardware_off_callback(self, entity=None, attribute=None, old=None, new=None, kwargs=None):
        self.log('Hardware turned off')
        self.media_entity.turn_off()