Compare commits

..

No commits in common. "c3cb93d903b9e0a6bc60c7f1e3bb7dae8a71118e" and "a00cfb688c5de58afca0df8c3a5eec2cfd4f1566" have entirely different histories.

40
main.py
View file

@ -3,7 +3,7 @@ import json
import sqlite3
from typing import Dict
from fastapi import FastAPI, Header, HTTPException, Query, WebSocket, WebSocketDisconnect
from fastapi import FastAPI, Header, HTTPException, Query
from fastapi.responses import JSONResponse
from fastapi.routing import APIRoute
from fastapi.staticfiles import StaticFiles
@ -11,8 +11,6 @@ from pydantic import BaseModel
from typing import Optional
from db import init_db, save_snapshot, DB_FILE
import asyncio
from starlette.concurrency import run_in_threadpool
# ------------------------------------------------------------------
app = FastAPI()
@ -196,42 +194,6 @@ def get_trails(
]
return JSONResponse(content={"trails": trails})
# -------------------- WebSocket endpoints -----------------------
browser_conns: set[WebSocket] = set()
async def _broadcast_to_browser_clients(snapshot: dict):
for ws in list(browser_conns):
try:
await ws.send_json(snapshot)
except WebSocketDisconnect:
browser_conns.remove(ws)
@app.websocket("/ws/position")
async def ws_receive_snapshots(websocket: WebSocket, secret: str = Query(...)):
await websocket.accept()
if secret != SHARED_SECRET:
await websocket.close(code=1008)
return
try:
while True:
data = await websocket.receive_json()
snap = TelemetrySnapshot.parse_obj(data)
live_snapshots[snap.character_name] = snap.dict()
await run_in_threadpool(save_snapshot, snap.dict())
await _broadcast_to_browser_clients(snap.dict())
except WebSocketDisconnect:
pass
@app.websocket("/ws/live")
async def ws_live_updates(websocket: WebSocket):
await websocket.accept()
browser_conns.add(websocket)
try:
while True:
await asyncio.sleep(3600)
except WebSocketDisconnect:
browser_conns.remove(websocket)
# -------------------- static frontend ---------------------------
app.mount("/", StaticFiles(directory="static", html=True), name="static")