Add headers support

1 files changed, 7 insertions, 1 deletions

diff --git a/src/main.rs b/src/main.rs
index c1fe2ab..67c6541 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -18,6 +18,8 @@ struct Args {
     #[clap(short, long)]
     headers: bool,
     #[clap(short, long)]
+    all_headers: bool,
+    #[clap(short, long)]
     body: bool,
     #[clap(short, long)]
     raw: bool,
@@ -26,7 +28,7 @@ struct Args {
 
 impl Args {
     pub fn print_headers(&self) -> bool {
-        self.headers || !self.body
+        self.headers || self.all_headers || !self.body
     }
 
     pub fn print_body(&self) -> bool {
@@ -36,6 +38,10 @@ impl Args {
     pub fn raw_body(&self) -> bool {
         self.raw
     }
+
+    pub fn req_headers(&self) -> bool {
+        self.all_headers
+    }
 }
 
 fn main() {