rustpbx 0.4.9

A SIP PBX implementation in Rust
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126

//! Per-call media state owned by the MediaEngine.
//!
//! [`MediaSession`] is the engine's per-session state container.  It owns:
//!
//! * the negotiated [`BridgePeer`] (WebRTC↔RTP bridge or plain RTP bridge)
//! * a shared [`Recorder`] handle (same `Arc` the bridge writes into)
//! * active playback tracks keyed by `track_id`
//! * a [`McuSwitch`] for dynamic Bridge↔MCU transitions
//!
//! The session is created by [`MediaCommand::CreateSession`] and destroyed by
//! [`MediaCommand::DestroySession`].  The `BridgePeer` is injected later,
//! once SDP negotiation is complete, via [`MediaCommand::AttachBridge`].

use std::collections::HashMap;
use std::sync::Arc;
use std::sync::atomic::AtomicBool;

use parking_lot::RwLock;

use crate::media::FileTrack;
use crate::media::bridge::{BridgeEndpoint, BridgePeer};
use crate::media::engine::mcu_switch::McuSwitch;
use crate::media::recorder::Recorder;

/// Per-session media state owned by the engine.
pub struct MediaSession {
    pub session_id: String,

    // ── Bridge ──────────────────────────────────────────────────────────
    /// The active in-process media bridge, set once SDP negotiation is complete.
    pub bridge: Option<Arc<BridgePeer>>,

    /// Whether the SIP caller maps to `BridgeEndpoint::Caller` (`true`) or
    /// `BridgeEndpoint::Callee` (`false`).  Set together with the bridge.
    pub caller_is_webrtc: bool,

    /// Codec info from the caller-facing SDP answer.
    /// Used to configure FileTrack payload type for bridge playback.
    pub caller_codec_info: Vec<crate::media::negotiate::CodecInfo>,

    // ── Recording ───────────────────────────────────────────────────────
    /// Shared recorder — the same `Arc` is passed to `BridgePeer::with_recorder`
    /// so the bridge's forwarding loop writes samples into it directly.
    pub recorder: Arc<RwLock<Option<Recorder>>>,

    /// Atomic flag that the recorder bridge task checks before writing samples.
    pub recording_paused: Arc<AtomicBool>,

    /// Wall-clock instant when the current recording started.  Used to
    /// compute duration in `StopRecording` (fixes the previous duration=0 bug).
    pub recording_started_at: Option<std::time::Instant>,

    // ── Playback ────────────────────────────────────────────────────────
    /// Active playback tracks keyed by caller-supplied `track_id`.
    pub playback_tracks: HashMap<String, FileTrack>,

    /// IDs of FileTrack(s) currently playing through the bridge, stored so
    /// StopPlayback can stop all of them (InjectAudio::Both creates two).
    pub bridge_playback_track_ids: Vec<String>,

    // ── MCU / TTS injection ─────────────────────────────────────────────
    /// Manages dynamic switching between direct bridge forwarding (low CPU)
    /// and MCU mixing (needed for TTS injection, conference).
    pub mcu: McuSwitch,
}

impl MediaSession {
    /// Create a new empty media session.
    pub fn new(session_id: String) -> Self {
        let mcu = McuSwitch::new(session_id.clone(), 8000);
        Self {
            session_id,
            bridge: None,
            caller_is_webrtc: false,
            caller_codec_info: vec![],
            recorder: Arc::new(RwLock::new(None)),
            recording_paused: Arc::new(AtomicBool::new(false)),
            recording_started_at: None,
            playback_tracks: HashMap::new(),
            bridge_playback_track_ids: Vec::new(),
            mcu,
        }
    }

    /// Resolve the bridge endpoint for the SIP caller leg.
    pub fn caller_endpoint(&self) -> BridgeEndpoint {
        if self.caller_is_webrtc {
            BridgeEndpoint::Caller
        } else {
            BridgeEndpoint::Callee
        }
    }

    /// Resolve the bridge endpoint for the SIP callee leg.
    pub fn callee_endpoint(&self) -> BridgeEndpoint {
        if self.caller_is_webrtc {
            BridgeEndpoint::Callee
        } else {
            BridgeEndpoint::Caller
        }
    }

    /// Map a `leg_id` string to the corresponding [`BridgeEndpoint`].
    /// Falls back to the caller endpoint for unknown IDs.
    pub fn endpoint_for_leg(&self, leg_id: &str) -> BridgeEndpoint {
        match leg_id {
            "callee" => self.callee_endpoint(),
            _ => self.caller_endpoint(),
        }
    }
}

impl std::fmt::Debug for MediaSession {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        f.debug_struct("MediaSession")
            .field("session_id", &self.session_id)
            .field("bridge_present", &self.bridge.is_some())
            .field("caller_is_webrtc", &self.caller_is_webrtc)
            .field("bridge_playback_track_ids", &self.bridge_playback_track_ids)
            .field(
                "playback_tracks",
                &self.playback_tracks.keys().collect::<Vec<_>>(),
            )
            .finish()
    }
}