use indexmap::IndexMap;
use std::fmt;
use std::str::FromStr;
use super::originate::OriginateError;
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
#[cfg_attr(feature = "serde", derive(serde::Serialize, serde::Deserialize))]
#[cfg_attr(feature = "serde", serde(rename_all = "lowercase"))]
#[non_exhaustive]
pub enum VariablesType {
Enterprise,
Default,
Channel,
}
impl VariablesType {
pub(super) fn delimiters(self) -> (char, char) {
match self {
Self::Enterprise => ('<', '>'),
Self::Default => ('{', '}'),
Self::Channel => ('[', ']'),
}
}
}
#[derive(Debug, Clone, PartialEq, Eq)]
pub struct Variables {
vars_type: VariablesType,
inner: IndexMap<String, String>,
separator: Option<char>,
}
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
#[cfg_attr(feature = "serde", derive(serde::Serialize, serde::Deserialize))]
#[non_exhaustive]
pub enum DialStringCarrier {
EslApi,
Dialplan,
}
const BACKSLASH: &str = r"\\\\\\\\";
const QUOTE_DIALPLAN: &str = r"\\'";
const QUOTE_ESL_API: &str = r"\\\'";
impl DialStringCarrier {
fn quote_escape(self) -> &'static str {
match self {
Self::EslApi => QUOTE_ESL_API,
Self::Dialplan => QUOTE_DIALPLAN,
}
}
}
fn check_representable(
key: &str,
value: &str,
vars_type: VariablesType,
) -> Result<(), OriginateError> {
if value.is_empty() {
return Err(OriginateError::ParseError(format!(
"variable {key} has an empty value: the switch discards such a pair \
without logging it, so it cannot be told from an absent variable on \
the wire. Give it a value or remove it -- and if its presence was \
itself the signal, that signal needs a home outside the dial string"
)));
}
let (open, close) = vars_type.delimiters();
let mut depth = 0i32;
for ch in value.chars() {
if ch == open {
depth += 1;
} else if ch == close {
depth -= 1;
if depth < 0 {
return Err(OriginateError::ParseError(format!(
"variable {key} closes a '{open}' it never opened, ending the block early"
)));
}
}
}
if depth != 0 {
return Err(OriginateError::ParseError(format!(
"variable {key} opens a '{open}' it never closes, swallowing the block's end"
)));
}
Ok(())
}
fn escape_value_keeping_commas(value: &str, carrier: DialStringCarrier) -> String {
let escaped = value
.replace('\\', BACKSLASH)
.replace('\'', carrier.quote_escape());
if escaped.contains(' ') {
format!("'{}'", escaped)
} else {
escaped
}
}
pub(super) fn escape_value(value: &str, carrier: DialStringCarrier) -> String {
let escaped = value
.replace('\\', BACKSLASH)
.replace('\'', carrier.quote_escape())
.replace(',', "\\,");
if escaped.contains(' ') {
format!("'{}'", escaped)
} else {
escaped
}
}
fn unescape_value(value: &str, carrier: DialStringCarrier) -> String {
let s = value
.strip_prefix('\'')
.and_then(|s| s.strip_suffix('\''))
.unwrap_or(value);
s.replace("\\,", ",")
.replace(carrier.quote_escape(), "'")
.replace(BACKSLASH, "\\")
}
impl Variables {
pub fn new(vars_type: VariablesType) -> Self {
Self {
vars_type,
inner: IndexMap::new(),
separator: None,
}
}
pub fn with_vars(
vars_type: VariablesType,
vars: impl IntoIterator<Item = (impl Into<String>, impl Into<String>)>,
) -> Self {
Self {
vars_type,
inner: vars
.into_iter()
.map(|(k, v)| (k.into(), v.into()))
.collect(),
separator: None,
}
}
pub fn with_separator(mut self, sep: char) -> Result<Self, OriginateError> {
let (_, close) = self
.vars_type
.delimiters();
if sep == close || sep == '=' || sep == '^' {
return Err(OriginateError::ParseError(format!(
"invalid ^^ separator: '{sep}'"
)));
}
if let Some((key, _)) = self
.inner
.iter()
.find(|(_, v)| v.contains(sep))
{
return Err(OriginateError::ParseError(format!(
"variable {key} contains the chosen '{sep}' separator"
)));
}
self.separator = Some(sep);
Ok(self)
}
pub fn separator(&self) -> Option<char> {
self.separator
}
pub fn insert(&mut self, key: impl Into<String>, value: impl Into<String>) {
self.inner
.insert(key.into(), value.into());
}
pub fn remove(&mut self, key: &str) -> Option<String> {
self.inner
.shift_remove(key)
}
pub fn get(&self, key: &str) -> Option<&str> {
self.inner
.get(key)
.map(|s| s.as_str())
}
pub fn is_empty(&self) -> bool {
self.inner
.is_empty()
}
pub fn len(&self) -> usize {
self.inner
.len()
}
pub fn scope(&self) -> VariablesType {
self.vars_type
}
pub fn set_scope(&mut self, scope: VariablesType) {
self.vars_type = scope;
}
pub fn iter(&self) -> impl Iterator<Item = (&str, &str)> {
self.inner
.iter()
.map(|(k, v)| (k.as_str(), v.as_str()))
}
pub fn iter_mut(&mut self) -> impl Iterator<Item = (&str, &mut String)> {
self.inner
.iter_mut()
.map(|(k, v)| (k.as_str(), v))
}
pub fn values_mut(&mut self) -> impl Iterator<Item = &mut String> {
self.inner
.values_mut()
}
}
#[cfg(feature = "serde")]
impl serde::Serialize for Variables {
fn serialize<S: serde::Serializer>(&self, serializer: S) -> Result<S::Ok, S::Error> {
if self.vars_type == VariablesType::Default {
self.inner
.serialize(serializer)
} else {
use serde::ser::SerializeStruct;
let mut s = serializer.serialize_struct("Variables", 2)?;
s.serialize_field("scope", &self.vars_type)?;
s.serialize_field("vars", &self.inner)?;
s.end()
}
}
}
#[cfg(feature = "serde")]
impl<'de> serde::Deserialize<'de> for Variables {
fn deserialize<D: serde::Deserializer<'de>>(deserializer: D) -> Result<Self, D::Error> {
#[derive(serde::Deserialize)]
#[serde(untagged)]
enum VariablesRepr {
Scoped {
scope: VariablesType,
vars: IndexMap<String, String>,
},
Flat(IndexMap<String, String>),
}
let (vars_type, inner) = match VariablesRepr::deserialize(deserializer)? {
VariablesRepr::Scoped { scope, vars } => (scope, vars),
VariablesRepr::Flat(map) => (VariablesType::Default, map),
};
for (key, value) in &inner {
check_representable(key, value, vars_type).map_err(serde::de::Error::custom)?;
}
Ok(Self {
vars_type,
inner,
separator: None,
})
}
}
#[derive(Debug, Clone, Copy)]
pub struct VariablesDisplay<'a> {
vars: &'a Variables,
carrier: DialStringCarrier,
}
impl fmt::Display for VariablesDisplay<'_> {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
self.vars
.write_for(f, self.carrier)
}
}
impl Variables {
pub fn display_for(&self, carrier: DialStringCarrier) -> VariablesDisplay<'_> {
VariablesDisplay {
vars: self,
carrier,
}
}
pub(super) fn write_for(
&self,
f: &mut fmt::Formatter<'_>,
carrier: DialStringCarrier,
) -> fmt::Result {
let (open, close) = self
.vars_type
.delimiters();
f.write_fmt(format_args!("{}", open))?;
if let Some(sep) = self.separator {
write!(f, "^^{sep}")?;
}
let sep = self
.separator
.unwrap_or(',');
for (i, (key, value)) in self
.inner
.iter()
.enumerate()
{
if i > 0 {
write!(f, "{sep}")?;
}
let value = match self.separator {
Some(_) => escape_value_keeping_commas(value, carrier),
None => escape_value(value, carrier),
};
write!(f, "{}={}", key, value)?;
}
f.write_fmt(format_args!("{}", close))
}
}
impl fmt::Display for Variables {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
self.write_for(f, DialStringCarrier::EslApi)
}
}
impl FromStr for Variables {
type Err = OriginateError;
fn from_str(s: &str) -> Result<Self, Self::Err> {
Self::parse_for(s, DialStringCarrier::EslApi)
}
}
impl Variables {
pub fn parse_for(s: &str, carrier: DialStringCarrier) -> Result<Self, OriginateError> {
let s = s.trim();
if s.len() < 2 {
return Err(OriginateError::ParseError(
"variable block too short".into(),
));
}
let (vars_type, inner_str) = match (s.as_bytes()[0], s.as_bytes()[s.len() - 1]) {
(b'{', b'}') => (VariablesType::Default, &s[1..s.len() - 1]),
(b'<', b'>') => (VariablesType::Enterprise, &s[1..s.len() - 1]),
(b'[', b']') => (VariablesType::Channel, &s[1..s.len() - 1]),
(open, close) => {
return Err(OriginateError::ParseError(format!(
"unknown variable delimiters: {:?}..{:?}",
open as char, close as char
)));
}
};
let mut inner = IndexMap::new();
if !inner_str.is_empty() {
if let Some(rest) = inner_str.strip_prefix("^^") {
let sep = rest
.chars()
.next()
.ok_or_else(|| {
OriginateError::ParseError("^^ without separator character".into())
})?;
let (_, close) = vars_type.delimiters();
if sep == close || sep == '=' {
return Err(OriginateError::ParseError(format!(
"invalid ^^ separator: '{sep}'"
)));
}
let var_str = &rest[sep.len_utf8()..];
if !var_str.is_empty() {
for (i, part) in var_str
.split(sep)
.enumerate()
{
let (key, value) = part
.split_once('=')
.ok_or_else(|| {
OriginateError::ParseError(format!("missing = in variable {i}"))
})?;
check_representable(key, value, vars_type)?;
inner.insert(key.to_string(), value.to_string());
}
}
} else {
for (i, part) in split_unescaped_commas(inner_str)
.into_iter()
.enumerate()
{
let (key, value) = part
.split_once('=')
.ok_or_else(|| {
OriginateError::ParseError(format!("missing = in variable {i}"))
})?;
let value = unescape_value(value, carrier);
check_representable(key, &value, vars_type)?;
inner.insert(key.to_string(), value);
}
}
}
Ok(Self {
vars_type,
inner,
separator: None,
})
}
}
pub(super) fn split_unescaped_commas(s: &str) -> Vec<&str> {
let mut parts = Vec::new();
let mut start = 0;
let bytes = s.as_bytes();
for i in 0..bytes.len() {
if bytes[i] == b',' {
let mut backslashes = 0;
let mut j = i;
while j > 0 && bytes[j - 1] == b'\\' {
backslashes += 1;
j -= 1;
}
if backslashes % 2 == 0 {
parts.push(&s[start..i]);
start = i + 1;
}
}
}
parts.push(&s[start..]);
parts
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn an_empty_value_is_refused_at_the_boundary() {
let err = "{k=,after=sentinel}"
.parse::<Variables>()
.unwrap_err()
.to_string();
assert!(err.contains('k'), "error does not name the variable: {err}");
assert!(
err.contains("remove it"),
"error does not name a remedy: {err}"
);
assert!(
err.contains("presence"),
"error prescribes a fix without allowing that presence meant something: {err}"
);
}
#[test]
fn an_unbalanced_bracket_is_refused_while_a_balanced_one_is_not() {
assert!("{k=oops}extra}"
.parse::<Variables>()
.is_err());
let balanced: Variables = "{k=${some_var}}"
.parse()
.expect("a balanced ${...} is ordinary and must parse");
assert_eq!(balanced.get("k"), Some("${some_var}"));
}
#[test]
fn chosen_separator_leaves_commas_alone() {
let mut vars = Variables::new(VariablesType::Default);
vars.insert("codecs", "PCMA,PCMU,G729");
let vars = vars
.with_separator(':')
.unwrap();
assert_eq!(vars.to_string(), "{^^:codecs=PCMA,PCMU,G729}");
assert_eq!(vars.separator(), Some(':'));
}
#[test]
fn separator_that_cannot_delimit_the_block_is_refused() {
let vars = Variables::new(VariablesType::Channel);
for sep in [']', '=', '^'] {
assert!(
vars.clone()
.with_separator(sep)
.is_err(),
"accepted {sep:?}"
);
}
}
#[test]
fn separator_already_present_in_a_value_is_refused() {
let mut vars = Variables::new(VariablesType::Default);
vars.insert("uri", "sip:bob@example.com");
let err = vars
.with_separator(':')
.unwrap_err()
.to_string();
assert!(
err.contains("uri"),
"error does not name the variable: {err}"
);
}
#[test]
fn escaping_pins_the_measured_wire_forms() {
let cases = [
(DialStringCarrier::Dialplan, "it's", r"it\\'s"),
(DialStringCarrier::EslApi, "it's", r"it\\\'s"),
(DialStringCarrier::Dialplan, r"a\nb", r"a\\\\\\\\nb"),
(DialStringCarrier::EslApi, r"a\nb", r"a\\\\\\\\nb"),
(DialStringCarrier::Dialplan, "a,b", r"a\,b"),
(DialStringCarrier::EslApi, "a,b", r"a\,b"),
];
for (carrier, value, want) in cases {
assert_eq!(
escape_value(value, carrier),
want,
"{value:?} for {carrier:?}"
);
}
}
#[test]
fn display_defaults_to_the_api_carrier() {
let mut vars = Variables::new(VariablesType::Default);
vars.insert("k", "it's");
assert_eq!(vars.to_string(), r"{k=it\\\'s}");
assert_eq!(
vars.display_for(DialStringCarrier::Dialplan)
.to_string(),
r"{k=it\\'s}"
);
}
#[test]
fn round_trips_at_either_carrier() {
for carrier in [DialStringCarrier::EslApi, DialStringCarrier::Dialplan] {
for value in ["it's", r"a\,b", r"C:\path", "a,b", "with space"] {
let mut vars = Variables::new(VariablesType::Default);
vars.insert("k", value);
vars.insert("after", "sentinel");
let rendered = vars
.display_for(carrier)
.to_string();
let back = Variables::parse_for(&rendered, carrier).unwrap_or_else(|e| {
panic!("{value:?} for {carrier:?} rendered {rendered}: {e}")
});
assert_eq!(back.get("k"), Some(value), "rendered {rendered}");
assert_eq!(back.get("after"), Some("sentinel"), "rendered {rendered}");
}
}
}
#[test]
fn value_with_backslash_round_trips() {
for value in [r"a\,b", r"C:\path", r"trailing\", r"\\", r"a\nb"] {
let mut vars = Variables::new(VariablesType::Default);
vars.insert("k", value);
vars.insert("after", "sentinel");
let rendered = vars.to_string();
let back: Variables = rendered
.parse()
.unwrap_or_else(|e| {
panic!("{value:?} rendered {rendered} and failed to parse: {e}")
});
assert_eq!(back.get("k"), Some(value), "rendered {rendered}");
assert_eq!(
back.get("after"),
Some("sentinel"),
"value {value:?} ate the next variable: {rendered}"
);
}
}
#[test]
fn missing_equals_error_omits_the_fragment() {
let msg = "{origination_caller_id_number=15551234567,15550009999}"
.parse::<Variables>()
.unwrap_err()
.to_string();
assert!(
!msg.contains("15550009999"),
"error quoted its input: {msg}"
);
assert!(
msg.contains("variable 1"),
"error does not name the part: {msg}"
);
}
#[test]
fn unknown_delimiters_error_omits_the_block() {
let msg = "(origination_caller_id_number=15551234567)"
.parse::<Variables>()
.unwrap_err()
.to_string();
assert!(
!msg.contains("15551234567"),
"error quoted its input: {msg}"
);
}
#[test]
fn variables_standard_chars() {
let mut vars = Variables::new(VariablesType::Default);
vars.insert("test_key", "this_value");
let result = vars.to_string();
assert!(result.contains("test_key"));
assert!(result.contains("this_value"));
}
#[test]
fn variables_comma_escaped() {
let mut vars = Variables::new(VariablesType::Default);
vars.insert("test_key", "this,is,a,value");
let result = vars.to_string();
assert!(result.contains("\\,"));
}
#[test]
fn variables_spaces_quoted() {
let mut vars = Variables::new(VariablesType::Default);
vars.insert("test_key", "this is a value");
let result = vars.to_string();
assert_eq!(
result
.matches('\'')
.count(),
2
);
}
#[test]
fn variables_single_quote_escaped() {
let mut vars = Variables::new(VariablesType::Default);
vars.insert("test_key", "let's_this_be_a_value");
let result = vars.to_string();
assert!(result.contains("\\'"));
}
#[test]
fn variables_enterprise_delimiters() {
let mut vars = Variables::new(VariablesType::Enterprise);
vars.insert("k", "v");
let result = vars.to_string();
assert!(result.starts_with('<'));
assert!(result.ends_with('>'));
}
#[test]
fn variables_channel_delimiters() {
let mut vars = Variables::new(VariablesType::Channel);
vars.insert("k", "v");
let result = vars.to_string();
assert!(result.starts_with('['));
assert!(result.ends_with(']'));
}
#[test]
fn variables_default_delimiters() {
let mut vars = Variables::new(VariablesType::Default);
vars.insert("k", "v");
let result = vars.to_string();
assert!(result.starts_with('{'));
assert!(result.ends_with('}'));
}
#[test]
fn variables_parse_round_trip() {
let mut vars = Variables::new(VariablesType::Default);
vars.insert("origination_caller_id_number", "9005551212");
vars.insert("sip_h_Call-Info", "<url>;meta=123,<uri>");
let s = vars.to_string();
let parsed: Variables = s
.parse()
.unwrap();
assert_eq!(
parsed.get("origination_caller_id_number"),
Some("9005551212")
);
assert_eq!(parsed.get("sip_h_Call-Info"), Some("<url>;meta=123,<uri>"));
}
#[test]
fn split_unescaped_commas_basic() {
assert_eq!(split_unescaped_commas("a,b,c"), vec!["a", "b", "c"]);
}
#[test]
fn split_unescaped_commas_escaped() {
assert_eq!(split_unescaped_commas(r"a\,b,c"), vec![r"a\,b", "c"]);
}
#[test]
fn split_unescaped_commas_double_backslash() {
assert_eq!(split_unescaped_commas(r"a\\,b"), vec![r"a\\", "b"]);
}
#[test]
fn split_unescaped_commas_triple_backslash() {
assert_eq!(split_unescaped_commas(r"a\\\,b"), vec![r"a\\\,b"]);
}
#[test]
fn variables_caret_caret_separator() {
let vars: Variables =
"[^^:sip_invite_domain=pbx.example.com:presence_id=1211@pbx.example.com]"
.parse()
.unwrap();
assert_eq!(vars.scope(), VariablesType::Channel);
assert_eq!(vars.get("sip_invite_domain"), Some("pbx.example.com"));
assert_eq!(vars.get("presence_id"), Some("1211@pbx.example.com"));
}
#[test]
fn variables_caret_caret_display_uses_canonical_comma() {
let vars: Variables = "[^^:a=1:b=2]"
.parse()
.unwrap();
assert_eq!(vars.to_string(), "[a=1,b=2]");
}
#[test]
fn variables_caret_caret_default_scope() {
let vars: Variables = "{^^|x=1|y=2}"
.parse()
.unwrap();
assert_eq!(vars.scope(), VariablesType::Default);
assert_eq!(vars.get("x"), Some("1"));
assert_eq!(vars.get("y"), Some("2"));
}
#[test]
fn variables_caret_caret_enterprise_scope() {
let vars: Variables = "<^^;a=1;b=2>"
.parse()
.unwrap();
assert_eq!(vars.scope(), VariablesType::Enterprise);
assert_eq!(vars.get("a"), Some("1"));
}
#[test]
fn variables_caret_caret_no_unescape() {
let vars: Variables = r"[^^:key=val\,ue:other=x]"
.parse()
.unwrap();
assert_eq!(vars.get("key"), Some(r"val\,ue"));
}
#[test]
fn variables_caret_caret_values_with_commas() {
let vars: Variables = "[^^|sip_h_X-Call-Info=<urn:foo>;purpose=bar,<urn:baz>|other=val]"
.parse()
.unwrap();
assert_eq!(
vars.get("sip_h_X-Call-Info"),
Some("<urn:foo>;purpose=bar,<urn:baz>")
);
assert_eq!(vars.get("other"), Some("val"));
}
#[test]
fn variables_caret_caret_empty_vars() {
let vars: Variables = "[^^:]"
.parse()
.unwrap();
assert!(vars.is_empty());
assert_eq!(vars.scope(), VariablesType::Channel);
}
#[test]
fn variables_caret_caret_missing_separator() {
assert!("[^^]"
.parse::<Variables>()
.is_err());
}
#[test]
fn variables_caret_caret_closing_bracket_as_sep() {
assert!("[^^]]"
.parse::<Variables>()
.is_err());
}
#[test]
fn variables_caret_caret_equals_as_sep() {
assert!("[^^=a=1]"
.parse::<Variables>()
.is_err());
}
#[test]
fn variables_from_str_empty_block() {
let result = "{}".parse::<Variables>();
assert!(
result.is_ok(),
"empty variable block should parse successfully"
);
let vars = result.unwrap();
assert!(
vars.is_empty(),
"parsed empty block should have no variables"
);
}
#[test]
fn variables_from_str_empty_channel_block() {
let result = "[]".parse::<Variables>();
assert!(result.is_ok());
let vars = result.unwrap();
assert!(vars.is_empty());
assert_eq!(vars.scope(), VariablesType::Channel);
}
#[test]
fn variables_from_str_empty_enterprise_block() {
let result = "<>".parse::<Variables>();
assert!(result.is_ok());
let vars = result.unwrap();
assert!(vars.is_empty());
assert_eq!(vars.scope(), VariablesType::Enterprise);
}
#[test]
fn serde_variables_type() {
let json = serde_json::to_string(&VariablesType::Enterprise).unwrap();
assert_eq!(json, "\"enterprise\"");
let parsed: VariablesType = serde_json::from_str(&json).unwrap();
assert_eq!(parsed, VariablesType::Enterprise);
}
#[test]
fn serde_variables_flat_default() {
let mut vars = Variables::new(VariablesType::Default);
vars.insert("key1", "val1");
vars.insert("key2", "val2");
let json = serde_json::to_string(&vars).unwrap();
let parsed: Variables = serde_json::from_str(&json).unwrap();
assert_eq!(parsed.scope(), VariablesType::Default);
assert_eq!(parsed.get("key1"), Some("val1"));
assert_eq!(parsed.get("key2"), Some("val2"));
}
#[test]
fn serde_variables_scoped_enterprise() {
let mut vars = Variables::new(VariablesType::Enterprise);
vars.insert("key1", "val1");
let json = serde_json::to_string(&vars).unwrap();
assert!(json.contains("\"enterprise\""));
let parsed: Variables = serde_json::from_str(&json).unwrap();
assert_eq!(parsed.scope(), VariablesType::Enterprise);
assert_eq!(parsed.get("key1"), Some("val1"));
}
#[test]
fn serde_variables_flat_map_deserializes_as_default() {
let json = r#"{"key1":"val1","key2":"val2"}"#;
let vars: Variables = serde_json::from_str(json).unwrap();
assert_eq!(vars.scope(), VariablesType::Default);
assert_eq!(vars.get("key1"), Some("val1"));
assert_eq!(vars.get("key2"), Some("val2"));
}
#[test]
fn serde_variables_scoped_deserializes() {
let json = r#"{"scope":"channel","vars":{"k":"v"}}"#;
let vars: Variables = serde_json::from_str(json).unwrap();
assert_eq!(vars.scope(), VariablesType::Channel);
assert_eq!(vars.get("k"), Some("v"));
}
}