From f683e2e7091d6dd252db034a357fa9ec01598180 Mon Sep 17 00:00:00 2001
From: Giovanni Harting <giovanni.harting@touchdata.net>
Date: Fri, 28 Aug 2015 18:26:21 +0200
Subject: [PATCH] added zeroconf service announcement closes #5

---
 ledd/daemon.py | 24 ++++++++++++++++++++++++
 setup.py       |  2 +-
 2 files changed, 25 insertions(+), 1 deletion(-)

diff --git a/ledd/daemon.py b/ledd/daemon.py
index 3839a51..4e057ae 100644
--- a/ledd/daemon.py
+++ b/ledd/daemon.py
@@ -26,6 +26,8 @@ import asyncio
 
 import spectra
 
+from zeroconf import Zeroconf, ServiceInfo
+
 from ledd import controller, VERSION
 from ledd.decorators import ledd_protocol
 from ledd.stripe import Stripe
@@ -46,6 +48,7 @@ class Daemon:
         Daemon.instance = self
 
         try:
+            # read config
             self.config = configparser.ConfigParser()
             try:
                 with open('ledd.config', 'w+') as f:
@@ -53,6 +56,11 @@ class Daemon:
             except FileNotFoundError:
                 log.info("No config file found!")
 
+            # create zeroconf service info
+            self.zinfo = ServiceInfo("_ledd._tcp", "LedD Daemon",
+                                     port=self.config.get(self.daemonSection, 'port', fallback=1425))
+
+            # SQL init
             self.sqldb = sqlite3.connect(self.config.get(self.databaseSection, 'name', fallback='ledd.sqlite'))
             self.sqldb.row_factory = sqlite3.Row
 
@@ -61,10 +69,15 @@ class Daemon:
 
             self.sqldb.commit()
 
+            # init controllers from db
             self.controllers = controller.Controller.from_db(self.sqldb)
             log.debug(self.controllers)
             logging.getLogger("asyncio").setLevel(logging.DEBUG)
 
+            # announce server to network
+            self.register_zeroconf()
+
+            # main loop
             self.loop = asyncio.get_event_loop()
             coro = self.loop.create_server(LedDProtocol,
                                            self.config.get(self.daemonSection, 'host', fallback='0.0.0.0'),
@@ -73,6 +86,7 @@ class Daemon:
             self.loop.run_forever()
         except (KeyboardInterrupt, SystemExit):
             log.info("Exiting")
+            self.deregister_zeroconf()
             self.sqldb.close()
             self.server.close()
             self.loop.run_until_complete(self.server.wait_closed())
@@ -112,6 +126,16 @@ class Daemon:
             c.close()
         self.check_db()
 
+    def register_zeroconf(self):
+        zeroconf = Zeroconf()
+        zeroconf.register_service(self.zinfo)
+        log.info("Registered ledd daemon with zeroconf")
+
+    def deregister_zeroconf(self):
+        zeroconf = Zeroconf()
+        zeroconf.unregister_service(self.zinfo)
+        log.info("Unregistered ledd daemon with zeroconf")
+
     @ledd_protocol(protocol)
     def set_color(self, req_json):
         """
diff --git a/setup.py b/setup.py
index e89ed7d..5690e1b 100644
--- a/setup.py
+++ b/setup.py
@@ -25,6 +25,6 @@ setup(name='LedD',
       license='GPLv3',
       packages=['ledd'],
       install_requires=[
-            'nose', 'spectra',
+          'nose', 'spectra', 'zeroconf',
       ],
       zip_safe=False)