1use super::Command;
2use crate::{
3 CallOption, ReferOption,
4 call::state::{ActorMsg, CallProgress, CallRuntime, Extras, LegShared, build_callrecord},
5 event::{EventReceiver, EventSender, SessionEvent},
6 media::{
7 TrackId,
8 engine::StreamEngine,
9 recorder::RecorderOption,
10 stream::{MediaStream, MediaStreamBuilder, SERVER_SIDE_TRACK_ID},
11 track::{
12 Track, TrackConfig, forwarding::ForwardingTrack, media_pass::MediaPassTrack,
13 tts::SynthesisHandle, websocket::WebsocketBytesReceiver,
14 },
15 },
16 synthesis::SynthesisCommand,
17 transcription::TranscriptionOption,
18};
19use crate::{
20 app::AppState,
21 call::{CommandReceiver, CommandSender, sip::Invitation},
22 callrecord::{CallRecord, CallRecordEvent, CallRecordEventType, CallRecordHangupReason},
23};
24use anyhow::Result;
25use arc_swap::{ArcSwap, ArcSwapOption};
26use chrono::{DateTime, Utc};
27use rsipstack::dialog::invite_dialog::InviteDialog;
28use serde::{Deserialize, Serialize};
29use std::{
30 collections::HashMap,
31 path::Path,
32 sync::{
33 Arc,
34 atomic::{AtomicBool, Ordering},
35 },
36 time::Duration,
37};
38use tokio::{fs::File, select, sync::mpsc};
39use tokio_util::sync::CancellationToken;
40use tracing::{debug, info, warn};
41
42pub enum PendingCallerTrack {
44 StartedForEarlyMedia,
47 NotStarted(Box<dyn Track>),
50}
51
52#[cfg(test)]
53mod tests {
54 use super::*;
55 use crate::app::AppStateBuilder;
56 use crate::config::Config;
57 use crate::media::track::tts::SynthesisHandle;
58 use crate::synthesis::SynthesisCommand;
59 use tokio::sync::mpsc;
60
61 async fn make_active_call_with_option(option: CallOption) -> Arc<ActiveCall> {
62 let mut config = Config::default();
63 config.udp_port = 0; config.media_cache_path = "/tmp/mediacache".to_string();
65 let app_state = AppStateBuilder::new()
66 .with_config(config)
67 .with_stream_engine(Arc::new(StreamEngine::default()))
68 .build()
69 .await
70 .unwrap();
71 let active_call = Arc::new(ActiveCall::new(CallSpec {
72 call_type: ActiveCallType::Sip,
73 cancel_token: CancellationToken::new(),
74 session_id: "test-session".to_string(),
75 invitation: app_state.invitation.clone(),
76 app_state: app_state.clone(),
77 track_config: TrackConfig::default(),
78 audio_receiver: None,
79 dump_events: false,
80 server_side_track_id: None,
81 extras: None,
82 }));
83 active_call.set_option(option);
84 active_call
85 }
86
87 #[tokio::test]
88 async fn test_tts_ssrc_reuse_for_autohangup() -> Result<()> {
89 let mut option = crate::CallOption::default();
90 option.tts = Some(crate::synthesis::SynthesisOption::default());
91 let active_call = make_active_call_with_option(option).await;
92
93 let (tx, mut rx) = mpsc::unbounded_channel::<SynthesisCommand>();
94 let initial_ssrc = 12345;
95 let handle = SynthesisHandle::new(tx, Some("play_1".to_string()), initial_ssrc);
96
97 active_call.tts_handle.store(Some(Arc::new(handle)));
99 active_call.set_current_play(Some("play_1".to_string()));
100
101 active_call
103 .do_tts(Command::Tts {
104 text: "hangup now".to_string(),
105 speaker: None,
106 play_id: Some("play_1".to_string()),
107 auto_hangup: Some(true),
108 streaming: Some(false),
109 end_of_stream: Some(true),
110 option: None,
111 wait_input_timeout: None,
112 base64: Some(false),
113 cache_key: None,
114 })
115 .await?;
116
117 let cmd = rx.try_recv().expect("Should have received tts command");
119 assert_eq!(cmd.text, "hangup now");
120 assert_eq!(cmd.auto_hangup, Some(true));
121
122 Ok(())
123 }
124
125 #[tokio::test]
126 async fn test_tts_new_ssrc_for_different_play_id() -> Result<()> {
127 let mut tts_opt = crate::synthesis::SynthesisOption::default();
128 tts_opt.provider = Some(crate::synthesis::SynthesisType::Aliyun);
129 let mut option = crate::CallOption::default();
130 option.tts = Some(tts_opt);
131 let active_call = make_active_call_with_option(option).await;
132
133 let (tx, _rx) = mpsc::unbounded_channel();
134 let initial_ssrc = 111;
135 let handle = SynthesisHandle::new(tx, Some("play_1".to_string()), initial_ssrc);
136
137 active_call.tts_handle.store(Some(Arc::new(handle)));
138 active_call.set_current_play(Some("play_1".to_string()));
139
140 active_call
142 .do_tts(Command::Tts {
143 text: "new play".to_string(),
144 speaker: None,
145 play_id: Some("play_2".to_string()),
146 auto_hangup: Some(true),
147 streaming: Some(false),
148 end_of_stream: Some(true),
149 option: None,
150 wait_input_timeout: None,
151 base64: Some(false),
152 cache_key: None,
153 })
154 .await?;
155
156 {
159 let handle = active_call.tts_handle.load_full();
160 assert!(handle.is_some(), "new tts handle should be stored");
161 let handle = handle.unwrap();
162 assert_ne!(
163 handle.ssrc, initial_ssrc,
164 "Should use a new SSRC for different play_id"
165 );
166 }
167
168 Ok(())
169 }
170
171 #[tokio::test]
173 async fn test_hangup_refer_true_cancels_refer_only() -> Result<()> {
174 let active_call = make_active_call_with_option(crate::CallOption::default()).await;
175
176 let refer_token = active_call.cancel_token.child_token();
177 let refer_leg = LegShared::new(1, true, CallProgress::default());
178 active_call.set_refer_call_token(refer_token.clone());
179 active_call.set_refer_leg(Some(refer_leg.clone()));
180
181 active_call.do_hangup(None, None, None, Some(true)).await?;
182
183 assert!(
184 refer_token.is_cancelled(),
185 "refer token should be cancelled"
186 );
187 assert!(
188 !active_call.media_stream.cancel_token.is_cancelled(),
189 "media stream should NOT stop"
190 );
191 assert!(
192 refer_leg.progress.load_full().hangup_reason.is_some(),
193 "hangup_reason should be set on refer leg"
194 );
195 Ok(())
196 }
197
198 #[tokio::test]
200 async fn test_hangup_none_cancels_refer_too() -> Result<()> {
201 let active_call = make_active_call_with_option(crate::CallOption::default()).await;
202
203 let refer_token = active_call.cancel_token.child_token();
204 active_call.set_refer_call_token(refer_token.clone());
205
206 active_call.do_hangup(None, None, None, None).await?;
207
208 assert!(
209 refer_token.is_cancelled(),
210 "refer token should be cancelled"
211 );
212 assert!(
213 active_call.media_stream.cancel_token.is_cancelled(),
214 "media stream should stop"
215 );
216 Ok(())
217 }
218
219 struct MockCallerTrack {
234 id: TrackId,
235 config: crate::media::track::TrackConfig,
236 processor_chain: crate::media::processor::ProcessorChain,
237 }
238
239 impl MockCallerTrack {
240 fn new(id: TrackId) -> Self {
241 Self {
242 id,
243 config: crate::media::track::TrackConfig::default(),
244 processor_chain: crate::media::processor::ProcessorChain::new(16000),
245 }
246 }
247 }
248
249 #[async_trait::async_trait]
250 impl crate::media::track::Track for MockCallerTrack {
251 fn ssrc(&self) -> u32 {
252 0
253 }
254 fn id(&self) -> &TrackId {
255 &self.id
256 }
257 fn config(&self) -> &crate::media::track::TrackConfig {
258 &self.config
259 }
260 fn processor_chain(&mut self) -> &mut crate::media::processor::ProcessorChain {
261 &mut self.processor_chain
262 }
263 async fn handshake(
264 &mut self,
265 _o: String,
266 _t: Option<tokio::time::Duration>,
267 ) -> Result<String> {
268 Ok(String::new())
269 }
270 async fn update_remote_description(&mut self, _a: &String) -> Result<()> {
271 Ok(())
272 }
273 async fn start(
274 &mut self,
275 _e: crate::event::EventSender,
276 _p: crate::media::track::TrackPacketSender,
277 ) -> Result<()> {
278 Ok(())
279 }
280 async fn stop(&self) -> Result<()> {
281 Ok(())
282 }
283 async fn send_packet(&mut self, _f: &crate::media::AudioFrame) -> Result<()> {
284 Ok(())
285 }
286 }
287
288 struct MockAsrClient;
289
290 #[async_trait::async_trait]
291 impl crate::transcription::TranscriptionClient for MockAsrClient {
292 fn send_audio(
293 &self,
294 _s: &[crate::media::Sample],
295 _src: Option<&crate::media::SourcePacket>,
296 ) -> Result<()> {
297 Ok(())
298 }
299 }
300
301 async fn make_active_call_with_engine_and_option(
302 engine: Arc<StreamEngine>,
303 cache_dir: &str,
304 option: crate::CallOption,
305 ) -> Arc<ActiveCall> {
306 let mut config = Config::default();
307 config.udp_port = 0;
308 config.media_cache_path = cache_dir.to_string();
309 let app_state = AppStateBuilder::new()
310 .with_config(config)
311 .with_stream_engine(engine)
312 .build()
313 .await
314 .unwrap();
315 let session_id = format!("test-{}-{}", cache_dir, uuid::Uuid::new_v4());
316 let active_call = Arc::new(ActiveCall::new(CallSpec {
317 call_type: ActiveCallType::Sip,
318 cancel_token: CancellationToken::new(),
319 session_id: session_id.clone(),
320 invitation: app_state.invitation.clone(),
321 app_state: app_state.clone(),
322 track_config: TrackConfig::default(),
323 audio_receiver: None,
324 dump_events: false,
325 server_side_track_id: None,
326 extras: None,
327 }));
328 active_call.set_option(option);
329 active_call
330 }
331
332 #[tokio::test]
333 async fn test_setup_track_with_stream_builds_processors_from_accept_option() -> Result<()> {
334 let (asr_created_tx, mut asr_created_rx) = mpsc::channel::<()>(1);
335
336 let mock_provider =
337 crate::transcription::TranscriptionType::Other("mock-ringing-asr".to_string());
338
339 let mut engine = StreamEngine::new();
340 engine.register_asr(
341 mock_provider.clone(),
342 Box::new(move |_tid, _tok, _opt, _es| {
343 let tx = asr_created_tx.clone();
344 Box::pin(async move {
345 let _ = tx.send(()).await;
346 Ok(Box::new(MockAsrClient)
347 as Box<dyn crate::transcription::TranscriptionClient>)
348 })
349 }),
350 );
351 let engine = Arc::new(engine);
352
353 let accept_option = crate::CallOption {
354 asr: Some(crate::transcription::TranscriptionOption {
355 provider: Some(mock_provider),
356 ..Default::default()
357 }),
358 ..Default::default()
359 };
360 let active_call = make_active_call_with_engine_and_option(
361 engine,
362 "mediacache_ringing_accept_test",
363 accept_option,
364 )
365 .await;
366 let cancel_token = active_call.cancel_token.clone();
367
368 let mock_track = Box::new(MockCallerTrack::new(active_call.session_id.clone()));
371
372 let accept_option = active_call.progress.load_full().option.clone().unwrap();
375 active_call
376 .setup_track_with_stream(&accept_option, mock_track)
377 .await?;
378
379 let received =
382 tokio::time::timeout(std::time::Duration::from_secs(3), asr_created_rx.recv()).await;
383 assert!(
384 received.is_ok() && received.unwrap().is_some(),
385 "ASR processor was NOT created — setup_track_with_stream did not build \
386 processors from the accept option (regression: ringing-before-accept)"
387 );
388
389 cancel_token.cancel();
390 Ok(())
391 }
392
393 #[tokio::test]
407 async fn test_update_track_wrapper_does_not_build_asr_processor() -> Result<()> {
408 let (asr_created_tx, mut asr_created_rx) = mpsc::channel::<()>(1);
409
410 let mock_provider =
411 crate::transcription::TranscriptionType::Other("mock-ringing-asr".to_string());
412
413 let mut engine = StreamEngine::new();
414 engine.register_asr(
415 mock_provider.clone(),
416 Box::new(move |_tid, _tok, _opt, _es| {
417 let tx = asr_created_tx.clone();
418 Box::pin(async move {
419 let _ = tx.send(()).await;
420 Ok(Box::new(MockAsrClient)
421 as Box<dyn crate::transcription::TranscriptionClient>)
422 })
423 }),
424 );
425 let engine = Arc::new(engine);
426
427 let accept_option = crate::CallOption {
430 asr: Some(crate::transcription::TranscriptionOption {
431 provider: Some(mock_provider),
432 ..Default::default()
433 }),
434 ..Default::default()
435 };
436 let active_call = make_active_call_with_engine_and_option(
437 engine,
438 "mediacache_update_track_wrapper_test",
439 accept_option,
440 )
441 .await;
442 let cancel_token = active_call.cancel_token.clone();
443
444 let mock_track = Box::new(MockCallerTrack::new(active_call.session_id.clone()));
445 active_call.update_track_wrapper(mock_track, None).await;
446
447 let received =
450 tokio::time::timeout(std::time::Duration::from_millis(500), asr_created_rx.recv())
451 .await;
452 assert!(
453 received.is_err(),
454 "ASR builder fired during track preparation — double-ASR regression"
455 );
456
457 cancel_token.cancel();
458 Ok(())
459 }
460}
461
462#[derive(Deserialize)]
463#[serde(rename_all = "camelCase")]
464pub struct CallParams {
465 pub id: Option<String>,
466 #[serde(rename = "dump")]
467 pub dump_events: Option<bool>,
468 #[serde(rename = "ping")]
469 pub ping_interval: Option<u32>,
470 pub server_side_track: Option<String>,
471 #[serde(default)]
476 pub forward: Option<bool>,
477 #[serde(default)]
479 pub visited: Option<String>,
480}
481
482impl CallParams {
483 pub fn to_forward_query(&self) -> String {
486 let mut parts: Vec<String> = Vec::new();
487 if let Some(id) = &self.id {
488 parts.push(format!("id={}", urlencoding::encode(id)));
489 }
490 if let Some(dump) = self.dump_events {
491 parts.push(format!("dump={}", dump));
492 }
493 if let Some(ping) = self.ping_interval {
494 parts.push(format!("ping={}", ping));
495 }
496 if let Some(track) = &self.server_side_track {
497 parts.push(format!("server_side_track={}", urlencoding::encode(track)));
498 }
499 parts.push("forward=true".to_string());
500 parts.join("&")
501 }
502}
503
504#[derive(Debug, Serialize, Deserialize, Clone, Default, PartialEq, Eq)]
505#[serde(rename_all = "camelCase")]
506pub enum ActiveCallType {
507 Webrtc,
508 B2bua,
509 WebSocket,
510 #[default]
511 Sip,
512}
513
514pub type ActiveCallRef = Arc<ActiveCall>;
515
516pub struct ActiveCall {
520 pub cancel_token: CancellationToken,
521 pub call_type: ActiveCallType,
522 pub session_id: String,
523 pub start_time: DateTime<Utc>,
524 pub media_stream: Arc<MediaStream>,
525 pub track_config: TrackConfig,
526 pub event_sender: EventSender,
527 pub app_state: AppState,
528 pub invitation: Invitation,
529 pub cmd_sender: CommandSender,
530 pub dump_events: bool,
531 pub server_side_track_id: TrackId,
532
533 pub ssrc: u32,
535 pub bridge_paused: Arc<AtomicBool>,
537 pub progress: Arc<ArcSwap<CallProgress>>,
539 pub extras: Extras,
541 pub moh: ArcSwapOption<String>,
543 pub current_play_id: ArcSwapOption<String>,
545 pub tts_handle: ArcSwapOption<SynthesisHandle>,
547 pub refer_leg: ArcSwapOption<LegShared>,
549 pub ready_to_answer: ArcSwapOption<ReadyAnswer>,
552 pub refer_call_token: ArcSwapOption<CancellationToken>,
554 pub wait_input_timeout: ArcSwapOption<u32>,
556 pub pending_asr_resume: ArcSwapOption<(u32, TranscriptionOption)>,
558 pub audio_receiver: std::sync::Mutex<Option<WebsocketBytesReceiver>>,
560}
561
562impl ActiveCall {
563 pub fn leg(&self) -> LegShared {
565 LegShared {
566 ssrc: self.ssrc,
567 is_refer: false,
568 progress: self.progress.clone(),
569 extras: self.extras.clone(),
570 }
571 }
572
573 pub fn set_option(&self, option: CallOption) {
575 self.progress.rcu(|p| {
576 let mut p = CallProgress::clone(p);
577 p.option = Some(option.clone());
578 p
579 });
580 }
581
582 pub fn moh_path(&self) -> Option<String> {
583 self.moh.load_full().map(|s| s.to_string())
584 }
585
586 pub fn set_moh(&self, v: Option<String>) {
587 self.moh.store(v.map(Arc::new));
588 }
589
590 pub fn current_play(&self) -> Option<String> {
591 self.current_play_id.load_full().map(|s| s.to_string())
592 }
593
594 pub fn set_current_play(&self, v: Option<String>) {
595 self.current_play_id.store(v.map(Arc::new));
596 }
597
598 pub fn refer_leg_value(&self) -> Option<LegShared> {
599 self.refer_leg.load_full().map(|l| l.as_ref().clone())
600 }
601
602 pub fn set_refer_leg(&self, v: Option<LegShared>) {
603 self.refer_leg.store(v.map(Arc::new));
604 }
605
606 pub fn set_ready_to_answer(&self, ready: ReadyAnswer) {
608 self.ready_to_answer.store(Some(Arc::new(ready)));
609 }
610
611 pub fn take_ready_to_answer(&self) -> Option<Arc<ReadyAnswer>> {
612 self.ready_to_answer.swap(None)
613 }
614
615 pub fn has_ready_to_answer(&self) -> bool {
616 self.ready_to_answer.load().is_some()
617 }
618
619 pub fn take_refer_call_token(&self) -> Option<CancellationToken> {
622 self.refer_call_token.swap(None).map(|t| (*t).clone())
623 }
624
625 pub fn set_refer_call_token(&self, token: CancellationToken) {
626 self.refer_call_token.store(Some(Arc::new(token)));
627 }
628
629 pub fn take_wait_input_timeout(&self) -> Option<u32> {
632 self.wait_input_timeout.swap(None).map(|t| *t)
633 }
634
635 pub fn set_wait_input_timeout(&self, v: Option<u32>) {
636 self.wait_input_timeout.store(v.map(Arc::new));
637 }
638
639 pub fn set_pending_asr_resume(&self, v: (u32, TranscriptionOption)) {
641 self.pending_asr_resume.store(Some(Arc::new(v)));
642 }
643
644 pub fn take_pending_asr_resume(&self) -> Option<(u32, TranscriptionOption)> {
645 self.pending_asr_resume.swap(None).map(|a| (*a).clone())
646 }
647
648 pub fn set_extra(&self, key: &str, value: serde_json::Value) {
650 self.leg().set_extra(key, value);
651 }
652
653 fn has_pending_invite(&self) -> bool {
655 self.invitation
656 .find_dialog_id_by_session_id(&self.session_id)
657 .is_some()
658 }
659
660 async fn hangup_now(&self, reason: Option<CallRecordHangupReason>) {
662 self.do_hangup(reason, None, None, None).await.ok();
663 }
664
665 async fn reject_now(&self, code: Option<rsipstack::rsip::StatusCode>, reason: Option<String>) {
667 self.do_reject(code, reason).await.ok();
668 }
669}
670
671pub struct ReadyAnswer {
674 pub answer: String,
675 pub track: PendingCallerTrack,
676 pub dialog: InviteDialog,
677}
678
679pub struct ActiveCallGuard {
680 pub call: ActiveCallRef,
681 pub active_calls: usize,
682}
683
684impl ActiveCallGuard {
685 pub fn new(call: ActiveCallRef) -> Self {
686 let active_calls = {
687 call.app_state
688 .total_calls
689 .fetch_add(1, std::sync::atomic::Ordering::Relaxed);
690 let mut calls = call.app_state.active_calls.lock().unwrap();
691 calls.insert(call.session_id.clone(), call.clone());
692 calls.len()
693 };
694 Self { call, active_calls }
695 }
696}
697
698impl Drop for ActiveCallGuard {
699 fn drop(&mut self) {
700 self.call
701 .app_state
702 .active_calls
703 .lock()
704 .unwrap()
705 .remove(&self.call.session_id);
706 }
707}
708
709pub struct ActiveCallReceiver {
710 pub cmd_receiver: CommandReceiver,
711 pub dump_cmd_receiver: CommandReceiver,
712 pub dump_event_receiver: EventReceiver,
713}
714
715pub struct CallSpec {
717 pub call_type: ActiveCallType,
718 pub cancel_token: CancellationToken,
719 pub session_id: String,
720 pub invitation: Invitation,
721 pub app_state: AppState,
722 pub track_config: TrackConfig,
723 pub audio_receiver: Option<WebsocketBytesReceiver>,
725 pub dump_events: bool,
726 pub server_side_track_id: Option<TrackId>,
728 pub extras: Option<HashMap<String, serde_json::Value>>,
731}
732
733impl ActiveCall {
734 pub fn new(spec: CallSpec) -> Self {
735 let CallSpec {
736 call_type,
737 cancel_token,
738 session_id,
739 invitation,
740 app_state,
741 track_config,
742 audio_receiver,
743 dump_events,
744 server_side_track_id,
745 extras,
746 } = spec;
747 let event_sender = crate::event::create_event_sender();
748 let cmd_sender = tokio::sync::broadcast::Sender::<Command>::new(32);
749 let server_side_track_id = server_side_track_id.unwrap_or(SERVER_SIDE_TRACK_ID.to_string());
750 let media_stream_builder = MediaStreamBuilder::new(event_sender.clone())
751 .with_id(session_id.clone())
752 .with_cancel_token(cancel_token.child_token());
753 let media_stream = Arc::new(media_stream_builder.build());
754 let start_time = Utc::now();
755 let call_type_str = match &call_type {
757 ActiveCallType::Sip => "sip",
758 ActiveCallType::WebSocket => "websocket",
759 ActiveCallType::Webrtc => "webrtc",
760 ActiveCallType::B2bua => "b2bua",
761 };
762 let mut extras = extras.unwrap_or_default();
763 extras
764 .entry(crate::playbook::BUILTIN_SESSION_ID.to_string())
765 .or_insert_with(|| serde_json::Value::String(session_id.clone()));
766 extras
767 .entry(crate::playbook::BUILTIN_CALL_TYPE.to_string())
768 .or_insert_with(|| serde_json::Value::String(call_type_str.to_string()));
769 extras
770 .entry(crate::playbook::BUILTIN_START_TIME.to_string())
771 .or_insert_with(|| serde_json::Value::String(start_time.to_rfc3339()));
772
773 let progress = CallProgress {
774 session_id: session_id.clone(),
775 start_time: Some(start_time),
776 ..Default::default()
777 };
778
779 Self {
780 cancel_token,
781 call_type,
782 session_id,
783 start_time,
784 media_stream,
785 track_config,
786 event_sender,
787 app_state,
788 invitation,
789 cmd_sender,
790 dump_events,
791 server_side_track_id,
792 ssrc: rand::random::<u32>(),
793 bridge_paused: Arc::new(AtomicBool::new(false)),
794 progress: Arc::new(ArcSwap::from_pointee(progress)),
795 extras: Arc::new(ArcSwap::from_pointee(extras)),
796 moh: ArcSwapOption::new(None),
797 current_play_id: ArcSwapOption::new(None),
798 tts_handle: ArcSwapOption::new(None),
799 refer_leg: ArcSwapOption::new(None),
800 ready_to_answer: ArcSwapOption::new(None),
801 refer_call_token: ArcSwapOption::new(None),
802 wait_input_timeout: ArcSwapOption::new(None),
803 pending_asr_resume: ArcSwapOption::new(None),
804 audio_receiver: std::sync::Mutex::new(audio_receiver),
805 }
806 }
807
808 pub async fn enqueue_command(&self, command: Command) -> Result<()> {
809 self.cmd_sender
810 .send(command)
811 .map_err(|e| anyhow::anyhow!("Failed to send command: {}", e))?;
812 Ok(())
813 }
814
815 pub fn new_receiver(&self) -> ActiveCallReceiver {
819 ActiveCallReceiver {
820 cmd_receiver: self.cmd_sender.subscribe(),
821 dump_cmd_receiver: self.cmd_sender.subscribe(),
822 dump_event_receiver: self.event_sender.subscribe(),
823 }
824 }
825
826 pub async fn serve(self: Arc<Self>, receiver: ActiveCallReceiver) -> Result<()> {
834 let ActiveCallReceiver {
835 mut cmd_receiver,
836 dump_cmd_receiver,
837 dump_event_receiver,
838 } = receiver;
839
840 let mut event_receiver = self.event_sender.subscribe();
841 let (actor_tx, mut actor_rx) = mpsc::channel::<ActorMsg>(16);
842 let mut runtime = CallRuntime::new(actor_tx);
843 runtime.me = Some(self.clone());
844
845 self.app_state
846 .total_calls
847 .fetch_add(1, std::sync::atomic::Ordering::Relaxed);
848
849 let me = self.clone();
850 let actor = async move {
851 let _cancel_on_exit = CancelOnExit(&me.cancel_token);
855 let mut ticker = tokio::time::interval(Duration::from_millis(100));
856 let mut media_serve = Box::pin(me.media_stream.serve());
858 loop {
859 tokio::select! {
860 cmd = cmd_receiver.recv() => {
861 match cmd {
862 Ok(command) => {
863 if let Err(e) = Box::pin(me.dispatch(&mut runtime, command)).await {
867 warn!(session_id = me.session_id, "{}", e);
868 me.event_sender
869 .send(SessionEvent::Error {
870 track_id: me.session_id.clone(),
871 timestamp: crate::media::get_timestamp(),
872 sender: "command".to_string(),
873 error: e.to_string(),
874 code: None,
875 })
876 .ok();
877 }
878 }
879 Err(tokio::sync::broadcast::error::RecvError::Lagged(_)) => continue,
880 Err(_) => {
881 info!(session_id = me.session_id, "command loop done");
882 break;
883 }
884 }
885 }
886 ev = event_receiver.recv() => {
887 match ev {
888 Ok(event) => Box::pin(me.handle_event(&mut runtime, event)).await,
889 Err(tokio::sync::broadcast::error::RecvError::Lagged(_)) => continue,
890 Err(_) => {
891 info!(session_id = me.session_id, "event loop done");
892 break;
893 }
894 }
895 }
896 Some(msg) = actor_rx.recv() => {
897 if let Err(e) = Box::pin(me.handle_actor_msg(msg)).await {
898 warn!(session_id = me.session_id, "{}", e);
899 me.event_sender
900 .send(SessionEvent::Error {
901 track_id: me.session_id.clone(),
902 timestamp: crate::media::get_timestamp(),
903 sender: "command".to_string(),
904 error: e.to_string(),
905 code: None,
906 })
907 .ok();
908 }
909 }
910 _ = ticker.tick() => {
911 Box::pin(me.check_input_timeout(&mut runtime)).await;
912 }
913 _ = &mut media_serve => {
914 info!(session_id = me.session_id, "media stream loop done");
915 break;
916 }
917 _ = me.cancel_token.cancelled() => {
918 info!(session_id = me.session_id, "call cancelled - cleaning up resources");
919 break;
920 }
921 }
922 }
923 };
924
925 tokio::join!(
926 self.dump_loop(self.dump_events, dump_cmd_receiver, dump_event_receiver),
927 actor
928 );
929 Ok(())
930 }
931
932 async fn check_input_timeout(&self, runtime: &mut CallRuntime) {
934 let (start_time, expire) = runtime.input_timeout_expire;
935 if expire > 0 && crate::media::get_timestamp() >= start_time + expire as u64 {
936 info!(session_id = self.session_id, "wait input timeout reached");
937 runtime.input_timeout_expire = (0, 0);
938 self.event_sender
939 .send(SessionEvent::Silence {
940 track_id: self.server_side_track_id.clone(),
941 timestamp: crate::media::get_timestamp(),
942 start_time,
943 duration: expire as u64,
944 samples: None,
945 refer: Some(false),
946 })
947 .ok();
948 }
949 }
950
951 async fn handle_event(&self, runtime: &mut CallRuntime, event: SessionEvent) {
953 match event {
954 SessionEvent::Speaking { .. }
955 | SessionEvent::Dtmf { .. }
956 | SessionEvent::AsrDelta { .. }
957 | SessionEvent::AsrFinal { .. }
958 | SessionEvent::TrackStart { .. } => {
959 runtime.input_timeout_expire = (0, 0);
960 }
961 SessionEvent::TrackEnd {
962 track_id,
963 play_id,
964 ssrc,
965 auto_hangup,
966 ..
967 } => {
968 if track_id != self.server_side_track_id {
969 return;
970 }
971
972 if play_id != self.current_play() {
973 debug!(
974 session_id = self.session_id,
975 ?play_id,
976 current = ?self.current_play(),
977 "ignoring interrupted track end"
978 );
979 return;
980 }
981 self.set_current_play(None);
982 let moh_path = self.moh_path();
983 let wait_timeout_val = self.take_wait_input_timeout();
984
985 if let Some(path) = moh_path {
986 info!(session_id = self.session_id, "looping moh: {}", path);
987 let ssrc = rand::random::<u32>();
988 let file_track = self.make_file_track(path.clone(), ssrc);
989 self.update_track_wrapper(Box::new(file_track), Some(path))
990 .await;
991 return;
992 }
993
994 if let Some(hangup_reason) = auto_hangup {
995 info!(
996 session_id = self.session_id,
997 ssrc, "auto hangup when track end track_id:{}", track_id
998 );
999 self.do_hangup(Some(hangup_reason), None, None, None)
1000 .await
1001 .ok();
1002 }
1003
1004 if let Some(timeout) = wait_timeout_val {
1005 runtime.input_timeout_expire = if timeout > 0 {
1006 (crate::media::get_timestamp(), timeout)
1007 } else {
1008 (0, 0)
1009 };
1010 }
1011 }
1012 SessionEvent::Interrupt { receiver } => {
1013 let track_id = receiver.unwrap_or_else(|| self.server_side_track_id.clone());
1014 if track_id == self.server_side_track_id {
1015 debug!(
1016 session_id = self.session_id,
1017 "received interrupt event, stopping playback"
1018 );
1019 self.do_interrupt(true).await.ok();
1020 }
1021 }
1022 SessionEvent::Inactivity { track_id, .. } => {
1023 info!(
1024 session_id = self.session_id,
1025 track_id, "inactivity timeout reached, hanging up"
1026 );
1027 self.do_hangup(
1028 Some(CallRecordHangupReason::InactivityTimeout),
1029 None,
1030 None,
1031 None,
1032 )
1033 .await
1034 .ok();
1035 }
1036 SessionEvent::Hangup { refer, .. } => {
1037 if refer == Some(true) {
1039 if let Some((refer_ssrc, asr_option)) = self.take_pending_asr_resume() {
1040 let is_refer_hangup = self
1042 .refer_leg
1043 .load_full()
1044 .map(|leg| leg.ssrc == refer_ssrc)
1045 .unwrap_or(false);
1046
1047 if is_refer_hangup {
1048 info!(
1049 session_id = self.session_id,
1050 "Refer call ended, resuming parent ASR"
1051 );
1052
1053 match self
1055 .app_state
1056 .stream_engine
1057 .create_asr_processor(
1058 self.server_side_track_id.clone(),
1059 self.cancel_token.child_token(),
1060 asr_option,
1061 self.event_sender.clone(),
1062 )
1063 .await
1064 {
1065 Ok(asr_processor) => {
1066 if let Err(e) = self
1067 .media_stream
1068 .append_processor(&self.server_side_track_id, asr_processor)
1069 .await
1070 {
1071 warn!(
1072 session_id = self.session_id,
1073 "Failed to resume ASR after refer: {}", e
1074 );
1075 }
1076 }
1077 Err(e) => {
1078 warn!(
1079 session_id = self.session_id,
1080 "Failed to create ASR processor for resume: {}", e
1081 );
1082 }
1083 }
1084 }
1085 }
1086 }
1087 }
1088 SessionEvent::Error { track_id, .. } => {
1089 if track_id != self.server_side_track_id {
1090 return;
1091 }
1092
1093 let moh_info = {
1094 let path = self.moh_path();
1095 path.map(|path| {
1096 let fallback = "./config/sounds/refer_moh.wav".to_string();
1097 if path != fallback && std::path::Path::new(&fallback).exists() {
1098 info!(
1099 session_id = self.session_id,
1100 "moh error, switching to fallback: {}", fallback
1101 );
1102 self.set_moh(Some(fallback.clone()));
1103 fallback
1104 } else {
1105 info!(
1106 session_id = self.session_id,
1107 "looping moh on error: {}", path
1108 );
1109 path
1110 }
1111 })
1112 };
1113
1114 if let Some(next_path) = moh_info {
1115 let ssrc = rand::random::<u32>();
1116 let file_track = self.make_file_track(next_path.clone(), ssrc);
1117 self.update_track_wrapper(Box::new(file_track), Some(next_path))
1118 .await;
1119 }
1120 }
1121 SessionEvent::Hold { on_hold, .. } => {
1122 self.bridge_paused.store(on_hold, Ordering::Relaxed);
1123 }
1124 _ => {}
1125 }
1126 }
1127
1128 async fn handle_actor_msg(&self, msg: ActorMsg) -> Result<()> {
1130 match msg {
1131 ActorMsg::ReferDone {
1132 track_id,
1133 forward_dtmf,
1134 result,
1135 } => match result {
1136 Ok(answer) => {
1137 self.media_stream
1138 .set_track_refer(&track_id, Some(true))
1139 .await;
1140 if !forward_dtmf {
1141 self.media_stream
1142 .set_track_dtmf_forward(&track_id, false)
1143 .await;
1144 }
1145 self.event_sender
1146 .send(SessionEvent::Answer {
1147 timestamp: crate::media::get_timestamp(),
1148 track_id,
1149 sdp: answer,
1150 refer: Some(true),
1151 })
1152 .ok();
1153 Ok(())
1154 }
1155 Err(e) => {
1156 warn!(
1157 session_id = self.session_id,
1158 "failed to create refer sip track: {}", e
1159 );
1160 self.emit_reject_from_rsip_error(track_id, true, &e);
1161 Err(e.into())
1162 }
1163 },
1164 }
1165 }
1166
1167 async fn dispatch(&self, runtime: &mut CallRuntime, command: Command) -> Result<()> {
1168 match command {
1169 Command::Invite { option } => self.do_invite(option).await,
1170 Command::Accept { option } => self.do_accept(option).await,
1171 Command::Reject { reason, code } => {
1172 self.do_reject(code.map(|c| (c as u16).into()), Some(reason))
1173 .await
1174 }
1175 Command::Ringing { .. } => self.do_ringing(command).await,
1176 Command::Tts { .. } => self.do_tts(command).await,
1177 Command::Play { .. } => self.do_play(command).await,
1178 Command::Hangup {
1179 reason,
1180 initiator,
1181 headers,
1182 refer,
1183 } => {
1184 let reason = reason.map(|r| {
1185 r.parse::<CallRecordHangupReason>()
1186 .unwrap_or(CallRecordHangupReason::BySystem)
1187 });
1188 self.do_hangup(reason, initiator, headers, refer).await
1189 }
1190 Command::Refer {
1191 caller,
1192 callee,
1193 options,
1194 } => self.do_refer(runtime, caller, callee, options).await,
1195 Command::Message {
1196 body,
1197 content_type,
1198 headers,
1199 refer,
1200 } => self.do_message(body, content_type, headers, refer).await,
1201 Command::Bridge { target_session_id } => self.do_bridge(target_session_id).await,
1202 Command::Unbridge { target_session_id } => self.do_unbridge(target_session_id).await,
1203 Command::Mute { track_id } => self.do_mute(track_id).await,
1204 Command::Unmute { track_id } => self.do_unmute(track_id).await,
1205 Command::Pause {} => self.do_pause().await,
1206 Command::Resume {} => self.do_resume().await,
1207 Command::Interrupt {
1208 graceful: passage,
1209 fade_out_ms: _,
1210 } => self.do_interrupt(passage.unwrap_or_default()).await,
1211 Command::History { speaker, text } => self.do_history(speaker, text).await,
1212 Command::Custom { sender, data } => self.do_custom(sender, data),
1213 Command::AddIceCandidate {
1214 candidate,
1215 sdp_mid,
1216 sdp_mline_index,
1217 } => {
1218 self.media_stream
1219 .add_ice_candidate(&candidate, sdp_mid.as_deref(), sdp_mline_index)
1220 .await
1221 }
1222 }
1223 }
1224
1225 fn build_record_option(&self, option: &CallOption) -> Option<RecorderOption> {
1226 if let Some(recorder_option) = &option.recorder {
1227 let mut recorder_file = recorder_option.recorder_file.clone();
1228 if recorder_file.contains("{id}") {
1229 recorder_file = recorder_file.replace("{id}", &self.session_id);
1230 }
1231
1232 let recorder_file = if recorder_file.is_empty() {
1233 self.app_state.get_recorder_file(&self.session_id)
1234 } else {
1235 let p = Path::new(&recorder_file);
1236 p.is_absolute()
1237 .then(|| recorder_file.clone())
1238 .unwrap_or_else(|| self.app_state.get_recorder_file(&recorder_file))
1239 };
1240 info!(
1241 session_id = self.session_id,
1242 recorder_file, "created recording file"
1243 );
1244
1245 let track_samplerate = self.track_config.samplerate;
1246 let recorder_samplerate = if track_samplerate > 0 {
1247 track_samplerate
1248 } else {
1249 recorder_option.samplerate
1250 };
1251 let recorder_ptime = if recorder_option.ptime == 0 {
1252 200
1253 } else {
1254 recorder_option.ptime
1255 };
1256 let requested_format = recorder_option
1257 .format
1258 .unwrap_or(self.app_state.config.recorder_format());
1259 let format = requested_format.effective();
1260 if requested_format != format {
1261 warn!(
1262 session_id = self.session_id,
1263 requested = requested_format.extension(),
1264 "Recorder format fallback to wav due to unsupported feature"
1265 );
1266 }
1267 let mut recorder_config = RecorderOption {
1268 recorder_file,
1269 samplerate: recorder_samplerate,
1270 ptime: recorder_ptime,
1271 format: Some(format),
1272 };
1273 recorder_config.ensure_path_extension(format);
1274 Some(recorder_config)
1275 } else {
1276 None
1277 }
1278 }
1279
1280 async fn invite_or_accept(&self, mut option: CallOption, sender: String) -> Result<CallOption> {
1281 {
1283 let state = self.progress.load_full();
1284 option = state.merge_option(option);
1285 }
1286
1287 option.check_default();
1288 if let Some(opt) = self.build_record_option(&option) {
1289 self.media_stream.update_recorder_option(opt).await;
1290 }
1291 self.ensure_call_ambiance(&option).await;
1292
1293 if let Some(opt) = &option.media_pass {
1294 let track_id = self.server_side_track_id.clone();
1295 let cancel_token = self.cancel_token.child_token();
1296 let ssrc = rand::random::<u32>();
1297 let media_pass_track = MediaPassTrack::new(
1298 self.session_id.clone(),
1299 ssrc,
1300 track_id,
1301 cancel_token,
1302 opt.clone(),
1303 );
1304 self.update_track_wrapper(Box::new(media_pass_track), None)
1305 .await;
1306 }
1307
1308 info!(
1309 session_id = self.session_id,
1310 call_type = ?self.call_type,
1311 sender,
1312 ?option,
1313 "caller with option"
1314 );
1315
1316 match self.setup_caller_track(&option).await {
1317 Ok(_) => return Ok(option),
1318 Err(e) => {
1319 self.app_state
1320 .total_failed_calls
1321 .fetch_add(1, std::sync::atomic::Ordering::Relaxed);
1322 let error_event = crate::event::SessionEvent::Error {
1323 track_id: self.session_id.clone(),
1324 timestamp: crate::media::get_timestamp(),
1325 sender,
1326 error: e.to_string(),
1327 code: None,
1328 };
1329 self.event_sender.send(error_event).ok();
1330 self.hangup_now(Some(CallRecordHangupReason::BySystem))
1331 .await;
1332 return Err(e);
1333 }
1334 }
1335 }
1336
1337 async fn do_invite(&self, option: CallOption) -> Result<()> {
1338 self.invite_or_accept(option, "invite".to_string())
1339 .await
1340 .map(|_| ())
1341 }
1342
1343 async fn do_accept(&self, mut option: CallOption) -> Result<()> {
1344 let has_pending = self.has_pending_invite();
1345 let ready_to_answer_val = !self.has_ready_to_answer();
1346
1347 if ready_to_answer_val {
1348 if !has_pending {
1349 warn!(session_id = self.session_id, "no pending call to accept");
1351 let rejet_event = crate::event::SessionEvent::Reject {
1352 track_id: self.session_id.clone(),
1353 timestamp: crate::media::get_timestamp(),
1354 reason: "no pending call".to_string(),
1355 refer: None,
1356 code: Some(486),
1357 };
1358 self.event_sender.send(rejet_event).ok();
1359 self.hangup_now(Some(CallRecordHangupReason::BySystem))
1360 .await;
1361 return Err(anyhow::anyhow!("no pending call to accept"));
1362 }
1363 option = self.invite_or_accept(option, "accept".to_string()).await?;
1364 } else {
1365 option.check_default();
1366 if let Some(opt) = self.build_record_option(&option) {
1367 self.media_stream.update_recorder_option(opt).await;
1368 }
1369 self.set_option(option.clone());
1370 self.ensure_call_ambiance(&option).await;
1371 }
1372 info!(session_id = self.session_id, ?option, "accepting call");
1373 let ready = self.take_ready_to_answer();
1374 if let Some(ready) = ready {
1375 let ReadyAnswer {
1378 answer,
1379 track: pending_track,
1380 dialog,
1381 } = match Arc::try_unwrap(ready) {
1382 Ok(ready) => ready,
1383 Err(_) => {
1384 warn!(
1385 session_id = self.session_id,
1386 "ready_to_answer held elsewhere; skipping accept"
1387 );
1388 return Ok(());
1389 }
1390 };
1391 info!(session_id = self.session_id, "ready to answer with track");
1392
1393 let headers = vec![rsipstack::rsip::Header::ContentType(
1394 "application/sdp".to_string().into(),
1395 )];
1396
1397 match dialog.accept(Some(headers), Some(answer.as_bytes().to_vec())) {
1398 Ok(_) => {
1399 self.leg().update_progress(|p| {
1400 p.answer = Some(answer.clone());
1401 p.answer_time.get_or_insert_with(Utc::now);
1402 });
1403 self.finish_caller_stack(&option, pending_track).await?;
1404 }
1405 Err(e) => {
1406 warn!(session_id = self.session_id, "failed to accept call: {}", e);
1407 return Err(anyhow::anyhow!("failed to accept call"));
1408 }
1409 }
1410 }
1411 return Ok(());
1412 }
1413
1414 async fn do_reject(
1415 &self,
1416 code: Option<rsipstack::rsip::StatusCode>,
1417 reason: Option<String>,
1418 ) -> Result<()> {
1419 match self
1420 .invitation
1421 .find_dialog_id_by_session_id(&self.session_id)
1422 {
1423 Some(id) => {
1424 info!(
1425 session_id = self.session_id,
1426 ?reason,
1427 ?code,
1428 "rejecting call"
1429 );
1430 let result = self.invitation.hangup(id, code, reason).await;
1431 if result.is_ok() {
1432 self.cancel_token.cancel();
1433 }
1434 result
1435 }
1436 None => {
1437 if let Some(ready) = self.take_ready_to_answer() {
1438 info!(
1439 session_id = self.session_id,
1440 ?reason,
1441 ?code,
1442 "rejecting call from ready_to_answer"
1443 );
1444 let dialog = &ready.dialog;
1445 let dialog_id = dialog.id();
1446 dialog.reject(code, reason).ok();
1447 self.invitation.dialog_layer.remove_dialog(&dialog_id);
1448 self.cancel_token.cancel();
1449 }
1450 Ok(())
1451 }
1452 }
1453 }
1454
1455 async fn do_ringing(&self, command: Command) -> Result<()> {
1456 let Command::Ringing {
1457 ringtone,
1458 recorder,
1459 early_media,
1460 } = command
1461 else {
1462 unreachable!("do_ringing called with non-Ringing command");
1463 };
1464
1465 if !self.has_ready_to_answer() {
1466 let option = CallOption {
1467 recorder,
1468 ..Default::default()
1469 };
1470 let _ = self.invite_or_accept(option, "ringing".to_string()).await?;
1471 }
1472
1473 if let Some(ready) = self.ready_to_answer.load_full() {
1474 let (headers, body) = if early_media.unwrap_or_default() || ringtone.is_some() {
1475 let headers = vec![rsipstack::rsip::Header::ContentType(
1476 "application/sdp".to_string().into(),
1477 )];
1478 (Some(headers), Some(ready.answer.as_bytes().to_vec()))
1479 } else {
1480 (None, None)
1481 };
1482
1483 ready.dialog.ringing(headers, body).ok();
1484 info!(
1485 session_id = self.session_id,
1486 ringtone, early_media, "playing ringtone"
1487 );
1488 if let Some(ringtone_url) = ringtone {
1489 self.do_play(Command::Play {
1490 url: ringtone_url,
1491 play_id: None,
1492 auto_hangup: None,
1493 wait_input_timeout: None,
1494 offset_ms: None,
1495 })
1496 .await
1497 .ok();
1498 } else {
1499 info!(session_id = self.session_id, "no ringtone to play");
1500 }
1501 }
1502 Ok(())
1503 }
1504
1505 async fn do_tts(&self, command: Command) -> Result<()> {
1506 let Command::Tts {
1507 text,
1508 speaker,
1509 play_id,
1510 auto_hangup,
1511 streaming,
1512 end_of_stream,
1513 option,
1514 wait_input_timeout,
1515 base64,
1516 cache_key,
1517 } = command
1518 else {
1519 unreachable!("do_tts called with non-Tts command");
1520 };
1521 let streaming = streaming.unwrap_or_default();
1522 let end_of_stream = end_of_stream.unwrap_or_default();
1523 let base64 = base64.unwrap_or_default();
1524
1525 let tts_option = {
1526 let call_state = self.progress.load_full();
1527 match call_state.option.clone().unwrap_or_default().tts {
1528 Some(opt) => opt.merge_with(option),
1529 None => {
1530 if let Some(opt) = option {
1531 opt
1532 } else {
1533 return Err(anyhow::anyhow!("no tts option available"));
1534 }
1535 }
1536 }
1537 };
1538 let speaker = match speaker {
1539 Some(s) => Some(s),
1540 None => tts_option.speaker.clone(),
1541 };
1542
1543 let mut play_command = SynthesisCommand {
1544 text,
1545 speaker,
1546 play_id: play_id.clone(),
1547 streaming,
1548 end_of_stream: if !streaming { true } else { end_of_stream },
1549 option: tts_option,
1550 base64,
1551 cache_key,
1552 auto_hangup,
1553 };
1554 info!(
1555 session_id = self.session_id,
1556 provider = ?play_command.option.provider,
1557 text = %play_command.text.chars().take(10).collect::<String>(),
1558 speaker = play_command.speaker.as_deref(),
1559 auto_hangup = auto_hangup.unwrap_or_default(),
1560 play_id = play_command.play_id.as_deref(),
1561 streaming = play_command.streaming,
1562 end_of_stream = play_command.end_of_stream,
1563 wait_input_timeout = wait_input_timeout.unwrap_or_default(),
1564 is_base64 = play_command.base64,
1565 cache_key = play_command.cache_key.as_deref(),
1566 "new synthesis"
1567 );
1568
1569 let ssrc = rand::random::<u32>();
1570 let (should_interrupt, picked_ssrc) = {
1571 let existing_handle = self.tts_handle.load_full();
1572 let current_play_id = self.current_play();
1573
1574 let (target_ssrc, changed) = if let Some(handle) = &existing_handle {
1575 if play_id.is_some() && current_play_id != play_id {
1576 (ssrc, true)
1577 } else {
1578 (handle.ssrc, false)
1579 }
1580 } else {
1581 (ssrc, false)
1582 };
1583
1584 self.set_wait_input_timeout(wait_input_timeout);
1587
1588 self.set_current_play(play_id.clone());
1589 (changed, target_ssrc)
1590 };
1591
1592 if should_interrupt {
1593 let _ = self.do_interrupt(false).await;
1594 }
1595
1596 let existing_handle = self.tts_handle.load_full();
1600 if let Some(tts_handle) = existing_handle {
1601 match tts_handle.try_send(play_command) {
1602 Ok(_) => return Ok(()),
1603 Err(e) => {
1604 play_command = e.0;
1605 }
1606 }
1607 }
1608
1609 let (new_handle, tts_track) = StreamEngine::create_tts_track(
1610 self.app_state.stream_engine.clone(),
1611 self.cancel_token.child_token(),
1612 self.session_id.clone(),
1613 self.server_side_track_id.clone(),
1614 picked_ssrc,
1615 play_id.clone(),
1616 streaming,
1617 &play_command.option,
1618 play_command.auto_hangup,
1619 )
1620 .await?;
1621
1622 new_handle.try_send(play_command)?;
1623 self.tts_handle.store(Some(Arc::new(new_handle)));
1624 self.update_track_wrapper(tts_track, play_id).await;
1625 Ok(())
1626 }
1627
1628 async fn do_play(&self, command: Command) -> Result<()> {
1629 let Command::Play {
1630 url,
1631 play_id,
1632 auto_hangup,
1633 wait_input_timeout,
1634 offset_ms,
1635 } = command
1636 else {
1637 unreachable!("do_play called with non-Play command");
1638 };
1639 let ssrc = rand::random::<u32>();
1640 info!(
1641 session_id = self.session_id,
1642 ssrc, url, play_id, auto_hangup, "play file track"
1643 );
1644
1645 let play_id = play_id.or(Some(url.clone()));
1646
1647 let mut file_track = self
1649 .make_file_track(url, ssrc)
1650 .with_play_id(play_id.clone())
1651 .with_auto_hangup(auto_hangup);
1652
1653 if let Some(offset) = offset_ms {
1654 file_track = file_track.with_offset_ms(offset);
1655 }
1656
1657 {
1658 self.tts_handle.store(None);
1659 self.set_wait_input_timeout(wait_input_timeout);
1660 }
1661
1662 self.update_track_wrapper(Box::new(file_track), play_id)
1663 .await;
1664 Ok(())
1665 }
1666
1667 async fn do_history(&self, speaker: String, text: String) -> Result<()> {
1668 self.event_sender
1669 .send(SessionEvent::AddHistory {
1670 sender: Some(self.session_id.clone()),
1671 timestamp: crate::media::get_timestamp(),
1672 speaker,
1673 text,
1674 })
1675 .map(|_| ())
1676 .map_err(Into::into)
1677 }
1678
1679 fn do_custom(&self, sender: Option<String>, data: serde_json::Value) -> Result<()> {
1680 self.event_sender
1681 .send(SessionEvent::Custom {
1682 timestamp: crate::media::get_timestamp(),
1683 sender,
1684 data,
1685 })
1686 .map(|_| ())
1687 .map_err(Into::into)
1688 }
1689
1690 async fn do_interrupt(&self, graceful: bool) -> Result<()> {
1691 {
1692 self.tts_handle.store(None);
1693 self.set_moh(None);
1694 }
1695 self.media_stream
1696 .remove_track(&self.server_side_track_id, graceful)
1697 .await;
1698 Ok(())
1699 }
1700 async fn do_pause(&self) -> Result<()> {
1701 self.media_stream
1702 .pause_playback(self.server_side_track_id.clone())
1703 .await?;
1704 Ok(())
1705 }
1706 async fn do_resume(&self) -> Result<()> {
1707 self.media_stream
1708 .resume_playback(self.server_side_track_id.clone())
1709 .await?;
1710 Ok(())
1711 }
1712 async fn do_hangup(
1713 &self,
1714 reason: Option<CallRecordHangupReason>,
1715 initiator: Option<String>,
1716 headers: Option<HashMap<String, String>>,
1717 refer: Option<bool>,
1718 ) -> Result<()> {
1719 info!(
1720 session_id = self.session_id,
1721 ?reason,
1722 ?initiator,
1723 ?headers,
1724 ?refer,
1725 "do_hangup"
1726 );
1727
1728 let hangup_reason = match initiator.as_deref() {
1729 Some("caller") => CallRecordHangupReason::ByCaller,
1730 Some("callee") => CallRecordHangupReason::ByCallee,
1731 Some("system") => CallRecordHangupReason::Autohangup,
1732 _ => reason.unwrap_or(CallRecordHangupReason::BySystem),
1733 };
1734
1735 match refer {
1736 Some(true) => {
1737 let refer_token = self.take_refer_call_token();
1739 let refer_leg = self.refer_leg_value();
1740 let has_refer_leg = refer_leg.is_some();
1741 if let Some(leg) = refer_leg {
1742 if let Some(headers) = headers {
1743 let h_val = serde_json::to_value(&headers).unwrap_or_default();
1744 leg.set_extra("_hangup_headers", h_val);
1745 }
1746 let reason = hangup_reason.clone();
1748 leg.update_progress(|p| p.set_hangup_reason(reason.clone()));
1749 }
1750 if let Some(token) = refer_token {
1751 token.cancel();
1752 }
1753 if has_refer_leg {
1754 self.media_stream
1755 .remove_track(&self.server_side_track_id, false)
1756 .await;
1757 }
1758 }
1759 _ => {
1760 if let Some(headers) = headers {
1761 let h_val = serde_json::to_value(&headers).unwrap_or_default();
1762 self.leg().set_extra("_hangup_headers", h_val);
1763 }
1764 self.leg()
1765 .update_progress(|p| p.set_hangup_reason(hangup_reason.clone()));
1766 let refer_token = self.take_refer_call_token();
1767 self.media_stream
1768 .stop(Some(hangup_reason.to_string()), initiator);
1769 if let Some(token) = refer_token {
1770 token.cancel();
1771 }
1772 }
1773 }
1774 tokio::task::yield_now().await;
1775 Ok(())
1776 }
1777
1778 async fn do_refer(
1784 &self,
1785 runtime: &mut CallRuntime,
1786 caller: String,
1787 callee: String,
1788 refer_option: Option<ReferOption>,
1789 ) -> Result<()> {
1790 self.do_interrupt(false).await.ok();
1791
1792 let pause_parent_asr = refer_option
1794 .as_ref()
1795 .and_then(|o| o.pause_parent_asr)
1796 .unwrap_or(false);
1797
1798 let original_asr_option = if pause_parent_asr {
1800 self.progress
1801 .load_full()
1802 .option
1803 .as_ref()
1804 .and_then(|o| o.asr.clone())
1805 } else {
1806 None
1807 };
1808
1809 if pause_parent_asr {
1811 info!(
1812 session_id = self.session_id,
1813 "Pausing parent call ASR during refer"
1814 );
1815 self.media_stream
1816 .remove_processor::<crate::media::asr_processor::AsrProcessor>(
1817 &self.server_side_track_id,
1818 )
1819 .await
1820 .ok();
1821 }
1822
1823 let mut moh = refer_option.as_ref().and_then(|o| o.moh.clone());
1824 if let Some(ref path) = moh {
1825 if !path.starts_with("http") && !std::path::Path::new(path).exists() {
1826 let fallback = "./config/sounds/refer_moh.wav";
1827 if std::path::Path::new(fallback).exists() {
1828 info!(
1829 session_id = self.session_id,
1830 "moh {} not found, using fallback {}", path, fallback
1831 );
1832 moh = Some(fallback.to_string());
1833 }
1834 }
1835 }
1836 let ref_call_id = refer_option
1837 .as_ref()
1838 .and_then(|o| o.call_id.clone())
1839 .unwrap_or_else(|| format!("ref-{}-{}", rand::random::<u32>(), self.session_id));
1840
1841 let session_id = self.session_id.clone();
1842 let track_id = self.server_side_track_id.clone();
1843
1844 let (recorder, parent_caller) = {
1845 let progress = self.progress.load_full();
1846 let option = progress.option.as_ref();
1847 (
1848 option.map(|o| o.recorder.clone()).unwrap_or_default(),
1849 option.and_then(|o| o.caller.clone()),
1850 )
1851 };
1852 let caller = if caller.trim().is_empty() {
1853 parent_caller.unwrap_or_default()
1854 } else {
1855 caller
1856 };
1857
1858 let mut call_option = CallOption {
1859 caller: Some(caller),
1860 callee: Some(callee.clone()),
1861 sip: refer_option.as_ref().and_then(|o| o.sip.clone()),
1862 vad: refer_option
1863 .as_ref()
1864 .and_then(|o| o.vad.clone())
1865 .map(|mut opts| {
1866 opts.refer = Some(true);
1867 opts
1868 }),
1869 asr: refer_option
1870 .as_ref()
1871 .and_then(|o| o.asr.clone())
1872 .map(|mut opts| {
1873 opts.refer = Some(true);
1874 opts
1875 }),
1876 denoise: refer_option.as_ref().and_then(|o| o.denoise.clone()),
1877 agc: refer_option.as_ref().and_then(|o| o.agc.clone()),
1878 recorder,
1879 ..Default::default()
1880 };
1881 call_option.check_default();
1882
1883 let mut invite_option = call_option.build_invite_option()?;
1884 invite_option.call_id = Some(ref_call_id.clone());
1885
1886 let headers = invite_option.headers.get_or_insert_with(|| Vec::new());
1887
1888 {
1889 let progress = self.progress.load_full();
1890 if let Some(opt) = progress.option.as_ref() {
1891 if let Some(callee) = opt.callee.as_ref() {
1892 headers.push(rsipstack::rsip::Header::Other(
1893 "X-Referred-To".to_string(),
1894 callee.clone(),
1895 ));
1896 }
1897 if let Some(caller) = opt.caller.as_ref() {
1898 headers.push(rsipstack::rsip::Header::Other(
1899 "X-Referred-From".to_string(),
1900 caller.clone(),
1901 ));
1902 }
1903 }
1904 }
1905
1906 headers.push(rsipstack::rsip::Header::Other(
1907 "X-Referred-Id".to_string(),
1908 self.session_id.clone(),
1909 ));
1910
1911 let ssrc = rand::random::<u32>();
1912 let refer_leg = LegShared::new(
1913 ssrc,
1914 true,
1915 CallProgress {
1916 session_id: ref_call_id.clone(),
1917 start_time: Some(Utc::now()),
1918 option: Some(call_option.clone()),
1919 ..Default::default()
1920 },
1921 );
1922 self.set_refer_leg(Some(refer_leg.clone()));
1923
1924 let auto_hangup_requested = refer_option
1925 .as_ref()
1926 .and_then(|o| o.auto_hangup)
1927 .unwrap_or(true);
1928
1929 if !auto_hangup_requested && pause_parent_asr && original_asr_option.is_some() {
1933 let asr_option = original_asr_option.unwrap();
1934 self.set_pending_asr_resume((ssrc, asr_option));
1935 }
1936
1937 let timeout_secs = refer_option.as_ref().and_then(|o| o.timeout).unwrap_or(30);
1938 let forward_dtmf = refer_option
1939 .as_ref()
1940 .and_then(|o| o.forward_dtmf)
1941 .unwrap_or(true);
1942
1943 info!(
1944 session_id = self.session_id,
1945 ssrc,
1946 auto_hangup = auto_hangup_requested,
1947 callee,
1948 timeout_secs,
1949 "do_refer"
1950 );
1951
1952 let refer_cancel_token = self.cancel_token.child_token();
1953 self.set_refer_call_token(refer_cancel_token.clone());
1954
1955 let me = runtime
1958 .me
1959 .clone()
1960 .ok_or_else(|| anyhow::anyhow!("refer is only supported inside serve()"))?;
1961 let actor_tx = runtime.actor_tx.clone();
1962 let event_sender = self.event_sender.clone();
1963 let log_session_id = session_id.clone();
1964 let reject_track_id = track_id.clone();
1965 crate::spawn(async move {
1966 let out = crate::call::tracks::OutgoingLeg {
1967 cancel_token: refer_cancel_token,
1968 leg: refer_leg,
1969 track_id: track_id.clone(),
1970 invite_option,
1971 call_option,
1972 moh,
1973 auto_hangup: auto_hangup_requested,
1974 };
1975 let result = match tokio::time::timeout(
1976 Duration::from_secs(timeout_secs as u64),
1977 me.create_outgoing_sip_track(out),
1978 )
1979 .await
1980 {
1981 Ok(res) => res,
1982 Err(_) => {
1983 warn!(
1984 session_id = log_session_id,
1985 "refer sip track creation timed out after {} seconds", timeout_secs
1986 );
1987 event_sender
1988 .send(SessionEvent::Reject {
1989 track_id: reject_track_id,
1990 timestamp: crate::media::get_timestamp(),
1991 reason: "Timeout when refer".into(),
1992 code: Some(408),
1993 refer: Some(true),
1994 })
1995 .ok();
1996 Err(rsipstack::Error::Error(
1997 "refer sip track creation timed out".to_string(),
1998 ))
1999 }
2000 };
2001 me.set_moh(None);
2002 actor_tx
2003 .send(ActorMsg::ReferDone {
2004 track_id,
2005 forward_dtmf,
2006 result,
2007 })
2008 .await
2009 .ok();
2010 });
2011
2012 Ok(())
2013 }
2014
2015 async fn do_message(
2016 &self,
2017 body: String,
2018 content_type: Option<String>,
2019 headers: Option<HashMap<String, String>>,
2020 refer: Option<bool>,
2021 ) -> Result<()> {
2022 if !matches!(self.call_type, ActiveCallType::Sip | ActiveCallType::B2bua) {
2023 return Err(anyhow::anyhow!(
2024 "message command is only supported for SIP calls"
2025 ));
2026 }
2027
2028 let dialog_key = if refer == Some(true) {
2029 self.refer_leg_value()
2030 .map(|leg| leg.progress.load_full().session_id.clone())
2031 } else {
2032 Some(self.progress.load_full().session_id.clone())
2033 };
2034
2035 let mut dialog = dialog_key
2036 .as_ref()
2037 .filter(|id| !id.is_empty())
2038 .and_then(|id| self.invitation.dialog_layer.get_dialog_with(id));
2039
2040 if dialog.is_none() {
2041 if let Some(target_id) = dialog_key.as_ref().filter(|id| !id.is_empty()) {
2042 dialog = self
2043 .invitation
2044 .dialog_layer
2045 .all_dialog_ids()
2046 .into_iter()
2047 .filter_map(|id| self.invitation.dialog_layer.get_dialog_with(&id))
2048 .find(|dialog| dialog.id().to_string() == *target_id);
2049 }
2050 }
2051
2052 if dialog.is_none() {
2055 let call_id = match (refer == Some(true), dialog_key.as_deref()) {
2056 (true, Some(id)) if !id.is_empty() => Some(id),
2057 (false, _) => Some(self.session_id.as_str()),
2058 _ => None,
2059 };
2060 if let Some(call_id) = call_id {
2061 dialog = self
2062 .invitation
2063 .dialog_layer
2064 .get_client_dialog_by_call_id(call_id)
2065 .into_iter()
2066 .find(|d| {
2067 matches!(
2068 d.state(),
2069 rsipstack::dialog::dialog::DialogState::Confirmed(_, _)
2070 )
2071 })
2072 .map(rsipstack::dialog::dialog::Dialog::Invite);
2073 }
2074 }
2075
2076 let dialog = dialog.ok_or_else(|| {
2077 anyhow::anyhow!(
2078 "no established SIP dialog found for message command, refer={}",
2079 refer.unwrap_or_default()
2080 )
2081 })?;
2082
2083 let mut sip_headers = vec![rsipstack::rsip::Header::ContentType(
2084 content_type
2085 .clone()
2086 .unwrap_or_else(|| "text/plain;charset=utf-8".to_string())
2087 .into(),
2088 )];
2089 if let Some(headers) = &headers {
2090 sip_headers.extend(crate::sip_util::sip_headers_from_map(headers));
2091 }
2092
2093 info!(
2094 session_id = self.session_id,
2095 dialog_id = %dialog.id(),
2096 content_type = content_type.as_deref().unwrap_or("text/plain;charset=utf-8"),
2097 refer = refer.unwrap_or_default(),
2098 body = %body.chars().take(64).collect::<String>(),
2099 "sending SIP MESSAGE"
2100 );
2101
2102 let response = dialog
2103 .message(Some(sip_headers), Some(body.into_bytes()))
2104 .await?;
2105 match response {
2106 Some(resp)
2107 if resp.status_code.kind() == rsipstack::rsip::StatusCodeKind::Successful =>
2108 {
2109 Ok(())
2110 }
2111 Some(resp) => Err(anyhow::anyhow!(
2112 "SIP MESSAGE rejected with status {}",
2113 resp.status_code
2114 )),
2115 None => Err(anyhow::anyhow!(
2116 "SIP MESSAGE was not sent because dialog is not confirmed"
2117 )),
2118 }
2119 }
2120
2121 fn bridge_track_id(source_session_id: &str, target_session_id: &str) -> TrackId {
2122 format!("bridge:{}:to:{}", source_session_id, target_session_id)
2123 }
2124
2125 async fn do_bridge(&self, target_session_id: String) -> Result<()> {
2126 let target = {
2127 let calls = self.app_state.active_calls.lock().unwrap();
2128 calls.get(&target_session_id).cloned()
2129 };
2130 let target = target.ok_or_else(|| {
2131 anyhow::anyhow!("bridge target session not found: {}", target_session_id)
2132 })?;
2133
2134 if target.session_id == self.session_id {
2135 return Err(anyhow::anyhow!("cannot bridge a call to itself").into());
2136 }
2137
2138 let self_bridge_track_id = Self::bridge_track_id(&self.session_id, &target.session_id);
2139 let target_bridge_track_id = Self::bridge_track_id(&target.session_id, &self.session_id);
2140
2141 self.media_stream
2142 .remove_track(&self_bridge_track_id, false)
2143 .await;
2144 target
2145 .media_stream
2146 .remove_track(&target_bridge_track_id, false)
2147 .await;
2148
2149 let (self_bridge_sender, self_bridge_receiver) = mpsc::channel(25);
2150 let (target_bridge_sender, target_bridge_receiver) = mpsc::channel(25);
2151
2152 let self_paused = self.bridge_paused.clone();
2153 let target_paused = target.bridge_paused.clone();
2154
2155 let self_forwarding_track = ForwardingTrack::new(
2156 self_bridge_track_id.clone(),
2157 self.session_id.clone(),
2158 target_bridge_sender,
2159 self_bridge_receiver,
2160 self.track_config.clone(),
2161 self.cancel_token.child_token(),
2162 rand::random::<u32>(),
2163 self_paused,
2164 );
2165
2166 let target_forwarding_track = ForwardingTrack::new(
2167 target_bridge_track_id.clone(),
2168 target.session_id.clone(),
2169 self_bridge_sender,
2170 target_bridge_receiver,
2171 target.track_config.clone(),
2172 target.cancel_token.child_token(),
2173 rand::random::<u32>(),
2174 target_paused,
2175 );
2176
2177 self.media_stream
2178 .update_track(Box::new(self_forwarding_track), None)
2179 .await;
2180 target
2181 .media_stream
2182 .update_track(Box::new(target_forwarding_track), None)
2183 .await;
2184
2185 info!(
2186 session_id = self.session_id,
2187 target = target_session_id,
2188 self_bridge_track_id,
2189 target_bridge_track_id,
2190 "audio bridge established"
2191 );
2192 Ok(())
2193 }
2194
2195 async fn do_unbridge(&self, target_session_id: String) -> Result<()> {
2196 let target = {
2197 let calls = self.app_state.active_calls.lock().unwrap();
2198 calls.get(&target_session_id).cloned()
2199 };
2200
2201 let self_bridge_track_id = Self::bridge_track_id(&self.session_id, &target_session_id);
2202 self.media_stream
2203 .remove_track(&self_bridge_track_id, false)
2204 .await;
2205
2206 if let Some(target) = target {
2207 let target_bridge_track_id =
2208 Self::bridge_track_id(&target.session_id, &self.session_id);
2209 target
2210 .media_stream
2211 .remove_track(&target_bridge_track_id, false)
2212 .await;
2213 info!(
2214 session_id = self.session_id,
2215 target = target.session_id,
2216 self_bridge_track_id,
2217 target_bridge_track_id,
2218 "audio bridge removed"
2219 );
2220 } else {
2221 info!(
2222 session_id = self.session_id,
2223 target = target_session_id,
2224 self_bridge_track_id,
2225 "audio bridge removed locally; target session not active"
2226 );
2227 }
2228
2229 Ok(())
2230 }
2231
2232 async fn do_mute(&self, track_id: Option<String>) -> Result<()> {
2233 self.media_stream.mute_track(track_id).await;
2234 Ok(())
2235 }
2236
2237 async fn do_unmute(&self, track_id: Option<String>) -> Result<()> {
2238 self.media_stream.unmute_track(track_id).await;
2239 Ok(())
2240 }
2241
2242 pub async fn cleanup(&self) -> Result<()> {
2243 if matches!(self.call_type, ActiveCallType::Sip | ActiveCallType::B2bua) {
2244 self.reject_now(
2245 Some(rsipstack::rsip::StatusCode::Decline),
2246 Some("handler disconnected".to_string()),
2247 )
2248 .await;
2249 }
2250 self.tts_handle.store(None);
2251 self.media_stream.cleanup().await.ok();
2252 Ok(())
2253 }
2254
2255 pub fn get_callrecord(&self) -> Option<CallRecord> {
2258 let progress = self.progress.load_full();
2259 let extras = self.extras.load_full();
2260 let refer_leg = self.refer_leg_value();
2261 Some(build_callrecord(
2262 &progress,
2263 &extras,
2264 refer_leg.as_ref(),
2265 &self.app_state,
2266 self.session_id.clone(),
2267 self.call_type.clone(),
2268 ))
2269 }
2270
2271 async fn dump_to_file(
2272 &self,
2273 dump_file: &mut File,
2274 cmd_receiver: &mut CommandReceiver,
2275 event_receiver: &mut EventReceiver,
2276 ) {
2277 loop {
2278 select! {
2279 _ = self.cancel_token.cancelled() => {
2280 break;
2281 }
2282 Ok(cmd) = cmd_receiver.recv() => {
2283 CallRecordEvent::write(CallRecordEventType::Command, cmd, dump_file)
2284 .await;
2285 }
2286 Ok(event) = event_receiver.recv() => {
2287 if matches!(event, SessionEvent::Binary{..}) {
2288 continue;
2289 }
2290 CallRecordEvent::write(CallRecordEventType::Event, event, dump_file)
2291 .await;
2292 }
2293 };
2294 }
2295 }
2296
2297 async fn dump_loop(
2298 &self,
2299 dump_events: bool,
2300 mut dump_cmd_receiver: CommandReceiver,
2301 mut dump_event_receiver: EventReceiver,
2302 ) {
2303 if !dump_events {
2304 return;
2305 }
2306
2307 let file_name = self.app_state.get_dump_events_file(&self.session_id);
2308 let mut dump_file = match File::options()
2309 .create(true)
2310 .append(true)
2311 .open(&file_name)
2312 .await
2313 {
2314 Ok(file) => file,
2315 Err(e) => {
2316 warn!(
2317 session_id = self.session_id,
2318 file_name, "failed to open dump events file: {}", e
2319 );
2320 return;
2321 }
2322 };
2323 self.dump_to_file(
2324 &mut dump_file,
2325 &mut dump_cmd_receiver,
2326 &mut dump_event_receiver,
2327 )
2328 .await;
2329
2330 while let Ok(event) = dump_event_receiver.try_recv() {
2331 if matches!(event, SessionEvent::Binary { .. }) {
2332 continue;
2333 }
2334 CallRecordEvent::write(CallRecordEventType::Event, event, &mut dump_file).await;
2335 }
2336 }
2337}
2338
2339struct CancelOnExit<'a>(&'a CancellationToken);
2342
2343impl Drop for CancelOnExit<'_> {
2344 fn drop(&mut self) {
2345 self.0.cancel();
2346 }
2347}
2348
2349impl Drop for ActiveCall {
2350 fn drop(&mut self) {
2351 info!(session_id = self.session_id, "dropping active call");
2352 if let Some(sender) = self.app_state.callrecord_sender.as_ref() {
2353 if let Some(record) = self.get_callrecord() {
2354 if let Err(e) = sender.send(record) {
2355 warn!(
2356 session_id = self.session_id,
2357 "failed to send call record: {}", e
2358 );
2359 }
2360 }
2361 }
2362 }
2363}