#!/usr/bin/python3
import hassapi as hass
import inspect
from threading import Timer

###############
# Action Tree #
###############

"""
1: States
    1: Lights
        1: Kitchen Lights (Toggle)
            1: Off
            2: 10%
            3: 55%
            4: 100%
        2: Main Room Lights (Toggle)
            [...]
        3: Living Room Lights (Toggle)
            [...]
        4: Bedroom Lights (Toggle)
            [...]
    2: A/C
        2: Main Room AC
            1: Power
                1: On
                2: On
                3: Off
                4: Off
            2: Mode
                1: Cool
                2: Heat
                3: Fan
            3: Fan Level
                1: Auto
                2: Quiet
                3: Medium
                4: High
            4: Temperature
                1: 72
                2: 74
                3: 76
                4: 78
        3: Living Room AC
            [...]
        4: Bedroom AC
            [...]
    3: Fans
        1: Kitchen Fan
            1: Off
            2: 1
            3: 2
            4: 3
        3: Living Room Fan
            [...]
        4: Bedroom Fan
            [...]
2: Music
    1: Join/unjoin speakers
        1. Unjoin speakers
        # Select master
        3. Living Room Sonos
        4. Bedroom Sonos
    2: Play Music
        3: Play on Living Room Sonos
            1: Spotify: Liked Songs
            2: Spotify: Getting Things Done
        4: Play on Bedroom Sonos
            1: Spotify: Liked Songs
            2: Spotify: Chill Sleep
    3: Media Controls
        3: Living Room Sonos
            1. Pause
            2. Play
            3. Previous
            4. Next
        4: Bedroom Sonos
            [...]
    4. Options:
        3: Living Room Sonos
            2: Crossfade
                1: Off
                2: On
            3: Shuffle
                1: Off
                2: On
            4: Repeat
                1: Off
                2: On
        4: Bedroom Sonos
            [...]
"""

tree = {
    1: { # State
        1: { # Lights
            1: { # Kitchen
                "function": lambda hass: hass.call_service("light/toggle",entity_id="light.kitchen_lights"),
                "description": "Kitchen Lights: Toggle",
                1: {
                    "function": lambda hass: set_brightness(hass,"light.kitchen_lights",0),
                    "description": "Kitchen Lights: 0%"
                },
                2: {
                    "function": lambda hass: set_brightness(hass,"light.kitchen_lights",10),
                    "description": "Kitchen Lights: 10%"
                },
                3: {
                    "function": lambda hass: set_brightness(hass,"light.kitchen_lights",55),
                    "description": "Kitchen Lights: 55%"
                },
                4: {
                    "function": lambda hass: set_brightness(hass,"light.kitchen_lights",100),
                    "description": "Kitchen Lights: 100%"
                }
            },
            2: { # Main Room
                "function": lambda hass: hass.call_service("light/toggle",entity_id="light.mainroom_lights"),
                "description": "Main Room Lights: Toggle",
                1: {
                    "function": lambda hass: set_brightness(hass,"light.mainroom_lights",0),
                    "description": "Main Room Lights: 0%"
                },
                2: {
                    "function": lambda hass: set_brightness(hass,"light.mainroom_lights",10),
                    "description": "Main Room Lights: 10%"
                },
                3: {
                    "function": lambda hass: set_brightness(hass,"light.mainroom_lights",55),
                    "description": "Main Room Lights: 55%"
                },
                4: {
                    "function": lambda hass: set_brightness(hass,"light.mainroom_lights",100),
                    "description": "Main Room Lights: 100%"
                }
            },
            3: { # Living Room
                "function": lambda hass: hass.call_service("light/toggle",entity_id="light.livingroom_lights"),
                "description": "Living Room Lights: Toggle",
                1: {
                    "function": lambda hass: set_brightness(hass,"light.livingroom_lights",0),
                    "description": "Living Room Lights: 0%"
                },
                2: {
                    "function": lambda hass: set_brightness(hass,"light.livingroom_lights",10),
                    "description": "Living Room Lights: 10%"
                },
                3: {
                    "function": lambda hass: set_brightness(hass,"light.livingroom_lights",55),
                    "description": "Living Room Lights: 55%"
                },
                4: {
                    "function": lambda hass: set_brightness(hass,"light.livingroom_lights",100),
                    "description": "Living Room Lights: 100%"
                }
            },
            4: { # Bedroom
                "function": lambda hass: hass.call_service("light/toggle",entity_id="light.bedroom_lights"),
                "description": "Bedroom Lights: Toggle",
                1: {
                    "function": lambda hass: set_brightness(hass,"light.bedroom_lights",0),
                    "description": "Bedroom Lights: 0%"
                },
                2: {
                    "function": lambda hass: set_brightness(hass,"light.bedroom_lights",10),
                    "description": "Bedroom Lights: 10%"
                },
                3: {
                    "function": lambda hass: set_brightness(hass,"light.bedroom_lights",55),
                    "description": "Bedroom Lights: 55%"
                },
                4: {
                    "function": lambda hass: set_brightness(hass,"light.bedroom_lights",100),
                    "description": "Bedroom Lights: 100%"
                }
            }
        },
        2: { # A/C
            2: { # Main Room
                1: { # Power
                    1: {
                        "function": lambda hass: hass.turn_on("climate.mainroom_ac"),
                        "description": "Main Room A/C Power: On"
                    },
                    2: {
                        "function": lambda hass: hass.turn_on("climate.mainroom_ac"),
                        "description": "Main Room A/C Power: On"
                    },
                    3: {
                        "function": lambda hass: hass.turn_off("climate.mainroom_ac"),
                        "description": "Main Room A/C Power: Off"
                    },
                    4: {
                        "function": lambda hass: hass.turn_off("climate.mainroom_ac"),
                        "description": "Main Room A/C Power: Off"
                    },
                },
                2: { # Mode
                    1: {
                        "function": lambda hass: gree_set_mode(hass,"mainroom_ac","cool"),
                        "description": "Main Room A/C Mode: Cool"
                    },
                    2: {
                        "function": lambda hass: gree_set_mode(hass,"mainroom_ac","heat"),
                        "description": "Main Room A/C Mode: Heat"
                    },
                    3: {
                        "function": lambda hass: gree_set_mode(hass,"mainroom_ac","fan_only"),
                        "description": "Main Room A/C Mode: Fan"
                    }
                },
                3: { # Fan Level
                    1: {
                        "function": lambda hass: gree_set_fan_level(hass,"mainroom_ac","auto"),
                        "description": "Main Room A/C Fan Level: Auto"
                    },
                    2: {
                        "function": lambda hass: gree_set_fan_level(hass,"mainroom_ac","quiet"),
                        "description": "Main Room A/C Fan Level: Quiet"
                    },
                    3: {
                        "function": lambda hass: gree_set_fan_level(hass,"mainroom_ac","medium"),
                        "description": "Main Room A/C Fan Level: Medium"
                    },
                    4: {
                        "function": lambda hass: gree_set_fan_level(hass,"mainroom_ac","high"),
                        "description": "Main Room A/C Fan Level: High"
                    }
                },
                4: { # Temperature
                    1: {
                        "function": lambda hass: set_temperature(hass,"mainroom_ac",72),
                        "description": "Main Room A/C Temperature: 72"
                    },
                    2: {
                        "function": lambda hass: set_temperature(hass,"mainroom_ac",74),
                        "description": "Main Room A/C Temperature: 74"
                    },
                    3: {
                        "function": lambda hass: set_temperature(hass,"mainroom_ac",76),
                        "description": "Main Room A/C Temperature: 76"
                    },
                    4: {
                        "function": lambda hass: set_temperature(hass,"mainroom_ac",78),
                        "description": "Main Room A/C Temperature: 78"
                    }
                }
            },
            3: { # Living Room
                1: { # Power
                    1: {
                        "function": lambda hass: hass.turn_on("climate.livingroom_ac"),
                        "description": "Living Room A/C Power: On"
                    },
                    2: {
                        "function": lambda hass: hass.turn_on("climate.livingroom_ac"),
                        "description": "Living Room A/C Power: On"
                    },
                    3: {
                        "function": lambda hass: hass.turn_off("climate.livingroom_ac"),
                        "description": "Living Room A/C Power: Off"
                    },
                    4: {
                        "function": lambda hass: hass.turn_off("climate.livingroom_ac"),
                        "description": "Living Room A/C Power: Off"
                    },
                },
                2: { # Mode
                    1: {
                        "function": lambda hass: gree_set_mode(hass,"livingroom_ac","cool"),
                        "description": "Living Room A/C Mode: Cool"
                    },
                    2: {
                        "function": lambda hass: gree_set_mode(hass,"livingroom_ac","heat"),
                        "description": "Living Room A/C Mode: Heat"
                    },
                    3: {
                        "function": lambda hass: gree_set_mode(hass,"livingroom_ac","fan_only"),
                        "description": "Living Room A/C Mode: Fan"
                    }
                },
                3: { # Fan Level
                    1: {
                        "function": lambda hass: gree_set_fan_level(hass,"livingroom_ac","auto"),
                        "description": "Living Room A/C Fan Level: Auto"
                    },
                    2: {
                        "function": lambda hass: gree_set_fan_level(hass,"livingroom_ac","quiet"),
                        "description": "Living Room A/C Fan Level: Quiet"
                    },
                    3: {
                        "function": lambda hass: gree_set_fan_level(hass,"livingroom_ac","medium"),
                        "description": "Living Room A/C Fan Level: Medium"
                    },
                    4: {
                        "function": lambda hass: gree_set_fan_level(hass,"livingroom_ac","high"),
                        "description": "Living Room A/C Fan Level: High"
                    }
                },
                4: { # Temperature
                    1: {
                        "function": lambda hass: set_temperature(hass,"livingroom_ac",72),
                        "description": "Living Room A/C Temperature: 72"
                    },
                    2: {
                        "function": lambda hass: set_temperature(hass,"livingroom_ac",74),
                        "description": "Living Room A/C Temperature: 74"
                    },
                    3: {
                        "function": lambda hass: set_temperature(hass,"livingroom_ac",76),
                        "description": "Living Room A/C Temperature: 76"
                    },
                    4: {
                        "function": lambda hass: set_temperature(hass,"livingroom_ac",78),
                        "description": "Living Room A/C Temperature: 78"
                    }
                }
            },
            4: { # Bedroom
                1: { # Power
                    1: {
                        "function": lambda hass: hass.turn_on("climate.bedroom_ac"),
                        "description": "Bedroom A/C Power: On"
                    },
                    2: {
                        "function": lambda hass: hass.turn_on("climate.bedroom_ac"),
                        "description": "Bedroom A/C Power: On"
                    },
                    3: {
                        "function": lambda hass: hass.turn_off("climate.bedroom_ac"),
                        "description": "Bedroom A/C Power: Off"
                    },
                    4: {
                        "function": lambda hass: hass.turn_off("climate.bedroom_ac"),
                        "description": "Bedroom A/C Power: Off"
                    },
                },
                2: { # Mode
                    1: {
                        "function": lambda hass: gree_set_mode(hass,"bedroom_ac","cool"),
                        "description": "Bedroom A/C Mode: Cool"
                    },
                    2: {
                        "function": lambda hass: gree_set_mode(hass,"bedroom_ac","heat"),
                        "description": "Bedroom A/C Mode: Heat"
                    },
                    3: {
                        "function": lambda hass: gree_set_mode(hass,"bedroom_ac","fan_only"),
                        "description": "Bedroom A/C Mode: Fan"
                    }
                },
                3: { # Fan Level
                    1: {
                        "function": lambda hass: gree_set_fan_level(hass,"bedroom_ac","auto"),
                        "description": "Bedroom A/C Fan Level: Auto"
                    },
                    2: {
                        "function": lambda hass: gree_set_fan_level(hass,"bedroom_ac","quiet"),
                        "description": "Bedroom A/C Fan Level: Quiet"
                    },
                    3: {
                        "function": lambda hass: gree_set_fan_level(hass,"bedroom_ac","medium"),
                        "description": "Bedroom A/C Fan Level: Medium"
                    },
                    4: {
                        "function": lambda hass: gree_set_fan_level(hass,"bedroom_ac","high"),
                        "description": "Bedroom A/C Fan Level: High"
                    }
                },
                4: { # Temperature
                    1: {
                        "function": lambda hass: set_temperature(hass,"bedroom_ac",72),
                        "description": "Bedroom A/C Temperature: 72"
                    },
                    2: {
                        "function": lambda hass: set_temperature(hass,"bedroom_ac",74),
                        "description": "Bedroom A/C Temperature: 74"
                    },
                    3: {
                        "function": lambda hass: set_temperature(hass,"bedroom_ac",76),
                        "description": "Bedroom A/C Temperature: 76"
                    },
                    4: {
                        "function": lambda hass: set_temperature(hass,"bedroom_ac",78),
                        "description": "Bedroom A/C Temperature: 78"
                    }
                }
            }
        },
        3: { # Fans
            1: { # Kitchen
                1: {"function": lambda hass: hass.set_value("input_number.kitchen_fan",0)},
                2: {"function": lambda hass: hass.set_value("input_number.kitchen_fan",1)},
                3: {"function": lambda hass: hass.set_value("input_number.kitchen_fan",2)},
                4: {"function": lambda hass: hass.set_value("input_number.kitchen_fan",3)}
            },
            3: { # Living Room
                1: {"function": lambda hass: hass.set_value("input_number.livingroom_fan",0)},
                2: {"function": lambda hass: hass.set_value("input_number.livingroom_fan",1)},
                3: {"function": lambda hass: hass.set_value("input_number.livingroom_fan",2)},
                4: {"function": lambda hass: hass.set_value("input_number.livingroom_fan",3)}
            },
            4: { # Bedroom
                1: {"function": lambda hass: hass.set_value("input_number.bedroom_fan",0)},
                2: {"function": lambda hass: hass.set_value("input_number.bedroom_fan",1)},
                3: {"function": lambda hass: hass.set_value("input_number.bedroom_fan",2)},
                4: {"function": lambda hass: hass.set_value("input_number.bedroom_fan",3)}
            }
        }
    },
    2: { # Music
        1: { # Join/unjoin
            1: {"function": lambda hass: hass.call_service("sonos/unjoin",
                entity_id=["media_player.livingroom_sonos","media_player.bedroom_sonos"])},
            3: {"function": lambda hass: hass.call_service("sonos/join",
                master="media_player.livingroom_sonos",entity_id="media_player.bedroom_sonos")},
            4: {"function": lambda hass: hass.call_service("sonos/join",
                master="media_player.livingroom_sonos",entity_id="media_player.bedroom_sonos")},
        },
        2: { # Play Music
            3: { # Living Room Sonos
                1: {"function": lambda hass: select_source(hass,"media_player.livingroom_sonos","Songs")},
                2: {"function": lambda hass: select_source(hass,"media_player.livingroom_sonos","Getting Things Done")}
            },
            4: { # Bedroom Sonos
                1: {"function": lambda hass: select_source(hass,"media_player.bedroom_sonos","Songs")},
                2: {"function": lambda hass: select_source(hass,"media_player.bedroom_sonos","Chill sleep")}
            }
        },
        3: { # Media Controls
            3: { # Living Room Sonos
                1: {"function": lambda hass: hass.call_service("media_player/media_pause",
                    entity_id="media_player.livingroom_sonos")},
                2: {"function": lambda hass: hass.call_service("media_player/media_play",
                    entity_id="media_player.livingroom_sonos")},
                3: {"function": lambda hass: hass.call_service("media_player/media_previous_track",
                    entity_id="media_player.livingroom_sonos")},
                4: {"function": lambda hass: hass.call_service("media_player/media_next_track",
                    entity_id="media_player.livingroom_sonos")}
            },
            4: { # Bedroom Sonos
                1: {"function": lambda hass: hass.call_service("media_player/media_pause",
                    entity_id="media_player.bedroom_sonos")},
                2: {"function": lambda hass: hass.call_service("media_player/media_play",
                    entity_id="media_player.bedroom_sonos")},
                3: {"function": lambda hass: hass.call_service("media_player/media_previous_track",
                    entity_id="media_player.bedroom_sonos")},
                4: {"function": lambda hass: hass.call_service("media_player/media_next_track",
                    entity_id="media_player.bedroom_sonos")}
            }
        },
        4: { # Options
            3: { # Living Room Sonos
                2: { # Crossfade
                    1: {"function": lambda hass: hass.turn_off("switch.livingroom_sonos_crossfade"),
                        "description": "Living Room Sonos: Crossfade: Off"},
                    2: {"function": lambda hass: hass.turn_on("switch.livingroom_sonos_crossfade"),
                        "description": "Living Room Sonos: Crossfade: On"}
                },
                3: { # Shuffle
                    1: {"function": lambda hass: hass.call_service("media_player/shuffle_set",
                        entity_id="media_player.livingroom_sonos",shuffle=False),
                        "description": "Living Room Sonos: Shuffle: Off"},
                    2: {"function": lambda hass: hass.call_service("media_player/shuffle_set",
                        entity_id="media_player.livingroom_sonos",shuffle=True),
                        "description": "Living Room Sonos: Shuffle: On"},
                },
                4: { # Repeat
                    1: {"function": lambda hass: hass.call_service("media_player/repeat_set",
                        entity_id="media_player.livingroom_sonos",repeat="off"),
                        "description": "Living Room Sonos: Repeat: Off"},
                    2: {"function": lambda hass: hass.call_service("media_player/repeat_set",
                        entity_id="media_player.livingroom_sonos",repeat="all"),
                        "description": "Living Room Sonos: Repeat: On"},
                }
            },
            4: { # Bedroom Sonos
                2: { # Crossfade
                    1: {"function": lambda hass: hass.turn_off("switch.bedroom_sonos_crossfade"),
                        "description": "Bedroom Sonos: Crossfade: Off"},
                    2: {"function": lambda hass: hass.turn_on("switch.bedroom_sonos_crossfade"),
                        "description": "Bedroom Sonos: Crossfade: On"}
                },
                3: { # Shuffle
                    1: {"function": lambda hass: hass.call_service("media_player/shuffle_set",
                        entity_id="media_player.bedroom_sonos",shuffle=False),
                        "description": "Bedroom Sonos: Shuffle: Off"},
                    2: {"function": lambda hass: hass.call_service("media_player/shuffle_set",
                        entity_id="media_player.bedroom_sonos",shuffle=True),
                        "description": "Bedroom Sonos: Shuffle: On"},
                },
                4: { # Repeat
                    1: {"function": lambda hass: hass.call_service("media_player/repeat_set",
                        entity_id="media_player.bedroom_sonos",repeat="off"),
                        "description": "Bedroom Sonos: Repeat: Off"},
                    2: {"function": lambda hass: hass.call_service("media_player/repeat_set",
                        entity_id="media_player.bedroom_sonos",repeat="all"),
                        "description": "Bedroom Sonos: Repeat: On"},
                }
            }
        }
    }
}

###########################
# Action Helper Functions #
###########################

def set_brightness(hass, device, brightness):
    # Convert brightness from percent to 0-255
    brightness = brightness * 255 / 100;
    hass.turn_on(device, brightness=brightness);

def set_temperature(hass, device, temperature):
    hass.call_service("climate/set_temperature",entity_id="climate."+device,temperature=temperature);

def gree_set_mode(hass, device, mode):
    hass.call_service("climate/set_hvac_mode",entity_id="climate."+device,hvac_mode=mode);

def gree_set_fan_level(hass, device, level):
    if (level == "quiet"):
        hass.turn_on(f"switch.{device}_quiet");
    else: 
        hass.turn_off(f"switch.{device}_quiet");
        hass.call_service("climate/set_fan_mode",entity_id="climate."+device,fan_mode=level);

def select_source(hass, device, source):
    hass.call_service("media_player/select_source",entity_id=device,source=source)

##################
# Sequence Logic #
##################

# Short presses will be recorded into a sequence array. Long presses are reserved for
# actions directly in hass for now; use KeyReleased event to capture them. A sequence 
# can be ended either immediately by recognizing a full sequence or by TIMEOUT seconds 
# passing. (The former is NYI but the code is capable as is.)

# - self.currentSequence contains an array of keys representing a sequence that is
#   actively being entered.
# - self.tot contains a timer for the process_sequence function, ending sequence entry.

TIMEOUT=1

class WallmotePlumbing(hass.Hass):
    def initialize(self):
        self.currentSequence = [];
        self.listen_event(self.onevent, "zwave_js_value_notification");

    def onevent(self, _, event,  kwargs):
        if (event.get('value') == 'KeyHeldDown'): return;
        if (event.get('device_id') != '3f7add36f94839b1446886b5dab35e6d'): return;
        key = int(event.get('property_key'));
        # We don't get a KeyPressed event at all for long presses
        evt = event.get('value');
        if (evt == "KeyPressed"): self.shortpress(key);

    def shortpress(self, key):
        self.currentSequence.append(key);
        self.reset_tot();

    def reset_tot(self):
        if (hasattr(self,"tot")): self.tot.cancel();
        self.tot = Timer(TIMEOUT,self.process_sequence);
        self.tot.start();

    def process_sequence(self):
        self.tot.cancel(); # So process_sequence can be called directly
        sequence = self.currentSequence;
        self.currentSequence = [];
        self.log(sequence);
        self.run_action(sequence);

    def run_action(self, sequence, tree=tree):
        if (len(sequence) > 0):
            # Deeper into the rabbit hole
            category = sequence.pop(0);
            if (category in tree): self.run_action(sequence,tree[category]);
            else: self.log("No match");
        else:
            # Perform an action from here and log the description
            if ("function" in tree): tree["function"](self);
            elif ("service" in tree): self.call_service(tree["service"]);
            else:
                self.log("No match");
                return;
            if ("description" in tree): self.log(tree["description"]);
            else: self.log(inspect.getsource(tree["function"]).strip());