diff options
| author | Fuwn <[email protected]> | 2022-06-28 08:03:40 +0000 |
|---|---|---|
| committer | Fuwn <[email protected]> | 2022-06-28 08:03:40 +0000 |
| commit | 50d118840448614b6309621faccc595a817782d4 (patch) | |
| tree | 1ad05b1d6c0f81ff6924e0dfdac8a51b3d2f4af1 /germ/src/meta.rs | |
| parent | fix(germ): don't default macros (diff) | |
| download | archived-germ-0.3.3.tar.xz archived-germ-0.3.3.zip | |
fix(crate): crate locations for readmev0.3.3
Diffstat (limited to 'germ/src/meta.rs')
| -rw-r--r-- | germ/src/meta.rs | 162 |
1 files changed, 162 insertions, 0 deletions
diff --git a/germ/src/meta.rs b/germ/src/meta.rs new file mode 100644 index 0000000..f8a9dfb --- /dev/null +++ b/germ/src/meta.rs @@ -0,0 +1,162 @@ +// 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, Default)] +pub struct Meta { + /// The mime type of a Gemini response + mime: String, + /// The parameters of a Gemini response + parameters: HashMap<String, String>, +} +impl ToString for Meta { + /// Convert a `Meta` into a `String` + /// + /// # Example + /// ```rust + /// let original_string = "text/gemini; hi=2; hi2=string=2"; + /// + /// assert_eq!( + /// germ::meta::Meta::from_string(original_string).to_string(), + /// original_string + /// ); + /// ``` + fn to_string(&self) -> String { + format!("{}{}", self.mime, { + let mut parameters = self + .parameters + .iter() + .map(|(k, v)| format!("{}={}", *k, v)) + .collect::<Vec<_>>(); + + parameters.sort(); + parameters.reverse(); + + if parameters.is_empty() { + "".to_string() + } else { + format!("; {}", parameters.join("; ")) + } + }) + } +} +impl Meta { + /// Create a new `Meta` + /// + /// # Example + /// + /// ```rust + /// let mut meta = germ::meta::Meta::new(); + /// ``` + #[must_use] + pub fn new() -> Self { Self::default() } + + /// Create a `Meta` from a string + /// + /// # Example + /// + /// ```rust + /// assert_eq!( + /// germ::meta::Meta::from_string("text/gemini; hi=2; hi2=string=2").mime(), + /// "text/gemini", + /// ); + /// ``` + #[must_use] + pub fn from_string(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, + } + } + + /// Obtain non-mutable access to the mime of the `Meta` + /// + /// # Example + /// + /// ```rust + /// assert_eq!( + /// germ::meta::Meta::from_string("text/gemini; hi=2; hi2=string=2").mime(), + /// "text/gemini", + /// ); + /// ``` + #[must_use] + pub fn mime(&self) -> &str { &self.mime } + + /// Obtain mutable access to the mime of the `Meta` + /// + /// # Example + /// + /// ```rust + /// let mut meta = germ::meta::Meta::new(); + /// + /// *meta.mime_mut() = "text/gemini".to_string(); + /// ``` + pub fn mime_mut(&mut self) -> &mut String { &mut self.mime } + + /// Obtain non-mutable access to the parameters of the `Meta` + /// + /// # Example + /// + /// ```rust + /// assert_eq!( + /// germ::meta::Meta::from_string("text/gemini; hi=2; hi2=string=2") + /// .parameters() + /// .get("hi2"), + /// Some(&"string=2".to_string()), + /// ); + /// ``` + #[must_use] + pub const fn parameters(&self) -> &HashMap<String, String> { + &self.parameters + } + + /// Obtain mutable access to the parameters of the `Meta` + /// + /// # Example + /// + /// ```rust + /// let mut meta = germ::meta::Meta::new(); + /// let mut parameters = std::collections::HashMap::new(); + /// + /// parameters.insert("hi".to_string(), "2".to_string()); + /// parameters.insert("hi2".to_string(), "string=2".to_string()); + /// + /// *meta.parameters_mut() = parameters; + /// ``` + pub fn parameters_mut(&mut self) -> &mut HashMap<String, String> { + &mut self.parameters + } +} |