feat(meta): meta-related utilities

6 files changed, 134 insertions, 0 deletions

diff --git a/Cargo.toml b/Cargo.toml
index b3b73f1..b826985 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -18,6 +18,7 @@ categories = ["encoding"]
 ast = []
 convert = []
 request = ["rustls", "url", "anyhow"]
+meta = []
 
 [dependencies]
 rustls = { version = "0.20.5", features = ["dangerous_configuration"], optional = true } # TLS
diff --git a/Makefile.toml b/Makefile.toml
index b4967e3..c9238b0 100644
--- a/Makefile.toml
+++ b/Makefile.toml
@@ -14,6 +14,10 @@ command = "cargo"
 args = ["clippy"]
 command = "cargo"
 
+[tasks.test]
+args = ["test", "--all-features"]
+command = "cargo"
+
 # -------------
 # | Executors |
 # -------------
diff --git a/examples/meta.rs b/examples/meta.rs
new file mode 100644
index 0000000..75d894b
--- /dev/null
+++ b/examples/meta.rs
@@ -0,0 +1,24 @@
+// This file is part of Germ <https://github.com/gemrest/germ>.
+// Copyright (C) 2022-2022 Fuwn <[email protected]>
+//
+// This program is free software: you can redistribute it and/or modify
+// it under the terms of the GNU General Public License as published by
+// the Free Software Foundation, version 3.
+//
+// This program is distributed in the hope that it will be useful, but
+// WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+// General Public License for more details.
+//
+// You should have received a copy of the GNU General Public License
+// along with this program. If not, see <http://www.gnu.org/licenses/>.
+//
+// Copyright (C) 2022-2022 Fuwn <[email protected]>
+// SPDX-License-Identifier: GPL-3.0-only
+
+fn main() {
+  println!(
+    "{:?}",
+    germ::meta::Meta::from_str("text/gemini; hi=2; hi2=string=2")
+  )
+}
diff --git a/src/lib.rs b/src/lib.rs
index 0b04370..032bbf3 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -37,3 +37,6 @@ pub mod convert;
 
 #[cfg(feature = "request")]
 pub mod request;
+
+#[cfg(feature = "meta")]
+pub mod meta;
diff --git a/src/meta.rs b/src/meta.rs
new file mode 100644
index 0000000..24a29f3
--- /dev/null
+++ b/src/meta.rs
@@ -0,0 +1,52 @@
+// This file is part of Germ <https://github.com/gemrest/germ>.
+// Copyright (C) 2022-2022 Fuwn <[email protected]>
+//
+// This program is free software: you can redistribute it and/or modify
+// it under the terms of the GNU General Public License as published by
+// the Free Software Foundation, version 3.
+//
+// This program is distributed in the hope that it will be useful, but
+// WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+// General Public License for more details.
+//
+// You should have received a copy of the GNU General Public License
+// along with this program. If not, see <http://www.gnu.org/licenses/>.
+//
+// Copyright (C) 2022-2022 Fuwn <[email protected]>
+// SPDX-License-Identifier: GPL-3.0-only
+
+use std::collections::HashMap;
+
+/// Structure-ize a Gemini response's meta section into it's mime type and it's
+/// parameters.
+#[derive(Debug)]
+pub struct Meta {
+  /// The mime type of a Gemini response
+  pub mime:       String,
+  /// The parameters of a Gemini response
+  pub parameters: HashMap<String, String>,
+}
+impl Meta {
+  pub fn from_str(meta: &str) -> Self {
+    let mut metas = meta.split(";");
+    let mime = metas.next().unwrap_or("").to_string();
+    let mut parameters = HashMap::new();
+
+    for parameter in metas {
+      let key_value = parameter
+        .trim_start()
+        .split_at(parameter.find("=").unwrap_or(0));
+
+      parameters.insert(
+        key_value.0.to_string().replace("=", ""),
+        key_value.1.to_string(),
+      );
+    }
+
+    Self {
+      mime,
+      parameters,
+    }
+  }
+}
diff --git a/tests/meta.rs b/tests/meta.rs
new file mode 100644
index 0000000..b7fffa1
--- /dev/null
+++ b/tests/meta.rs
@@ -0,0 +1,50 @@
+// This file is part of Germ <https://github.com/gemrest/germ>.
+// Copyright (C) 2022-2022 Fuwn <[email protected]>
+//
+// This program is free software: you can redistribute it and/or modify
+// it under the terms of the GNU General Public License as published by
+// the Free Software Foundation, version 3.
+//
+// This program is distributed in the hope that it will be useful, but
+// WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+// General Public License for more details.
+//
+// You should have received a copy of the GNU General Public License
+// along with this program. If not, see <http://www.gnu.org/licenses/>.
+//
+// Copyright (C) 2022-2022 Fuwn <[email protected]>
+// SPDX-License-Identifier: GPL-3.0-only
+
+#[cfg(test)]
+mod test {
+  use germ::meta::Meta;
+
+  #[test]
+  fn meta_to_map_mime() {
+    assert_eq!(
+      Meta::from_str("text/gemini; hi=2; hi2=string=2").mime,
+      "text/gemini",
+    );
+  }
+
+  #[test]
+  fn meta_to_map_with_parameters() {
+    assert_eq!(
+      Meta::from_str("text/gemini; hi=2; hi2=string=2")
+        .parameters
+        .get("hi2"),
+      Some(&"string=2".to_string()),
+    );
+  }
+
+  #[test]
+  fn meta_to_map_length() {
+    assert_eq!(
+      Meta::from_str("text/gemini; hi=2; hi2=string=2")
+        .parameters
+        .len(),
+      2,
+    );
+  }
+}