ce353f8b49
All checks were successful
CI/CD Pipeline / VM Test - backend-integration (push) Successful in 12s
CI/CD Pipeline / VM Test - full-stack (push) Successful in 9s
CI/CD Pipeline / VM Test - performance (push) Successful in 9s
CI/CD Pipeline / VM Test - security (push) Successful in 9s
CI/CD Pipeline / Backend Linting (push) Successful in 3s
CI/CD Pipeline / Frontend Linting (push) Successful in 24s
CI/CD Pipeline / Nix Flake Check (push) Successful in 53s
CI/CD Pipeline / CI Summary (push) Successful in 1s
CI/CD Pipeline / VM Test - backend-integration (pull_request) Successful in 2s
CI/CD Pipeline / VM Test - full-stack (pull_request) Successful in 2s
CI/CD Pipeline / VM Test - performance (pull_request) Successful in 2s
CI/CD Pipeline / VM Test - security (pull_request) Successful in 2s
CI/CD Pipeline / Backend Linting (pull_request) Successful in 2s
CI/CD Pipeline / Frontend Linting (pull_request) Successful in 16s
CI/CD Pipeline / Nix Flake Check (pull_request) Successful in 38s
CI/CD Pipeline / CI Summary (pull_request) Successful in 0s
80 lines
2.2 KiB
Python
80 lines
2.2 KiB
Python
"""Connection quality detection and testing endpoints."""
|
|
|
|
import time
|
|
|
|
from fastapi import APIRouter
|
|
from pydantic import BaseModel
|
|
|
|
router = APIRouter(tags=["quality"])
|
|
|
|
|
|
class ConnectionTestRequest(BaseModel):
|
|
"""Request schema for connection test."""
|
|
|
|
test_size_bytes: int = 100000 # 100KB default test size
|
|
|
|
|
|
class ConnectionTestResponse(BaseModel):
|
|
"""Response schema for connection test results."""
|
|
|
|
speed_mbps: float
|
|
latency_ms: float
|
|
quality_tier: str # 'low', 'medium', 'high'
|
|
recommended_thumbnail: str # 'low', 'medium', 'high'
|
|
|
|
|
|
@router.post("/connection/test", response_model=ConnectionTestResponse)
|
|
async def test_connection_speed(request: ConnectionTestRequest) -> ConnectionTestResponse:
|
|
"""
|
|
Test connection speed and return quality recommendation.
|
|
|
|
This endpoint helps determine appropriate thumbnail quality.
|
|
The client measures download time of test data to calculate speed.
|
|
|
|
Args:
|
|
request: Test configuration
|
|
|
|
Returns:
|
|
Connection quality information and recommendations
|
|
"""
|
|
# Record start time for latency measurement
|
|
start_time = time.time()
|
|
|
|
# Simulate latency measurement (in real implementation, client measures this)
|
|
latency_ms = (time.time() - start_time) * 1000
|
|
|
|
# Client will measure actual download time
|
|
# Here we just provide the test data size for calculation
|
|
# The client calculates: speed_mbps = (test_size_bytes * 8) / (download_time_seconds * 1_000_000)
|
|
|
|
# For now, we return a standard response
|
|
# In practice, the client does the speed calculation
|
|
return ConnectionTestResponse(
|
|
speed_mbps=0.0, # Client calculates this
|
|
latency_ms=latency_ms,
|
|
quality_tier="medium",
|
|
recommended_thumbnail="medium",
|
|
)
|
|
|
|
|
|
@router.get("/connection/test-data")
|
|
async def get_test_data(size: int = 100000) -> bytes:
|
|
"""
|
|
Serve test data for connection speed measurement.
|
|
|
|
Client downloads this and measures time to calculate speed.
|
|
|
|
Args:
|
|
size: Size of test data in bytes (max 500KB)
|
|
|
|
Returns:
|
|
Random bytes for speed testing
|
|
"""
|
|
import secrets
|
|
|
|
# Cap size at 500KB to prevent abuse
|
|
size = min(size, 500000)
|
|
|
|
# Generate random bytes
|
|
return secrets.token_bytes(size)
|