Merge lp:~sil/lolocopter/to-protect-and-serve into lp:lolocopter

[Stuart Langridge]

[Stuart Langridge (sil)]

Here I have added the server. No integration work done yet.

[Jono Bacon (jonobacon)]

Looks good to me. Thanks.

review: Approve

=== added file 'lolocopter/lolocopter_server.py'
--- lolocopter/lolocopter_server.py	1970-01-01 00:00:00 +0000
+++ lolocopter/lolocopter_server.py	2009-11-14 17:40:21 +0000
@@ -0,0 +1,94 @@
+import gtk, BaseHTTPServer, threading, gobject, json, shutil, avahi, dbus, os
+
+class MyHandler(BaseHTTPServer.BaseHTTPRequestHandler):
+    counts = 0
+    def do_GET(self):
+        if self.path == "/":
+            self.send_summary()
+        elif self.path == "/loldata":
+            self.send_lolfile()
+        else:
+            self.send_error()
+    def send_summary(self):
+        self.send_response(200)
+        self.send_header("Content-type", "text/plain")
+        self.end_headers()
+        self.wfile.write(json.dumps({
+            "lol_data": "/loldata",
+            "network_clock_udp_port": self.network_clock_udp_port
+        }))
+    def send_error(self):
+        self.send_response(404)
+        self.send_header("Content-type", "text/plain")
+        self.end_headers()
+        self.wfile.write("fail.")
+    def send_lolfile(self):
+        self.send_response(200)
+        self.send_header("Content-type", "application/octet-stream")
+        self.end_headers()
+        fp = open(self.lol_to_advertise)
+        shutil.copyfileobj(fp, self.wfile)
+        fp.close()
+        # tell lolocopter that someone's downloaded the file
+        if self.joiner_callback:
+            # call it in idle so it's on the main thread
+            gobject.idle_add(self.joiner_callback)
+
+class LolocopterServer(object):
+    def __init__(self, network_clock_udp_port,
+            lol_to_advertise, public_name, joiner_callback=None):
+        self.network_clock_udp_port = network_clock_udp_port
+        self.lol_to_advertise = lol_to_advertise
+        self.public_name = public_name
+        self.joiner_callback = joiner_callback
+    
+    def __advertise_service(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)
+        shortlol = os.path.splitext(
+            os.path.split(self.lol_to_advertise)[1])[0]
+        advertise_name = "%s on %s's lolocopter" % (shortlol, 
+            self.public_name)
+        g.AddService(avahi.IF_UNSPEC, avahi.PROTO_UNSPEC,dbus.UInt32(0),
+                     advertise_name, 
+                     "_http._tcp", "", "",
+                     dbus.UInt16(self.httpd_port), "")
+        g.Commit()
+        self.group = g
+        
+        print "serving on ", self.httpd_port
+    
+    def __starthttp(self, network_clock_udp_port,
+            lol_to_advertise, public_name, joiner_callback=None):
+        MyHandler.protocol_version = "HTTP/1.0"
+        MyHandler.network_clock_udp_port = network_clock_udp_port
+        MyHandler.joiner_callback = joiner_callback
+        MyHandler.lol_to_advertise = lol_to_advertise
+        httpd = BaseHTTPServer.HTTPServer(('', 0), MyHandler)
+        sa = httpd.socket.getsockname()
+        self.httpd_port = sa[1]
+        self.__advertise_service()
+        httpd.serve_forever()
+    
+    # start http server in a thread
+    def start_server(self):
+        self.thr = threading.Thread(target=self.__starthttp, args=( 
+            self.network_clock_udp_port,
+            self.lol_to_advertise, 
+            self.public_name, 
+            self.joiner_callback))
+        self.thr.daemon = True
+        self.thr.start()
+    
+    def stop_server(self):
+        # unpublish
+        self.group.Reset()
+        # kill thread
+        # erm, not sure how to stop a serve_forever
+        # so do nothing here. epic fail.
+        # should kill the thread.
+