from homeassistant import bootstrap
from homeassistant.loader import get_component
from homeassistant.const import (
    EVENT_HOMEASSISTANT_START, EVENT_HOMEASSISTANT_STOP,
    EVENT_PLATFORM_DISCOVERED, ATTR_SERVICE, ATTR_DISCOVERED)

DOMAIN = "zwave"
DEPENDENCIES = []

CONF_USB_STICK_PATH = "usb_path"
DEFAULT_CONF_USB_STICK_PATH = "/zwaveusbstick"
CONF_DEBUG = "debug"

DISCOVER_SENSORS = "zwave.sensors"

VALUE_SENSOR = 72057594076463104
VALUE_TEMPERATURE = 72057594076479506
VALUE_LUMINANCE = 72057594076479538
VALUE_RELATIVE_HUMIDITY = 72057594076479570
VALUE_BATTERY_LEVEL = 72057594077773825

NETWORK = None


def get_node_value(node, key):
    """ Helper function to get a node value. """
    return node.values[key].data if key in node.values else None


def setup(hass, config):
    """
    Setup Z-wave.
    Will automatically load components to support devices found on the network.
    """
    global NETWORK

    from louie import connect
    from openzwave.option import ZWaveOption
    from openzwave.network import ZWaveNetwork

    # Setup options
    options = ZWaveOption(
        config[DOMAIN].get(CONF_USB_STICK_PATH, DEFAULT_CONF_USB_STICK_PATH),
        user_path=hass.config_dir)

    if config[DOMAIN].get(CONF_DEBUG) == '1':
        options.set_console_output(True)

    options.lock()

    NETWORK = ZWaveNetwork(options, autostart=False)

    def log_all(signal):
        print("")
        print("LOG ALL")
        print(signal)
        print("")
        print("")
        print("")

    connect(log_all, weak=False)

    def zwave_init_done(network):
        """ Called when Z-Wave has initialized. """
        init_sensor = False

        # This should be rewritten more efficient when supporting more types
        for node in network.nodes.values():
            if get_node_value(node, VALUE_SENSOR) and not init_sensor:
                init_sensor = True

                component = get_component('sensor')

                # Ensure component is loaded
                if component.DOMAIN not in hass.components:
                    bootstrap.setup_component(hass, component.DOMAIN, config)

                # Fire discovery event
                hass.bus.fire(EVENT_PLATFORM_DISCOVERED, {
                    ATTR_SERVICE: DISCOVER_SENSORS,
                    ATTR_DISCOVERED: {}
                })

    connect(
        zwave_init_done, ZWaveNetwork.SIGNAL_NETWORK_READY, weak=False)

    def stop_zwave(event):
        """ Stop Z-wave. """
        NETWORK.stop()

    def start_zwave(event):
        """ Called when Home Assistant starts up. """
        NETWORK.start()

        hass.bus.listen_once(EVENT_HOMEASSISTANT_STOP, stop_zwave)

    hass.bus.listen_once(EVENT_HOMEASSISTANT_START, start_zwave)