Get sandstorm user id in launcher script

.sandstorm/get-sandstorm-id.py

@@ -0,0 +1,46 @@
+#!/usr/bin/python3
+from http.server import BaseHTTPRequestHandler, HTTPServer
+import os, tempfile, sys
+
+os.chdir(os.path.split(__file__)[0])
+
+[_, serverPort, reloadDelay, envOutputPath] = sys.argv
+serverPort = int(serverPort)
+
+hostName = "0.0.0.0"
+
+class MyServer(BaseHTTPRequestHandler):
+    def write_id(self, sandstorm_id):
+        dir_path = os.path.split(envOutputPath)[0]
+        f = tempfile.NamedTemporaryFile(mode='w', dir=dir_path, delete=False)
+        f.write(sandstorm_id)
+        f.close()
+        os.rename(f.name, envOutputPath)
+
+    def do_GET(self):
+        self.write_id(self.headers['X-Sandstorm-User-Id'])
+        self.send_response(200)
+        self.send_header("Content-type", "text/html")
+        self.end_headers()
+
+        # reloadDelay ms "hopefully" is enough time for the next server to start
+        # can make this marginally faster and more reliable by returning
+        # 503 and using Caddy's load balancer to wait for the service to start
+        self.wfile.write(bytes(f"<script>setTimeout(() => window.location = '/', {reloadDelay})</script>", "utf-8"))
+        # Quit so that the service can start with the sandstorm id
+        exit(0)
+
+if __name__ == "__main__":
+    if os.path.exists(envOutputPath):
+        exit(0)
+
+    webServer = HTTPServer((hostName, serverPort), MyServer)
+    print("Server started http://%s:%s" % (hostName, serverPort))
+
+    try:
+        webServer.serve_forever()
+    except KeyboardInterrupt:
+        pass
+
+    webServer.server_close()
+    print("Server stopped.")

.sandstorm/launcher.sh

@@ -35,6 +35,9 @@ done
 
 redis-server --daemonize yes
 
+./get-sandstorm-id.py 8000 1000 /var/sandstorm-id
+echo "Sandstorm User ID: $(cat /var/sandstorm-id)" # whatever you need to do with it
+
 cd /opt/paperless/src
 $VENV/bin/python3 manage.py migrate
-$VENV/bin/python3 manage.py runserver
+$VENV/bin/python3 manage.py runserver