98 lines
4.3 KiB
Python
98 lines
4.3 KiB
Python
"""Канал наблюдателя: внешний монитор и пульт преподавателя.
|
||
|
||
**На этом сокете нет ни одного обработчика входящих сообщений.** Внешний
|
||
монитор физически не может повлиять на занятие: у него нет ни аудио, ни канала
|
||
записи. Преподаватель смотрит здесь, а пишет через отдельный `control`.
|
||
|
||
Входящие кадры читаются и выбрасываются, не разбираясь: чтение нужно ровно
|
||
затем, чтобы заметить разрыв соединения. Без него задача сокета висела бы
|
||
на очереди до первой неудачной отправки, а закрытая вкладка монитора оставляла
|
||
бы за собой подписку. Разбора, диспетчеризации и эффекта у входящих нет.
|
||
"""
|
||
|
||
import asyncio
|
||
from uuid import UUID
|
||
|
||
from fastapi import APIRouter, WebSocket, WebSocketDisconnect
|
||
|
||
from app.domain.events import ErrorEvent, ErrorKind
|
||
from app.api.auth import principal_of, websocket_origin_allowed
|
||
from app.domain.roles import Role
|
||
from app.session.hub import LEASE_FENCED_MESSAGE, hub
|
||
|
||
router = APIRouter()
|
||
|
||
|
||
async def _pump(ws: WebSocket, queue: asyncio.Queue) -> None:
|
||
while True:
|
||
event = await queue.get()
|
||
await ws.send_text(event.model_dump_json())
|
||
if (isinstance(event, ErrorEvent) and event.code is ErrorKind.INTERNAL
|
||
and event.message == LEASE_FENCED_MESSAGE):
|
||
await ws.close(code=1012)
|
||
return
|
||
|
||
|
||
async def _wait_for_disconnect(ws: WebSocket) -> None:
|
||
"""Единственное назначение — дождаться разрыва. Содержимое кадров
|
||
не читается и никуда не передаётся."""
|
||
while True:
|
||
message = await ws.receive()
|
||
if message["type"] == "websocket.disconnect":
|
||
return
|
||
|
||
|
||
async def _reject(ws: WebSocket, message: str) -> None:
|
||
"""Отказ до входа в цикл: сокет закрывается с объяснением, а не молча."""
|
||
await ws.send_text(
|
||
ErrorEvent(code=ErrorKind.FORBIDDEN, message=message).model_dump_json()
|
||
)
|
||
await ws.close()
|
||
|
||
|
||
@router.websocket("/ws/observe/{session_id}")
|
||
async def observe(ws: WebSocket, session_id: UUID) -> None:
|
||
if not websocket_origin_allowed(ws):
|
||
await ws.close(code=1008)
|
||
return
|
||
if hub.is_lease_fenced(session_id):
|
||
await ws.close(code=1012)
|
||
return
|
||
await ws.accept()
|
||
|
||
# Наблюдение за чужим занятием — не для обучающегося.
|
||
who = principal_of(ws)
|
||
if who is None or who.role not in (Role.INSTRUCTOR, Role.ADMIN):
|
||
await _reject(ws, "Недостаточно прав для этого экрана")
|
||
return
|
||
|
||
state = hub.get(session_id)
|
||
if state is None:
|
||
await ws.send_text(
|
||
ErrorEvent(code=ErrorKind.SESSION_NOT_FOUND, message="Занятие не запущено").model_dump_json()
|
||
)
|
||
await ws.close()
|
||
return
|
||
|
||
# Live state is process-local, so authorize against the owner snapshot on
|
||
# the state itself. Instructors may observe only their own sessions;
|
||
# administrators retain the cross-owner diagnostic view.
|
||
if who.role is Role.INSTRUCTOR and state.owner_login != who.login:
|
||
await ws.send_text(
|
||
ErrorEvent(code=ErrorKind.SESSION_NOT_FOUND, message="Занятие не запущено").model_dump_json()
|
||
)
|
||
await ws.close()
|
||
return
|
||
|
||
# Снимок при подключении обязателен: монитор в классе включают посреди
|
||
# занятия, и он должен показать текущее состояние, а не ждать событий.
|
||
await ws.send_text(state.snapshot().model_dump_json())
|
||
|
||
with hub.observer(session_id) as queue:
|
||
sender = asyncio.create_task(_pump(ws, queue))
|
||
try:
|
||
await _wait_for_disconnect(ws)
|
||
except WebSocketDisconnect:
|
||
pass
|
||
finally:
|
||
sender.cancel()
|