from __future__ import annotations

from fastapi import APIRouter, Request
from fastapi.responses import RedirectResponse

from app.core.db.common import CommonSessionLocal
from app.core.http_responses import ui_access_denied
from app.core.security.csrf import get_or_create_csrf_token
from app.core.security.session_auth import get_current_user
from app.core.templating import templates
from app.modules.core.rbac.deps import get_user_permissions, get_user_roles
from app.modules.partner_dashboard.service import build_partner_dashboard_payload, can_access_partner_dashboard

router = APIRouter(prefix="/partner", tags=["partner-dashboard-v2-ui"])

VALID_TABS = {
    "overview": "modules/partner_dashboard/templates/partner_dashboard/partials/overview.html",
    "branch-work": "modules/partner_dashboard/templates/partner_dashboard/partials/branch_work.html",
    "clients": "modules/partner_dashboard/templates/partner_dashboard/partials/clients.html",
    "staff": "modules/partner_dashboard/templates/partner_dashboard/partials/staff.html",
    "review": "modules/partner_dashboard/templates/partner_dashboard/partials/review.html",
    "billing": "modules/partner_dashboard/templates/partner_dashboard/partials/billing.html",
    "reports": "modules/partner_dashboard/templates/partner_dashboard/partials/reports.html",
    "wizards": "modules/partner_dashboard/templates/partner_dashboard/partials/wizards.html",
}


def _ctx(request: Request, db, current_user, *, active_tab: str = "overview"):
    payload = build_partner_dashboard_payload(db, request, current_user)
    return {
        "request": request,
        "current_user": current_user,
        "current_user_roles": get_user_roles(db, current_user.id),
        "current_user_permissions": get_user_permissions(db, current_user.id),
        "csrf_token": get_or_create_csrf_token(request),
        "title": "Partner Operations",
        "active_tab": active_tab,
        **payload,
    }


@router.get("/dashboard")
def dashboard(request: Request, tab: str = "overview"):
    db = CommonSessionLocal()
    try:
        current_user = get_current_user(request, db=db)
        if not current_user:
            return RedirectResponse(url="/login", status_code=303)
        if not can_access_partner_dashboard(db, current_user):
            return ui_access_denied()
        active_tab = tab if tab in VALID_TABS else "overview"
        return templates.TemplateResponse(
            "modules/partner_dashboard/templates/partner_dashboard/dashboard.html",
            _ctx(request, db, current_user, active_tab=active_tab),
        )
    finally:
        db.close()


@router.get("/dashboard/tab/{tab_name}")
def dashboard_tab(request: Request, tab_name: str):
    db = CommonSessionLocal()
    try:
        current_user = get_current_user(request, db=db)
        if not current_user:
            return RedirectResponse(url="/login", status_code=303)
        if not can_access_partner_dashboard(db, current_user):
            return ui_access_denied()
        active_tab = tab_name if tab_name in VALID_TABS else "overview"
        return templates.TemplateResponse(
            VALID_TABS[active_tab],
            _ctx(request, db, current_user, active_tab=active_tab),
        )
    finally:
        db.close()
