Hook up fake authenticator backend for testing

1 files changed, 56 insertions, 6 deletions

diff --git a/server/src/fake_auth.rs b/server/src/fake_auth.rs
index 5ef9cc8..d19d858 100644
--- a/server/src/fake_auth.rs
+++ b/server/src/fake_auth.rs
@@ -1,13 +1,31 @@
-use std::collections::HashMap;
+use std::{
+    collections::HashMap,
+    env,
+    sync::{Arc, Mutex},
+};
 
 use async_trait::async_trait;
+use chrono::{DateTime, Utc};
+use openidconnect::{RefreshToken, AccessToken};
+use reqwest::Url;
+use tracing::info;
 
 use crate::{
-    auth::{AuthenticatedSession, Authenticator, SessionId},
+    auth::{AuthenticatedSession, Authenticator, SessionId, store_authenticated_session},
     error::{self, BridgeError},
 };
 
-pub struct FakeAuthenticator {}
+pub struct FakeAuthenticator {
+    login_cache: Arc<Mutex<HashMap<SessionId, Option<String>>>>,
+}
+
+impl FakeAuthenticator {
+    pub fn new() -> Self {
+        Self {
+            login_cache: Arc::new(Mutex::new(HashMap::new())),
+        }
+    }
+}
 
 #[async_trait]
 impl Authenticator for FakeAuthenticator {
@@ -15,7 +33,15 @@ impl Authenticator for FakeAuthenticator {
         &self,
         session: &mut AuthenticatedSession,
     ) -> Result<String, error::BridgeError> {
-        Err(BridgeError::NotLoggedIn)
+        let user: String = self
+            .login_cache
+            .lock()
+            .unwrap()
+            .get(&session.session_id)
+            .and_then(|u| u.as_ref())
+            .ok_or(BridgeError::NotLoggedIn)?
+            .clone();
+        Ok(user)
     }
 
     async fn authenticate(
@@ -24,10 +50,34 @@ impl Authenticator for FakeAuthenticator {
         session_id: SessionId,
         auth_params: HashMap<String, String>,
     ) -> Result<AuthenticatedSession, error::BridgeError> {
-        Err(BridgeError::Internal("not implemented".to_string()))
+        let user = auth_params
+            .get("user")
+            .ok_or(BridgeError::Internal("missing 'user' param".to_string()))?;
+        self.login_cache
+            .lock()
+            .unwrap()
+            .insert(session_id.clone(), Some(user.clone()));
+
+        let mut session = AuthenticatedSession::new(
+            user.clone(),
+            session_id,
+            Utc::now() + chrono::Duration::days(10000),
+            AccessToken::new("fake".to_string()),
+            RefreshToken::new("fake".to_string()),
+        );
+
+        store_authenticated_session(pool, &mut session).await?;
+        Ok(session)
     }
 
     async fn get_login_url(&self) -> (SessionId, reqwest::Url) {
-        todo!()
+        let auth_url =
+            Url::parse(&format!("{}/api/fake_login", env::var("APP_URL").unwrap())).unwrap();
+        let user_id = SessionId::new();
+        self.login_cache
+            .lock()
+            .unwrap()
+            .insert(user_id.clone(), None);
+        (user_id, auth_url)
     }
 }