1 files changed, 173 insertions, 0 deletions
diff --git a/avahi-python/avahi-bookmarks.in b/avahi-python/avahi-bookmarks.in
new file mode 100755
index 0000000..2de9186
--- /dev/null
+++ b/avahi-python/avahi-bookmarks.in
@@ -0,0 +1,173 @@
+#!@PYTHON@
+# -*-python-*-
+# $Id$ 
+
+# This file is part of avahi.
+#
+# avahi is free software; you can redistribute it and/or modify it
+# under the terms of the GNU Lesser General Public License as
+# published by the Free Software Foundation; either version 2 of the
+# License, or (at your option) any later version.
+#
+# avahi is distributed in the hope that it will be useful, but WITHOUT
+# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
+# or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public
+# License for more details.
+#
+# You should have received a copy of the GNU Lesser General Public
+# License along with avahi; if not, write to the Free Software
+# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
+# USA.
+
+import sys, getopt
+
+try:
+    import avahi, gobject, dbus
+except ImportError:
+    print "Sorry, to use this tool you need to install Avahi, pygtk and python-dbus."
+    sys.exit(1)
+
+try:
+    import dbus.glib
+except ImportError:
+    pass
+
+try:
+    from twisted.internet import gtk2reactor
+    gtk2reactor.install()
+    from twisted.internet import reactor
+    from twisted.web import server, resource
+except ImportError:
+    print "Sorry, to use this tool you need to install twisted and twisted.web."
+    sys.exit(1)
+
+urlproto = { "_http._tcp" : "http",  "_https._tcp" : "https", "_ftp._tcp" : "ftp" }
+
+port = 8080
+address = "127.0.0.1"
+use_host_names = False
+domain = "local"
+
+class AvahiBookmarks(resource.Resource):
+    isLeaf = True
+
+    services = {}
+
+    def __init__(self):
+        resource.Resource.__init__(self)
+
+        self.bus = dbus.SystemBus()
+        self.server = dbus.Interface(self.bus.get_object(avahi.DBUS_NAME, avahi.DBUS_PATH_SERVER), avahi.DBUS_INTERFACE_SERVER)
+
+        self.version_string = self.server.GetVersionString()
+
+        self.browse_service_type("_http._tcp")
+
+        # Hurrah! if I enable one of the following lines, python segfaults.
+        #self.browse_service_type("_https._tcp")
+        #self.browse_service_type("_ftp._tcp")
+
+    def browse_service_type(self, stype):
+
+        global domain
+
+        browser = dbus.Interface(self.bus.get_object(avahi.DBUS_NAME, self.server.ServiceBrowserNew(avahi.IF_UNSPEC, avahi.PROTO_UNSPEC, stype, domain, dbus.UInt32(0))), avahi.DBUS_INTERFACE_SERVICE_BROWSER)
+        browser.connect_to_signal('ItemNew', self.new_service)
+        browser.connect_to_signal('ItemRemove', self.remove_service)
+
+    def find_path(self, txt):
+
+        l = avahi.txt_array_to_string_array(txt)
+
+        for k in l:
+            if k[:5] == "path=":
+                if k[5:].startswith("/"):
+                    return k[5:]
+                else:
+                    return "/" + k[5:]
+
+        return "/"
+
+    def render_GET(self, request):
+
+        t = '<html><head><title>Zeroconf Bookmarks</title></head><body><h1>Zeroconf Bookmarks</h1>'
+
+        if len(self.services) == 0:
+            t += '<p>Sorry, no web services have been registered on the local LAN.</p>'
+        else:
+            t += '<ul style="padding: 0px; margin: 20px; list-style-type: none">'
+
+            for k, v in self.services.iteritems():
+            
+                if v[3] == 80:
+                    port = ''
+                else:
+                    port = ':%i' % v[3]
+
+                path = self.find_path(v[4])
+                t += '<li><a href="%s://%s%s%s">%s</a></li>' % (urlproto[k[3]], v[2], port, path, k[2])
+                
+            t += '</ul>'
+        
+        t += '<hr noshade/><p style="font-size: 8; font-family: sans-serif">Served by %s</p></body></html>' % self.version_string
+
+        return t
+
+
+    def new_service(self, interface, protocol, name, type, domain, flags):
+
+        interface, protocol, name, type, domain, host, aprotocol, address, port, txt, flags = self.server.ResolveService(interface, protocol, name, type, domain, avahi.PROTO_UNSPEC, dbus.UInt32(0))
+
+        if use_host_names:
+            h = host
+        else:
+            if aprotocol == avahi.PROTO_INET6:
+                h = "[" + address + "]"
+            else:
+                h = address
+
+        self.services[(interface, protocol, name, type, domain)] = (host, aprotocol, h, port, txt)
+
+    def remove_service(self, interface, protocol, name, type, domain):
+        del self.services[(interface, protocol, name, type, domain)]
+
+
+def usage(retval = 0):
+    print "%s [options]\n" % sys.argv[0]
+    print "   -h --help             Show this help"
+    print "   -p --port PORT        Specify the port to use (default %u)" % port
+    print "   -a --address ADDRESS  Specify the address to bind to (default %s)" % address
+    print "   -H --host-names       Show all services, regardless of the type"
+    print "   -d --domain DOMAIN    Specify the domain to browse" 
+    sys.exit(retval)
+
+try:
+    opts, args = getopt.getopt(sys.argv[1:], "hp:a:Hd:", ["help", "port=", "address=", "host-names", "domain="])
+except getopt.GetoptError:
+    usage(2)
+
+for o, a in opts:
+    if o in ("-h", "--help"):
+        usage()
+
+    if o in ("-p", "--port"):
+        port = int(a)
+
+    if o in ("-a", "--address"):
+        address = a
+
+    if o in ("-H", "--host-names"):
+        use_host_names = True
+
+    if o in ("-d", "--domain"):
+        domain = a
+    
+site = server.Site(AvahiBookmarks())
+reactor.listenTCP(port, site, interface=address)
+
+print "Now point your web browser to http://%s:%u/!" % (address, port)
+
+try:
+    reactor.run()
+except KeyboardInterrupt, k:
+    pass