Implement Authenthication

src/main.rs

@@ -57,6 +57,21 @@ fn main() -> Result<(), io::Error> {
         //       request.url(),
         //       request.headers()
         //   );
+        match config.get("auth") { 
+            Some(toml::Value::String(server_auth)) => {
+                match get_authkey(request.url().clone()) {
+                    Ok(client_auth) => {
+                        if client_auth != server_auth.clone() {
+                            println!("invalid auth:{}, server auth: {}", client_auth, server_auth);
+                            request.respond(Response::from_string(""));
+                            continue;
+                        }
+                    },
+                    _ => {}
+                }
+            },
+            _ => {}
+        }
         let query = get_query(request.url());
         //println!("got query:{}", query.clone().unwrap());
         match query {
@@ -95,6 +110,18 @@ fn get_query(request_url: &str) -> Result<String, url::ParseError> {
     }
     Ok(String::from_str("").unwrap())
 }
+fn get_authkey(request_url: &str) -> Result<String, url::ParseError> {
+    let parsed_url = request_url.split_once('?').map_or(request_url, |(_, after)| after);
+    //println!("parsed_url:{}", parsed_url);
+    let query_pairs = url::form_urlencoded::parse(parsed_url.as_bytes());
+    for (key, value) in query_pairs {
+        //println!("key:{}, value: {}", key, value);
+        if key == "auth" {
+            return Ok(value.into_owned());
+        }
+    }
+    Ok(String::from_str("").unwrap())
+}
 
 fn build_markov_chain(file_path: &str) -> Result<MarkovChain, io::Error> {
     let file = File::open(file_path)?;