"""Auth, users, audit logs, and API token routes.""" from __future__ import annotations from typing import Annotated, Any from fastapi import APIRouter, Depends, Query, Request from sqlalchemy.orm import Session from .auth_deps import AuthContext, require_admin, require_user from .auth_schemas import ( ApiTokenCreateRequest, ChangePasswordRequest, LoginRequest, UserCreateRequest, UserUpdateRequest, ) from .auth_service import ( authenticate_user, change_password, create_api_token, create_user, list_api_tokens, list_audit_logs, list_users, login_issue_token, revoke_api_token, update_user, user_public, write_audit, ) from .db import get_db router = APIRouter(tags=["auth"]) def _client_meta(request: Request) -> tuple[str, str]: ip = str(request.client.host if request.client else "") ua = str(request.headers.get("user-agent") or "")[:512] return ip, ua @router.post("/v1/auth/login") def api_login(body: LoginRequest, request: Request, db: Session = Depends(get_db)) -> dict[str, Any]: ip, ua = _client_meta(request) user = authenticate_user(db, body.username, body.password) if user is None: write_audit( db, action="auth.login_failed", actor_username=str(body.username or "").strip(), method="POST", path="/v1/auth/login", status_code=401, client_ip=ip, user_agent=ua, detail={}, ) from fastapi import HTTPException raise HTTPException(status_code=401, detail="invalid_credentials") out = login_issue_token(user) write_audit( db, action="auth.login", actor_user_id=user.id, actor_username=user.username, method="POST", path="/v1/auth/login", status_code=200, client_ip=ip, user_agent=ua, detail={"role": user.role}, ) return out @router.post("/v1/auth/logout") def api_logout( request: Request, ctx: Annotated[AuthContext, Depends(require_user)], db: Session = Depends(get_db), ) -> dict[str, Any]: ip, ua = _client_meta(request) write_audit( db, action="auth.logout", actor_user_id=ctx.user.id, actor_username=ctx.user.username, method="POST", path="/v1/auth/logout", status_code=200, client_ip=ip, user_agent=ua, detail={"auth_via": ctx.auth_via}, ) return {"ok": True} @router.get("/v1/auth/me") def api_me(ctx: Annotated[AuthContext, Depends(require_user)]) -> dict[str, Any]: from .auth_scopes import ALL_SCOPES return { "user": user_public(ctx.user), "auth_via": ctx.auth_via, "scopes": sorted(ctx.scopes), "all_scopes": sorted(ALL_SCOPES), } @router.post("/v1/auth/change-password") def api_change_password( body: ChangePasswordRequest, request: Request, ctx: Annotated[AuthContext, Depends(require_user)], db: Session = Depends(get_db), ) -> dict[str, Any]: change_password(db, user=ctx.user, old_password=body.old_password, new_password=body.new_password) ip, ua = _client_meta(request) write_audit( db, action="auth.change_password", actor_user_id=ctx.user.id, actor_username=ctx.user.username, method="POST", path="/v1/auth/change-password", status_code=200, client_ip=ip, user_agent=ua, detail={}, ) return {"ok": True} @router.get("/v1/users") def api_list_users(ctx: Annotated[AuthContext, Depends(require_admin)], db: Session = Depends(get_db)) -> dict[str, Any]: del ctx return {"items": list_users(db)} @router.post("/v1/users") def api_create_user( body: UserCreateRequest, request: Request, ctx: Annotated[AuthContext, Depends(require_admin)], db: Session = Depends(get_db), ) -> dict[str, Any]: user = create_user( db, username=body.username, password=body.password, role=body.role, actor=ctx.user, scopes=body.scopes, ) ip, ua = _client_meta(request) write_audit( db, action="users.create", actor_user_id=ctx.user.id, actor_username=ctx.user.username, method="POST", path="/v1/users", status_code=200, client_ip=ip, user_agent=ua, detail={"target_username": user.username, "role": user.role}, ) return {"user": user_public(user)} @router.patch("/v1/users/{user_id}") def api_update_user( user_id: str, body: UserUpdateRequest, request: Request, ctx: Annotated[AuthContext, Depends(require_admin)], db: Session = Depends(get_db), ) -> dict[str, Any]: user = update_user( db, user_id=user_id, actor=ctx.user, is_active=body.is_active, role=body.role, password=body.password, scopes=body.scopes, ) ip, ua = _client_meta(request) write_audit( db, action="users.update", actor_user_id=ctx.user.id, actor_username=ctx.user.username, method="PATCH", path=f"/v1/users/{user_id}", status_code=200, client_ip=ip, user_agent=ua, detail={ "target_username": user.username, "is_active": user.is_active, "role": user.role, "password_reset": body.password is not None, }, ) return {"user": user_public(user)} @router.get("/v1/audit-logs") def api_audit_logs( ctx: Annotated[AuthContext, Depends(require_user)], db: Session = Depends(get_db), page: int = Query(default=1, ge=1), page_size: int = Query(default=50, ge=1, le=200), username: str = Query(default=""), action: str = Query(default=""), ) -> dict[str, Any]: return list_audit_logs( db, actor=ctx.user, page=page, page_size=page_size, username=username, action=action, ) @router.get("/v1/api-tokens") def api_list_tokens( ctx: Annotated[AuthContext, Depends(require_user)], db: Session = Depends(get_db), ) -> dict[str, Any]: user_id = None if ctx.user.role == "admin" else ctx.user.id return {"items": list_api_tokens(db, user_id=user_id)} @router.post("/v1/api-tokens") def api_create_token( body: ApiTokenCreateRequest, request: Request, ctx: Annotated[AuthContext, Depends(require_user)], db: Session = Depends(get_db), ) -> dict[str, Any]: from .auth_service import get_user_by_id target = ctx.user target_user_id = str(body.user_id or "").strip() if target_user_id and target_user_id != ctx.user.id: if ctx.user.role != "admin": from fastapi import HTTPException raise HTTPException(status_code=403, detail="admin_required") other = get_user_by_id(db, target_user_id) if other is None or not other.is_active: from fastapi import HTTPException raise HTTPException(status_code=404, detail="user_not_found") target = other expires_in_days = body.expires_in_days if expires_in_days is None: expires_in_days = 90 row, plaintext = create_api_token( db, user=target, name=body.name, expires_in_days=expires_in_days, scopes=body.scopes, ) ip, ua = _client_meta(request) write_audit( db, action="api_tokens.create", actor_user_id=ctx.user.id, actor_username=ctx.user.username, method="POST", path="/v1/api-tokens", status_code=200, client_ip=ip, user_agent=ua, detail={ "token_id": row.id, "name": row.name, "owner_user_id": target.id, "owner_username": target.username, "scopes": getattr(row, "scopes", None) or [], "expires_at": row.expires_at.isoformat() if row.expires_at else None, }, ) return { "token": { "id": row.id, "name": row.name, "user_id": row.user_id, "username": target.username, "scopes": getattr(row, "scopes", None) or [], "created_at": row.created_at.isoformat() if row.created_at else None, "expires_at": row.expires_at.isoformat() if row.expires_at else None, "token": plaintext, } } @router.delete("/v1/api-tokens/{token_id}") def api_revoke_token( token_id: str, request: Request, ctx: Annotated[AuthContext, Depends(require_user)], db: Session = Depends(get_db), ) -> dict[str, Any]: row = revoke_api_token(db, token_id=token_id, actor=ctx.user) ip, ua = _client_meta(request) write_audit( db, action="api_tokens.revoke", actor_user_id=ctx.user.id, actor_username=ctx.user.username, method="DELETE", path=f"/v1/api-tokens/{token_id}", status_code=200, client_ip=ip, user_agent=ua, detail={"token_id": row.id, "name": row.name}, ) return {"ok": True, "id": row.id}