From 94cf8afdff00f0d24f6f37edf530b793f5293c5c Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Maxime=20=E2=80=9Cpep=E2=80=9D=20Buquet?= <pep@bouah.net>
Date: Sat, 25 Dec 2021 22:55:36 +0100
Subject: [PATCH] parsers: XEP-0363: HTTP File Upload
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Signed-off-by: Maxime “pep” Buquet <pep@bouah.net>
---
 parsers/src/http_upload.rs | 120 +++++++++++++++++++++++++++++++++++++
 parsers/src/lib.rs         |   3 +
 parsers/src/ns.rs          |   3 +
 3 files changed, 126 insertions(+)
 create mode 100644 parsers/src/http_upload.rs

diff --git a/parsers/src/http_upload.rs b/parsers/src/http_upload.rs
new file mode 100644
index 00000000..740904bf
--- /dev/null
+++ b/parsers/src/http_upload.rs
@@ -0,0 +1,120 @@
+// Copyright (c) 2021 Maxime “pep” Buquet <pep@bouah.net>
+//
+// This Source Code Form is subject to the terms of the Mozilla Public
+// 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/.
+
+use crate::iq::{IqGetPayload, IqResultPayload};
+use crate::util::helpers::Text;
+
+generate_element!(
+    /// Requesting a slot
+    SlotRequest, "request", HTTP_UPLOAD,
+    attributes: [
+        /// The filename to be uploaded.
+        filename: Required<String> = "filename",
+
+        /// Size of the file to be uploaded.
+        size: Required<u64> = "size",
+
+        /// Content-Type of the file.
+        content_type: Option<String> = "content-type",
+    ]
+);
+
+impl IqGetPayload for SlotRequest {}
+
+generate_element!(
+    /// Slot header
+    Header, "header", HTTP_UPLOAD,
+    attributes: [
+        /// Name of the header
+        name: Required<String> = "name"
+    ],
+    text: (
+        /// Content of the header
+        data: Text<String>
+    )
+);
+
+generate_element!(
+    /// Put URL
+    Put, "put", HTTP_UPLOAD,
+    attributes: [
+        /// URL
+        url: Required<String> = "url",
+    ],
+    children: [
+        /// Header list
+        headers: Vec<Header> = ("header", HTTP_UPLOAD) => Header
+    ]
+);
+
+generate_element!(
+    /// Get URL
+    Get, "get", HTTP_UPLOAD,
+    attributes: [
+        /// URL
+        url: Required<String> = "url",
+    ]
+);
+
+generate_element!(
+    /// Requesting a slot
+    SlotResult, "slot", HTTP_UPLOAD,
+    children: [
+        /// Put URL and headers
+        put: Required<Put> = ("put", HTTP_UPLOAD) => Put,
+        /// Get URL
+        get: Required<Get> = ("get", HTTP_UPLOAD) => Get
+    ]
+);
+
+impl IqResultPayload for SlotResult {}
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+    use crate::Element;
+    use std::convert::TryFrom;
+
+    #[test]
+    fn test_slot_request() {
+        let elem: Element = "<request xmlns='urn:xmpp:http:upload:0'
+            filename='très cool.jpg'
+            size='23456'
+            content-type='image/jpeg' />"
+            .parse()
+            .unwrap();
+        let slot = SlotRequest::try_from(elem).unwrap();
+        assert_eq!(slot.filename, String::from("très cool.jpg"));
+        assert_eq!(slot.size, 23456);
+        assert_eq!(slot.content_type, Some(String::from("image/jpeg")));
+    }
+
+    #[test]
+    fn test_slot_result() {
+        let elem: Element = "<slot xmlns='urn:xmpp:http:upload:0'>
+            <put url='https://upload.montague.tld/4a771ac1-f0b2-4a4a-9700-f2a26fa2bb67/tr%C3%A8s%20cool.jpg'>
+              <header name='Authorization'>Basic Base64String==</header>
+              <header name='Cookie'>foo=bar; user=romeo</header>
+            </put>
+            <get url='https://download.montague.tld/4a771ac1-f0b2-4a4a-9700-f2a26fa2bb67/tr%C3%A8s%20cool.jpg' />
+          </slot>"
+            .parse()
+            .unwrap();
+        let slot = SlotResult::try_from(elem).unwrap();
+        assert_eq!(slot.put.url, String::from("https://upload.montague.tld/4a771ac1-f0b2-4a4a-9700-f2a26fa2bb67/tr%C3%A8s%20cool.jpg"));
+        assert_eq!(slot.put.headers[0].name, String::from("Authorization"));
+        assert_eq!(
+            slot.put.headers[0].data,
+            String::from("Basic Base64String==")
+        );
+        assert_eq!(slot.put.headers[1].name, String::from("Cookie"));
+        assert_eq!(
+            slot.put.headers[1].data,
+            String::from("foo=bar; user=romeo")
+        );
+        assert_eq!(slot.get.url, String::from("https://download.montague.tld/4a771ac1-f0b2-4a4a-9700-f2a26fa2bb67/tr%C3%A8s%20cool.jpg"));
+    }
+}
diff --git a/parsers/src/lib.rs b/parsers/src/lib.rs
index d5aae810..03087406 100644
--- a/parsers/src/lib.rs
+++ b/parsers/src/lib.rs
@@ -210,6 +210,9 @@ pub mod jingle_message;
 /// XEP-0359: Unique and Stable Stanza IDs
 pub mod stanza_id;
 
+/// XEP-0363: HTTP File Upload
+pub mod http_upload;
+
 /// XEP-0369: Mediated Information eXchange (MIX)
 pub mod mix;
 
diff --git a/parsers/src/ns.rs b/parsers/src/ns.rs
index ad11298b..18a3ed56 100644
--- a/parsers/src/ns.rs
+++ b/parsers/src/ns.rs
@@ -219,6 +219,9 @@ pub const JINGLE_MESSAGE: &str = "urn:xmpp:jingle-message:0";
 /// XEP-0359: Unique and Stable Stanza IDs
 pub const SID: &str = "urn:xmpp:sid:0";
 
+/// XEP-0363: HTTP File Upload
+pub const HTTP_UPLOAD: &str = "urn:xmpp:http:upload:0";
+
 /// XEP-0369: Mediated Information eXchange (MIX)
 pub const MIX_CORE: &str = "urn:xmpp:mix:core:1";
 /// XEP-0369: Mediated Information eXchange (MIX)