[+] Json server

server/host.py

@@ -0,0 +1,71 @@
+import json
+import random
+from fastapi.responses import RedirectResponse
+import uvicorn
+from subprocess import check_output
+from fastapi import FastAPI, Request
+from fastapi.middleware.cors import CORSMiddleware
+from pathlib import Path
+
+
+app = FastAPI()
+src = Path(__file__).parent
+data = src / "data"
+
+origins = ["*"]
+
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=origins,
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+keys = {
+    "rows": "int",
+    "cols": "int",
+    "hStates": "list[int]",
+    "vStates": "list[int]",
+    "hColors": "list[int]",
+    "vColors": "list[int]",
+    "numbers": "list[int]",
+    "nMask": "list[int]",
+    "colors": "list[string]",
+}
+
+
+# Post / with a json: Create a puzzle
+# Store the json in the data directory
+@app.post("/")
+async def input_request(request: Request):
+    body = (await request.body()).decode().replace("\r\n", "\n")
+    
+    # Parse json, ensure only keys in the whitelist exist, and ensure they are correct types
+    try:
+        obj = json.loads(body)
+        k = [k for k in obj if k not in keys]
+        assert not k, f"Invalid keys: {k}"
+    except Exception as e:
+        return str(e)
+
+    # Randomly generate a 8-letter id for the puzzle
+    id = "".join(random.choices("abcdefghijklmnopqrstuvwxyz0123456789", k=8))
+    tf = data / f"{id}.json"
+    tf.write_text(body)
+    return {"id": id}
+
+
+# Get /{id}: Get the puzzle with the given id
+@app.get("/{id}")
+async def get_puzzle(id: str):
+    tf = data / f"{id}.json"
+    if not tf.exists():
+        return {"error": "Puzzle not found"}
+    return tf.read_text()
+
+
+# Getting / redirects to main page
+@app.get("/")
+async def get_main():
+    return RedirectResponse("https://slither.hydev.org")