jingle: Update to Into and TryFrom.

This commit is contained in:
Emmanuel Gil Peyrot 2017-05-04 01:20:28 +01:00
commit 6c4fd8da01
2 changed files with 208 additions and 194 deletions

View file

@ -5,6 +5,8 @@
// License, v. 2.0. If a copy of the MPL was not distributed with this // License, v. 2.0. If a copy of the MPL was not distributed with this
// file, You can obtain one at http://mozilla.org/MPL/2.0/. // file, You can obtain one at http://mozilla.org/MPL/2.0/.
use std::convert::TryFrom;
use minidom::Element; use minidom::Element;
use minidom::IntoAttributeValue; use minidom::IntoAttributeValue;
@ -17,7 +19,7 @@ use ns;
use stanza_error; use stanza_error;
use disco; use disco;
use ibb; use ibb;
use jingle; use jingle::Jingle;
use ping; use ping;
/// Lists every known payload of a `<iq/>`. /// Lists every known payload of a `<iq/>`.
@ -25,7 +27,7 @@ use ping;
pub enum IqPayload { pub enum IqPayload {
Disco(disco::Disco), Disco(disco::Disco),
IBB(ibb::IBB), IBB(ibb::IBB),
Jingle(jingle::Jingle), Jingle(Jingle),
Ping(ping::Ping), Ping(ping::Ping),
} }
@ -97,7 +99,7 @@ pub fn parse_iq(root: &Element) -> Result<Iq, Error> {
Some(IqPayload::Disco(disco)) Some(IqPayload::Disco(disco))
} else if let Ok(ibb) = ibb::parse_ibb(elem) { } else if let Ok(ibb) = ibb::parse_ibb(elem) {
Some(IqPayload::IBB(ibb)) Some(IqPayload::IBB(ibb))
} else if let Ok(jingle) = jingle::parse_jingle(elem) { } else if let Ok(jingle) = Jingle::try_from(elem) {
Some(IqPayload::Jingle(jingle)) Some(IqPayload::Jingle(jingle))
} else if let Ok(ping) = ping::parse_ping(elem) { } else if let Ok(ping) = ping::parse_ping(elem) {
Some(IqPayload::Ping(ping)) Some(IqPayload::Ping(ping))
@ -152,7 +154,7 @@ pub fn serialise_payload(payload: &IqPayload) -> Element {
match *payload { match *payload {
IqPayload::Disco(ref disco) => disco::serialise_disco(disco), IqPayload::Disco(ref disco) => disco::serialise_disco(disco),
IqPayload::IBB(ref ibb) => ibb::serialise(ibb), IqPayload::IBB(ref ibb) => ibb::serialise(ibb),
IqPayload::Jingle(ref jingle) => jingle::serialise(jingle), IqPayload::Jingle(ref jingle) => jingle.into(),
IqPayload::Ping(_) => ping::serialise_ping(), IqPayload::Ping(_) => ping::serialise_ping(),
} }
} }

View file

@ -4,10 +4,10 @@
// License, v. 2.0. If a copy of the MPL was not distributed with this // License, v. 2.0. If a copy of the MPL was not distributed with this
// file, You can obtain one at http://mozilla.org/MPL/2.0/. // file, You can obtain one at http://mozilla.org/MPL/2.0/.
use std::convert::TryFrom;
use std::str::FromStr; use std::str::FromStr;
use minidom::{Element, IntoElements}; use minidom::Element;
use minidom::convert::ElementEmitter;
use error::Error; use error::Error;
use ns; use ns;
@ -204,9 +204,9 @@ impl FromStr for Reason {
} }
} }
impl IntoElements for Reason { impl<'a> Into<Element> for &'a Reason {
fn into_elements(self, emitter: &mut ElementEmitter) { fn into(self) -> Element {
let elem = Element::builder(match self { Element::builder(match *self {
Reason::AlternativeSession => "alternative-session", Reason::AlternativeSession => "alternative-session",
Reason::Busy => "busy", Reason::Busy => "busy",
Reason::Cancel => "cancel", Reason::Cancel => "cancel",
@ -224,8 +224,7 @@ impl IntoElements for Reason {
Reason::Timeout => "timeout", Reason::Timeout => "timeout",
Reason::UnsupportedApplications => "unsupported-applications", Reason::UnsupportedApplications => "unsupported-applications",
Reason::UnsupportedTransports => "unsupported-transports", Reason::UnsupportedTransports => "unsupported-transports",
}).build(); }).build()
emitter.append_child(elem);
} }
} }
@ -246,7 +245,10 @@ pub struct Jingle {
pub other: Vec<Element>, pub other: Vec<Element>,
} }
pub fn parse_jingle(root: &Element) -> Result<Jingle, Error> { impl<'a> TryFrom<&'a Element> for Jingle {
type Error = Error;
fn try_from(root: &'a Element) -> Result<Jingle, Error> {
if !root.is("jingle", ns::JINGLE) { if !root.is("jingle", ns::JINGLE) {
return Err(Error::ParseError("This is not a Jingle element.")); return Err(Error::ParseError("This is not a Jingle element."));
} }
@ -378,63 +380,73 @@ pub fn parse_jingle(root: &Element) -> Result<Jingle, Error> {
other: other, other: other,
}) })
} }
}
pub fn serialise_content(content: &Content) -> Element { impl<'a> Into<Element> for &'a Content {
fn into(self) -> Element {
let mut root = Element::builder("content") let mut root = Element::builder("content")
.ns(ns::JINGLE) .ns(ns::JINGLE)
.attr("creator", String::from(content.creator.clone())) .attr("creator", String::from(self.creator.clone()))
.attr("disposition", content.disposition.clone()) .attr("disposition", self.disposition.clone())
.attr("name", content.name.clone()) .attr("name", self.name.clone())
.attr("senders", String::from(content.senders.clone())) .attr("senders", String::from(self.senders.clone()))
.build(); .build();
root.append_child(content.description.1.clone()); root.append_child(self.description.1.clone());
root.append_child(content.transport.1.clone()); root.append_child(self.transport.1.clone());
if let Some(security) = content.security.clone() { if let Some(security) = self.security.clone() {
root.append_child(security.1.clone()); root.append_child(security.1.clone());
} }
root root
} }
}
pub fn serialise(jingle: &Jingle) -> Element { impl<'a> Into<Element> for &'a Jingle {
fn into(self) -> Element {
let mut root = Element::builder("jingle") let mut root = Element::builder("jingle")
.ns(ns::JINGLE) .ns(ns::JINGLE)
.attr("action", String::from(jingle.action.clone())) .attr("action", String::from(self.action.clone()))
.attr("initiator", jingle.initiator.clone()) .attr("initiator", self.initiator.clone())
.attr("responder", jingle.responder.clone()) .attr("responder", self.responder.clone())
.attr("sid", jingle.sid.clone()) .attr("sid", self.sid.clone())
.build(); .build();
for content in jingle.contents.clone() { for content in self.contents.clone() {
let content_elem = serialise_content(&content); let content_elem = (&content).into();
root.append_child(content_elem); root.append_child(content_elem);
} }
if let Some(ref reason) = jingle.reason { if let Some(ref reason) = self.reason {
let reason2: Element = (&reason.reason).into();
let reason_elem = Element::builder("reason") let reason_elem = Element::builder("reason")
.append(reason.reason.clone()) .append(reason2)
.append(reason.text.clone()) .append(reason.text.clone())
.build(); .build();
root.append_child(reason_elem); root.append_child(reason_elem);
} }
root root
} }
}
impl Into<Element> for Jingle {
fn into(self) -> Element {
(&self).into()
}
}
#[cfg(test)] #[cfg(test)]
mod tests { mod tests {
use minidom::Element; use super::*;
use error::Error;
use jingle;
#[test] #[test]
fn test_simple() { fn test_simple() {
let elem: Element = "<jingle xmlns='urn:xmpp:jingle:1' action='session-initiate' sid='coucou'/>".parse().unwrap(); let elem: Element = "<jingle xmlns='urn:xmpp:jingle:1' action='session-initiate' sid='coucou'/>".parse().unwrap();
let jingle = jingle::parse_jingle(&elem).unwrap(); let jingle = Jingle::try_from(&elem).unwrap();
assert_eq!(jingle.action, jingle::Action::SessionInitiate); assert_eq!(jingle.action, Action::SessionInitiate);
assert_eq!(jingle.sid, "coucou"); assert_eq!(jingle.sid, "coucou");
} }
#[test] #[test]
fn test_invalid_jingle() { fn test_invalid_jingle() {
let elem: Element = "<jingle xmlns='urn:xmpp:jingle:1'/>".parse().unwrap(); let elem: Element = "<jingle xmlns='urn:xmpp:jingle:1'/>".parse().unwrap();
let error = jingle::parse_jingle(&elem).unwrap_err(); let error = Jingle::try_from(&elem).unwrap_err();
let message = match error { let message = match error {
Error::ParseError(string) => string, Error::ParseError(string) => string,
_ => panic!(), _ => panic!(),
@ -442,7 +454,7 @@ mod tests {
assert_eq!(message, "Jingle must have an 'action' attribute."); assert_eq!(message, "Jingle must have an 'action' attribute.");
let elem: Element = "<jingle xmlns='urn:xmpp:jingle:1' action='session-info'/>".parse().unwrap(); let elem: Element = "<jingle xmlns='urn:xmpp:jingle:1' action='session-info'/>".parse().unwrap();
let error = jingle::parse_jingle(&elem).unwrap_err(); let error = Jingle::try_from(&elem).unwrap_err();
let message = match error { let message = match error {
Error::ParseError(string) => string, Error::ParseError(string) => string,
_ => panic!(), _ => panic!(),
@ -450,7 +462,7 @@ mod tests {
assert_eq!(message, "Jingle must have a 'sid' attribute."); assert_eq!(message, "Jingle must have a 'sid' attribute.");
let elem: Element = "<jingle xmlns='urn:xmpp:jingle:1' action='coucou' sid='coucou'/>".parse().unwrap(); let elem: Element = "<jingle xmlns='urn:xmpp:jingle:1' action='coucou' sid='coucou'/>".parse().unwrap();
let error = jingle::parse_jingle(&elem).unwrap_err(); let error = Jingle::try_from(&elem).unwrap_err();
let message = match error { let message = match error {
Error::ParseError(string) => string, Error::ParseError(string) => string,
_ => panic!(), _ => panic!(),
@ -461,25 +473,25 @@ mod tests {
#[test] #[test]
fn test_content() { fn test_content() {
let elem: Element = "<jingle xmlns='urn:xmpp:jingle:1' action='session-initiate' sid='coucou'><content creator='initiator' name='coucou'><description/><transport/></content></jingle>".parse().unwrap(); let elem: Element = "<jingle xmlns='urn:xmpp:jingle:1' action='session-initiate' sid='coucou'><content creator='initiator' name='coucou'><description/><transport/></content></jingle>".parse().unwrap();
let jingle = jingle::parse_jingle(&elem).unwrap(); let jingle = Jingle::try_from(&elem).unwrap();
assert_eq!(jingle.contents[0].creator, jingle::Creator::Initiator); assert_eq!(jingle.contents[0].creator, Creator::Initiator);
assert_eq!(jingle.contents[0].name, "coucou"); assert_eq!(jingle.contents[0].name, "coucou");
assert_eq!(jingle.contents[0].senders, jingle::Senders::Both); assert_eq!(jingle.contents[0].senders, Senders::Both);
assert_eq!(jingle.contents[0].disposition, "session"); assert_eq!(jingle.contents[0].disposition, "session");
let elem: Element = "<jingle xmlns='urn:xmpp:jingle:1' action='session-initiate' sid='coucou'><content creator='initiator' name='coucou' senders='both'><description/><transport/></content></jingle>".parse().unwrap(); let elem: Element = "<jingle xmlns='urn:xmpp:jingle:1' action='session-initiate' sid='coucou'><content creator='initiator' name='coucou' senders='both'><description/><transport/></content></jingle>".parse().unwrap();
let jingle = jingle::parse_jingle(&elem).unwrap(); let jingle = Jingle::try_from(&elem).unwrap();
assert_eq!(jingle.contents[0].senders, jingle::Senders::Both); assert_eq!(jingle.contents[0].senders, Senders::Both);
let elem: Element = "<jingle xmlns='urn:xmpp:jingle:1' action='session-initiate' sid='coucou'><content creator='initiator' name='coucou' disposition='early-session'><description/><transport/></content></jingle>".parse().unwrap(); let elem: Element = "<jingle xmlns='urn:xmpp:jingle:1' action='session-initiate' sid='coucou'><content creator='initiator' name='coucou' disposition='early-session'><description/><transport/></content></jingle>".parse().unwrap();
let jingle = jingle::parse_jingle(&elem).unwrap(); let jingle = Jingle::try_from(&elem).unwrap();
assert_eq!(jingle.contents[0].disposition, "early-session"); assert_eq!(jingle.contents[0].disposition, "early-session");
} }
#[test] #[test]
fn test_invalid_content() { fn test_invalid_content() {
let elem: Element = "<jingle xmlns='urn:xmpp:jingle:1' action='session-initiate' sid='coucou'><content/></jingle>".parse().unwrap(); let elem: Element = "<jingle xmlns='urn:xmpp:jingle:1' action='session-initiate' sid='coucou'><content/></jingle>".parse().unwrap();
let error = jingle::parse_jingle(&elem).unwrap_err(); let error = Jingle::try_from(&elem).unwrap_err();
let message = match error { let message = match error {
Error::ParseError(string) => string, Error::ParseError(string) => string,
_ => panic!(), _ => panic!(),
@ -487,7 +499,7 @@ mod tests {
assert_eq!(message, "Content must have a 'creator' attribute."); assert_eq!(message, "Content must have a 'creator' attribute.");
let elem: Element = "<jingle xmlns='urn:xmpp:jingle:1' action='session-initiate' sid='coucou'><content creator='initiator'/></jingle>".parse().unwrap(); let elem: Element = "<jingle xmlns='urn:xmpp:jingle:1' action='session-initiate' sid='coucou'><content creator='initiator'/></jingle>".parse().unwrap();
let error = jingle::parse_jingle(&elem).unwrap_err(); let error = Jingle::try_from(&elem).unwrap_err();
let message = match error { let message = match error {
Error::ParseError(string) => string, Error::ParseError(string) => string,
_ => panic!(), _ => panic!(),
@ -495,7 +507,7 @@ mod tests {
assert_eq!(message, "Content must have a 'name' attribute."); assert_eq!(message, "Content must have a 'name' attribute.");
let elem: Element = "<jingle xmlns='urn:xmpp:jingle:1' action='session-initiate' sid='coucou'><content creator='coucou' name='coucou'/></jingle>".parse().unwrap(); let elem: Element = "<jingle xmlns='urn:xmpp:jingle:1' action='session-initiate' sid='coucou'><content creator='coucou' name='coucou'/></jingle>".parse().unwrap();
let error = jingle::parse_jingle(&elem).unwrap_err(); let error = Jingle::try_from(&elem).unwrap_err();
let message = match error { let message = match error {
Error::ParseError(string) => string, Error::ParseError(string) => string,
_ => panic!(), _ => panic!(),
@ -503,7 +515,7 @@ mod tests {
assert_eq!(message, "Unknown creator."); assert_eq!(message, "Unknown creator.");
let elem: Element = "<jingle xmlns='urn:xmpp:jingle:1' action='session-initiate' sid='coucou'><content creator='initiator' name='coucou' senders='coucou'/></jingle>".parse().unwrap(); let elem: Element = "<jingle xmlns='urn:xmpp:jingle:1' action='session-initiate' sid='coucou'><content creator='initiator' name='coucou' senders='coucou'/></jingle>".parse().unwrap();
let error = jingle::parse_jingle(&elem).unwrap_err(); let error = Jingle::try_from(&elem).unwrap_err();
let message = match error { let message = match error {
Error::ParseError(string) => string, Error::ParseError(string) => string,
_ => panic!(), _ => panic!(),
@ -511,7 +523,7 @@ mod tests {
assert_eq!(message, "Unknown senders."); assert_eq!(message, "Unknown senders.");
let elem: Element = "<jingle xmlns='urn:xmpp:jingle:1' action='session-initiate' sid='coucou'><content creator='initiator' name='coucou' senders=''/></jingle>".parse().unwrap(); let elem: Element = "<jingle xmlns='urn:xmpp:jingle:1' action='session-initiate' sid='coucou'><content creator='initiator' name='coucou' senders=''/></jingle>".parse().unwrap();
let error = jingle::parse_jingle(&elem).unwrap_err(); let error = Jingle::try_from(&elem).unwrap_err();
let message = match error { let message = match error {
Error::ParseError(string) => string, Error::ParseError(string) => string,
_ => panic!(), _ => panic!(),
@ -519,7 +531,7 @@ mod tests {
assert_eq!(message, "Unknown senders."); assert_eq!(message, "Unknown senders.");
let elem: Element = "<jingle xmlns='urn:xmpp:jingle:1' action='session-initiate' sid='coucou'><content creator='initiator' name='coucou'/></jingle>".parse().unwrap(); let elem: Element = "<jingle xmlns='urn:xmpp:jingle:1' action='session-initiate' sid='coucou'><content creator='initiator' name='coucou'/></jingle>".parse().unwrap();
let error = jingle::parse_jingle(&elem).unwrap_err(); let error = Jingle::try_from(&elem).unwrap_err();
let message = match error { let message = match error {
Error::ParseError(string) => string, Error::ParseError(string) => string,
_ => panic!(), _ => panic!(),
@ -527,7 +539,7 @@ mod tests {
assert_eq!(message, "Content must have one description."); assert_eq!(message, "Content must have one description.");
let elem: Element = "<jingle xmlns='urn:xmpp:jingle:1' action='session-initiate' sid='coucou'><content creator='initiator' name='coucou'><description/></content></jingle>".parse().unwrap(); let elem: Element = "<jingle xmlns='urn:xmpp:jingle:1' action='session-initiate' sid='coucou'><content creator='initiator' name='coucou'><description/></content></jingle>".parse().unwrap();
let error = jingle::parse_jingle(&elem).unwrap_err(); let error = Jingle::try_from(&elem).unwrap_err();
let message = match error { let message = match error {
Error::ParseError(string) => string, Error::ParseError(string) => string,
_ => panic!(), _ => panic!(),
@ -538,22 +550,22 @@ mod tests {
#[test] #[test]
fn test_reason() { fn test_reason() {
let elem: Element = "<jingle xmlns='urn:xmpp:jingle:1' action='session-initiate' sid='coucou'><reason><success/></reason></jingle>".parse().unwrap(); let elem: Element = "<jingle xmlns='urn:xmpp:jingle:1' action='session-initiate' sid='coucou'><reason><success/></reason></jingle>".parse().unwrap();
let jingle = jingle::parse_jingle(&elem).unwrap(); let jingle = Jingle::try_from(&elem).unwrap();
let reason = jingle.reason.unwrap(); let reason = jingle.reason.unwrap();
assert_eq!(reason.reason, jingle::Reason::Success); assert_eq!(reason.reason, Reason::Success);
assert_eq!(reason.text, None); assert_eq!(reason.text, None);
let elem: Element = "<jingle xmlns='urn:xmpp:jingle:1' action='session-initiate' sid='coucou'><reason><success/><text>coucou</text></reason></jingle>".parse().unwrap(); let elem: Element = "<jingle xmlns='urn:xmpp:jingle:1' action='session-initiate' sid='coucou'><reason><success/><text>coucou</text></reason></jingle>".parse().unwrap();
let jingle = jingle::parse_jingle(&elem).unwrap(); let jingle = Jingle::try_from(&elem).unwrap();
let reason = jingle.reason.unwrap(); let reason = jingle.reason.unwrap();
assert_eq!(reason.reason, jingle::Reason::Success); assert_eq!(reason.reason, Reason::Success);
assert_eq!(reason.text, Some(String::from("coucou"))); assert_eq!(reason.text, Some(String::from("coucou")));
} }
#[test] #[test]
fn test_invalid_reason() { fn test_invalid_reason() {
let elem: Element = "<jingle xmlns='urn:xmpp:jingle:1' action='session-initiate' sid='coucou'><reason/></jingle>".parse().unwrap(); let elem: Element = "<jingle xmlns='urn:xmpp:jingle:1' action='session-initiate' sid='coucou'><reason/></jingle>".parse().unwrap();
let error = jingle::parse_jingle(&elem).unwrap_err(); let error = Jingle::try_from(&elem).unwrap_err();
let message = match error { let message = match error {
Error::ParseError(string) => string, Error::ParseError(string) => string,
_ => panic!(), _ => panic!(),
@ -561,7 +573,7 @@ mod tests {
assert_eq!(message, "Reason doesnt contain a valid reason."); assert_eq!(message, "Reason doesnt contain a valid reason.");
let elem: Element = "<jingle xmlns='urn:xmpp:jingle:1' action='session-initiate' sid='coucou'><reason><a/></reason></jingle>".parse().unwrap(); let elem: Element = "<jingle xmlns='urn:xmpp:jingle:1' action='session-initiate' sid='coucou'><reason><a/></reason></jingle>".parse().unwrap();
let error = jingle::parse_jingle(&elem).unwrap_err(); let error = Jingle::try_from(&elem).unwrap_err();
let message = match error { let message = match error {
Error::ParseError(string) => string, Error::ParseError(string) => string,
_ => panic!(), _ => panic!(),
@ -569,7 +581,7 @@ mod tests {
assert_eq!(message, "Unknown reason."); assert_eq!(message, "Unknown reason.");
let elem: Element = "<jingle xmlns='urn:xmpp:jingle:1' action='session-initiate' sid='coucou'><reason><a xmlns='http://www.w3.org/1999/xhtml'/></reason></jingle>".parse().unwrap(); let elem: Element = "<jingle xmlns='urn:xmpp:jingle:1' action='session-initiate' sid='coucou'><reason><a xmlns='http://www.w3.org/1999/xhtml'/></reason></jingle>".parse().unwrap();
let error = jingle::parse_jingle(&elem).unwrap_err(); let error = Jingle::try_from(&elem).unwrap_err();
let message = match error { let message = match error {
Error::ParseError(string) => string, Error::ParseError(string) => string,
_ => panic!(), _ => panic!(),
@ -577,7 +589,7 @@ mod tests {
assert_eq!(message, "Reason contains a foreign element."); assert_eq!(message, "Reason contains a foreign element.");
let elem: Element = "<jingle xmlns='urn:xmpp:jingle:1' action='session-initiate' sid='coucou'><reason><decline/></reason><reason/></jingle>".parse().unwrap(); let elem: Element = "<jingle xmlns='urn:xmpp:jingle:1' action='session-initiate' sid='coucou'><reason><decline/></reason><reason/></jingle>".parse().unwrap();
let error = jingle::parse_jingle(&elem).unwrap_err(); let error = Jingle::try_from(&elem).unwrap_err();
let message = match error { let message = match error {
Error::ParseError(string) => string, Error::ParseError(string) => string,
_ => panic!(), _ => panic!(),
@ -585,7 +597,7 @@ mod tests {
assert_eq!(message, "Jingle must not have more than one reason."); assert_eq!(message, "Jingle must not have more than one reason.");
let elem: Element = "<jingle xmlns='urn:xmpp:jingle:1' action='session-initiate' sid='coucou'><reason><decline/><text/><text/></reason></jingle>".parse().unwrap(); let elem: Element = "<jingle xmlns='urn:xmpp:jingle:1' action='session-initiate' sid='coucou'><reason><decline/><text/><text/></reason></jingle>".parse().unwrap();
let error = jingle::parse_jingle(&elem).unwrap_err(); let error = Jingle::try_from(&elem).unwrap_err();
let message = match error { let message = match error {
Error::ParseError(string) => string, Error::ParseError(string) => string,
_ => panic!(), _ => panic!(),