Ticket #8: autodiscovery

diff --git a/generate-cert.sh b/generate-cert.sh
index 4daf76a..ce28e03 100755
--- a/generate-cert.sh
+++ b/generate-cert.sh
@@ -5,4 +5,5 @@ if [ ! -x /usr/bin/certtool ] ; then
 fi
 
 certtool --generate-privkey --outfile ca-key.pem
-certtool --generate-self-signed --load-privkey ca-key.pem --outfile ca-cert.pem
\ No newline at end of file
+certtool --generate-self-signed --load-privkey ca-key.pem --outfile ca-cert.pem
+
diff --git a/manage.py b/manage.py
old mode 100644
new mode 100755
diff --git a/pydra_server/cluster/master.py b/pydra_server/cluster/master.py
index d782e23..59a54c2 100755
--- a/pydra_server/cluster/master.py
+++ b/pydra_server/cluster/master.py
@@ -46,6 +46,12 @@ import datetime
 
 from threading import Lock
 
+# should be executed before any other reactor stuff to prevent from using non
+# glib2 event loop which we need for dbus
+
+from twisted.internet import glib2reactor
+glib2reactor.install()
+
 from zope.interface import implements
 from twisted.cred import portal, checkers
 from twisted.spread import pb
@@ -56,6 +62,9 @@ from twisted.web import server, resource
 from twisted.cred import credentials
 from django.utils import simplejson
 import settings
+import dbus, avahi
+from dbus import DBusException
+from dbus.mainloop.glib import DBusGMainLoop
 
 from pydra_server.models import Node, TaskInstance
 from pydra_server.cluster.constants import *
@@ -143,6 +152,42 @@ class Master(object):
 
         self.host = 'localhost'
         self.port = 18800
+        self.autodiscovery()
+
+    def autodiscovery(self, callback=None):
+        """shamelessly stolen from http://avahi.org/wiki/PythonBrowseExample
+        """
+        def service_resolved(*args):
+            if not  Node.objects.filter(host=args[7], port=args[8]):
+                node = Node.objects.create(host=args[7], port=args[8])
+
+        def print_error(*args):
+            logger.info("Couldn't resolve avahi name: %s" % args)
+
+        def myhandler(interface, protocol, name, stype, domain, flags):
+
+            if flags & avahi.LOOKUP_RESULT_LOCAL:
+                    # local service, skip
+                    pass
+
+            server.ResolveService(interface, protocol, name, stype,
+                domain, avahi.PROTO_UNSPEC, dbus.UInt32(0),
+                reply_handler=service_resolved, error_handler=print_error)
+
+        loop = DBusGMainLoop()
+
+        bus = dbus.SystemBus(mainloop=loop)
+
+        server = dbus.Interface( bus.get_object(avahi.DBUS_NAME, '/'),
+                'org.freedesktop.Avahi.Server')
+
+        sbrowser = dbus.Interface(bus.get_object(avahi.DBUS_NAME,
+                server.ServiceBrowserNew(avahi.IF_UNSPEC,
+                    avahi.PROTO_UNSPEC, '_pydra._tcp', 'local', dbus.UInt32(0))),
+                avahi.DBUS_INTERFACE_SERVICE_BROWSER)
+
+        sbrowser.connect_to_signal("ItemNew", myhandler)
+
 
     def get_services(self):
         """
diff --git a/pydra_server/cluster/node.py b/pydra_server/cluster/node.py
index fd37c50..f37609b 100755
--- a/pydra_server/cluster/node.py
+++ b/pydra_server/cluster/node.py
@@ -46,6 +46,7 @@ from twisted.application import service, internet
 
 import os
 from subprocess import Popen
+import platform, dbus, avahi
 from pydra_server.cluster.auth.rsa_auth import load_crypto
 from pydra_server.cluster.auth.master_avatar import MasterAvatar
 
@@ -55,6 +56,46 @@ import settings
 from pydra_server.logging.logger import init_logging
 logger = init_logging(settings.LOG_FILENAME_NODE)
 
+class ZeroconfService:
+    """A simple class to publish a network service with zeroconf using
+    avahi.
+
+    Shamelessly stolen from http://stackp.online.fr/?p=35 
+    """
+
+    def __init__(self, name, port, stype="_http._tcp",
+                 domain="", host="", text=""):
+        self.name = name
+        self.stype = stype
+        self.domain = domain
+        self.host = host
+        self.port = port
+        self.text = text
+
+    def publish(self):
+        bus = dbus.SystemBus()
+        server = dbus.Interface(
+                         bus.get_object(
+                                 avahi.DBUS_NAME,
+                                 avahi.DBUS_PATH_SERVER),
+                        avahi.DBUS_INTERFACE_SERVER)
+
+        g = dbus.Interface(
+                    bus.get_object(avahi.DBUS_NAME,
+                                   server.EntryGroupNew()),
+                    avahi.DBUS_INTERFACE_ENTRY_GROUP)
+
+        g.AddService(avahi.IF_UNSPEC, avahi.PROTO_UNSPEC,dbus.UInt32(0),
+                     self.name, self.stype, self.domain, self.host,
+                     dbus.UInt16(self.port), self.text)
+
+        g.Commit()
+        self.group = g
+
+    def unpublish(self):
+        self.group.Reset()
+
+
 
 class NodeServer:
     """
@@ -80,6 +121,10 @@ class NodeServer:
 
         # get information about the server
         self.determine_info()
+        service = ZeroconfService(name=platform.node(), port=self.port_base,
+            stype="_pydra._tcp")
+        service.publish()
+
 
         logger.info('Node - starting server on port %s' % self.port_base)
 
@@ -221,4 +266,4 @@ node_server = NodeServer()
 
 # attach service
 service = node_server.get_service()
-service.setServiceParent(application)
\ No newline at end of file
+service.setServiceParent(application)