git.meshlink.io Git - catta/blob - avahi-utils/avahi-bookmarks.in

   1 #!@PYTHON@
   2 # -*-python-*-
   3 # $Id$
   4
   5 # This file is part of avahi.
   6 #
   7 # avahi is free software; you can redistribute it and/or modify it
   8 # under the terms of the GNU Lesser General Public License as
   9 # published by the Free Software Foundation; either version 2 of the
  10 # License, or (at your option) any later version.
  11 #
  12 # avahi is distributed in the hope that it will be useful, but WITHOUT
  13 # ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
  14 # or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public
  15 # License for more details.
  16 #
  17 # You should have received a copy of the GNU Lesser General Public
  18 # License along with avahi; if not, write to the Free Software
  19 # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
  20 # USA.
  21
  22 import sys
  23
  24 try:
  25     import avahi, gobject, dbus
  26 except ImportError:
  27     print "Sorry, to use this tool you need to install Avahi, pygtk and python-dbus."
  28     sys.exit(1)
  29
  30 try:
  31     import dbus.glib
  32 except ImportError:
  33     pass
  34
  35 try:
  36     from twisted.internet import gtk2reactor
  37     gtk2reactor.install()
  38     from twisted.internet import reactor
  39     from twisted.web import server, resource
  40 except ImportError:
  41     print "Sorry, to use this tool you need to install twisted and twisted.web."
  42     sys.exit(1)
  43
  44 urlproto = { "_http._tcp" : "http",  "_https._tcp" : "https", "_ftp._tcp" : "ftp" }
  45
  46 class AvahiBookmarks(resource.Resource):
  47     isLeaf = True
  48
  49     services = {}
  50
  51     def __init__(self):
  52         resource.Resource.__init__(self)
  53
  54         self.bus = dbus.SystemBus()
  55         self.server = dbus.Interface(self.bus.get_object(avahi.DBUS_NAME, avahi.DBUS_PATH_SERVER), avahi.DBUS_INTERFACE_SERVER)
  56
  57         self.version_string = self.server.GetVersionString()
  58
  59         self.browse_service_type("_http._tcp")
  60
  61         # Hurrah! if I enable one of the following lines, python segfaults.
  62         #self.browse_service_type("_https._tcp")
  63         #self.browse_service_type("_ftp._tcp")
  64
  65     def browse_service_type(self, stype):
  66
  67         browser = dbus.Interface(self.bus.get_object(avahi.DBUS_NAME, self.server.ServiceBrowserNew(avahi.IF_UNSPEC, avahi.PROTO_UNSPEC, stype, "local")), avahi.DBUS_INTERFACE_SERVICE_BROWSER)
  68         browser.connect_to_signal('ItemNew', self.new_service)
  69         browser.connect_to_signal('ItemRemove', self.remove_service)
  70
  71     def find_path(self, txt):
  72
  73         l = avahi.txt_array_to_string_array(txt)
  74
  75         for k in l:
  76             if k[:5] == "path=":
  77                 return k[5:]
  78
  79         return "/"
  80
  81     def render_GET(self, request):
  82
  83         t = '<html><head><title>Zeroconf Bookmarks</title></head><body><h1>Zeroconf Bookmarks</h1>'
  84
  85         if len(self.services) == 0:
  86             t += '<p>Sorry, no web services have been registered on the local LAN.</p>'
  87         else:
  88             t += '<ul style="padding: 0px; margin: 20px; list-style-type: none">'
  89
  90             for k, v in self.services.iteritems():
  91
  92                 if v[3] == 80:
  93                     port = ''
  94                 else:
  95                     port = ':%i' % v[3]
  96
  97                 path = self.find_path(v[4])
  98
  99                 if v[1] == avahi.PROTO_INET6:
 100                     ip = "[" + v[2] + "]"
 101                 else:
 102                     ip = v[2]
 103
 104                 t += '<li><a href="%s://%s%s%s">%s</a></li>' % (urlproto[k[3]], ip, port, path, k[2])
 105
 106             t += '</ul>'
 107
 108         t += '<hr noshade/><p style="font-size: 8; font-family: sans-serif">Served by %s</p></body></html>' % self.version_string
 109
 110         return t
 111
 112
 113     def new_service(self, interface, protocol, name, type, domain):
 114
 115         interface, protocol, name, type, domain, host, aprotocol, address, port, txt = self.server.ResolveService(interface, protocol, name, type, domain, avahi.PROTO_UNSPEC)
 116
 117         self.services[(interface, protocol, name, type, domain)] = (host, aprotocol, address, port, txt)
 118
 119     def remove_service(self, interface, protocol, name, type, domain):
 120         del self.services[(interface, protocol, name, type, domain)]
 121
 122
 123 port = 8080
 124
 125 if __name__ == '__main__':
 126     site = server.Site(AvahiBookmarks())
 127     reactor.listenTCP(port, site, interface="127.0.0.1")
 128
 129     print "Now point your web browser to http://localhost:%u/!" % port
 130
 131     try:
 132         reactor.run()
 133     except KeyboardInterrupt, k:
 134         pass