From 1e2f32f3d0242f083e954182f20378fe72f4b1d7 Mon Sep 17 00:00:00 2001
From: Horhik <horhik@tuta.io>
Date: Tue, 12 Jan 2021 17:43:00 +0200
Subject: [PATCH] impl FcpRequest trait for some types 	String, bool,
 VerbosityPut

---
 src/lib.rs                    |  14 ++++
 src/types/client/fcp_types.rs | 138 ++++++++++++++++++++++++++++++++--
 src/types/client/types.rs     |   4 +
 3 files changed, 151 insertions(+), 5 deletions(-)

diff --git a/src/lib.rs b/src/lib.rs
index 33285db..b76239b 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -3,6 +3,20 @@ use types::fcp_types;
 
 #[cfg(test)]
 mod tests {
+
+    #[macro_export]
+    macro_rules! vec {
+    ( $( $x:expr ),* ) => {
+        {
+            let mut temp_vec = Vec::new();
+            $(
+                temp_vec.push($x);
+            )*
+            temp_vec
+        }
+    };
+}
+
     #[test]
     fn it_works() {
         assert_eq!(2 + 2, 4);
diff --git a/src/types/client/fcp_types.rs b/src/types/client/fcp_types.rs
index 2ac6483..7523cce 100644
--- a/src/types/client/fcp_types.rs
+++ b/src/types/client/fcp_types.rs
@@ -1,4 +1,5 @@
 use super::types::*;
+
 impl ClientHello {
     fn new(name: String, exp_ver: f32) -> Self {
         ClientHello {
@@ -6,7 +7,10 @@ impl ClientHello {
             expected_version: exp_ver,
         }
     }
-    fn string(self) -> String {
+}
+
+impl FcpRequest for ClientHello {
+    fn parse(&self) -> String {
         return format!(
             "ClientHello\n\
              Name={}\n\
@@ -26,7 +30,7 @@ pub struct ClientHello {
 fn client_hello_parses() {
     let hello = ClientHello::new("user name".to_string(), 2.0);
     assert_eq!(
-        hello.string(),
+        hello.parse(),
         "ClientHello\nName=user name\nExpectedVersion=2\nEndMessage\n\n"
     );
 }
@@ -89,8 +93,9 @@ pub struct GenerateSSK {
 }
 
 pub struct ClientPut {
-    message_name: String,
     uri: String, //TODO create key type
+    data_length: u64,
+    filename: String,
     content_type: Option<String>,
     identifier: Option<String>,
     verbosity: Option<VerbosityPut>,
@@ -104,8 +109,6 @@ pub struct ClientPut {
     persistence: Option<Box<OsStr>>,
     early_encode: Option<bool>,
     upload_ffrom: Option<UploadForm>,
-    data_length: u64,
-    filename: String,
     target_uri: Option<String>, // cloning  uri if does not exists
     file_hash: Option<String>,  //TODO SHAA256 type
     binary_blob: Option<bool>,
@@ -119,6 +122,131 @@ pub struct ClientPut {
     metadata_threshold: Option<i64>,
     data: Option<String>, // Data fromdirect
 }
+/*
+impl FcpRequest for ClientPut {
+    fn parse(self) -> String {
+        format!(
+            "ClientPut\n\
+                 {}\
+                 {}\
+                 {}\
+                 {}\
+                 {}\
+                 {}\
+                 {}\
+                 {}\
+                 {}\
+                 {}\
+                 {}\
+                 {}\
+                 {}\
+                 {}\
+                 {}\
+                 {}\
+                 {}\
+                 {}\
+                 {}\
+                 {}\
+                 {}\
+                 {}\
+                 {}\
+                 {}\
+                 {}\
+                 {}\
+                 {}\
+                 {}\
+                 EndMessage\n\
+                 {}\
+                 ",
+            self.uri,
+            self.data_length,
+            self.filename,
+            self.content_type.unwrap_or("".to_string()),
+            self.identifier.unwrap_or("".to_string()),
+            self.verbosity.unwrap_or("".to_string()),
+            self.max_retries.unwrap_or("".to_string()),
+            self.priority_class.unwrap_or("".to_string()),
+            self.get_chk_only.unwrap_or("".to_string()),
+            self.global.unwrap_or("".to_string()),
+            self.dont_compress.unwrap_or("".to_string()),
+            self.codecs.unwrap_or("".to_string()),
+            self.identifier.unwrap_or("".to_string()),
+            self.identifier.unwrap_or("".to_string()),
+            self.identifier.unwrap_or("".to_string()),
+            self.identifier.unwrap_or("".to_string()),
+            self.identifier.unwrap_or("".to_string()),
+            self.identifier.unwrap_or("".to_string()),
+            self.identifier.unwrap_or("".to_string()),
+            self.identifier.unwrap_or("".to_string()),
+            self.identifier.unwrap_or("".to_string()),
+            self.identifier.unwrap_or("".to_string()),
+            self.identifier.unwrap_or("".to_string()),
+            self.identifier.unwrap_or("".to_string()),
+            self.identifier.unwrap_or("".to_string()),
+            self.identifier.unwrap_or("".to_string()),
+            self.identifier.unwrap_or("".to_string()),
+            self.identifier.unwrap_or("".to_string()),
+            self.identifier.unwrap_or("".to_string()),
+            self.identifier.unwrap_or("".to_string()),
+        )
+    }
+}*/
+
+pub fn default_unwrap<T: FcpRequest>(fcp_type: Option<&T>) -> String {
+    match fcp_type {
+        Some(val) => val.parse(),
+        None => String::from(""),
+    }
+}
+impl FcpRequest for String {
+    fn parse(&self) -> String {
+        self.to_string()
+    }
+}
+
+impl FcpRequest for bool {
+    fn parse(&self) -> String {
+        if *self {
+            "true".to_string()
+        } else {
+            "false".to_string()
+        }
+    }
+}
+
+impl FcpRequest for VerbosityPut {
+    fn parse(&self) -> String {
+        match self {
+            VerbosityPut::SimpleProgress => 0.to_string(),
+            VerbosityPut::ExpectedHashes => 3.to_string(),
+            VerbosityPut::PutFetchable => 8.to_string(),
+            VerbosityPut::StartedCompressionANDFinishedCompression => 9.to_string(),
+        }
+    }
+}
+
+#[test]
+fn is_berbosity_put_parsing() {
+    assert_eq!(default_unwrap::<VerbosityPut>(None), "".to_string());
+    assert_eq!(
+        default_unwrap::<VerbosityPut>(Some(&VerbosityPut::SimpleProgress)),
+        "0".to_string()
+    );
+    assert_eq!(
+        default_unwrap::<VerbosityPut>(Some(&VerbosityPut::ExpectedHashes)),
+        "3".to_string()
+    );
+    assert_eq!(
+        default_unwrap::<VerbosityPut>(Some(&VerbosityPut::PutFetchable)),
+        "8".to_string()
+    );
+    assert_eq!(
+        default_unwrap::<VerbosityPut>(Some(
+            &VerbosityPut::StartedCompressionANDFinishedCompression
+        )),
+        "9".to_string()
+    );
+}
 
 pub struct ClientGet {
     message_name: String,
diff --git a/src/types/client/types.rs b/src/types/client/types.rs
index c98ae51..cf95016 100644
--- a/src/types/client/types.rs
+++ b/src/types/client/types.rs
@@ -71,3 +71,7 @@ pub enum Priority {
     F, // 5
     G, // 6
 }
+
+pub trait FcpRequest {
+    fn parse(&self) -> String;
+}