From e5d33932b19bf73d642b8d60773657c4c6b5f6d4 Mon Sep 17 00:00:00 2001
From: Eric Van Albert <eric@van.al>
Date: Mon, 19 Jun 2017 00:07:13 -0400
Subject: [PATCH] add simple keyserver

---
 bin/serve.py | 62 ++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 62 insertions(+)
 create mode 100755 bin/serve.py

diff --git a/bin/serve.py b/bin/serve.py
new file mode 100755
index 0000000..281273f
--- /dev/null
+++ b/bin/serve.py
@@ -0,0 +1,62 @@
+#!/usr/bin/env python3
+
+from http.server import BaseHTTPRequestHandler, HTTPServer
+from socketserver import ForkingMixIn
+from urllib.parse import urlparse, parse_qs
+import tempfile
+import subprocess
+import os
+import shutil
+
+PORT_NUMBER = 8080
+MAX_STR_LEN = 100
+
+class MyHandler(BaseHTTPRequestHandler):
+    def do_GET(self):
+        query_components = parse_qs(urlparse(self.path).query)
+        if "key" not in query_components or len(query_components["key"]) != 1:
+            with open("build/keys.json") as f:
+                j = f.read()
+            b = bytes(j, encoding="utf8")
+            self.send_response(200)
+            self.send_header("Content-type", "text/json; charset=utf-8")
+            self.send_header("Content-length", len(b))
+            self.end_headers()
+            self.wfile.write(b)
+        else:
+            opts = [str(query_components["key"][0])]
+            if "bitting" in query_components and len(query_components["bitting"]) == 1:
+                opts += ["-b", str(query_components["bitting"][0])[0:MAX_STR_LEN]]
+            if "outline" in query_components and len(query_components["outline"]) == 1:
+                opts += ["-u", str(query_components["outline"][0])[0:MAX_STR_LEN]]
+            if "warding" in query_components and len(query_components["warding"]) == 1:
+                opts += ["-w", str(query_components["warding"][0])[0:MAX_STR_LEN]]
+
+            with tempfile.NamedTemporaryFile(suffix=".stl") as tf:
+                opts += ["-o", tf.name]
+                r = subprocess.call(["bin/keygen.py"] + opts)
+                if r != 0:
+                    self.send_response(500)
+                    self.send_header("Content-type", "text/plain; charset=utf-8")
+                    self.end_headers()
+                    self.wfile.write(b"Command exited with non-zero return code")
+                else:
+                    length = os.stat(tf.name).st_size
+                    self.send_response(200)
+                    self.send_header("Content-type", "application/sla")
+                    self.send_header("Content-length", str(length))
+                    self.send_header("Content-Disposition", 'inline; filename="key.stl"')
+                    self.end_headers()
+                    with open(tf.name, 'rb') as stl:
+                        shutil.copyfileobj(stl, self.wfile)
+
+class ForkingSimpleServer(ForkingMixIn, HTTPServer):
+    pass
+
+if __name__ == '__main__':
+    httpd = ForkingSimpleServer(("", PORT_NUMBER), MyHandler)
+    try:
+        httpd.serve_forever()
+    except KeyboardInterrupt:
+        pass
+    httpd.server_close()