hold-slayer/api/call_flows.py

"""
Call Flows API — Store and manage IVR navigation trees.

The system gets smarter every time you call somewhere.
"""

import uuid
from datetime import datetime

from fastapi import APIRouter, Depends, HTTPException
from slugify import slugify
from sqlalchemy import select
from sqlalchemy.ext.asyncio import AsyncSession

from api.deps import get_gateway
from core.gateway import AIPSTNGateway
from db.database import StoredCallFlow, get_db
from models.call_flow import (
    CallFlow,
    CallFlowCreate,
    CallFlowStep,
    CallFlowSummary,
    CallFlowUpdate,
)

router = APIRouter()


@router.post("/", response_model=CallFlow)
async def create_call_flow(
    flow: CallFlowCreate,
    db: AsyncSession = Depends(get_db),
):
    """Store a new call flow for a phone number."""
    flow_id = slugify(flow.name)

    # Check if ID already exists
    existing = await db.execute(
        select(StoredCallFlow).where(StoredCallFlow.id == flow_id)
    )
    if existing.scalar_one_or_none():
        raise HTTPException(
            status_code=409,
            detail=f"Call flow '{flow_id}' already exists. Use PUT to update.",
        )

    db_flow = StoredCallFlow(
        id=flow_id,
        name=flow.name,
        phone_number=flow.phone_number,
        description=flow.description,
        steps=[s.model_dump() for s in flow.steps],
        tags=flow.tags,
        notes=flow.notes,
        last_verified=datetime.now(),
    )
    db.add(db_flow)
    await db.flush()

    return CallFlow(
        id=flow_id,
        name=flow.name,
        phone_number=flow.phone_number,
        description=flow.description,
        steps=flow.steps,
        tags=flow.tags,
        notes=flow.notes,
        last_verified=datetime.now(),
    )


@router.get("/", response_model=list[CallFlowSummary])
async def list_call_flows(
    db: AsyncSession = Depends(get_db),
):
    """List all stored call flows."""
    result = await db.execute(select(StoredCallFlow))
    rows = result.scalars().all()

    return [
        CallFlowSummary(
            id=row.id,
            name=row.name,
            phone_number=row.phone_number,
            description=row.description or "",
            step_count=len(row.steps) if row.steps else 0,
            avg_hold_time=row.avg_hold_time,
            success_rate=row.success_rate,
            last_used=row.last_used,
            times_used=row.times_used or 0,
            tags=row.tags or [],
        )
        for row in rows
    ]


@router.get("/{flow_id}", response_model=CallFlow)
async def get_call_flow(
    flow_id: str,
    db: AsyncSession = Depends(get_db),
):
    """Get a stored call flow by ID."""
    result = await db.execute(
        select(StoredCallFlow).where(StoredCallFlow.id == flow_id)
    )
    row = result.scalar_one_or_none()
    if not row:
        raise HTTPException(status_code=404, detail=f"Call flow '{flow_id}' not found")

    return CallFlow(
        id=row.id,
        name=row.name,
        phone_number=row.phone_number,
        description=row.description or "",
        steps=[CallFlowStep(**s) for s in row.steps],
        tags=row.tags or [],
        notes=row.notes,
        avg_hold_time=row.avg_hold_time,
        success_rate=row.success_rate,
        last_used=row.last_used,
        times_used=row.times_used or 0,
    )


@router.get("/by-number/{phone_number}", response_model=CallFlow)
async def get_flow_for_number(
    phone_number: str,
    db: AsyncSession = Depends(get_db),
):
    """Look up stored call flow by phone number."""
    result = await db.execute(
        select(StoredCallFlow).where(StoredCallFlow.phone_number == phone_number)
    )
    row = result.scalar_one_or_none()
    if not row:
        raise HTTPException(
            status_code=404,
            detail=f"No call flow found for {phone_number}",
        )

    return CallFlow(
        id=row.id,
        name=row.name,
        phone_number=row.phone_number,
        description=row.description or "",
        steps=[CallFlowStep(**s) for s in row.steps],
        tags=row.tags or [],
        notes=row.notes,
        avg_hold_time=row.avg_hold_time,
        success_rate=row.success_rate,
        last_used=row.last_used,
        times_used=row.times_used or 0,
    )


@router.put("/{flow_id}", response_model=CallFlow)
async def update_call_flow(
    flow_id: str,
    update: CallFlowUpdate,
    db: AsyncSession = Depends(get_db),
):
    """Update an existing call flow."""
    result = await db.execute(
        select(StoredCallFlow).where(StoredCallFlow.id == flow_id)
    )
    row = result.scalar_one_or_none()
    if not row:
        raise HTTPException(status_code=404, detail=f"Call flow '{flow_id}' not found")

    if update.name is not None:
        row.name = update.name
    if update.description is not None:
        row.description = update.description
    if update.steps is not None:
        row.steps = [s.model_dump() for s in update.steps]
    if update.tags is not None:
        row.tags = update.tags
    if update.notes is not None:
        row.notes = update.notes
    if update.last_verified is not None:
        row.last_verified = update.last_verified

    await db.flush()

    return CallFlow(
        id=row.id,
        name=row.name,
        phone_number=row.phone_number,
        description=row.description or "",
        steps=[CallFlowStep(**s) for s in row.steps],
        tags=row.tags or [],
        notes=row.notes,
        avg_hold_time=row.avg_hold_time,
        success_rate=row.success_rate,
        last_used=row.last_used,
        times_used=row.times_used or 0,
    )


@router.delete("/{flow_id}")
async def delete_call_flow(
    flow_id: str,
    db: AsyncSession = Depends(get_db),
):
    """Delete a stored call flow."""
    result = await db.execute(
        select(StoredCallFlow).where(StoredCallFlow.id == flow_id)
    )
    row = result.scalar_one_or_none()
    if not row:
        raise HTTPException(status_code=404, detail=f"Call flow '{flow_id}' not found")

    await db.delete(row)
    return {"status": "deleted", "flow_id": flow_id}