mirror of
https://github.com/bellingcat/auto-archiver-api.git
synced 2026-06-08 03:28:35 +03:00
60 lines
2.1 KiB
Python
60 lines
2.1 KiB
Python
|
|
from typing import Dict
|
|
from fastapi import APIRouter, Depends, Request, HTTPException
|
|
from fastapi.responses import FileResponse, JSONResponse
|
|
|
|
from app.shared.config import VERSION, BREAKING_CHANGES
|
|
from app.shared.log import log_error
|
|
from app.shared.db import crud
|
|
from app.shared.schemas import ActiveUser, UsageResponse
|
|
from app.web.db.user_state import UserState
|
|
from app.web.security import get_user_auth, bearer_security, get_user_state
|
|
from app.shared.user_groups import GroupInfo
|
|
|
|
default_router = APIRouter()
|
|
|
|
|
|
@default_router.get("/")
|
|
async def home(request: Request):
|
|
# TODO: maybe split into 2 routes: one non authenticated and one authenticated for the groups info only, necessary only for the extension
|
|
status = {"version": VERSION, "breakingChanges": BREAKING_CHANGES}
|
|
try:
|
|
email = await get_user_auth(await bearer_security(request))
|
|
status["groups"] = crud.get_user_groups(email)
|
|
except HTTPException: pass # not authenticated is fine
|
|
except Exception as e: log_error(e)
|
|
return JSONResponse(status)
|
|
|
|
|
|
@default_router.get("/health")
|
|
async def health():
|
|
return JSONResponse({"status": "ok"})
|
|
|
|
|
|
@default_router.get("/user/active", summary="Check if the user is active and can use the tool.")
|
|
async def active(
|
|
user: UserState = Depends(get_user_state),
|
|
) -> ActiveUser:
|
|
return {"active": user.active}
|
|
|
|
|
|
@default_router.get("/user/permissions", summary="Get the user's global 'all' permissions and the permissions for each group they belong to.")
|
|
def get_user_permissions(
|
|
user: UserState = Depends(get_user_state),
|
|
) -> Dict[str, GroupInfo]:
|
|
return user.permissions
|
|
|
|
@default_router.get("/user/usage", summary="Get the user's monthly URLs/MBs usage along with the total active sheets, breakdown by group.")
|
|
def get_user_usage(
|
|
user: UserState = Depends(get_user_state),
|
|
) -> UsageResponse:
|
|
if not user.active:
|
|
raise HTTPException(status_code=403, detail="User is not active.")
|
|
return user.usage()
|
|
|
|
|
|
|
|
@default_router.get('/favicon.ico', include_in_schema=False)
|
|
async def favicon() -> FileResponse:
|
|
return FileResponse("web/static/favicon.ico")
|