HR-ATS-Portal/backend/org_settings/app.py

from typing import Any

from fastapi import APIRouter, Depends, HTTPException, Query
from fastapi.responses import JSONResponse
from pydantic import BaseModel
from sqlalchemy.ext.asyncio import AsyncSession

from db_setup import get_session
from org_settings.views import Exclusion, OrgSetting
from users.permissions import PermissionTag, require_permission

router = APIRouter()


class OrgSettingItem(BaseModel):
    key: str
    value: Any = None
    category: str


class OrgSettingsUpdate(BaseModel):
    settings: list[OrgSettingItem]


@router.get("/org-settings/fetch")
async def fetch_org_settings(
    current_user: dict = Depends(require_permission(PermissionTag.SETTINGS_VIEW)),
    category: str | None = Query(None),
    session: AsyncSession = Depends(get_session),
):
    try:
        service = OrgSetting(session=session)
        data, total = await service.get_settings(category)
        return JSONResponse(content={"data": data, "total": total, "status_code": 200})
    except HTTPException:
        raise
    except Exception as e:
        raise HTTPException(status_code=500, detail=str(e))


@router.put("/org-settings/update")
async def update_org_settings(
    payload: OrgSettingsUpdate,
    current_user: dict = Depends(require_permission(PermissionTag.SETTINGS_CONFIGURE)),
    session: AsyncSession = Depends(get_session),
):
    try:
        service = OrgSetting(session=session)
        data = await service.update_settings(payload.model_dump(exclude_unset=True), current_user)
        return JSONResponse(content={"data": data, "total": len(data), "status_code": 200})
    except HTTPException:
        raise
    except Exception as e:
        raise HTTPException(status_code=500, detail=str(e))


class ExcludeUniversityCreate(BaseModel):
    name: str


class ExcludeUniversityUpdate(BaseModel):
    name: str | None = None


class ExcludeCompanyCreate(BaseModel):
    name: str
    linkedin_url: str | None = None


class ExcludeCompanyUpdate(BaseModel):
    name: str | None = None
    linkedin_url: str | None = None


@router.get("/org-settings/exclude-university/fetch")
async def fetch_exclude_universities(
    current_user: dict = Depends(require_permission(PermissionTag.SETTINGS_VIEW)),
    session: AsyncSession = Depends(get_session),
):
    try:
        service = Exclusion(session=session)
        data, total = await service.get_universities()
        return JSONResponse(content={"data": data, "total": total, "status_code": 200})
    except HTTPException:
        raise
    except Exception as e:
        raise HTTPException(status_code=500, detail=str(e))


@router.post("/org-settings/exclude-university/create")
async def create_exclude_university(
    payload: ExcludeUniversityCreate,
    current_user: dict = Depends(require_permission(PermissionTag.SETTINGS_CONFIGURE)),
    session: AsyncSession = Depends(get_session),
):
    try:
        service = Exclusion(session=session)
        data = await service.create_university(payload.model_dump(exclude_unset=True), current_user)
        return JSONResponse(content={"data": data, "status_code": 200})
    except HTTPException:
        raise
    except Exception as e:
        raise HTTPException(status_code=500, detail=str(e))


@router.patch("/org-settings/exclude-university/update")
async def update_exclude_university(
    payload: ExcludeUniversityUpdate,
    current_user: dict = Depends(require_permission(PermissionTag.SETTINGS_CONFIGURE)),
    record_id: str = Query(...),
    session: AsyncSession = Depends(get_session),
):
    try:
        service = Exclusion(session=session)
        data = await service.update_university(
            record_id, payload.model_dump(exclude_unset=True), current_user
        )
        return JSONResponse(content={"data": data, "status_code": 200})
    except HTTPException:
        raise
    except Exception as e:
        raise HTTPException(status_code=500, detail=str(e))


@router.delete("/org-settings/exclude-university/delete")
async def delete_exclude_university(
    current_user: dict = Depends(require_permission(PermissionTag.SETTINGS_CONFIGURE)),
    record_id: str = Query(...),
    session: AsyncSession = Depends(get_session),
):
    try:
        service = Exclusion(session=session)
        data = await service.delete_university(record_id, current_user)
        return JSONResponse(content={"data": data, "status_code": 200})
    except HTTPException:
        raise
    except Exception as e:
        raise HTTPException(status_code=500, detail=str(e))


@router.get("/org-settings/exclude-company/fetch")
async def fetch_exclude_companies(
    current_user: dict = Depends(require_permission(PermissionTag.SETTINGS_VIEW)),
    session: AsyncSession = Depends(get_session),
):
    try:
        service = Exclusion(session=session)
        data, total = await service.get_companies()
        return JSONResponse(content={"data": data, "total": total, "status_code": 200})
    except HTTPException:
        raise
    except Exception as e:
        raise HTTPException(status_code=500, detail=str(e))


@router.post("/org-settings/exclude-company/create")
async def create_exclude_company(
    payload: ExcludeCompanyCreate,
    current_user: dict = Depends(require_permission(PermissionTag.SETTINGS_CONFIGURE)),
    session: AsyncSession = Depends(get_session),
):
    try:
        service = Exclusion(session=session)
        data = await service.create_company(payload.model_dump(exclude_unset=True), current_user)
        return JSONResponse(content={"data": data, "status_code": 200})
    except HTTPException:
        raise
    except Exception as e:
        raise HTTPException(status_code=500, detail=str(e))


@router.patch("/org-settings/exclude-company/update")
async def update_exclude_company(
    payload: ExcludeCompanyUpdate,
    current_user: dict = Depends(require_permission(PermissionTag.SETTINGS_CONFIGURE)),
    record_id: str = Query(...),
    session: AsyncSession = Depends(get_session),
):
    try:
        service = Exclusion(session=session)
        data = await service.update_company(
            record_id, payload.model_dump(exclude_unset=True), current_user
        )
        return JSONResponse(content={"data": data, "status_code": 200})
    except HTTPException:
        raise
    except Exception as e:
        raise HTTPException(status_code=500, detail=str(e))


@router.delete("/org-settings/exclude-company/delete")
async def delete_exclude_company(
    current_user: dict = Depends(require_permission(PermissionTag.SETTINGS_CONFIGURE)),
    record_id: str = Query(...),
    session: AsyncSession = Depends(get_session),
):
    try:
        service = Exclusion(session=session)
        data = await service.delete_company(record_id, current_user)
        return JSONResponse(content={"data": data, "status_code": 200})
    except HTTPException:
        raise
    except Exception as e:
        raise HTTPException(status_code=500, detail=str(e))