Bandito / server-fastrtc.py
makinuh's picture
Rename server.py to server-fastrtc.py
e7a8f4f verified
raw
history blame contribute delete
970 Bytes
# server.py (using FastAPI and FastRTC)
import asyncio
import json
import uuid
from fastapi import FastAPI, WebSocket, WebSocketDisconnect
from fastrtc import RTCRoom, RTCRoomManager
app = FastAPI()
room_manager = RTCRoomManager()
@app.websocket("/ws/{room_id}/{client_id}")
async def websocket_endpoint(websocket: WebSocket, room_id: str, client_id: str):
await websocket.accept()
room = room_manager.get_or_create_room(room_id)
peer = await room.join(client_id, websocket)
try:
while True:
message = await websocket.receive_text()
data = json.loads(message)
if data["type"] == "offer" or data["type"] == "answer" or data["type"] == "candidate":
await peer.handle_sdp(data)
else:
print(f"Received unknown message: {data}")
except WebSocketDisconnect:
await room.leave(client_id)
print(f"Client {client_id} disconnected from room {room_id}")