From bdd87a4d54703b8f6e3844afd32ba88926e68d58 Mon Sep 17 00:00:00 2001
From: Tyler Murphy <tylerm@tylerm.dev>
Date: Sun, 2 Jul 2023 23:44:48 -0400
Subject: status header

---
 src/script.rs | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

(limited to 'src/script.rs')

diff --git a/src/script.rs b/src/script.rs
index 4a94ba4..064bc67 100644
--- a/src/script.rs
+++ b/src/script.rs
@@ -37,13 +37,21 @@ impl Script {
             Err(err) => return Err(format!("{err}"))
         };
 
-        let status = output.status.code().unwrap_or(500) as u16;
-
         let response = String::from_utf8_lossy(&output.stdout).into_owned();
         let mut lines = response.split('\n').into_iter();
 
         res.headers.serialize(&mut lines);
 
+        let status = match res.headers.get("Status") {
+            Some(header) => {
+                match header.value.parse::<u16>() {
+                    Ok(status) => status,
+                    Err(err) => return Err(format!("{err}"))
+                }
+            },
+            None => 200
+        };
+
         let body: String = lines.collect::<Vec<&str>>().join("\n");
         res.headers.put(Header::new("Content-Length", &format!("{}", body.len())));
         
-- 
cgit v1.2.3-freya