agate

commit 2cc5f791aa760503ce6e1edb6cc5204d7f9d3668
parent c12b2ab6567f92cd3e29d0a11fddbc68ee0d1a25
Author: Matt Brubeck <mbrubeck@limpet.net>
Date:   Thu, 16 Jul 2020 08:50:36 -0700

Code cleanup

Diffstat:
M
src/main.rs
 | 
3
++-

1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/src/main.rs b/src/main.rs
@@ -130,11 +130,12 @@ async fn send_response<W: Write + Unpin>(url: &Url, mut stream: W) -> Result {
         }
     }
 
+    // Make sure the file opens successfully before sending the success header.
     let mut file = async_std::fs::File::open(&path).await?;
     
     // Send header.
     if path.extension() == Some(OsStr::new("gmi")) {
-        stream.write_all(b"20 text/gemini\r\n").await?;
+        respond(&mut stream, "20", &["text/gemini"]).await?;
     } else {
         let mime = tree_magic_mini::from_filepath(&path).ok_or("Can't read file")?;
         respond(&mut stream, "20", &[mime]).await?;