import json
import re

from appdaemon.plugins.hass.hassapi import Hass
from room_control import Motion


class SceneDetector(Hass):
    def initialize(self):
        self.scene_entity = (
            self.args['scene']
            if self.args['scene'].startswith('scene.')
            else f'scene.{self.args["scene"]}'
        )
        self.scene_entity = self.get_entity(self.scene_entity)

        self.listen_event(
            self.event_callback, event='call_service', domain='scene', service='turn_on'
        )
        self.log(f'Waiting for {self.scene_entity.friendly_name} to activate')

    async def event_callback(self, event_name, data, cb_args):
        entity_id = data['service_data']['entity_id']
        if entity_id == self.scene_entity.entity_id:
            await self.scene_detected()

    async def scene_detected(self):
        self.log(f'Detected scene activation: {self.scene_entity.friendly_name}')


class MotionCanceller(SceneDetector):
    async def scene_detected(self):
        await super().scene_detected()
        app: Motion = await self.get_app(self.args['app'])
        all_callbacks = await self.get_callback_entries()
        app_callbacks = all_callbacks[app.name]

        for handle, info in app_callbacks.items():
            if info['entity'] == app.sensor.entity_id and 'new=off' in info['kwargs']:
                self.log(f'Cancelling motion callback for {app.name}')
                await self.AD.state.cancel_state_callback(handle, app.name)
                # self.log(json.dumps(info, indent=4))
                break
        else:
            self.log('Did not cancel anything', level='WARNING')