From e12ba5be8f49fb06a1be182c8f940a54535b5ab3 Mon Sep 17 00:00:00 2001
From: Jonathan Rothberg <jonnyrpi@gmail.com>
Date: Sat, 28 Sep 2019 19:03:10 -0700
Subject: [PATCH] Added support for more `post` headers.

---
 src/commands/post.rs | 38 +++++++++++++++++++++++++++++++++++++-
 1 file changed, 37 insertions(+), 1 deletion(-)

diff --git a/src/commands/post.rs b/src/commands/post.rs
index 1bff755d9e..ac41709eca 100644
--- a/src/commands/post.rs
+++ b/src/commands/post.rs
@@ -11,6 +11,11 @@ use std::path::PathBuf;
 use std::str::FromStr;
 use surf::mime;
 
+pub enum HeaderKind {
+    ContentType(String),
+    ContentLength(String),
+}
+
 pub struct Post;
 
 impl PerItemCommand for Post {
@@ -24,6 +29,8 @@ impl PerItemCommand for Post {
             .required("body", SyntaxShape::Any)
             .named("user", SyntaxShape::Any)
             .named("password", SyntaxShape::Any)
+            .named("content-type", SyntaxShape::Any)
+            .named("content-length", SyntaxShape::Any)
             .switch("raw")
     }
 
@@ -73,9 +80,30 @@ fn run(
     let registry = registry.clone();
     let raw_args = raw_args.clone();
 
+    let content_type = call_info
+        .args
+        .get("content-type")
+        .map(|x| x.as_string().unwrap());
+
+    let content_length = call_info
+        .args
+        .get("content-length")
+        .map(|x| x.as_string().unwrap());
+
+    let mut headers = vec![];
+    match content_type {
+        Some(ct) => headers.push(HeaderKind::ContentType(ct)),
+        None => {}
+    };
+
+    match content_length {
+        Some(cl) => headers.push(HeaderKind::ContentLength(cl)),
+        None => {}
+    };
+
     let stream = async_stream! {
         let (file_extension, contents, contents_tag, span_source) =
-            post(&path_str, &body, user, password, path_span, &registry, &raw_args).await.unwrap();
+            post(&path_str, &body, user, password, &headers, path_span, &registry, &raw_args).await.unwrap();
 
         let file_extension = if has_raw {
             None
@@ -143,6 +171,7 @@ pub async fn post(
     body: &Tagged<Value>,
     user: Option<String>,
     password: Option<String>,
+    headers: &Vec<HeaderKind>,
     tag: Tag,
     registry: &CommandRegistry,
     raw_args: &RawCommandArgs,
@@ -164,6 +193,13 @@ pub async fn post(
                 if let Some(login) = login {
                     s = s.set_header("Authorization", format!("Basic {}", login));
                 }
+
+                for h in headers {
+                    s = match h {
+                        HeaderKind::ContentType(ct) => s.set_header("Content-Type", ct),
+                        HeaderKind::ContentLength(cl) => s.set_header("Content-Length", cl),
+                    };
+                }
                 s.await
             }
             Tagged {