1// Copyright (C) 2019 Maxime “pep” Buquet <pep@bouah.net>
2// This Source Code Form is subject to the terms of the Mozilla Public
3// License, v. 2.0. If a copy of the MPL was not distributed with this
4// file, You can obtain one at http://mozilla.org/MPL/2.0/.
5
6use crate::data_forms::{DataForm, DataFormType, Field, FieldType};
7use crate::ns;
8use xso::error::Error;
9
10/// Structure representing a `http://jabber.org/network/serverinfo` form type.
11#[derive(Debug, Clone, PartialEq, Default)]
12pub struct ServerInfo {
13 /// Abuse addresses
14 pub abuse: Vec<String>,
15
16 /// Admin addresses
17 pub admin: Vec<String>,
18
19 /// Feedback addresses
20 pub feedback: Vec<String>,
21
22 /// Sales addresses
23 pub sales: Vec<String>,
24
25 /// Security addresses
26 pub security: Vec<String>,
27
28 /// Support addresses
29 pub support: Vec<String>,
30}
31
32impl TryFrom<DataForm> for ServerInfo {
33 type Error = Error;
34
35 fn try_from(form: DataForm) -> Result<ServerInfo, Error> {
36 if form.type_ != DataFormType::Result_ {
37 return Err(Error::Other("Wrong type of form."));
38 }
39 if form.form_type() != Some(ns::SERVER_INFO) {
40 return Err(Error::Other("Wrong FORM_TYPE for form."));
41 }
42 let mut server_info = ServerInfo::default();
43 for field in form.fields {
44 if field.var.as_deref().unwrap_or("") == "FORM_TYPE" {
45 continue;
46 }
47 if field.type_ != FieldType::ListMulti {
48 return Err(Error::Other("Field is not of the required type."));
49 }
50 if field.var.as_deref() == Some("abuse-addresses") {
51 server_info.abuse = field.values;
52 } else if field.var.as_deref() == Some("admin-addresses") {
53 server_info.admin = field.values;
54 } else if field.var.as_deref() == Some("feedback-addresses") {
55 server_info.feedback = field.values;
56 } else if field.var.as_deref() == Some("sales-addresses") {
57 server_info.sales = field.values;
58 } else if field.var.as_deref() == Some("security-addresses") {
59 server_info.security = field.values;
60 } else if field.var.as_deref() == Some("support-addresses") {
61 server_info.support = field.values;
62 } else {
63 return Err(Error::Other("Unknown form field var."));
64 }
65 }
66
67 Ok(server_info)
68 }
69}
70
71impl From<ServerInfo> for DataForm {
72 fn from(server_info: ServerInfo) -> DataForm {
73 let mut form = DataForm {
74 type_: DataFormType::Result_,
75 title: None,
76 instructions: None,
77 fields: vec![
78 generate_address_field("abuse-addresses", server_info.abuse),
79 generate_address_field("admin-addresses", server_info.admin),
80 generate_address_field("feedback-addresses", server_info.feedback),
81 generate_address_field("sales-addresses", server_info.sales),
82 generate_address_field("security-addresses", server_info.security),
83 generate_address_field("support-addresses", server_info.support),
84 ],
85 };
86 form.set_form_type(ns::SERVER_INFO.to_owned());
87 form
88 }
89}
90
91/// Generate `Field` for addresses
92pub fn generate_address_field<S: Into<String>>(var: S, values: Vec<String>) -> Field {
93 Field {
94 var: Some(var.into()),
95 type_: FieldType::ListMulti,
96 label: None,
97 required: false,
98 desc: None,
99 options: vec![],
100 values,
101 media: vec![],
102 validate: None,
103 }
104}
105
106#[cfg(test)]
107mod tests {
108 use super::*;
109
110 #[cfg(target_pointer_width = "32")]
111 #[test]
112 fn test_size() {
113 assert_size!(ServerInfo, 72);
114 }
115
116 #[cfg(target_pointer_width = "64")]
117 #[test]
118 fn test_size() {
119 assert_size!(ServerInfo, 144);
120 }
121
122 #[test]
123 fn test_simple() {
124 let form = DataForm::new(
125 DataFormType::Result_,
126 ns::SERVER_INFO,
127 vec![
128 Field::new("abuse-addresses", FieldType::ListMulti),
129 Field::new("admin-addresses", FieldType::ListMulti)
130 .with_value("xmpp:admin@foo.bar")
131 .with_value("https://foo.bar/chat/")
132 .with_value("mailto:admin@foo.bar"),
133 Field::new("feedback-addresses", FieldType::ListMulti),
134 Field::new("sales-addresses", FieldType::ListMulti),
135 Field::new("security-addresses", FieldType::ListMulti)
136 .with_value("xmpp:security@foo.bar")
137 .with_value("mailto:security@foo.bar"),
138 Field::new("support-addresses", FieldType::ListMulti)
139 .with_value("mailto:support@foo.bar"),
140 ],
141 );
142
143 let server_info = ServerInfo {
144 abuse: vec![],
145 admin: vec![
146 String::from("xmpp:admin@foo.bar"),
147 String::from("https://foo.bar/chat/"),
148 String::from("mailto:admin@foo.bar"),
149 ],
150 feedback: vec![],
151 sales: vec![],
152 security: vec![
153 String::from("xmpp:security@foo.bar"),
154 String::from("mailto:security@foo.bar"),
155 ],
156 support: vec![String::from("mailto:support@foo.bar")],
157 };
158
159 // assert_eq!(DataForm::from(server_info), form);
160 assert_eq!(ServerInfo::try_from(form).unwrap(), server_info);
161 }
162}