save

src/webutils/auth.rs

@@ -1,13 +1,19 @@
+use actix_session::Session;
+use actix_web::Error;
 use actix_web_httpauth::extractors::basic::BasicAuth;
-use std::borrow::Cow;
 
 pub trait AuthValidator {
-        fn check_user(&self, name : &String, pwd : &String) ->  bool;
+        fn check_user(&self, name : &String, pwd : &String) -> Option<User>;
 
-        fn check_basic(&self, basic : BasicAuth) -> bool {
-                match basic.password() {
-                        None => {false}
-                        Some(pwd) => {self.check_user(&basic.user_id().to_string(), &pwd.to_string())}
+        fn check_basic(&self, basic : &BasicAuth) -> Option<User> {
+                basic.password().and_then(|pwd| self.check_user(&basic.user_id().to_string(), &pwd.to_string()))
+        }
+
+        fn check_session(&self, session : &Session) -> Option<User> {
+                let result = session.get::<String>("user");
+                match result {
+                        Ok(username) => {username.map(|u| User(u))}
+                        Err(e) => {None}
                 }
         }
 }
@@ -15,7 +21,13 @@ pub trait AuthValidator {
 pub struct TestValidator;
 
 impl AuthValidator for TestValidator {
-        fn check_user(&self, name: &String, pwd: &String) -> bool {
-                pwd.eq(&(name.clone() + "pwd")) //stub!
+        fn check_user(&self, name: &String, pwd: &String) -> Option<User> {
+                if pwd.eq(&(name.clone() + "pwd")) {
+                        Some(User(name.clone()))
+                } else {
+                        None
+                }
         }
-}
+}
+
+pub struct User(String);

src/webutils/mod.rs

@@ -1 +1,2 @@
-pub mod auth;
+pub mod auth;
+pub mod user;