changed main

api/app/main.py

@@ -1,26 +1,101 @@
-from fastapi import FastAPI, Depends, HTTPException
+# Add to api/app/main.py (after the existing imports and app definition)
-from .auth import init_db, verify_api_key
+import os
-from .models import Event, BatchEvents
+from fastapi import Request, Query
-from .enrichment import enrich_event
+from fastapi.responses import JSONResponse
-from .influx import write_event
+from .influx import client, INFLUX_ORG, INFLUX_BUCKET
-import logging
+import sqlite3
+import datetime
 
-logging.basicConfig(level=logging.INFO)
+ADMIN_KEY = os.getenv("ADMIN_API_KEY", "admin-secret-change-me")
-init_db()
 
-app = FastAPI(title="Signal - Roblox Telemetry")
+# ---------- simple admin auth helper ----------
+def require_admin(x_admin_key: str = Header(None)):
+    if not x_admin_key or x_admin_key != ADMIN_KEY:
+        raise HTTPException(status_code=403, detail="Forbidden")
+    return True
 
-@app.get("/health")
+# ---------- API Keys Management ----------
-async def health():
+@app.get("/admin/keys")
-    return {"status": "ok"}
+async def list_keys(admin: bool = Depends(require_admin)):
+    conn = get_db()
+    rows = conn.execute("SELECT key, game, active, created_at FROM api_keys").fetchall()
+    conn.close()
+    return [dict(r) for r in rows]
 
-@app.post("/api/log")
+@app.post("/admin/keys")
-async def ingest_event(payload: Event | BatchEvents, game: str = Depends(verify_api_key)):
+async def create_key(game: str = Query(...), key: str = Query(None), admin: bool = Depends(require_admin)):
-    if isinstance(payload, BatchEvents):
+    if not key:
-        for event in payload.events:
+        import secrets
-            enriched = enrich_event(event, game)
+        key = secrets.token_hex(16)   # auto-generate if not provided
-            write_event(enriched)
+    conn = get_db()
-    else:
+    conn.execute("INSERT OR REPLACE INTO api_keys(key, game, active) VALUES(?, ?, 1)", (key, game))
-        enriched = enrich_event(payload, game)
+    conn.commit()
-        write_event(enriched)
+    conn.close()
-    return {"success": True}
+    return {"key": key, "game": game, "active": True}
+
+@app.put("/admin/keys/{key}/toggle")
+async def toggle_key(key: str, admin: bool = Depends(require_admin)):
+    conn = get_db()
+    conn.execute("UPDATE api_keys SET active = NOT active WHERE key = ?", (key,))
+    conn.commit()
+    conn.close()
+    return {"ok": True}
+
+@app.delete("/admin/keys/{key}")
+async def delete_key(key: str, admin: bool = Depends(require_admin)):
+    conn = get_db()
+    conn.execute("DELETE FROM api_keys WHERE key = ?", (key,))
+    conn.commit()
+    conn.close()
+    return {"ok": True}
+
+# ---------- InfluxDB Bucket Info ----------
+@app.get("/admin/buckets")
+async def list_buckets(admin: bool = Depends(require_admin)):
+    if not client:
+        return JSONResponse({"error": "InfluxDB client not initialised"}, status_code=503)
+    buckets_api = client.buckets_api()
+    buckets = buckets_api.find_buckets().buckets
+    return [{"name": b.name, "id": b.id, "org": b.org_id} for b in buckets]
+
+# ---------- View recent events ----------
+@app.get("/admin/events")
+async def list_events(hours: float = 24, limit: int = 100, admin: bool = Depends(require_admin)):
+    if not client:
+        return JSONResponse({"error": "InfluxDB client not initialised"}, status_code=503)
+    query = f'''
+    from(bucket: "{INFLUX_BUCKET}")
+      |> range(start: -{hours}h)
+      |> limit(n: {limit})
+    '''
+    tables = client.query_api().query(query, org=INFLUX_ORG)
+    events = []
+    for table in tables:
+        for record in table.records:
+            events.append({
+                "time": record.get_time().isoformat(),
+                "measurement": record.get_measurement(),
+                "fields": {k: v for k, v in record.values.items() if k not in ('_start','_stop','_time')},
+                "tags": dict(record.values.get("_tags", {}))
+            })
+    return events[:limit]
+
+# ---------- Delete events (by time range + optional measurement) ----------
+@app.delete("/admin/events")
+async def delete_events(
+    start: str = Query(..., description="ISO timestamp, e.g. 2026-04-27T00:00:00Z"),
+    stop: str = Query(..., description="ISO timestamp"),
+    measurement: str = Query(None),
+    admin: bool = Depends(require_admin)
+):
+    if not client:
+        return JSONResponse({"error": "InfluxDB client not initialised"}, status_code=503)
+    try:
+        predicate = f'_measurement="{measurement}"' if measurement else ""
+        delete_api = client.delete_api()
+        delete_api.delete(
+            start=start, stop=stop, bucket=INFLUX_BUCKET, org=INFLUX_ORG, predicate=predicate
+        )
+        return {"ok": True}
+    except Exception as e:
+        return JSONResponse({"error": str(e)}, status_code=500)