Initial commit: webmail client (M0/M1)
Three-pane webmail client. - Backend: FastAPI + async SQLAlchemy + arq worker (Postgres/Redis), IMAP/SMTP sync, threading, search, drafts/outbox with undo, attachments, 2FA. - Frontend: React 18 + Vite + TypeScript + TanStack Query + Tailwind; three-pane UI with rich-text composer, light/dark themes, resizable panels. Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
This commit is contained in:
@@ -0,0 +1,70 @@
|
||||
"""WebSocket hub (SPEC §9.2).
|
||||
|
||||
Auth via session cookie. Relays every §8.3 event belonging to the user as
|
||||
``{event, payload}`` frames; the client refetches via TanStack Query invalidation.
|
||||
Client -> server: ``{"type":"ping"}`` only.
|
||||
"""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
import asyncio
|
||||
import json
|
||||
import logging
|
||||
|
||||
from fastapi import APIRouter, WebSocket, WebSocketDisconnect
|
||||
|
||||
from app.identity.cookies import SESSION_COOKIE_INSECURE, SESSION_COOKIE_SECURE
|
||||
from app.identity.sessions import resolve_session
|
||||
from app.redis_client import get_redis
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
router = APIRouter()
|
||||
|
||||
|
||||
@router.websocket("/ws")
|
||||
async def ws_endpoint(websocket: WebSocket) -> None:
|
||||
token = websocket.cookies.get(SESSION_COOKIE_SECURE) or websocket.cookies.get(
|
||||
SESSION_COOKIE_INSECURE, ""
|
||||
)
|
||||
session = await resolve_session(token)
|
||||
if session is None:
|
||||
await websocket.close(code=1008) # policy violation
|
||||
return
|
||||
|
||||
await websocket.accept()
|
||||
user_id = str(session.user_id)
|
||||
pubsub = get_redis().pubsub()
|
||||
await pubsub.psubscribe("evt:*")
|
||||
|
||||
async def forward() -> None:
|
||||
async for message in pubsub.listen():
|
||||
if message.get("type") != "pmessage":
|
||||
continue
|
||||
try:
|
||||
data = json.loads(message["data"])
|
||||
except (ValueError, TypeError):
|
||||
continue
|
||||
payload = data.get("payload", {})
|
||||
if str(payload.get("user_id")) == user_id:
|
||||
await websocket.send_json(data)
|
||||
|
||||
async def receive() -> None:
|
||||
while True:
|
||||
raw = await websocket.receive_text()
|
||||
try:
|
||||
if json.loads(raw).get("type") == "ping":
|
||||
await websocket.send_json({"type": "pong"})
|
||||
except (ValueError, TypeError):
|
||||
continue
|
||||
|
||||
forward_task = asyncio.create_task(forward())
|
||||
receive_task = asyncio.create_task(receive())
|
||||
try:
|
||||
await asyncio.wait({forward_task, receive_task}, return_when=asyncio.FIRST_COMPLETED)
|
||||
except WebSocketDisconnect:
|
||||
pass
|
||||
finally:
|
||||
for task in (forward_task, receive_task):
|
||||
task.cancel()
|
||||
await pubsub.aclose() # type: ignore[no-untyped-call]
|
||||
Reference in New Issue
Block a user