From 3f80205783bcf6a2ed682f6f21e5b1877d597328 Mon Sep 17 00:00:00 2001
From: metamuffin <metamuffin@disroot.org>
Date: Thu, 4 Jul 2024 17:15:25 +0200
Subject: rfiles

---
 src/encoding/headermap.rs | 34 ++++++++++++++++++++++++++++++++++
 1 file changed, 34 insertions(+)
 create mode 100644 src/encoding/headermap.rs

(limited to 'src/encoding/headermap.rs')

diff --git a/src/encoding/headermap.rs b/src/encoding/headermap.rs
new file mode 100644
index 0000000..313d8b5
--- /dev/null
+++ b/src/encoding/headermap.rs
@@ -0,0 +1,34 @@
+use super::headers::Header;
+use anyhow::Result;
+use std::fmt::Display;
+
+#[derive(Debug)]
+pub struct HeaderMap(pub Vec<(String, String)>);
+
+impl HeaderMap {
+    pub fn new() -> Self {
+        Self(vec![])
+    }
+    pub fn add<H: Header>(mut self, h: H) -> Self {
+        self.0.push((H::NAME.to_string(), format!("{h}")));
+        self
+    }
+    pub fn get<H: Header>(&self) -> Option<Result<H>> {
+        self.0
+            .iter()
+            .find(|(k, _)| k == H::NAME)
+            .map(|(_, v)| H::from_str(v))
+    }
+    pub fn insert_raw(&mut self, key: String, value: String) {
+        self.0.push((key, value))
+    }
+}
+
+impl Display for HeaderMap {
+    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
+        for (k, v) in &self.0 {
+            write!(f, "{k}: {v}\r\n")?;
+        }
+        Ok(())
+    }
+}
-- 
cgit v1.2.3-70-g09d2