AGPLv3+CAFv2

[hippotat.git] / hippotatd
diff --git a/hippotatd b/hippotatd

index 32d9a343763e6a41fb6baab2e773aeb211f18beb..49521683b250d503f2e61dddf34f4ef70843134a 100755 (executable)
--- a/hippotatd
+++ b/hippotatd
@@ -3,15 +3,25 @@
  from hippotatlib import *
  
  import os
+import tempfile
+import atexit
+import shutil
  
  import twisted.internet
  from twisted.web.server import NOT_DONE_YET
  
+import twisted.web.static
+
+import hippotatlib.ownsource
+from hippotatlib.ownsource import SourceShipmentPreparer
+
  #import twisted.web.server import Site
  #from twisted.web.resource import Resource
  
  import syslog
  
+cleanups = [ ]
+
  clients = { }
  
  #---------- "router" ----------
@@ -200,10 +210,36 @@ class IphttpResource(NotStupidResource):
  
    def render_GET(self, request):
      log_debug(DBG.HTTP, 'GET request')
-    return b'<html><body>hippotat</body></html>'
+    return b'''
+<html><body>
+hippotat
+<p>
+<a href="source">source</a> available
+</body></html>
+'''
+
+class SourceResource(twisted.web.static.File):
+  def __init__(self):
+    td = tempfile.mkdtemp()
+
+    def cleanup():
+      try: shutil.rmtree(td)
+      except FileNotFoundError: pass
+
+    cleanups.append(cleanup)
+
+    self._ssp = SourceShipmentPreparer(td)
+    self._ssp.logger = self.log
+    self._ssp.generate()
+
+    super().__init__(self._ssp.output_path)
+
+  def log(self, m):
+    log_debug(DBG.OWNSOURCE, m)
  
  def start_http():
    resource = IphttpResource()
+  resource.putChild(b'source',SourceResource())
    site = twisted.web.server.Site(resource)
    for sa in c.saddrs:
      ep = sa.make_endpoint()
@@ -211,7 +247,7 @@ def start_http():
      log_debug(DBG.INIT, 'listening on %s' % sa)
  
  #---------- config and setup ----------
-        
+
  def process_cfg(putative_servers, putative_clients):
    global c
    c = ConfigResults()
@@ -244,7 +280,25 @@ def process_cfg(putative_servers, putative_clients):
                      ('peer', 'vrelay'),
                      ('rnets','vnetwork')))
  
+def catch_termination():
+  def run_cleanups():
+    for cleanup in cleanups:
+      cleanup()
+
+  atexit.register(run_cleanups)
+
+  def signal_handler(name, sig, *args):
+    signal.signal(sig, signal.SIG_DFL)
+    print('exiting due to %s' % name, file=sys.stderr)
+    run_cleanups()
+    os.kill(os.getpid(), sig)
+    raise RuntimeError('did not die due to signal %s !' % name)
+
+  for sig in (signal.SIGINT, signal.SIGTERM):
+    signal.signal(sig, partial(signal_handler, sig.name))
+
  common_startup(process_cfg)
+catch_termination()
  ipif = start_ipif(c.ipif_command, (lambda p,s,d: route(p,"[ipif]",s,d)))
  start_http()
  common_run()