Add HR agent onboarding model and actor enforcement

backend/app/api/hr.py

@@ -3,10 +3,10 @@ from __future__ import annotations
 from fastapi import APIRouter, Depends, HTTPException
 from sqlmodel import Session, select
 
-from app.api.utils import log_activity
+from app.api.utils import log_activity, get_actor_employee_id
 from app.db.session import get_session
-from app.models.hr import EmploymentAction, HeadcountRequest
-from app.schemas.hr import EmploymentActionCreate, HeadcountRequestCreate, HeadcountRequestUpdate
+from app.models.hr import EmploymentAction, HeadcountRequest, AgentOnboarding
+from app.schemas.hr import EmploymentActionCreate, HeadcountRequestCreate, HeadcountRequestUpdate, AgentOnboardingCreate, AgentOnboardingUpdate
 
 router = APIRouter(prefix="/hr", tags=["hr"])
 
@@ -17,18 +17,18 @@ def list_headcount_requests(session: Session = Depends(get_session)):
 
 
 @router.post("/headcount", response_model=HeadcountRequest)
-def create_headcount_request(payload: HeadcountRequestCreate, session: Session = Depends(get_session)):
+def create_headcount_request(payload: HeadcountRequestCreate, session: Session = Depends(get_session), actor_employee_id: int = Depends(get_actor_employee_id)):
     req = HeadcountRequest(**payload.model_dump())
     session.add(req)
     session.commit()
     session.refresh(req)
-    log_activity(session, actor_employee_id=req.requested_by_manager_id, entity_type="headcount_request", entity_id=req.id, verb="submitted")
+    log_activity(session, actor_employee_id=actor_employee_id, entity_type="headcount_request", entity_id=req.id, verb="submitted")
     session.commit()
     return req
 
 
 @router.patch("/headcount/{request_id}", response_model=HeadcountRequest)
-def update_headcount_request(request_id: int, payload: HeadcountRequestUpdate, session: Session = Depends(get_session)):
+def update_headcount_request(request_id: int, payload: HeadcountRequestUpdate, session: Session = Depends(get_session), actor_employee_id: int = Depends(get_actor_employee_id)):
     req = session.get(HeadcountRequest, request_id)
     if not req:
         raise HTTPException(status_code=404, detail="Request not found")
@@ -40,7 +40,7 @@ def update_headcount_request(request_id: int, payload: HeadcountRequestUpdate, s
     session.add(req)
     session.commit()
     session.refresh(req)
-    log_activity(session, actor_employee_id=req.requested_by_manager_id, entity_type="headcount_request", entity_id=req.id, verb="updated", payload=data)
+    log_activity(session, actor_employee_id=actor_employee_id, entity_type="headcount_request", entity_id=req.id, verb="updated", payload=data)
     session.commit()
     return req
 
@@ -51,11 +51,47 @@ def list_employment_actions(session: Session = Depends(get_session)):
 
 
 @router.post("/actions", response_model=EmploymentAction)
-def create_employment_action(payload: EmploymentActionCreate, session: Session = Depends(get_session)):
+def create_employment_action(payload: EmploymentActionCreate, session: Session = Depends(get_session), actor_employee_id: int = Depends(get_actor_employee_id)):
     action = EmploymentAction(**payload.model_dump())
     session.add(action)
     session.commit()
     session.refresh(action)
-    log_activity(session, actor_employee_id=action.issued_by_employee_id, entity_type="employment_action", entity_id=action.id, verb=action.action_type, payload={"employee_id": action.employee_id})
+    log_activity(session, actor_employee_id=actor_employee_id, entity_type="employment_action", entity_id=action.id, verb=action.action_type, payload={"employee_id": action.employee_id})
     session.commit()
     return action
+
+@router.get("/onboarding", response_model=list[AgentOnboarding])
+def list_agent_onboarding(session: Session = Depends(get_session)):
+    return session.exec(select(AgentOnboarding).order_by(AgentOnboarding.id.desc())).all()
+
+
+@router.post("/onboarding", response_model=AgentOnboarding)
+def create_agent_onboarding(payload: AgentOnboardingCreate, session: Session = Depends(get_session), actor_employee_id: int = Depends(get_actor_employee_id)):
+    item = AgentOnboarding(**payload.model_dump())
+    session.add(item)
+    session.commit()
+    session.refresh(item)
+    log_activity(session, actor_employee_id=actor_employee_id, entity_type="agent_onboarding", entity_id=item.id, verb="created", payload={"agent_name": item.agent_name, "status": item.status})
+    session.commit()
+    return item
+
+
+@router.patch("/onboarding/{onboarding_id}", response_model=AgentOnboarding)
+def update_agent_onboarding(onboarding_id: int, payload: AgentOnboardingUpdate, session: Session = Depends(get_session), actor_employee_id: int = Depends(get_actor_employee_id)):
+    item = session.get(AgentOnboarding, onboarding_id)
+    if not item:
+        raise HTTPException(status_code=404, detail="Onboarding record not found")
+
+    data = payload.model_dump(exclude_unset=True)
+    for k, v in data.items():
+        setattr(item, k, v)
+    from datetime import datetime
+    item.updated_at = datetime.utcnow()
+
+    session.add(item)
+    session.commit()
+    session.refresh(item)
+    log_activity(session, actor_employee_id=actor_employee_id, entity_type="agent_onboarding", entity_id=item.id, verb="updated", payload=data)
+    session.commit()
+    return item
+