diff --git a/src/iq.rs b/src/iq.rs index 9420b44c..976d79f8 100644 --- a/src/iq.rs +++ b/src/iq.rs @@ -24,15 +24,35 @@ use jingle::Jingle; use ping::Ping; use mam::{Query as MamQuery, Fin as MamFin, Prefs as MamPrefs}; -/// Lists every known payload of a ``. +/// Lists every known payload of an ``. #[derive(Debug, Clone)] -pub enum IqPayload { +pub enum IqGetPayload { + Roster(Roster), + DiscoInfo(DiscoInfoQuery), + Ping(Ping), + MamQuery(MamQuery), + MamPrefs(MamPrefs), + + Unknown(Element), +} + +/// Lists every known payload of an ``. +#[derive(Debug, Clone)] +pub enum IqSetPayload { Roster(Roster), - DiscoInfoResult(DiscoInfoResult), - DiscoInfoQuery(DiscoInfoQuery), IBB(IBB), Jingle(Jingle), - Ping(Ping), + MamQuery(MamQuery), + MamPrefs(MamPrefs), + + Unknown(Element), +} + +/// Lists every known payload of an ``. +#[derive(Debug, Clone)] +pub enum IqResultPayload { + Roster(Roster), + DiscoInfo(DiscoInfoResult), MamQuery(MamQuery), MamFin(MamFin), MamPrefs(MamPrefs), @@ -40,45 +60,117 @@ pub enum IqPayload { Unknown(Element), } -impl TryFrom for IqPayload { +impl TryFrom for IqGetPayload { type Error = Error; - fn try_from(elem: Element) -> Result { + fn try_from(elem: Element) -> Result { Ok(match (elem.name().as_ref(), elem.ns().unwrap().as_ref()) { // RFC-6121 - ("query", ns::ROSTER) => IqPayload::Roster(Roster::try_from(elem)?), + ("query", ns::ROSTER) => IqGetPayload::Roster(Roster::try_from(elem)?), // XEP-0030 - ("query", ns::DISCO_INFO) => { - // TODO: separate all three types of payloads. - match DiscoInfoQuery::try_from(elem.clone()) { - Ok(payload) => IqPayload::DiscoInfoQuery(payload), - // TODO: put that error somewhere too? - Err(_) => IqPayload::DiscoInfoResult(DiscoInfoResult::try_from(elem)?), - } - }, + ("query", ns::DISCO_INFO) => IqGetPayload::DiscoInfo(DiscoInfoQuery::try_from(elem)?), + + // XEP-0199 + ("ping", ns::PING) => IqGetPayload::Ping(Ping::try_from(elem)?), + + // XEP-0313 + ("query", ns::MAM) => IqGetPayload::MamQuery(MamQuery::try_from(elem)?), + ("prefs", ns::MAM) => IqGetPayload::MamPrefs(MamPrefs::try_from(elem)?), + + _ => IqGetPayload::Unknown(elem), + }) + } +} + +impl Into for IqGetPayload { + fn into(self) -> Element { + match self { + IqGetPayload::Roster(roster) => roster.into(), + IqGetPayload::DiscoInfo(disco) => disco.into(), + IqGetPayload::Ping(ping) => ping.into(), + IqGetPayload::MamQuery(query) => query.into(), + IqGetPayload::MamPrefs(prefs) => prefs.into(), + + IqGetPayload::Unknown(elem) => elem, + } + } +} + +impl TryFrom for IqSetPayload { + type Error = Error; + + fn try_from(elem: Element) -> Result { + Ok(match (elem.name().as_ref(), elem.ns().unwrap().as_ref()) { + // RFC-6121 + ("query", ns::ROSTER) => IqSetPayload::Roster(Roster::try_from(elem)?), // XEP-0047 ("open", ns::IBB) | ("data", ns::IBB) - | ("close", ns::IBB) => IqPayload::IBB(IBB::try_from(elem)?), + | ("close", ns::IBB) => IqSetPayload::IBB(IBB::try_from(elem)?), // XEP-0166 - ("jingle", ns::JINGLE) => IqPayload::Jingle(Jingle::try_from(elem)?), - - // XEP-0199 - ("ping", ns::PING) => IqPayload::Ping(Ping::try_from(elem)?), + ("jingle", ns::JINGLE) => IqSetPayload::Jingle(Jingle::try_from(elem)?), // XEP-0313 - ("query", ns::MAM) => IqPayload::MamQuery(MamQuery::try_from(elem)?), - ("fin", ns::MAM) => IqPayload::MamFin(MamFin::try_from(elem)?), - ("prefs", ns::MAM) => IqPayload::MamPrefs(MamPrefs::try_from(elem)?), + ("query", ns::MAM) => IqSetPayload::MamQuery(MamQuery::try_from(elem)?), + ("prefs", ns::MAM) => IqSetPayload::MamPrefs(MamPrefs::try_from(elem)?), - _ => IqPayload::Unknown(elem), + _ => IqSetPayload::Unknown(elem), }) } } +impl Into for IqSetPayload { + fn into(self) -> Element { + match self { + IqSetPayload::Roster(roster) => roster.into(), + IqSetPayload::IBB(ibb) => ibb.into(), + IqSetPayload::Jingle(jingle) => jingle.into(), + IqSetPayload::MamQuery(query) => query.into(), + IqSetPayload::MamPrefs(prefs) => prefs.into(), + + IqSetPayload::Unknown(elem) => elem, + } + } +} + +impl TryFrom for IqResultPayload { + type Error = Error; + + fn try_from(elem: Element) -> Result { + Ok(match (elem.name().as_ref(), elem.ns().unwrap().as_ref()) { + // RFC-6121 + ("query", ns::ROSTER) => IqResultPayload::Roster(Roster::try_from(elem)?), + + // XEP-0030 + ("query", ns::DISCO_INFO) => IqResultPayload::DiscoInfo(DiscoInfoResult::try_from(elem)?), + + // XEP-0313 + ("query", ns::MAM) => IqResultPayload::MamQuery(MamQuery::try_from(elem)?), + ("fin", ns::MAM) => IqResultPayload::MamFin(MamFin::try_from(elem)?), + ("prefs", ns::MAM) => IqResultPayload::MamPrefs(MamPrefs::try_from(elem)?), + + _ => IqResultPayload::Unknown(elem), + }) + } +} + +impl Into for IqResultPayload { + fn into(self) -> Element { + match self { + IqResultPayload::Roster(roster) => roster.into(), + IqResultPayload::DiscoInfo(disco) => disco.into(), + IqResultPayload::MamQuery(query) => query.into(), + IqResultPayload::MamFin(fin) => fin.into(), + IqResultPayload::MamPrefs(prefs) => prefs.into(), + + IqResultPayload::Unknown(elem) => elem, + } + } +} + #[derive(Debug, Clone)] pub enum IqType { Get(Element), @@ -175,24 +267,6 @@ impl TryFrom for Iq { } } -impl Into for IqPayload { - fn into(self) -> Element { - match self { - IqPayload::Roster(roster) => roster.into(), - IqPayload::DiscoInfoResult(disco) => disco.into(), - IqPayload::DiscoInfoQuery(disco) => disco.into(), - IqPayload::IBB(ibb) => ibb.into(), - IqPayload::Jingle(jingle) => jingle.into(), - IqPayload::Ping(ping) => ping.into(), - IqPayload::MamQuery(query) => query.into(), - IqPayload::MamFin(fin) => fin.into(), - IqPayload::MamPrefs(prefs) => prefs.into(), - - IqPayload::Unknown(elem) => elem, - } - } -} - impl Into for Iq { fn into(self) -> Element { let mut stanza = Element::builder("iq") @@ -344,11 +418,11 @@ mod tests { let elem: Element = "".parse().unwrap(); let iq = Iq::try_from(elem).unwrap(); let payload = match iq.payload { - IqType::Get(payload) => IqPayload::try_from(payload).unwrap(), + IqType::Get(payload) => IqGetPayload::try_from(payload).unwrap(), _ => panic!(), }; assert!(match payload { - IqPayload::DiscoInfoQuery(DiscoInfoQuery { .. }) => true, + IqGetPayload::DiscoInfo(DiscoInfoQuery { .. }) => true, _ => false, }); }