use crate::channel::{
AnswerState, CallDirection, CallState, ChannelState, ChannelTimetable, HangupCause,
ParseAnswerStateError, ParseCallDirectionError, ParseCallStateError, ParseChannelStateError,
ParseHangupCauseError, ParseTimetableError,
};
#[cfg(feature = "esl")]
use crate::event::{EslEventPriority, ParsePriorityError};
use crate::headers::EventHeader;
use crate::sofia::{
GatewayPingStatus, GatewayRegState, ParseGatewayPingStatusError, ParseGatewayRegStateError,
ParseSipUserPingStatusError, ParseSofiaEventSubclassError, SipUserPingStatus,
SofiaEventSubclass,
};
use crate::variables::VariableName;
use sip_header::SipHeaderLookup;
use std::str::FromStr;
#[inline]
fn parse_opt<T: FromStr>(raw: Option<&str>) -> Result<Option<T>, T::Err> {
raw.map(str::parse)
.transpose()
}
pub trait HeaderLookup: SipHeaderLookup {
fn header_str(&self, name: &str) -> Option<&str>;
fn variable_str(&self, name: &str) -> Option<&str>;
fn header(&self, name: EventHeader) -> Option<&str> {
self.header_str(name.as_str())
}
fn variable(&self, name: impl VariableName) -> Option<&str> {
self.variable_str(name.as_str())
}
fn unique_id(&self) -> Option<&str> {
self.header(EventHeader::UniqueId)
.or_else(|| self.header(EventHeader::CallerUniqueId))
}
fn job_uuid(&self) -> Option<&str> {
self.header(EventHeader::JobUuid)
}
fn channel_name(&self) -> Option<&str> {
self.header(EventHeader::ChannelName)
}
fn caller_id_number(&self) -> Option<&str> {
self.header(EventHeader::CallerCallerIdNumber)
}
fn caller_id_name(&self) -> Option<&str> {
self.header(EventHeader::CallerCallerIdName)
}
fn destination_number(&self) -> Option<&str> {
self.header(EventHeader::CallerDestinationNumber)
}
fn callee_id_number(&self) -> Option<&str> {
self.header(EventHeader::CallerCalleeIdNumber)
}
fn callee_id_name(&self) -> Option<&str> {
self.header(EventHeader::CallerCalleeIdName)
}
fn channel_presence_id(&self) -> Option<&str> {
self.header(EventHeader::ChannelPresenceId)
}
fn presence_call_direction(&self) -> Result<Option<CallDirection>, ParseCallDirectionError> {
parse_opt(self.header(EventHeader::PresenceCallDirection))
}
fn event_date_timestamp(&self) -> Option<&str> {
self.header(EventHeader::EventDateTimestamp)
}
fn event_sequence(&self) -> Option<&str> {
self.header(EventHeader::EventSequence)
}
fn dtmf_duration(&self) -> Option<&str> {
self.header(EventHeader::DtmfDuration)
}
fn dtmf_source(&self) -> Option<&str> {
self.header(EventHeader::DtmfSource)
}
fn hangup_cause(&self) -> Result<Option<HangupCause>, ParseHangupCauseError> {
parse_opt(self.header(EventHeader::HangupCause))
}
fn event_subclass(&self) -> Option<&str> {
self.header(EventHeader::EventSubclass)
}
fn sofia_event_subclass(
&self,
) -> Result<Option<SofiaEventSubclass>, ParseSofiaEventSubclassError> {
parse_opt(self.event_subclass())
}
fn gateway(&self) -> Option<&str> {
self.header(EventHeader::Gateway)
}
fn profile_name(&self) -> Option<&str> {
self.header(EventHeader::ProfileName)
}
fn phrase(&self) -> Option<&str> {
self.header(EventHeader::Phrase)
}
fn sip_status_code(&self) -> Result<Option<u16>, std::num::ParseIntError> {
parse_opt(self.header(EventHeader::Status))
}
fn gateway_reg_state(&self) -> Result<Option<GatewayRegState>, ParseGatewayRegStateError> {
parse_opt(self.header(EventHeader::State))
}
fn gateway_ping_status(
&self,
) -> Result<Option<GatewayPingStatus>, ParseGatewayPingStatusError> {
parse_opt(self.header(EventHeader::PingStatus))
}
fn sip_user_ping_status(
&self,
) -> Result<Option<SipUserPingStatus>, ParseSipUserPingStatusError> {
parse_opt(self.header(EventHeader::PingStatus))
}
fn pl_data(&self) -> Option<&str> {
self.header(EventHeader::PlData)
}
fn sip_event(&self) -> Option<&str> {
self.header(EventHeader::SipEvent)
}
fn gateway_name(&self) -> Option<&str> {
self.header(EventHeader::GatewayName)
}
fn channel_state(&self) -> Result<Option<ChannelState>, ParseChannelStateError> {
parse_opt(self.header(EventHeader::ChannelState))
}
fn channel_state_number(&self) -> Result<Option<ChannelState>, ParseChannelStateError> {
match self.header(EventHeader::ChannelStateNumber) {
Some(s) => {
let n: u8 = s
.parse()
.map_err(|_| ParseChannelStateError(s.to_string()))?;
ChannelState::from_number(n)
.ok_or_else(|| ParseChannelStateError(s.to_string()))
.map(Some)
}
None => Ok(None),
}
}
fn call_state(&self) -> Result<Option<CallState>, ParseCallStateError> {
parse_opt(self.header(EventHeader::ChannelCallState))
}
fn answer_state(&self) -> Result<Option<AnswerState>, ParseAnswerStateError> {
parse_opt(self.header(EventHeader::AnswerState))
}
fn call_direction(&self) -> Result<Option<CallDirection>, ParseCallDirectionError> {
parse_opt(self.header(EventHeader::CallDirection))
}
#[cfg(feature = "esl")]
fn priority(&self) -> Result<Option<EslEventPriority>, ParsePriorityError> {
parse_opt(self.header(EventHeader::Priority))
}
fn timetable(&self, prefix: &str) -> Result<Option<ChannelTimetable>, ParseTimetableError> {
ChannelTimetable::from_lookup(prefix, |key| self.header_str(key))
}
fn caller_timetable(&self) -> Result<Option<ChannelTimetable>, ParseTimetableError> {
self.timetable("Caller")
}
fn other_leg_timetable(&self) -> Result<Option<ChannelTimetable>, ParseTimetableError> {
self.timetable("Other-Leg")
}
}
impl HeaderLookup for std::collections::HashMap<String, String> {
fn header_str(&self, name: &str) -> Option<&str> {
self.get(name)
.map(|s| s.as_str())
}
fn variable_str(&self, name: &str) -> Option<&str> {
self.get(&format!("variable_{name}"))
.map(|s| s.as_str())
}
}
#[cfg(test)]
mod tests {
use super::*;
use crate::variables::ChannelVariable;
use std::collections::HashMap;
struct TestStore(HashMap<String, String>);
impl SipHeaderLookup for TestStore {
fn sip_header_str(&self, name: &str) -> Option<&str> {
self.0
.get(name)
.map(|s| s.as_str())
}
}
impl HeaderLookup for TestStore {
fn header_str(&self, name: &str) -> Option<&str> {
self.0
.get(name)
.map(|s| s.as_str())
}
fn variable_str(&self, name: &str) -> Option<&str> {
self.0
.get(&format!("variable_{}", name))
.map(|s| s.as_str())
}
}
fn store_with(pairs: &[(&str, &str)]) -> TestStore {
let map: HashMap<String, String> = pairs
.iter()
.map(|(k, v)| (k.to_string(), v.to_string()))
.collect();
TestStore(map)
}
#[test]
fn header_str_direct() {
let s = store_with(&[("Unique-ID", "abc-123")]);
assert_eq!(s.header_str("Unique-ID"), Some("abc-123"));
assert_eq!(s.header_str("Missing"), None);
}
#[test]
fn header_by_enum() {
let s = store_with(&[("Unique-ID", "abc-123")]);
assert_eq!(s.header(EventHeader::UniqueId), Some("abc-123"));
}
#[test]
fn variable_str_direct() {
let s = store_with(&[("variable_read_codec", "PCMU")]);
assert_eq!(s.variable_str("read_codec"), Some("PCMU"));
assert_eq!(s.variable_str("missing"), None);
}
#[test]
fn variable_by_enum() {
let s = store_with(&[("variable_read_codec", "PCMU")]);
assert_eq!(s.variable(ChannelVariable::ReadCodec), Some("PCMU"));
}
#[test]
fn unique_id_primary() {
let s = store_with(&[("Unique-ID", "uuid-1")]);
assert_eq!(s.unique_id(), Some("uuid-1"));
}
#[test]
fn unique_id_fallback() {
let s = store_with(&[("Caller-Unique-ID", "uuid-2")]);
assert_eq!(s.unique_id(), Some("uuid-2"));
}
#[test]
fn unique_id_none() {
let s = store_with(&[]);
assert_eq!(s.unique_id(), None);
}
#[test]
fn job_uuid() {
let s = store_with(&[("Job-UUID", "job-1")]);
assert_eq!(s.job_uuid(), Some("job-1"));
}
#[test]
fn channel_name() {
let s = store_with(&[("Channel-Name", "sofia/internal/1000@example.com")]);
assert_eq!(s.channel_name(), Some("sofia/internal/1000@example.com"));
}
#[test]
fn caller_id_number_and_name() {
let s = store_with(&[
("Caller-Caller-ID-Number", "1000"),
("Caller-Caller-ID-Name", "Alice"),
]);
assert_eq!(s.caller_id_number(), Some("1000"));
assert_eq!(s.caller_id_name(), Some("Alice"));
}
#[test]
fn hangup_cause_typed() {
let s = store_with(&[("Hangup-Cause", "NORMAL_CLEARING")]);
assert_eq!(
s.hangup_cause()
.unwrap(),
Some(crate::channel::HangupCause::NormalClearing)
);
}
#[test]
fn hangup_cause_invalid_is_error() {
let s = store_with(&[("Hangup-Cause", "BOGUS_CAUSE")]);
assert!(s
.hangup_cause()
.is_err());
}
#[test]
fn destination_number() {
let s = store_with(&[("Caller-Destination-Number", "1000")]);
assert_eq!(s.destination_number(), Some("1000"));
}
#[test]
fn callee_id() {
let s = store_with(&[
("Caller-Callee-ID-Number", "2000"),
("Caller-Callee-ID-Name", "Bob"),
]);
assert_eq!(s.callee_id_number(), Some("2000"));
assert_eq!(s.callee_id_name(), Some("Bob"));
}
#[test]
fn event_subclass() {
let s = store_with(&[("Event-Subclass", "sofia::register")]);
assert_eq!(s.event_subclass(), Some("sofia::register"));
}
#[test]
fn sofia_event_subclass_typed() {
let s = store_with(&[("Event-Subclass", "sofia::gateway_state")]);
assert_eq!(
s.sofia_event_subclass()
.unwrap(),
Some(crate::sofia::SofiaEventSubclass::GatewayState)
);
}
#[test]
fn sofia_event_subclass_absent() {
let s = store_with(&[]);
assert_eq!(
s.sofia_event_subclass()
.unwrap(),
None
);
}
#[test]
fn sofia_event_subclass_non_sofia_is_error() {
let s = store_with(&[("Event-Subclass", "conference::maintenance")]);
assert!(s
.sofia_event_subclass()
.is_err());
}
#[test]
fn gateway_reg_state_typed() {
let s = store_with(&[("State", "REGED")]);
assert_eq!(
s.gateway_reg_state()
.unwrap(),
Some(crate::sofia::GatewayRegState::Reged)
);
}
#[test]
fn gateway_reg_state_invalid_is_error() {
let s = store_with(&[("State", "BOGUS")]);
assert!(s
.gateway_reg_state()
.is_err());
}
#[test]
fn gateway_ping_status_typed() {
let s = store_with(&[("Ping-Status", "UP")]);
assert_eq!(
s.gateway_ping_status()
.unwrap(),
Some(crate::sofia::GatewayPingStatus::Up)
);
}
#[test]
fn sip_user_ping_status_typed() {
let s = store_with(&[("Ping-Status", "REACHABLE")]);
assert_eq!(
s.sip_user_ping_status()
.unwrap(),
Some(crate::sofia::SipUserPingStatus::Reachable)
);
}
#[test]
fn gateway_accessor() {
let s = store_with(&[("Gateway", "my-gateway")]);
assert_eq!(s.gateway(), Some("my-gateway"));
}
#[test]
fn profile_name_accessor() {
let s = store_with(&[("profile-name", "internal")]);
assert_eq!(s.profile_name(), Some("internal"));
}
#[test]
fn phrase_accessor() {
let s = store_with(&[("Phrase", "OK")]);
assert_eq!(s.phrase(), Some("OK"));
}
#[test]
fn channel_state_typed() {
let s = store_with(&[("Channel-State", "CS_EXECUTE")]);
assert_eq!(
s.channel_state()
.unwrap(),
Some(ChannelState::CsExecute)
);
}
#[test]
fn channel_state_number_typed() {
let s = store_with(&[("Channel-State-Number", "4")]);
assert_eq!(
s.channel_state_number()
.unwrap(),
Some(ChannelState::CsExecute)
);
}
#[test]
fn call_state_typed() {
let s = store_with(&[("Channel-Call-State", "ACTIVE")]);
assert_eq!(
s.call_state()
.unwrap(),
Some(CallState::Active)
);
}
#[test]
fn answer_state_typed() {
let s = store_with(&[("Answer-State", "answered")]);
assert_eq!(
s.answer_state()
.unwrap(),
Some(AnswerState::Answered)
);
}
#[test]
fn call_direction_typed() {
let s = store_with(&[("Call-Direction", "inbound")]);
assert_eq!(
s.call_direction()
.unwrap(),
Some(CallDirection::Inbound)
);
}
#[test]
fn priority_typed() {
let s = store_with(&[("priority", "HIGH")]);
assert_eq!(
s.priority()
.unwrap(),
Some(EslEventPriority::High)
);
}
#[test]
fn timetable_extraction() {
let s = store_with(&[
("Caller-Channel-Created-Time", "1700000001000000"),
("Caller-Channel-Answered-Time", "1700000005000000"),
]);
let tt = s
.caller_timetable()
.unwrap()
.expect("should have timetable");
assert_eq!(tt.created, Some(1700000001000000));
assert_eq!(tt.answered, Some(1700000005000000));
assert_eq!(tt.hungup, None);
}
#[test]
fn timetable_other_leg() {
let s = store_with(&[("Other-Leg-Channel-Created-Time", "1700000001000000")]);
let tt = s
.other_leg_timetable()
.unwrap()
.expect("should have timetable");
assert_eq!(tt.created, Some(1700000001000000));
}
#[test]
fn timetable_none_when_absent() {
let s = store_with(&[]);
assert_eq!(
s.caller_timetable()
.unwrap(),
None
);
}
#[test]
fn timetable_invalid_is_error() {
let s = store_with(&[("Caller-Channel-Created-Time", "not_a_number")]);
let err = s
.caller_timetable()
.unwrap_err();
assert_eq!(err.header, "Caller-Channel-Created-Time");
}
#[test]
fn missing_headers_return_none() {
let s = store_with(&[]);
assert_eq!(
s.channel_state()
.unwrap(),
None
);
assert_eq!(
s.channel_state_number()
.unwrap(),
None
);
assert_eq!(
s.call_state()
.unwrap(),
None
);
assert_eq!(
s.answer_state()
.unwrap(),
None
);
assert_eq!(
s.call_direction()
.unwrap(),
None
);
assert_eq!(
s.priority()
.unwrap(),
None
);
assert_eq!(
s.hangup_cause()
.unwrap(),
None
);
assert_eq!(s.channel_name(), None);
assert_eq!(s.caller_id_number(), None);
assert_eq!(s.caller_id_name(), None);
assert_eq!(s.destination_number(), None);
assert_eq!(s.callee_id_number(), None);
assert_eq!(s.callee_id_name(), None);
assert_eq!(s.event_subclass(), None);
assert_eq!(s.job_uuid(), None);
assert_eq!(s.pl_data(), None);
assert_eq!(s.sip_event(), None);
assert_eq!(s.gateway_name(), None);
assert_eq!(s.channel_presence_id(), None);
assert_eq!(
s.presence_call_direction()
.unwrap(),
None
);
assert_eq!(s.event_date_timestamp(), None);
assert_eq!(s.event_sequence(), None);
assert_eq!(s.dtmf_duration(), None);
assert_eq!(s.dtmf_source(), None);
}
#[test]
fn notify_in_headers() {
let s = store_with(&[
("pl_data", r#"{"invite":"INVITE ..."}"#),
("event", "emergency-AbandonedCall"),
("gateway_name", "ng911-bcf"),
]);
assert_eq!(s.pl_data(), Some(r#"{"invite":"INVITE ..."}"#));
assert_eq!(s.sip_event(), Some("emergency-AbandonedCall"));
assert_eq!(s.gateway_name(), Some("ng911-bcf"));
}
#[test]
fn channel_presence_id() {
let s = store_with(&[("Channel-Presence-ID", "1000@example.com")]);
assert_eq!(s.channel_presence_id(), Some("1000@example.com"));
}
#[test]
fn presence_call_direction_typed() {
let s = store_with(&[("Presence-Call-Direction", "outbound")]);
assert_eq!(
s.presence_call_direction()
.unwrap(),
Some(CallDirection::Outbound)
);
}
#[test]
fn event_date_timestamp() {
let s = store_with(&[("Event-Date-Timestamp", "1700000001000000")]);
assert_eq!(s.event_date_timestamp(), Some("1700000001000000"));
}
#[test]
fn event_sequence() {
let s = store_with(&[("Event-Sequence", "12345")]);
assert_eq!(s.event_sequence(), Some("12345"));
}
#[test]
fn dtmf_duration() {
let s = store_with(&[("DTMF-Duration", "2000")]);
assert_eq!(s.dtmf_duration(), Some("2000"));
}
#[test]
fn dtmf_source() {
let s = store_with(&[("DTMF-Source", "rtp")]);
assert_eq!(s.dtmf_source(), Some("rtp"));
}
#[test]
fn invalid_values_return_err() {
let s = store_with(&[
("Channel-State", "BOGUS"),
("Channel-State-Number", "999"),
("Channel-Call-State", "BOGUS"),
("Answer-State", "bogus"),
("Call-Direction", "bogus"),
("Presence-Call-Direction", "bogus"),
("priority", "BOGUS"),
("Hangup-Cause", "BOGUS"),
]);
assert!(s
.channel_state()
.is_err());
assert!(s
.channel_state_number()
.is_err());
assert!(s
.call_state()
.is_err());
assert!(s
.answer_state()
.is_err());
assert!(s
.call_direction()
.is_err());
assert!(s
.presence_call_direction()
.is_err());
assert!(s
.priority()
.is_err());
assert!(s
.hangup_cause()
.is_err());
}
}