1package eu.siacs.conversations.utils;
2
3import android.net.Uri;
4import android.support.annotation.NonNull;
5
6import com.google.common.collect.ImmutableList;
7import com.google.common.collect.ImmutableMap;
8
9import java.io.UnsupportedEncodingException;
10import java.net.URLDecoder;
11import java.util.ArrayList;
12import java.util.Collections;
13import java.util.List;
14import java.util.Locale;
15import java.util.Map;
16
17import rocks.xmpp.addr.Jid;
18
19public class XmppUri {
20
21 protected Uri uri;
22 protected String jid;
23 private List<Fingerprint> fingerprints = new ArrayList<>();
24 private Map<String,String> parameters = Collections.emptyMap();
25 private boolean safeSource = true;
26
27 private static final String OMEMO_URI_PARAM = "omemo-sid-";
28
29 public static final String ACTION_JOIN = "join";
30 public static final String ACTION_MESSAGE = "message";
31
32 public XmppUri(String uri) {
33 try {
34 parse(Uri.parse(uri));
35 } catch (IllegalArgumentException e) {
36 try {
37 jid = Jid.of(uri).asBareJid().toString();
38 } catch (IllegalArgumentException e2) {
39 jid = null;
40 }
41 }
42 }
43
44 public XmppUri(Uri uri) {
45 parse(uri);
46 }
47
48 public XmppUri(Uri uri, boolean safeSource) {
49 this.safeSource = safeSource;
50 parse(uri);
51 }
52
53 public boolean isSafeSource() {
54 return safeSource;
55 }
56
57 protected void parse(final Uri uri) {
58 if (uri == null) {
59 return;
60 }
61 this.uri = uri;
62 String scheme = uri.getScheme();
63 String host = uri.getHost();
64 List<String> segments = uri.getPathSegments();
65 if ("https".equalsIgnoreCase(scheme) && "conversations.im".equalsIgnoreCase(host)) {
66 if (segments.size() >= 2 && segments.get(1).contains("@")) {
67 // sample : https://conversations.im/i/foo@bar.com
68 try {
69 jid = Jid.of(lameUrlDecode(segments.get(1))).toString();
70 } catch (Exception e) {
71 jid = null;
72 }
73 } else if (segments.size() >= 3) {
74 // sample : https://conversations.im/i/foo/bar.com
75 jid = segments.get(1) + "@" + segments.get(2);
76 }
77 if (segments.size() > 1 && "j".equalsIgnoreCase(segments.get(0))) {
78 this.parameters = ImmutableMap.of(ACTION_JOIN, "");
79 }
80 final Map<String,String> parameters = parseParameters(uri.getQuery(), '&');
81 this.fingerprints = parseFingerprints(parameters);
82 } else if ("xmpp".equalsIgnoreCase(scheme)) {
83 // sample: xmpp:foo@bar.com
84 this.parameters = parseParameters(uri.getQuery(), ';');
85 if (uri.getAuthority() != null) {
86 jid = uri.getAuthority();
87 } else {
88 final String[] parts = uri.getSchemeSpecificPart().split("\\?");
89 if (parts.length > 0) {
90 jid = parts[0];
91 } else {
92 return;
93 }
94 }
95 this.fingerprints = parseFingerprints(parameters);
96 } else if ("imto".equalsIgnoreCase(scheme)) {
97 // sample: imto://xmpp/foo@bar.com
98 try {
99 jid = URLDecoder.decode(uri.getEncodedPath(), "UTF-8").split("/")[1].trim();
100 } catch (final UnsupportedEncodingException ignored) {
101 jid = null;
102 }
103 } else {
104 try {
105 jid = Jid.of(uri.toString()).asBareJid().toString();
106 } catch (final IllegalArgumentException ignored) {
107 jid = null;
108 }
109 }
110 }
111
112
113 private static Map<String,String> parseParameters(final String query, final char seperator) {
114 final ImmutableMap.Builder<String,String> builder = new ImmutableMap.Builder<>();
115 final String[] pairs = query == null ? new String[0] : query.split(String.valueOf(seperator));
116 for (String pair : pairs) {
117 final String[] parts = pair.split("=", 2);
118 if (parts.length == 0) {
119 continue;
120 }
121 final String key = parts[0].toLowerCase(Locale.US);
122 final String value;
123 if (parts.length == 2) {
124 String decoded;
125 try {
126 decoded = URLDecoder.decode(parts[1],"UTF-8");
127 } catch (UnsupportedEncodingException e) {
128 decoded = "";
129 }
130 value = decoded;
131 } else {
132 value = "";
133 }
134 builder.put(key, value);
135 }
136 return builder.build();
137 }
138
139 @Override
140 @NonNull
141 public String toString() {
142 if (uri != null) {
143 return uri.toString();
144 }
145 return "";
146 }
147
148 private static List<Fingerprint> parseFingerprints(Map<String,String> parameters) {
149 ImmutableList.Builder<Fingerprint> builder = new ImmutableList.Builder<>();
150 for (Map.Entry<String, String> parameter : parameters.entrySet()) {
151 final String key = parameter.getKey();
152 final String value = parameter.getValue().toLowerCase(Locale.US);
153 if (key.startsWith(OMEMO_URI_PARAM)) {
154 try {
155 final int id = Integer.parseInt(key.substring(OMEMO_URI_PARAM.length()));
156 builder.add(new Fingerprint(FingerprintType.OMEMO, value, id));
157 } catch (Exception e) {
158 //ignoring invalid device id
159 }
160 }
161 }
162 return builder.build();
163 }
164
165 public boolean isAction(final String action) {
166 return parameters.containsKey(action);
167 }
168
169 public Jid getJid() {
170 try {
171 return this.jid == null ? null : Jid.of(this.jid);
172 } catch (IllegalArgumentException e) {
173 return null;
174 }
175 }
176
177 public boolean isJidValid() {
178 if (jid == null) {
179 return false;
180 }
181 try {
182 Jid.of(jid);
183 return true;
184 } catch (IllegalArgumentException e) {
185 return false;
186 }
187 }
188
189 public String getBody() {
190 return parameters.get("body");
191 }
192
193 public String getName() {
194 return parameters.get("name");
195 }
196
197 public List<Fingerprint> getFingerprints() {
198 return this.fingerprints;
199 }
200
201 public boolean hasFingerprints() {
202 return fingerprints.size() > 0;
203 }
204
205 public enum FingerprintType {
206 OMEMO
207 }
208
209 public static String getFingerprintUri(String base, List<XmppUri.Fingerprint> fingerprints, char separator) {
210 StringBuilder builder = new StringBuilder(base);
211 builder.append('?');
212 for (int i = 0; i < fingerprints.size(); ++i) {
213 XmppUri.FingerprintType type = fingerprints.get(i).type;
214 if (type == XmppUri.FingerprintType.OMEMO) {
215 builder.append(XmppUri.OMEMO_URI_PARAM);
216 builder.append(fingerprints.get(i).deviceId);
217 }
218 builder.append('=');
219 builder.append(fingerprints.get(i).fingerprint);
220 if (i != fingerprints.size() - 1) {
221 builder.append(separator);
222 }
223 }
224 return builder.toString();
225 }
226
227 public static class Fingerprint {
228 public final FingerprintType type;
229 public final String fingerprint;
230 final int deviceId;
231
232 public Fingerprint(FingerprintType type, String fingerprint, int deviceId) {
233 this.type = type;
234 this.fingerprint = fingerprint;
235 this.deviceId = deviceId;
236 }
237
238 @NonNull
239 @Override
240 public String toString() {
241 return type.toString() + ": " + fingerprint + (deviceId != 0 ? " " + deviceId : "");
242 }
243 }
244
245 private static String lameUrlDecode(String url) {
246 return url.replace("%23", "#").replace("%25", "%");
247 }
248
249 public static String lameUrlEncode(String url) {
250 return url.replace("%", "%25").replace("#", "%23");
251 }
252}