from fastapi import APIRouter, BackgroundTasks, Depends, Form, Request, Response from fastapi.responses import HTMLResponse, RedirectResponse from fastapi.templating import Jinja2Templates from sqlalchemy import desc, func, or_, select from sqlalchemy.orm import Session from app.config import Settings, get_settings from app.db import SessionLocal, get_db from app.models import CrawlError, CrawlRun, Employee from app.security import SESSION_COOKIE, require_admin, sign_session, verify_admin from app.services.crawler import run_crawl from app.version import BACKEND_VERSION, FRONTEND_VERSION router = APIRouter(prefix="/admin") templates = Jinja2Templates(directory="app/templates") @router.get("", response_class=HTMLResponse) def dashboard(request: Request, db: Session = Depends(get_db), settings: Settings = Depends(get_settings)): require_admin(request, settings) counts = { "active": db.scalar(select(func.count()).select_from(Employee).where(Employee.status == "active")) or 0, "dismissed": db.scalar(select(func.count()).select_from(Employee).where(Employee.status == "dismissed")) or 0, "runs": db.scalar(select(func.count()).select_from(CrawlRun)) or 0, "errors": db.scalar(select(func.count()).select_from(CrawlError)) or 0, } runs = db.scalars(select(CrawlRun).order_by(desc(CrawlRun.started_at)).limit(10)).all() return _render(request, "dashboard.html", {"counts": counts, "runs": runs}) @router.get("/login", response_class=HTMLResponse) def login_form(request: Request): return _render(request, "login.html", {"error": None}) @router.post("/login") def login( response: Response, request: Request, username: str = Form(...), password: str = Form(...), settings: Settings = Depends(get_settings), ): if not verify_admin(username, password, settings): return _render(request, "login.html", {"error": "Неверный логин или пароль"}, status_code=401) redirect = RedirectResponse("/admin", status_code=303) redirect.set_cookie(SESSION_COOKIE, sign_session(username, settings), httponly=True, samesite="lax") return redirect @router.post("/logout") def logout(): redirect = RedirectResponse("/admin/login", status_code=303) redirect.delete_cookie(SESSION_COOKIE) return redirect @router.get("/employees", response_class=HTMLResponse) def employees( request: Request, status: str | None = None, q: str | None = None, db: Session = Depends(get_db), settings: Settings = Depends(get_settings), ): require_admin(request, settings) stmt = select(Employee) if status: stmt = stmt.where(Employee.status == status) if q: pattern = f"%{q}%" stmt = stmt.where(or_(Employee.full_name.ilike(pattern), Employee.canonical_url.ilike(pattern))) items = db.scalars(stmt.order_by(Employee.full_name).limit(200)).all() return _render(request, "employees.html", {"employees": items, "status": status or "", "q": q or ""}) @router.get("/employees/{employee_id}", response_class=HTMLResponse) def employee_detail( employee_id: int, request: Request, db: Session = Depends(get_db), settings: Settings = Depends(get_settings), ): require_admin(request, settings) employee = db.get(Employee, employee_id) if not employee: return RedirectResponse("/admin/employees", status_code=303) snapshots = sorted(employee.snapshots, key=lambda item: item.captured_at, reverse=True)[:20] return _render(request, "employee_detail.html", {"employee": employee, "snapshots": snapshots}) @router.get("/runs", response_class=HTMLResponse) def runs(request: Request, db: Session = Depends(get_db), settings: Settings = Depends(get_settings)): require_admin(request, settings) items = db.scalars(select(CrawlRun).order_by(desc(CrawlRun.started_at)).limit(50)).all() errors = db.scalars(select(CrawlError).order_by(desc(CrawlError.created_at)).limit(50)).all() return _render(request, "runs.html", {"runs": items, "errors": errors}) @router.post("/runs") def trigger_run( request: Request, background_tasks: BackgroundTasks, settings: Settings = Depends(get_settings), ): require_admin(request, settings) def _crawl() -> None: with SessionLocal() as db: run_crawl(db, settings) background_tasks.add_task(_crawl) return RedirectResponse("/admin/runs", status_code=303) def _render(request: Request, template: str, context: dict, status_code: int = 200) -> HTMLResponse: payload = { "request": request, "backend_version": BACKEND_VERSION, "frontend_version": FRONTEND_VERSION, **context, } return templates.TemplateResponse(template, payload, status_code=status_code)