active_call/handler/

handler.rs

use crate::{
    app::AppState,
    call::{
        ActiveCall, ActiveCallType, Command,
        active_call::{ActiveCallGuard, CallParams},
    },
    handler::api,
    playbook::{Playbook, PlaybookRunner},
};
use axum::{
    Router,
    extract::{Query, State, WebSocketUpgrade, ws::Message},
    response::Response,
    routing::get,
};
use bytes::Bytes;
use chrono::Utc;
use futures::{SinkExt, StreamExt};
use std::{path::PathBuf, sync::Arc, time::Duration};
use tokio::{join, select};
use tokio_util::sync::CancellationToken;
use tracing::{debug, info, warn};
use uuid::Uuid;
use voice_engine::{event::SessionEvent, media::track::TrackConfig};

pub fn call_router() -> Router<AppState> {
    Router::new()
        .route("/call", get(ws_handler))
        .route("/call/webrtc", get(webrtc_handler))
        .route("/call/sip", get(sip_handler))
}

pub fn playbook_router() -> Router<AppState> {
    Router::new()
        .route("/api/playbooks", get(api::list_playbooks))
        .route(
            "/api/playbooks/{name}",
            get(api::get_playbook).post(api::save_playbook),
        )
        .route("/api/playbook/run", axum::routing::post(api::run_playbook))
        .route("/api/records", get(api::list_records))
}

pub async fn ws_handler(
    ws: WebSocketUpgrade,
    State(state): State<AppState>,
    Query(params): Query<CallParams>,
) -> Response {
    call_handler(ActiveCallType::WebSocket, ws, state, params).await
}

pub async fn sip_handler(
    ws: WebSocketUpgrade,
    State(state): State<AppState>,
    Query(params): Query<CallParams>,
) -> Response {
    call_handler(ActiveCallType::Sip, ws, state, params).await
}

pub async fn webrtc_handler(
    ws: WebSocketUpgrade,
    State(state): State<AppState>,
    Query(params): Query<CallParams>,
) -> Response {
    call_handler(ActiveCallType::Webrtc, ws, state, params).await
}

pub async fn call_handler(
    call_type: ActiveCallType,
    ws: WebSocketUpgrade,
    app_state: AppState,
    params: CallParams,
) -> Response {
    let session_id = params
        .id
        .unwrap_or_else(|| format!("s.{}", Uuid::new_v4().to_string()));
    let server_side_track = params.server_side_track.clone();
    let dump_events = params.dump_events.unwrap_or(true);
    let ping_interval = params.ping_interval.unwrap_or(20);

    let resp = ws.on_upgrade(move |socket| async move {
        let (mut ws_sender, mut ws_receiver) = socket.split();
        let (audio_sender, audio_receiver) = tokio::sync::mpsc::unbounded_channel::<Bytes>();
        let cancel_token = CancellationToken::new();
        let track_config = TrackConfig::default();
        let active_call = Arc::new(ActiveCall::new(
            call_type.clone(),
            cancel_token.clone(),
            session_id.clone(),
            app_state.invitation.clone(),
            app_state.clone(),
            track_config,
            Some(audio_receiver),
            dump_events,
            server_side_track,
            None, // No extra data for now
        ));

        // Check for pending playbook
        {
            let mut pending = app_state.pending_playbooks.lock().await;
            if let Some(name) = pending.remove(&session_id) {
                let path = PathBuf::from("config/playbook").join(&name);
                match Playbook::load(path).await {
                    Ok(playbook) => match PlaybookRunner::new(playbook, active_call.clone()) {
                        Ok(runner) => {
                            tokio::spawn(async move {
                                runner.run().await;
                            });
                            info!(session_id, "Playbook runner started for {}", name);
                        }
                        Err(e) => warn!(session_id, "Failed to create runner {}: {}", name, e),
                    },
                    Err(e) => {
                        warn!(session_id, "Failed to load playbook {}: {}", name, e);
                    }
                }
            }
        }

        let recv_from_ws_loop = async {
            while let Some(Ok(message)) = ws_receiver.next().await {
                match message {
                    Message::Text(text) => {
                        let command = match serde_json::from_str::<Command>(&text) {
                            Ok(cmd) => cmd,
                            Err(e) => {
                                warn!(session_id, %text, "Failed to parse command {}",e);
                                continue;
                            }
                        };
                        if let Err(_) = active_call.enqueue_command(command).await {
                            break;
                        }
                    }
                    Message::Binary(bin) => {
                        audio_sender.send(bin.into()).ok();
                    }
                    Message::Close(_) => {
                        info!(session_id, "WebSocket closed by client");
                        break;
                    }
                    _ => {}
                }
            }
        };

        let mut event_receiver = active_call.event_sender.subscribe();
        let send_to_ws_loop = async {
            while let Ok(event) = event_receiver.recv().await {
                let message = match event.into_ws_message() {
                    Ok(msg) => msg,
                    Err(_) => continue,
                };
                if let Err(_) = ws_sender.send(message).await {
                    break;
                }
            }
        };

        let send_ping_loop = async {
            if ping_interval == 0 {
                active_call.cancel_token.cancelled().await;
                return;
            }
            let mut ticker = tokio::time::interval(Duration::from_secs(ping_interval.into()));
            loop {
                ticker.tick().await;
                let payload = Utc::now().to_rfc3339();
                let event = SessionEvent::Ping {
                    timestamp: voice_engine::media::get_timestamp(),
                    payload: Some(payload),
                };
                if let Err(_) = active_call.event_sender.send(event) {
                    break;
                }
            }
        };
        let guard = ActiveCallGuard::new(active_call.clone());
        info!(
            session_id,
            active_calls = guard.active_calls,
            ?call_type,
            "new call started"
        );
        let receiver = active_call.new_receiver();

        let (r, _) = join! {
            active_call.serve(receiver),
            async {
                select!{
                    _ = send_ping_loop => {},
                    _ = cancel_token.cancelled() => {},
                    _ = send_to_ws_loop => { },
                    _ = recv_from_ws_loop => {
                        info!(session_id, "WebSocket closed by client");
                    },
                }
                cancel_token.cancel();
            }
        };
        match r {
            Ok(_) => info!(session_id, "call ended successfully"),
            Err(e) => warn!(session_id, "call ended with error: {}", e),
        }

        active_call.cleanup().await.ok();
        // Drain remaining events
        while let Ok(event) = event_receiver.try_recv() {
            let message = match event.into_ws_message() {
                Ok(msg) => msg,
                Err(_) => continue,
            };
            if let Err(_) = ws_sender.send(message).await {
                break;
            }
        }
        ws_sender.flush().await.ok();
        ws_sender.close().await.ok();
        debug!(session_id, "WebSocket connection closed");
    });
    resp
}

trait IntoWsMessage {
    fn into_ws_message(self) -> Result<Message, serde_json::Error>;
}

impl IntoWsMessage for voice_engine::event::SessionEvent {
    fn into_ws_message(self) -> Result<Message, serde_json::Error> {
        match self {
            SessionEvent::Binary { data, .. } => Ok(Message::Binary(data.into())),
            SessionEvent::Ping { timestamp, payload } => {
                let payload = payload.unwrap_or_else(|| timestamp.to_string());
                Ok(Message::Ping(payload.into()))
            }
            event => serde_json::to_string(&event).map(|payload| Message::Text(payload.into())),
        }
    }
}