115 lines
4.3 KiB
Python
115 lines
4.3 KiB
Python
from fastapi import APIRouter, Form, HTTPException, Request
|
|
from fastapi.responses import HTMLResponse
|
|
from sqlalchemy import func, select
|
|
|
|
from app.auth.admin import require_admin
|
|
from app.auth.csrf import ensure_csrf_token, validate_csrf
|
|
from app.db.models import Activity
|
|
from app.db.repositories import ActivityRepository, UserRepository
|
|
from app.sync.manager import SyncAlreadyRunning
|
|
from app.web.routes import templates
|
|
|
|
router = APIRouter()
|
|
|
|
APP_VERSION = "1.0.0"
|
|
|
|
|
|
def _normalize_outcome(item):
|
|
if isinstance(item, Exception):
|
|
return {
|
|
"status": "error",
|
|
"user_id": None,
|
|
"discovered": 0,
|
|
"imported": 0,
|
|
"skipped": 0,
|
|
"failed": 0,
|
|
"message": str(item),
|
|
}
|
|
return {
|
|
"status": item.status,
|
|
"user_id": item.user_id,
|
|
"discovered": item.discovered,
|
|
"imported": item.imported,
|
|
"skipped": item.skipped,
|
|
"failed": item.failed,
|
|
"message": item.message,
|
|
}
|
|
|
|
|
|
@router.post("/users/{user_id}/sync", response_class=HTMLResponse)
|
|
async def manual_sync(request: Request, user_id: int, csrf_token: str = Form(...)):
|
|
require_admin(request)
|
|
validate_csrf(request, csrf_token)
|
|
try:
|
|
outcome = await request.app.state.sync_manager.sync_user(user_id)
|
|
except SyncAlreadyRunning:
|
|
return HTMLResponse("Sync already running for this user", status_code=409)
|
|
return templates.TemplateResponse(
|
|
request, "fragments/sync_result.html", {"outcomes": [_normalize_outcome(outcome)]}
|
|
)
|
|
|
|
|
|
@router.post("/sync-all", response_class=HTMLResponse)
|
|
async def manual_sync_all(request: Request, csrf_token: str = Form(...)):
|
|
require_admin(request)
|
|
validate_csrf(request, csrf_token)
|
|
outcomes = await request.app.state.sync_manager.sync_all_enabled()
|
|
return templates.TemplateResponse(
|
|
request, "fragments/sync_result.html", {"outcomes": [_normalize_outcome(o) for o in outcomes]}
|
|
)
|
|
|
|
|
|
@router.post("/users/{user_id}/garmin-mfa", response_class=HTMLResponse)
|
|
async def garmin_mfa(request: Request, user_id: int, csrf_token: str = Form(...), code: str = Form(...)):
|
|
require_admin(request)
|
|
validate_csrf(request, csrf_token)
|
|
stripped = code.strip()
|
|
if not stripped or len(stripped) > 20:
|
|
raise HTTPException(status_code=400, detail="Invalid MFA code")
|
|
try:
|
|
outcome = await request.app.state.sync_manager.sync_user(user_id, mfa_code=stripped)
|
|
except SyncAlreadyRunning:
|
|
return HTMLResponse("Sync already running for this user", status_code=409)
|
|
return templates.TemplateResponse(
|
|
request, "fragments/sync_result.html", {"outcomes": [_normalize_outcome(outcome)]}
|
|
)
|
|
|
|
|
|
@router.post("/activities/{activity_id}/retry", response_class=HTMLResponse)
|
|
async def retry_activity(request: Request, activity_id: int, csrf_token: str = Form(...)):
|
|
require_admin(request)
|
|
validate_csrf(request, csrf_token)
|
|
with request.app.state.session_factory() as session:
|
|
activity_repo = ActivityRepository(session)
|
|
try:
|
|
activity = activity_repo.reset_retryable_failure(activity_id)
|
|
except ValueError:
|
|
return HTMLResponse("Activity is not retryable", status_code=409)
|
|
user_id = activity.user_id
|
|
try:
|
|
outcome = await request.app.state.sync_manager.sync_user(user_id)
|
|
except SyncAlreadyRunning:
|
|
return HTMLResponse("Sync already running for this user", status_code=409)
|
|
return templates.TemplateResponse(
|
|
request, "fragments/sync_result.html", {"outcomes": [_normalize_outcome(outcome)]}
|
|
)
|
|
|
|
|
|
@router.get("/system", response_class=HTMLResponse)
|
|
def system_page(request: Request):
|
|
require_admin(request)
|
|
settings = request.app.state.settings
|
|
scheduler = request.app.state.scheduler
|
|
with request.app.state.session_factory() as session:
|
|
user_count = len(UserRepository(session).list_all())
|
|
activity_count = session.scalar(select(func.count()).select_from(Activity)) or 0
|
|
return templates.TemplateResponse(request, "system.html", {
|
|
"csrf_token": ensure_csrf_token(request),
|
|
"app_version": APP_VERSION,
|
|
"sync_interval_minutes": settings.sync_interval_minutes,
|
|
"last_tick": scheduler.last_tick,
|
|
"next_tick": scheduler.next_tick,
|
|
"user_count": user_count,
|
|
"activity_count": activity_count,
|
|
})
|