1// Copyright (c) 2017 Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
2//
3// This Source Code Form is subject to the terms of the Mozilla Public
4// License, v. 2.0. If a copy of the MPL was not distributed with this
5// file, You can obtain one at http://mozilla.org/MPL/2.0/.
6
7use std::collections::HashMap;
8use try_from::TryFrom;
9
10use minidom::Element;
11
12use error::Error;
13
14use iq::{IqGetPayload, IqSetPayload, IqResultPayload};
15use data_forms::DataForm;
16
17use ns;
18
19/// Query for registering against a service.
20#[derive(Debug, Clone)]
21pub struct Query {
22 /// Deprecated fixed list of possible fields to fill before the user can
23 /// register.
24 pub fields: HashMap<String, String>,
25
26 /// Whether this account is already registered.
27 pub registered: bool,
28
29 /// Whether to remove this account.
30 pub remove: bool,
31
32 /// A data form the user must fill before being allowed to register.
33 pub form: Option<DataForm>,
34
35 // Not yet implemented.
36 //pub oob: Option<Oob>,
37}
38
39impl IqGetPayload for Query {}
40impl IqSetPayload for Query {}
41impl IqResultPayload for Query {}
42
43impl TryFrom<Element> for Query {
44 type Err = Error;
45
46 fn try_from(elem: Element) -> Result<Query, Error> {
47 check_self!(elem, "query", REGISTER, "IBR query");
48 let mut query = Query {
49 registered: false,
50 fields: HashMap::new(),
51 remove: false,
52 form: None,
53 };
54 for child in elem.children() {
55 let namespace = child.ns().unwrap();
56 if namespace == ns::REGISTER {
57 let name = child.name();
58 let fields = vec!["address", "city", "date", "email", "first", "instructions",
59 "key", "last", "misc", "name", "nick", "password", "phone",
60 "state", "text", "url", "username", "zip"];
61 if fields.binary_search(&name).is_ok() {
62 query.fields.insert(name.to_owned(), child.text());
63 } else if name == "registered" {
64 query.registered = true;
65 } else if name == "remove" {
66 query.remove = true;
67 } else {
68 return Err(Error::ParseError("Wrong field in ibr element."));
69 }
70 } else if child.is("x", ns::DATA_FORMS) {
71 query.form = Some(DataForm::try_from(child.clone())?);
72 } else {
73 return Err(Error::ParseError("Unknown child in ibr element."));
74 }
75 }
76 Ok(query)
77 }
78}
79
80impl From<Query> for Element {
81 fn from(query: Query) -> Element {
82 Element::builder("query")
83 .ns(ns::REGISTER)
84 .append(if query.registered { Some(Element::builder("registered").ns(ns::REGISTER)) } else { None })
85 .append(query.fields.into_iter().map(|(name, value)| {
86 Element::builder(name).ns(ns::REGISTER).append(value)
87 }).collect::<Vec<_>>())
88 .append(if query.remove { Some(Element::builder("remove").ns(ns::REGISTER)) } else { None })
89 .append(query.form)
90 .build()
91 }
92}
93
94#[cfg(test)]
95mod tests {
96 use super::*;
97 use compare_elements::NamespaceAwareCompare;
98
99 #[test]
100 fn test_size() {
101 assert_size!(Query, 152);
102 }
103
104 #[test]
105 fn test_simple() {
106 let elem: Element = "<query xmlns='jabber:iq:register'/>".parse().unwrap();
107 Query::try_from(elem).unwrap();
108 }
109
110 #[test]
111 fn test_ex2() {
112 let elem: Element = r#"
113<query xmlns='jabber:iq:register'>
114 <instructions>
115 Choose a username and password for use with this service.
116 Please also provide your email address.
117 </instructions>
118 <username/>
119 <password/>
120 <email/>
121</query>
122"#.parse().unwrap();
123 let query = Query::try_from(elem).unwrap();
124 assert_eq!(query.registered, false);
125 assert_eq!(query.fields["instructions"], "\n Choose a username and password for use with this service.\n Please also provide your email address.\n ");
126 assert_eq!(query.fields["username"], "");
127 assert_eq!(query.fields["password"], "");
128 assert_eq!(query.fields["email"], "");
129 assert_eq!(query.fields.contains_key("name"), false);
130
131 // FIXME: HashMap doesn’t keep the order right.
132 //let elem2 = query.into();
133 //assert_eq!(elem, elem2);
134 }
135
136 #[test]
137 fn test_ex9() {
138 let elem: Element = r#"
139<query xmlns='jabber:iq:register'>
140 <instructions>
141 Use the enclosed form to register. If your Jabber client does not
142 support Data Forms, visit http://www.shakespeare.lit/contests.php
143 </instructions>
144 <x xmlns='jabber:x:data' type='form'>
145 <title>Contest Registration</title>
146 <instructions>
147 Please provide the following information
148 to sign up for our special contests!
149 </instructions>
150 <field type='hidden' var='FORM_TYPE'>
151 <value>jabber:iq:register</value>
152 </field>
153 <field label='Given Name' var='first'>
154 <required/>
155 </field>
156 <field label='Family Name' var='last'>
157 <required/>
158 </field>
159 <field label='Email Address' var='email'>
160 <required/>
161 </field>
162 <field type='list-single' label='Gender' var='x-gender'>
163 <option label='Male'><value>M</value></option>
164 <option label='Female'><value>F</value></option>
165 </field>
166 </x>
167</query>
168"#.parse().unwrap();
169 let elem1 = elem.clone();
170 let query = Query::try_from(elem).unwrap();
171 assert_eq!(query.registered, false);
172 assert!(!query.fields["instructions"].is_empty());
173 let form = query.form.clone().unwrap();
174 assert!(!form.instructions.unwrap().is_empty());
175 assert!(form.fields.binary_search_by(|field| field.var.cmp(&String::from("first"))).is_ok());
176 assert!(form.fields.binary_search_by(|field| field.var.cmp(&String::from("x-gender"))).is_ok());
177 assert!(form.fields.binary_search_by(|field| field.var.cmp(&String::from("coucou"))).is_err());
178 let elem2 = query.into();
179 assert!(elem1.compare_to(&elem2));
180 }
181
182 #[test]
183 fn test_ex10() {
184 let elem: Element = r#"
185<query xmlns='jabber:iq:register'>
186 <x xmlns='jabber:x:data' type='submit'>
187 <field type='hidden' var='FORM_TYPE'>
188 <value>jabber:iq:register</value>
189 </field>
190 <field label='Given Name' var='first'>
191 <value>Juliet</value>
192 </field>
193 <field label='Family Name' var='last'>
194 <value>Capulet</value>
195 </field>
196 <field label='Email Address' var='email'>
197 <value>juliet@capulet.com</value>
198 </field>
199 <field type='list-single' label='Gender' var='x-gender'>
200 <value>F</value>
201 </field>
202 </x>
203</query>
204"#.parse().unwrap();
205 let elem1 = elem.clone();
206 let query = Query::try_from(elem).unwrap();
207 assert_eq!(query.registered, false);
208 for _ in &query.fields {
209 panic!();
210 }
211 let elem2 = query.into();
212 assert!(elem1.compare_to(&elem2));
213 }
214}