from __future__ import annotations from fastapi import APIRouter, Depends, HTTPException from sqlalchemy.exc import IntegrityError from sqlmodel import Session, select from app.api.utils import get_actor_employee_id, log_activity from app.db.session import get_session from app.models.projects import Project, ProjectMember from app.schemas.projects import ProjectCreate, ProjectUpdate router = APIRouter(prefix="/projects", tags=["projects"]) @router.get("", response_model=list[Project]) def list_projects(session: Session = Depends(get_session)): return session.exec(select(Project).order_by(Project.name.asc())).all() @router.post("", response_model=Project) def create_project( payload: ProjectCreate, session: Session = Depends(get_session), actor_employee_id: int = Depends(get_actor_employee_id), ): """Create a project. Keep operation atomic: flush to get id, log activity, then commit once. Translate DB integrity errors to 409s. """ proj = Project(**payload.model_dump()) session.add(proj) try: session.flush() log_activity( session, actor_employee_id=actor_employee_id, entity_type="project", entity_id=proj.id, verb="created", payload={"name": proj.name}, ) session.commit() except IntegrityError: session.rollback() raise HTTPException( status_code=409, detail="Project already exists or violates constraints" ) session.refresh(proj) return proj @router.patch("/{project_id}", response_model=Project) def update_project( project_id: int, payload: ProjectUpdate, session: Session = Depends(get_session), actor_employee_id: int = Depends(get_actor_employee_id), ): proj = session.get(Project, project_id) if not proj: raise HTTPException(status_code=404, detail="Project not found") data = payload.model_dump(exclude_unset=True) for k, v in data.items(): setattr(proj, k, v) session.add(proj) session.commit() session.refresh(proj) log_activity( session, actor_employee_id=actor_employee_id, entity_type="project", entity_id=proj.id, verb="updated", payload=data, ) session.commit() return proj @router.get("/{project_id}/members", response_model=list[ProjectMember]) def list_project_members(project_id: int, session: Session = Depends(get_session)): return session.exec( select(ProjectMember) .where(ProjectMember.project_id == project_id) .order_by(ProjectMember.id.asc()) ).all() @router.post("/{project_id}/members", response_model=ProjectMember) def add_project_member( project_id: int, payload: ProjectMember, session: Session = Depends(get_session), actor_employee_id: int = Depends(get_actor_employee_id), ): existing = session.exec( select(ProjectMember).where( ProjectMember.project_id == project_id, ProjectMember.employee_id == payload.employee_id ) ).first() if existing: raise HTTPException(status_code=409, detail="Member already added") member = ProjectMember( project_id=project_id, employee_id=payload.employee_id, role=payload.role ) session.add(member) session.commit() session.refresh(member) log_activity( session, actor_employee_id=actor_employee_id, entity_type="project_member", entity_id=member.id, verb="added", payload={"project_id": project_id, "employee_id": member.employee_id, "role": member.role}, ) session.commit() return member @router.delete("/{project_id}/members/{member_id}") def remove_project_member( project_id: int, member_id: int, session: Session = Depends(get_session), actor_employee_id: int = Depends(get_actor_employee_id), ): member = session.get(ProjectMember, member_id) if not member or member.project_id != project_id: raise HTTPException(status_code=404, detail="Project member not found") session.delete(member) session.commit() log_activity( session, actor_employee_id=actor_employee_id, entity_type="project_member", entity_id=member_id, verb="removed", payload={"project_id": project_id}, ) session.commit() return {"ok": True} @router.patch("/{project_id}/members/{member_id}", response_model=ProjectMember) def update_project_member( project_id: int, member_id: int, payload: ProjectMember, session: Session = Depends(get_session), actor_employee_id: int = Depends(get_actor_employee_id), ): member = session.get(ProjectMember, member_id) if not member or member.project_id != project_id: raise HTTPException(status_code=404, detail="Project member not found") if payload.role is not None: member.role = payload.role session.add(member) session.commit() session.refresh(member) log_activity( session, actor_employee_id=actor_employee_id, entity_type="project_member", entity_id=member.id, verb="updated", payload={"project_id": project_id, "role": member.role}, ) session.commit() return member