adding url_extract to servme

src/lib.rs

@@ -3,7 +3,9 @@ mod config;
 mod requester;
 mod responder;
 mod server;
+mod url_extract;
 
 pub use requester::Requester;
 pub use responder::Responder;
 pub use server::Server;
+pub use url_extract::UrlExtract;

src/url_extract.rs

@@ -0,0 +1,61 @@
+use http::Uri;
+use std::collections::HashMap;
+
+pub struct UrlExtract {
+    params: HashMap<String, String>,
+}
+
+impl UrlExtract {
+    pub fn new(uri: &Uri) -> Self {
+        Self {
+            params: extract_params(uri),
+        }
+    }
+
+    pub fn param_str(&self, param: &str) -> Option<String> {
+        self.params.get(param).map(|param| Self::parse_param(param))
+    }
+
+    pub fn param_i64(&self, param: &str) -> Option<i64> {
+        self.params.get(param)?.parse().ok()
+    }
+
+    pub fn param_f64(&self, param: &str) -> Option<f64> {
+        self.params.get(param)?.parse().ok()
+    }
+
+    pub fn url_i64(uri: &Uri, match_path: &str) -> Option<i64> {
+        Self::extract_url(uri, match_path).first()?.parse().ok()
+    }
+
+    pub fn url_str(uri: &Uri, match_path: &str) -> Option<String> {
+        Self::extract_url(uri, match_path).first()?.parse().ok()
+    }
+
+    fn extract_url(uri: &Uri, r#match: &str) -> Vec<String> {
+        uri.path()
+            .trim_start_matches(r#match)
+            .split('/')
+            .map(|s| s.to_string())
+            .collect()
+    }
+
+    fn parse_param(param: &str) -> String {
+        param.replace("%20", " ")
+    }
+}
+
+pub fn extract_params(uri: &Uri) -> HashMap<String, String> {
+    uri.query()
+        .map(|q| {
+            q.split('&')
+                .map(|pair| {
+                    let mut split = pair.splitn(2, '=');
+                    let key = split.next().unwrap_or_default().to_string();
+                    let value = split.next().unwrap_or_default().to_string();
+                    (key, value)
+                })
+                .collect()
+        })
+        .unwrap_or_default()
+}