refactor: общий вход в сокет занятия и выполнение команды с fencing вместо копий в четырёх каналах

backend/app/api/ws/observe.py

@@ -15,24 +15,13 @@ 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.api.ws.session import pump, session_socket
 from app.domain.roles import Role
-from app.session.hub import LEASE_FENCED_MESSAGE, hub
+from app.session.hub import 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:
     """Единственное назначение — дождаться разрыва. Содержимое кадров
     не читается и никуда не передаётся."""
@@ -42,54 +31,21 @@ async def _wait_for_disconnect(ws: WebSocket) -> None:
             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()
+    # Наблюдение за чужим занятием — не для обучающегося. Преподаватель видит
+    # только свои занятия, администратор — любые, для диагностики.
+    entered = await session_socket(ws, session_id, (Role.INSTRUCTOR, Role.ADMIN))
+    if entered is None:
         return
+    _who, state = entered
 
     # Снимок при подключении обязателен: монитор в классе включают посреди
     # занятия, и он должен показать текущее состояние, а не ждать событий.
     await ws.send_text(state.snapshot().model_dump_json())
 
     with hub.observer(session_id) as queue:
-        sender = asyncio.create_task(_pump(ws, queue))
+        sender = asyncio.create_task(pump(ws, queue))
         try:
             await _wait_for_disconnect(ws)
         except WebSocketDisconnect: