refactor: voice-service entfernt (verschoben nach breakpilot-core)
This commit is contained in:
Benjamin Boenisch
@@ -1,220 +0,0 @@
|
||||
"""
|
||||
Session Management API
|
||||
Handles voice session lifecycle
|
||||
|
||||
Endpoints:
|
||||
- POST /api/v1/sessions # Session erstellen
|
||||
- GET /api/v1/sessions/{id} # Session Status
|
||||
- DELETE /api/v1/sessions/{id} # Session beenden
|
||||
- GET /api/v1/sessions/{id}/tasks # Pending Tasks
|
||||
"""
|
||||
import structlog
|
||||
from fastapi import APIRouter, HTTPException, Request, Depends
|
||||
from typing import List, Optional
|
||||
from datetime import datetime, timedelta
|
||||
|
||||
from config import settings
|
||||
from models.session import (
|
||||
VoiceSession,
|
||||
SessionCreate,
|
||||
SessionResponse,
|
||||
SessionStatus,
|
||||
)
|
||||
from models.task import TaskResponse, TaskState
|
||||
|
||||
logger = structlog.get_logger(__name__)
|
||||
|
||||
router = APIRouter()
|
||||
|
||||
|
||||
# In-memory session store (will be replaced with Valkey in production)
|
||||
# This is transient - sessions are never persisted to disk
|
||||
_sessions: dict[str, VoiceSession] = {}
|
||||
|
||||
|
||||
async def get_session(session_id: str) -> VoiceSession:
|
||||
"""Get session by ID or raise 404."""
|
||||
session = _sessions.get(session_id)
|
||||
if not session:
|
||||
raise HTTPException(status_code=404, detail="Session not found")
|
||||
return session
|
||||
|
||||
|
||||
@router.post("", response_model=SessionResponse)
|
||||
async def create_session(request: Request, session_data: SessionCreate):
|
||||
"""
|
||||
Create a new voice session.
|
||||
|
||||
Returns a session ID and WebSocket URL for audio streaming.
|
||||
The client must connect to the WebSocket within 30 seconds.
|
||||
"""
|
||||
logger.info(
|
||||
"Creating voice session",
|
||||
namespace_id=session_data.namespace_id[:8] + "...",
|
||||
device_type=session_data.device_type,
|
||||
)
|
||||
|
||||
# Verify namespace key hash
|
||||
orchestrator = request.app.state.orchestrator
|
||||
encryption = request.app.state.encryption
|
||||
|
||||
if settings.encryption_enabled:
|
||||
if not encryption.verify_key_hash(session_data.key_hash):
|
||||
logger.warning("Invalid key hash", namespace_id=session_data.namespace_id[:8])
|
||||
raise HTTPException(status_code=401, detail="Invalid encryption key hash")
|
||||
|
||||
# Check rate limits
|
||||
namespace_sessions = [
|
||||
s for s in _sessions.values()
|
||||
if s.namespace_id == session_data.namespace_id
|
||||
and s.status not in [SessionStatus.CLOSED, SessionStatus.ERROR]
|
||||
]
|
||||
if len(namespace_sessions) >= settings.max_sessions_per_user:
|
||||
raise HTTPException(
|
||||
status_code=429,
|
||||
detail=f"Maximum {settings.max_sessions_per_user} concurrent sessions allowed"
|
||||
)
|
||||
|
||||
# Create session
|
||||
session = VoiceSession(
|
||||
namespace_id=session_data.namespace_id,
|
||||
key_hash=session_data.key_hash,
|
||||
device_type=session_data.device_type,
|
||||
client_version=session_data.client_version,
|
||||
)
|
||||
|
||||
# Store session (in RAM only)
|
||||
_sessions[session.id] = session
|
||||
|
||||
logger.info(
|
||||
"Voice session created",
|
||||
session_id=session.id[:8],
|
||||
namespace_id=session_data.namespace_id[:8],
|
||||
)
|
||||
|
||||
# Build WebSocket URL
|
||||
# Use X-Forwarded-Proto if behind a reverse proxy (nginx), otherwise use request scheme
|
||||
forwarded_proto = request.headers.get("x-forwarded-proto", request.url.scheme)
|
||||
host = request.headers.get("host", f"localhost:{settings.port}")
|
||||
ws_scheme = "wss" if forwarded_proto == "https" else "ws"
|
||||
ws_url = f"{ws_scheme}://{host}/ws/voice?session_id={session.id}"
|
||||
|
||||
return SessionResponse(
|
||||
id=session.id,
|
||||
namespace_id=session.namespace_id,
|
||||
status=session.status,
|
||||
created_at=session.created_at,
|
||||
websocket_url=ws_url,
|
||||
)
|
||||
|
||||
|
||||
@router.get("/{session_id}", response_model=SessionResponse)
|
||||
async def get_session_status(session_id: str, request: Request):
|
||||
"""
|
||||
Get session status.
|
||||
|
||||
Returns current session state including message count and pending tasks.
|
||||
"""
|
||||
session = await get_session(session_id)
|
||||
|
||||
# Check if session expired
|
||||
session_age = datetime.utcnow() - session.created_at
|
||||
if session_age > timedelta(hours=settings.session_ttl_hours):
|
||||
session.status = SessionStatus.CLOSED
|
||||
logger.info("Session expired", session_id=session_id[:8])
|
||||
|
||||
# Build WebSocket URL
|
||||
# Use X-Forwarded-Proto if behind a reverse proxy (nginx), otherwise use request scheme
|
||||
forwarded_proto = request.headers.get("x-forwarded-proto", request.url.scheme)
|
||||
host = request.headers.get("host", f"localhost:{settings.port}")
|
||||
ws_scheme = "wss" if forwarded_proto == "https" else "ws"
|
||||
ws_url = f"{ws_scheme}://{host}/ws/voice?session_id={session.id}"
|
||||
|
||||
return SessionResponse(
|
||||
id=session.id,
|
||||
namespace_id=session.namespace_id,
|
||||
status=session.status,
|
||||
created_at=session.created_at,
|
||||
websocket_url=ws_url,
|
||||
)
|
||||
|
||||
|
||||
@router.delete("/{session_id}")
|
||||
async def close_session(session_id: str):
|
||||
"""
|
||||
Close and delete a session.
|
||||
|
||||
All transient data (messages, audio state) is discarded.
|
||||
This is the expected cleanup path.
|
||||
"""
|
||||
session = await get_session(session_id)
|
||||
|
||||
logger.info(
|
||||
"Closing session",
|
||||
session_id=session_id[:8],
|
||||
messages_count=len(session.messages),
|
||||
tasks_count=len(session.pending_tasks),
|
||||
)
|
||||
|
||||
# Mark as closed
|
||||
session.status = SessionStatus.CLOSED
|
||||
|
||||
# Remove from active sessions
|
||||
del _sessions[session_id]
|
||||
|
||||
return {"status": "closed", "session_id": session_id}
|
||||
|
||||
|
||||
@router.get("/{session_id}/tasks", response_model=List[TaskResponse])
|
||||
async def get_session_tasks(session_id: str, request: Request, state: Optional[TaskState] = None):
|
||||
"""
|
||||
Get tasks for a session.
|
||||
|
||||
Optionally filter by task state.
|
||||
"""
|
||||
session = await get_session(session_id)
|
||||
|
||||
# Get tasks from the in-memory task store
|
||||
from api.tasks import _tasks
|
||||
|
||||
# Filter tasks by session_id and optionally by state
|
||||
tasks = [
|
||||
task for task in _tasks.values()
|
||||
if task.session_id == session_id
|
||||
and (state is None or task.state == state)
|
||||
]
|
||||
|
||||
return [
|
||||
TaskResponse(
|
||||
id=task.id,
|
||||
session_id=task.session_id,
|
||||
type=task.type,
|
||||
state=task.state,
|
||||
created_at=task.created_at,
|
||||
updated_at=task.updated_at,
|
||||
result_available=task.result_ref is not None,
|
||||
error_message=task.error_message,
|
||||
)
|
||||
for task in tasks
|
||||
]
|
||||
|
||||
|
||||
@router.get("/{session_id}/stats")
|
||||
async def get_session_stats(session_id: str):
|
||||
"""
|
||||
Get session statistics (for debugging/monitoring).
|
||||
|
||||
No PII is returned - only aggregate counts.
|
||||
"""
|
||||
session = await get_session(session_id)
|
||||
|
||||
return {
|
||||
"session_id_truncated": session_id[:8],
|
||||
"status": session.status.value,
|
||||
"age_seconds": (datetime.utcnow() - session.created_at).total_seconds(),
|
||||
"message_count": len(session.messages),
|
||||
"pending_tasks_count": len(session.pending_tasks),
|
||||
"audio_chunks_received": session.audio_chunks_received,
|
||||
"audio_chunks_processed": session.audio_chunks_processed,
|
||||
"device_type": session.device_type,
|
||||
}
|
||||
Reference in New Issue
Block a user