WebRTC – Modern Browser-to-Browser Streaming

from fastapi import FastAPI, WebSocket, WebSocketDisconnect
from typing import dict
import json

app = FastAPI()

# 룸 단위로 피어 관리: {"room_id": [websocket1, websocket2]}
rooms: dict[str, list[WebSocket]] = {}

@app.websocket("/signal/{room_id}")
async def signaling(websocket: WebSocket, room_id: str):
    """WebRTC 시그널링 서버 — SDP와 ICE 후보를 피어 간 중계합니다."""
    await websocket.accept()

    # 룸에 참여
    if room_id not in rooms:
        rooms[room_id] = []
    rooms[room_id].append(websocket)
    peers = rooms[room_id]
    print(f"[{room_id}] 새 피어 참여. 현재 {len(peers)}명")

    try:
        async for raw in websocket.iter_text():
            msg = json.loads(raw)
            msg_type = msg.get("type")

            # SDP Offer/Answer, ICE Candidate를 같은 룸의 다른 피어에게 중계
            if msg_type in ("offer", "answer", "ice-candidate"):
                for peer in peers:
                    if peer != websocket:       # 자신 제외
                        await peer.send_text(raw)
                        print(f"[{room_id}] {msg_type} 중계 완료")

    except WebSocketDisconnect:
        peers.remove(websocket)
        print(f"[{room_id}] 피어 퇴장. 남은 인원: {len(peers)}")
        # 상대방에게 연결 종료 알림
        for peer in peers:
            await peer.send_json({"type": "peer-left"})

# 클라이언트 측 JavaScript (참고):
# const pc = new RTCPeerConnection({iceServers: [{urls: 'stun:stun.l.google.com:19302'}]});
# const ws = new WebSocket('ws://localhost:8000/signal/room1');
# // Offer 생성 후 시그널링 서버로 전송
# const offer = await pc.createOffer();
# await pc.setLocalDescription(offer);
# ws.send(JSON.stringify({type: 'offer', sdp: offer.sdp}));