|
| 1 | +from fastapi import APIRouter, Depends, HTTPException |
| 2 | +from pydantic import BaseModel |
| 3 | +from typing import List, Optional |
| 4 | +from middleware.auth import get_current_user |
| 5 | +from services.sequence_service import ( |
| 6 | + list_sequences, create_sequence, delete_sequence, |
| 7 | + enroll_in_sequence, list_enrollments, cancel_enrollment, |
| 8 | +) |
| 9 | + |
| 10 | +router = APIRouter(prefix="/sequences", tags=["sequences"]) |
| 11 | + |
| 12 | + |
| 13 | +@router.get("") |
| 14 | +async def get_sequences(current_user: dict = Depends(get_current_user)): |
| 15 | + return {"sequences": list_sequences(current_user["id"])} |
| 16 | + |
| 17 | + |
| 18 | +class SequenceStep(BaseModel): |
| 19 | + delay_days: int |
| 20 | + subject_template: str |
| 21 | + body_template: str |
| 22 | + |
| 23 | + |
| 24 | +class CreateSequenceBody(BaseModel): |
| 25 | + name: str |
| 26 | + steps: List[SequenceStep] |
| 27 | + |
| 28 | + |
| 29 | +@router.post("", status_code=201) |
| 30 | +async def post_sequence(body: CreateSequenceBody, current_user: dict = Depends(get_current_user)): |
| 31 | + seq = create_sequence( |
| 32 | + user_id=current_user["id"], |
| 33 | + name=body.name, |
| 34 | + steps=[s.model_dump() for s in body.steps], |
| 35 | + ) |
| 36 | + return seq |
| 37 | + |
| 38 | + |
| 39 | +@router.delete("/{seq_id}", status_code=204) |
| 40 | +async def del_sequence(seq_id: str, current_user: dict = Depends(get_current_user)): |
| 41 | + ok = delete_sequence(seq_id, current_user["id"]) |
| 42 | + if not ok: |
| 43 | + raise HTTPException(status_code=404, detail="Sequence not found") |
| 44 | + |
| 45 | + |
| 46 | +class EnrollBody(BaseModel): |
| 47 | + contact_email: str |
| 48 | + email_id: Optional[str] = None |
| 49 | + |
| 50 | + |
| 51 | +@router.post("/{seq_id}/enroll", status_code=201) |
| 52 | +async def enroll(seq_id: str, body: EnrollBody, current_user: dict = Depends(get_current_user)): |
| 53 | + enrollment = enroll_in_sequence( |
| 54 | + user_id=current_user["id"], |
| 55 | + sequence_id=seq_id, |
| 56 | + contact_email=body.contact_email, |
| 57 | + email_id=body.email_id, |
| 58 | + ) |
| 59 | + if not enrollment: |
| 60 | + raise HTTPException(status_code=404, detail="Sequence not found") |
| 61 | + return enrollment |
| 62 | + |
| 63 | + |
| 64 | +@router.get("/enrollments") |
| 65 | +async def get_enrollments(current_user: dict = Depends(get_current_user)): |
| 66 | + return {"enrollments": list_enrollments(current_user["id"])} |
| 67 | + |
| 68 | + |
| 69 | +@router.post("/enrollments/{enrollment_id}/cancel", status_code=204) |
| 70 | +async def cancel(enrollment_id: str, current_user: dict = Depends(get_current_user)): |
| 71 | + cancel_enrollment(enrollment_id, current_user["id"]) |
0 commit comments