From c8ad9c4daa7cbc2f596a2a3c00278d6539cd79b3 Mon Sep 17 00:00:00 2001
From: Marco Orovecchia <Oro@users.noreply.github.com>
Date: Sun, 20 May 2018 20:53:57 +0200
Subject: [PATCH] Add auto discovery for nanoleaf aurora lights (#14301)

* auto discovery added for nanoleaf aurora lights

* changes requested by review

* visual indentation

* line too long

* hide autocreated config
---
 homeassistant/components/discovery.py         |  1 +
 .../components/light/nanoleaf_aurora.py       | 47 ++++++++++++++++---
 2 files changed, 41 insertions(+), 7 deletions(-)

diff --git a/homeassistant/components/discovery.py b/homeassistant/components/discovery.py
index a24e82da106..69447b81cd4 100644
--- a/homeassistant/components/discovery.py
+++ b/homeassistant/components/discovery.py
@@ -83,6 +83,7 @@ SERVICE_HANDLERS = {
     'songpal': ('media_player', 'songpal'),
     'kodi': ('media_player', 'kodi'),
     'volumio': ('media_player', 'volumio'),
+    'nanoleaf_aurora': ('light', 'nanoleaf_aurora'),
 }
 
 OPTIONAL_SERVICE_HANDLERS = {
diff --git a/homeassistant/components/light/nanoleaf_aurora.py b/homeassistant/components/light/nanoleaf_aurora.py
index 99c07166037..8b0b7c053c8 100644
--- a/homeassistant/components/light/nanoleaf_aurora.py
+++ b/homeassistant/components/light/nanoleaf_aurora.py
@@ -17,6 +17,7 @@ import homeassistant.helpers.config_validation as cv
 from homeassistant.util import color as color_util
 from homeassistant.util.color import \
     color_temperature_mired_to_kelvin as mired_to_kelvin
+from homeassistant.util.json import load_json, save_json
 
 REQUIREMENTS = ['nanoleaf==0.4.1']
 
@@ -24,6 +25,10 @@ _LOGGER = logging.getLogger(__name__)
 
 DEFAULT_NAME = 'Aurora'
 
+DATA_NANOLEAF_AURORA = 'nanoleaf_aurora'
+
+CONFIG_FILE = '.nanoleaf_aurora.conf'
+
 ICON = 'mdi:triangle-outline'
 
 SUPPORT_AURORA = (SUPPORT_BRIGHTNESS | SUPPORT_COLOR_TEMP | SUPPORT_EFFECT |
@@ -39,31 +44,59 @@ PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend({
 def setup_platform(hass, config, add_devices, discovery_info=None):
     """Set up the Nanoleaf Aurora device."""
     import nanoleaf
-    host = config.get(CONF_HOST)
-    name = config.get(CONF_NAME)
-    token = config.get(CONF_TOKEN)
+    import nanoleaf.setup
+    if DATA_NANOLEAF_AURORA not in hass.data:
+        hass.data[DATA_NANOLEAF_AURORA] = dict()
+
+    token = ''
+    if discovery_info is not None:
+        host = discovery_info['host']
+        name = discovery_info['hostname']
+        # if device already exists via config, skip discovery setup
+        if host in hass.data[DATA_NANOLEAF_AURORA]:
+            return
+        _LOGGER.info("Discovered a new Aurora: %s", discovery_info)
+        conf = load_json(hass.config.path(CONFIG_FILE))
+        if conf.get(host, {}).get('token'):
+            token = conf[host]['token']
+    else:
+        host = config[CONF_HOST]
+        name = config[CONF_NAME]
+        token = config[CONF_TOKEN]
+
+    if not token:
+        token = nanoleaf.setup.generate_auth_token(host)
+        if not token:
+            _LOGGER.error("Could not generate the auth token, did you press "
+                          "and hold the power button on %s"
+                          "for 5-7 seconds?", name)
+            return
+        conf = load_json(hass.config.path(CONFIG_FILE))
+        conf[host] = {'token': token}
+        save_json(hass.config.path(CONFIG_FILE), conf)
+
     aurora_light = nanoleaf.Aurora(host, token)
-    aurora_light.hass_name = name
 
     if aurora_light.on is None:
         _LOGGER.error(
             "Could not connect to Nanoleaf Aurora: %s on %s", name, host)
         return
 
-    add_devices([AuroraLight(aurora_light)], True)
+    hass.data[DATA_NANOLEAF_AURORA][host] = aurora_light
+    add_devices([AuroraLight(aurora_light, name)], True)
 
 
 class AuroraLight(Light):
     """Representation of a Nanoleaf Aurora."""
 
-    def __init__(self, light):
+    def __init__(self, light, name):
         """Initialize an Aurora light."""
         self._brightness = None
         self._color_temp = None
         self._effect = None
         self._effects_list = None
         self._light = light
-        self._name = light.hass_name
+        self._name = name
         self._hs_color = None
         self._state = None