from fastapi import APIRouter, Depends, HTTPException, Query, status
from sqlalchemy.orm import Session

from app.core.config import get_settings
from app.db.session import get_db
from app.models.api_client import ApiClient
from app.repositories.companies_repo import CompaniesRepository
from app.repositories.modules_repo import ModulesRepository
from app.repositories.permission_groups_repo import PermissionGroupsRepository
from app.repositories.sites_repo import SitesRepository
from app.repositories.systems_repo import SystemsRepository
from app.repositories.users_repo import UsersRepository
from app.schemas.catalog import (
    CompanyCreateRequest,
    CompanyItem,
    CompanyUpdateRequest,
    MemberItem,
    MemberUpdateRequest,
    MemberUpsertRequest,
    ModuleCreateRequest,
    ModuleItem,
    ModuleUpdateRequest,
    PermissionGroupCreateRequest,
    PermissionGroupItem,
    PermissionGroupUpdateRequest,
    SiteCreateRequest,
    SiteItem,
    SiteUpdateRequest,
    SystemCreateRequest,
    SystemItem,
    SystemUpdateRequest,
)
from app.schemas.permissions import PermissionGrantRequest, PermissionRevokeRequest
from app.security.api_client_auth import require_api_client
from app.services.authentik_admin_service import AuthentikAdminService

router = APIRouter(prefix="/admin", tags=["admin"])


def _resolve_module_id(db: Session, system_key: str, module_key: str | None) -> str:
    systems_repo = SystemsRepository(db)
    modules_repo = ModulesRepository(db)
    system = systems_repo.get_by_key(system_key)
    if not system:
        raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="system_not_found")
    target_module_key = f"{system_key}.{module_key}" if module_key else f"{system_key}.__system__"
    module = modules_repo.get_by_key(target_module_key)
    if not module:
        module = modules_repo.create(module_key=target_module_key, name=target_module_key, status="active")
    return module.id


def _resolve_scope_ids(db: Session, scope_type: str, scope_id: str) -> tuple[str | None, str | None]:
    companies_repo = CompaniesRepository(db)
    sites_repo = SitesRepository(db)
    if scope_type == "company":
        company = companies_repo.get_by_key(scope_id)
        if not company:
            raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="company_not_found")
        return company.id, None
    if scope_type == "site":
        site = sites_repo.get_by_key(scope_id)
        if not site:
            raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="site_not_found")
        return None, site.id
    raise HTTPException(status_code=status.HTTP_400_BAD_REQUEST, detail="invalid_scope_type")


def _sync_member_to_authentik(
    *,
    authentik_sub: str,
    email: str | None,
    display_name: str | None,
    is_active: bool,
) -> dict[str, str | int]:
    if not email:
        raise HTTPException(status_code=status.HTTP_400_BAD_REQUEST, detail="email_required_for_authentik_sync")
    settings = get_settings()
    service = AuthentikAdminService(settings=settings)
    result = service.ensure_user(
        sub=authentik_sub,
        email=email,
        display_name=display_name,
        is_active=is_active,
    )
    return {"authentik_user_id": result.user_id, "sync_action": result.action}


@router.get("/systems")
def list_systems(
    _: ApiClient = Depends(require_api_client),
    db: Session = Depends(get_db),
    limit: int = Query(default=100, ge=1, le=500),
    offset: int = Query(default=0, ge=0),
) -> dict:
    repo = SystemsRepository(db)
    items, total = repo.list(limit=limit, offset=offset)
    return {"items": [SystemItem(id=i.id, system_key=i.system_key, name=i.name, status=i.status).model_dump() for i in items], "total": total, "limit": limit, "offset": offset}


@router.post("/systems", response_model=SystemItem)
def create_system(
    payload: SystemCreateRequest,
    _: ApiClient = Depends(require_api_client),
    db: Session = Depends(get_db),
) -> SystemItem:
    repo = SystemsRepository(db)
    if repo.get_by_key(payload.system_key):
        raise HTTPException(status_code=status.HTTP_409_CONFLICT, detail="system_key_already_exists")
    row = repo.create(system_key=payload.system_key, name=payload.name, status=payload.status)
    return SystemItem(id=row.id, system_key=row.system_key, name=row.name, status=row.status)


@router.patch("/systems/{system_key}", response_model=SystemItem)
def update_system(
    system_key: str,
    payload: SystemUpdateRequest,
    _: ApiClient = Depends(require_api_client),
    db: Session = Depends(get_db),
) -> SystemItem:
    repo = SystemsRepository(db)
    row = repo.get_by_key(system_key)
    if not row:
        raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="system_not_found")
    row = repo.update(row, name=payload.name, status=payload.status)
    return SystemItem(id=row.id, system_key=row.system_key, name=row.name, status=row.status)


@router.get("/modules")
def list_modules(
    _: ApiClient = Depends(require_api_client),
    db: Session = Depends(get_db),
    limit: int = Query(default=200, ge=1, le=500),
    offset: int = Query(default=0, ge=0),
) -> dict:
    modules_repo = ModulesRepository(db)
    items, total = modules_repo.list(limit=limit, offset=offset)
    out = []
    for i in items:
        system_key = i.module_key.split(".", 1)[0] if "." in i.module_key else None
        out.append(
            ModuleItem(
                id=i.id,
                system_key=system_key,
                module_key=i.module_key,
                name=i.name,
                status=i.status,
            ).model_dump()
        )
    return {"items": out, "total": total, "limit": limit, "offset": offset}


@router.post("/modules", response_model=ModuleItem)
def create_module(
    payload: ModuleCreateRequest,
    _: ApiClient = Depends(require_api_client),
    db: Session = Depends(get_db),
) -> ModuleItem:
    systems_repo = SystemsRepository(db)
    modules_repo = ModulesRepository(db)
    system = systems_repo.get_by_key(payload.system_key)
    if not system:
        raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="system_not_found")
    full_module_key = f"{payload.system_key}.{payload.module_key}"
    if modules_repo.get_by_key(full_module_key):
        raise HTTPException(status_code=status.HTTP_409_CONFLICT, detail="module_key_already_exists")
    row = modules_repo.create(
        module_key=full_module_key,
        name=payload.name,
        status=payload.status,
    )
    return ModuleItem(id=row.id, system_key=payload.system_key, module_key=row.module_key, name=row.name, status=row.status)


@router.patch("/modules/{module_key}")
def update_module(
    module_key: str,
    payload: ModuleUpdateRequest,
    _: ApiClient = Depends(require_api_client),
    db: Session = Depends(get_db),
) -> ModuleItem:
    modules_repo = ModulesRepository(db)
    row = modules_repo.get_by_key(module_key)
    if not row:
        raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="module_not_found")
    row = modules_repo.update(row, name=payload.name, status=payload.status)
    system_key = row.module_key.split(".", 1)[0] if "." in row.module_key else None
    return ModuleItem(id=row.id, system_key=system_key, module_key=row.module_key, name=row.name, status=row.status)


@router.get("/companies")
def list_companies(
    _: ApiClient = Depends(require_api_client),
    db: Session = Depends(get_db),
    keyword: str | None = Query(default=None),
    limit: int = Query(default=100, ge=1, le=500),
    offset: int = Query(default=0, ge=0),
) -> dict:
    repo = CompaniesRepository(db)
    items, total = repo.list(keyword=keyword, limit=limit, offset=offset)
    return {"items": [CompanyItem(id=i.id, company_key=i.company_key, name=i.name, status=i.status).model_dump() for i in items], "total": total, "limit": limit, "offset": offset}


@router.post("/companies", response_model=CompanyItem)
def create_company(
    payload: CompanyCreateRequest,
    _: ApiClient = Depends(require_api_client),
    db: Session = Depends(get_db),
) -> CompanyItem:
    repo = CompaniesRepository(db)
    if repo.get_by_key(payload.company_key):
        raise HTTPException(status_code=status.HTTP_409_CONFLICT, detail="company_key_already_exists")
    row = repo.create(company_key=payload.company_key, name=payload.name, status=payload.status)
    return CompanyItem(id=row.id, company_key=row.company_key, name=row.name, status=row.status)


@router.patch("/companies/{company_key}", response_model=CompanyItem)
def update_company(
    company_key: str,
    payload: CompanyUpdateRequest,
    _: ApiClient = Depends(require_api_client),
    db: Session = Depends(get_db),
) -> CompanyItem:
    repo = CompaniesRepository(db)
    row = repo.get_by_key(company_key)
    if not row:
        raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="company_not_found")
    row = repo.update(row, name=payload.name, status=payload.status)
    return CompanyItem(id=row.id, company_key=row.company_key, name=row.name, status=row.status)


@router.get("/sites")
def list_sites(
    _: ApiClient = Depends(require_api_client),
    db: Session = Depends(get_db),
    company_key: str | None = Query(default=None),
    keyword: str | None = Query(default=None),
    limit: int = Query(default=100, ge=1, le=500),
    offset: int = Query(default=0, ge=0),
) -> dict:
    companies_repo = CompaniesRepository(db)
    sites_repo = SitesRepository(db)
    company_lookup: dict[str, str] = {}
    all_companies, _ = companies_repo.list(limit=1000, offset=0)
    for c in all_companies:
        company_lookup[c.id] = c.company_key
    company_id = None
    if company_key:
        company = companies_repo.get_by_key(company_key)
        if not company:
            raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="company_not_found")
        company_id = company.id
    items, total = sites_repo.list(keyword=keyword, company_id=company_id, limit=limit, offset=offset)
    return {
        "items": [
            SiteItem(
                id=i.id,
                site_key=i.site_key,
                company_key=company_lookup.get(i.company_id, ""),
                name=i.name,
                status=i.status,
            ).model_dump()
            for i in items
        ],
        "total": total,
        "limit": limit,
        "offset": offset,
    }


@router.post("/sites", response_model=SiteItem)
def create_site(
    payload: SiteCreateRequest,
    _: ApiClient = Depends(require_api_client),
    db: Session = Depends(get_db),
) -> SiteItem:
    companies_repo = CompaniesRepository(db)
    sites_repo = SitesRepository(db)
    company = companies_repo.get_by_key(payload.company_key)
    if not company:
        raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="company_not_found")
    if sites_repo.get_by_key(payload.site_key):
        raise HTTPException(status_code=status.HTTP_409_CONFLICT, detail="site_key_already_exists")
    row = sites_repo.create(site_key=payload.site_key, company_id=company.id, name=payload.name, status=payload.status)
    return SiteItem(id=row.id, site_key=row.site_key, company_key=payload.company_key, name=row.name, status=row.status)


@router.patch("/sites/{site_key}", response_model=SiteItem)
def update_site(
    site_key: str,
    payload: SiteUpdateRequest,
    _: ApiClient = Depends(require_api_client),
    db: Session = Depends(get_db),
) -> SiteItem:
    companies_repo = CompaniesRepository(db)
    sites_repo = SitesRepository(db)
    row = sites_repo.get_by_key(site_key)
    if not row:
        raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="site_not_found")
    company_id = None
    company_key = None
    if payload.company_key is not None:
        company = companies_repo.get_by_key(payload.company_key)
        if not company:
            raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="company_not_found")
        company_id = company.id
        company_key = company.company_key
    row = sites_repo.update(row, company_id=company_id, name=payload.name, status=payload.status)
    if company_key is None:
        current_company = companies_repo.get_by_id(row.company_id)
        company_key = current_company.company_key if current_company else ""
    return SiteItem(id=row.id, site_key=row.site_key, company_key=company_key, name=row.name, status=row.status)


@router.get("/members")
def list_members(
    _: ApiClient = Depends(require_api_client),
    db: Session = Depends(get_db),
    keyword: str | None = Query(default=None),
    limit: int = Query(default=100, ge=1, le=500),
    offset: int = Query(default=0, ge=0),
) -> dict:
    users_repo = UsersRepository(db)
    items, total = users_repo.list(keyword=keyword, limit=limit, offset=offset)
    return {"items": [MemberItem(id=i.id, authentik_sub=i.authentik_sub, email=i.email, display_name=i.display_name, is_active=i.is_active).model_dump() for i in items], "total": total, "limit": limit, "offset": offset}


@router.post("/members/upsert", response_model=MemberItem)
def upsert_member(
    payload: MemberUpsertRequest,
    _: ApiClient = Depends(require_api_client),
    db: Session = Depends(get_db),
) -> MemberItem:
    users_repo = UsersRepository(db)
    authentik_user_id = None
    if payload.sync_to_authentik:
        sync = _sync_member_to_authentik(
            authentik_sub=payload.authentik_sub,
            email=payload.email,
            display_name=payload.display_name,
            is_active=payload.is_active,
        )
        authentik_user_id = int(sync["authentik_user_id"])
    row = users_repo.upsert_by_sub(
        authentik_sub=payload.authentik_sub,
        email=payload.email,
        display_name=payload.display_name,
        is_active=payload.is_active,
        authentik_user_id=authentik_user_id,
    )
    return MemberItem(
        id=row.id,
        authentik_sub=row.authentik_sub,
        email=row.email,
        display_name=row.display_name,
        is_active=row.is_active,
    )


@router.patch("/members/{authentik_sub}", response_model=MemberItem)
def update_member(
    authentik_sub: str,
    payload: MemberUpdateRequest,
    _: ApiClient = Depends(require_api_client),
    db: Session = Depends(get_db),
) -> MemberItem:
    users_repo = UsersRepository(db)
    row = users_repo.get_by_sub(authentik_sub)
    if not row:
        raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="user_not_found")

    next_email = payload.email if payload.email is not None else row.email
    next_display_name = payload.display_name if payload.display_name is not None else row.display_name
    next_is_active = payload.is_active if payload.is_active is not None else row.is_active

    authentik_user_id = row.authentik_user_id
    if payload.sync_to_authentik:
        sync = _sync_member_to_authentik(
            authentik_sub=row.authentik_sub,
            email=next_email,
            display_name=next_display_name,
            is_active=next_is_active,
        )
        authentik_user_id = int(sync["authentik_user_id"])

    row = users_repo.upsert_by_sub(
        authentik_sub=row.authentik_sub,
        email=next_email,
        display_name=next_display_name,
        is_active=next_is_active,
        authentik_user_id=authentik_user_id,
    )
    return MemberItem(
        id=row.id,
        authentik_sub=row.authentik_sub,
        email=row.email,
        display_name=row.display_name,
        is_active=row.is_active,
    )


@router.get("/permission-groups")
def list_permission_groups(
    _: ApiClient = Depends(require_api_client),
    db: Session = Depends(get_db),
    limit: int = Query(default=100, ge=1, le=500),
    offset: int = Query(default=0, ge=0),
) -> dict:
    repo = PermissionGroupsRepository(db)
    items, total = repo.list(limit=limit, offset=offset)
    return {"items": [PermissionGroupItem(id=i.id, group_key=i.group_key, name=i.name, status=i.status).model_dump() for i in items], "total": total, "limit": limit, "offset": offset}


@router.post("/permission-groups", response_model=PermissionGroupItem)
def create_permission_group(
    payload: PermissionGroupCreateRequest,
    _: ApiClient = Depends(require_api_client),
    db: Session = Depends(get_db),
) -> PermissionGroupItem:
    repo = PermissionGroupsRepository(db)
    if repo.get_by_key(payload.group_key):
        raise HTTPException(status_code=status.HTTP_409_CONFLICT, detail="group_key_already_exists")
    row = repo.create(group_key=payload.group_key, name=payload.name, status=payload.status)
    return PermissionGroupItem(id=row.id, group_key=row.group_key, name=row.name, status=row.status)


@router.patch("/permission-groups/{group_key}", response_model=PermissionGroupItem)
def update_permission_group(
    group_key: str,
    payload: PermissionGroupUpdateRequest,
    _: ApiClient = Depends(require_api_client),
    db: Session = Depends(get_db),
) -> PermissionGroupItem:
    repo = PermissionGroupsRepository(db)
    row = repo.get_by_key(group_key)
    if not row:
        raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="group_not_found")
    row = repo.update(row, name=payload.name, status=payload.status)
    return PermissionGroupItem(id=row.id, group_key=row.group_key, name=row.name, status=row.status)


@router.post("/permission-groups/{group_key}/members/{authentik_sub}")
def add_group_member(
    group_key: str,
    authentik_sub: str,
    _: ApiClient = Depends(require_api_client),
    db: Session = Depends(get_db),
) -> dict[str, str]:
    groups_repo = PermissionGroupsRepository(db)
    group = groups_repo.get_by_key(group_key)
    if not group:
        raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="group_not_found")
    row = groups_repo.add_member_if_not_exists(group.id, authentik_sub)
    return {"membership_id": row.id, "result": "added"}


@router.delete("/permission-groups/{group_key}/members/{authentik_sub}")
def remove_group_member(
    group_key: str,
    authentik_sub: str,
    _: ApiClient = Depends(require_api_client),
    db: Session = Depends(get_db),
) -> dict[str, int | str]:
    groups_repo = PermissionGroupsRepository(db)
    group = groups_repo.get_by_key(group_key)
    if not group:
        raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="group_not_found")
    deleted = groups_repo.remove_member(group.id, authentik_sub)
    return {"deleted": deleted, "result": "removed"}


@router.post("/permission-groups/{group_key}/permissions/grant")
def grant_group_permission(
    group_key: str,
    payload: PermissionGrantRequest,
    _: ApiClient = Depends(require_api_client),
    db: Session = Depends(get_db),
) -> dict[str, str]:
    groups_repo = PermissionGroupsRepository(db)
    group = groups_repo.get_by_key(group_key)
    if not group:
        raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="group_not_found")
    _resolve_module_id(db, payload.system, payload.module)
    _resolve_scope_ids(db, payload.scope_type, payload.scope_id)
    module_key = f"{payload.system}.{payload.module}" if payload.module else f"{payload.system}.__system__"
    row = groups_repo.grant_group_permission(
        group_id=group.id,
        system=payload.system,
        module=module_key,
        action=payload.action,
        scope_type=payload.scope_type,
        scope_id=payload.scope_id,
    )
    return {"permission_id": row.id, "result": "granted"}


@router.post("/permission-groups/{group_key}/permissions/revoke")
def revoke_group_permission(
    group_key: str,
    payload: PermissionRevokeRequest,
    _: ApiClient = Depends(require_api_client),
    db: Session = Depends(get_db),
) -> dict[str, int | str]:
    groups_repo = PermissionGroupsRepository(db)
    group = groups_repo.get_by_key(group_key)
    if not group:
        raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="group_not_found")
    _resolve_module_id(db, payload.system, payload.module)
    _resolve_scope_ids(db, payload.scope_type, payload.scope_id)
    module_key = f"{payload.system}.{payload.module}" if payload.module else f"{payload.system}.__system__"
    deleted = groups_repo.revoke_group_permission(
        group_id=group.id,
        system=payload.system,
        module=module_key,
        action=payload.action,
        scope_type=payload.scope_type,
        scope_id=payload.scope_id,
    )
    return {"deleted": deleted, "result": "revoked"}