intiial

main.py

@@ -0,0 +1,69 @@
+from uuid import UUID, uuid4
+import fastapi
+from fastapi.responses import HTMLResponse, RedirectResponse
+from fastapi.templating import Jinja2Templates
+from pydantic import BaseModel
+import uvicorn
+
+app = fastapi.FastAPI(debug=True)
+templates = Jinja2Templates(directory="templates")
+
+
+calls = {}
+
+
+class CallPostResponse(BaseModel):
+    call_id: UUID
+
+
+@app.post("/call")
+async def create_call() -> CallPostResponse:
+    call_id = uuid4()
+    calls[call_id] = {}
+    return CallPostResponse(call_id=call_id)
+
+
+@app.get("/call", response_class=HTMLResponse)
+async def call(request: fastapi.Request, call_id: UUID = fastapi.Query(), session_id: UUID | None = fastapi.Cookie(None)):
+    if not session_id:
+        session_id = uuid4()
+        response = RedirectResponse(f"/call?call_id={call_id}")
+        response.set_cookie("session_id", str(session_id))
+        return response
+    if call_id not in calls:
+        return RedirectResponse("/")
+
+    return templates.TemplateResponse(
+        "call.html",
+        context={"request": request}
+    )
+
+
+@app.websocket("/connect")
+async def connect(websocket: fastapi.WebSocket, call_id: UUID = fastapi.Query(), session_id: UUID | None = fastapi.Cookie(None)):
+    await websocket.accept()
+    calls[call_id][session_id] = websocket
+    another_websocket: fastapi.WebSocket | None = None
+    while True:
+        data = await websocket.receive_bytes()
+        print("RECEIVED")
+        if not another_websocket:
+            if len(calls[call_id]) == 1:
+                continue
+            for sess_id, ws in calls[call_id].items():
+                if sess_id != session_id:
+                    another_websocket = ws
+        print("SENDING DATA")
+        await another_websocket.send_bytes(data)
+
+
+@app.get("/", response_class=HTMLResponse)
+async def home(request: fastapi.Request):
+    return templates.TemplateResponse(
+        "index.html",
+        context={"request": request}
+    )
+
+
+if __name__ == '__main__':
+    uvicorn.run(app, host="0.0.0.0", port=1239)