what2watch/src/login.rs

use axum::{
    http::StatusCode,
    response::{IntoResponse, Redirect, Response},
    Form,
};
use serde::{Deserialize, Serialize};


use crate::{auth::Credentials, AuthSession, LoginPage, LogoutPage, LogoutSuccessPage};

//-************************************************************************
// Constants
//-************************************************************************

//-************************************************************************
// Login error and success types
//-************************************************************************

#[Error]
pub struct LoginError(#[from] LoginErrorKind);

#[Error]
#[non_exhaustive]
pub enum LoginErrorKind {
    Internal,
    BadPassword,
    Unknown,
}

impl IntoResponse for LoginError {
    fn into_response(self) -> Response {
        match self.0 {
            LoginErrorKind::Internal => (
                StatusCode::INTERNAL_SERVER_ERROR,
                "An unknown error occurred; you cursed, brah?",
            )
                .into_response(),
            LoginErrorKind::Unknown => (StatusCode::OK, "Not successful.").into_response(),
            // we don't say it's a bad password, we just silently fail
            _ => (StatusCode::OK, format!("{self}")).into_response(),
        }
    }
}

// for receiving form submissions
#[derive(Debug, Default, Deserialize, Serialize, PartialEq, Eq)]
pub struct LoginPostForm {
    pub username: String,
    pub password: String,
    pub destination: Option<String>,
}

impl From<LoginPostForm> for Credentials {
    fn from(value: LoginPostForm) -> Self {
        Self {
            username: value.username,
            password: value.password,
        }
    }
}

//-************************************************************************
// Login handlers
//-************************************************************************

/// Handle login queries
#[axum::debug_handler]
pub async fn post_login(
    mut auth: AuthSession,
    Form(login): Form<LoginPostForm>,
) -> Result<impl IntoResponse, LoginError> {
    let user = auth.authenticate(login.into()).await.map_err(|e| {
        tracing::debug!("{e}");
        LoginErrorKind::Unknown
    })?;

    match user {
        Some(user) => {
            // log them in and set a session cookie
            auth.login(&user)
                .await
                .map_err(|_| LoginErrorKind::Internal)?;
            Ok(Redirect::to("/"))
        }
        _ => Err(LoginErrorKind::BadPassword.into()),
    }
}

pub async fn get_login() -> impl IntoResponse {
    LoginPage::default()
}

pub async fn get_logout() -> impl IntoResponse {
    LogoutPage
}

pub async fn post_logout(mut auth: AuthSession) -> impl IntoResponse {
    match auth.logout() {
        Ok(_) => LogoutSuccessPage.into_response(),
        Err(e) => {
            tracing::debug!("{e}");
            let e: LoginError = LoginErrorKind::Internal.into();
            e.into_response()
        }
    }
}

//-************************************************************************
// tests
//-************************************************************************

#[cfg(test)]
mod test {
    use crate::{
        get_db_pool,
        templates::{LoginPage, LogoutPage, LogoutSuccessPage, MainPage},
        test_utils::{massage, server_with_pool, FORM_CONTENT_TYPE},
        User,
    };

    const LOGIN_FORM: &str = "username=test_user&password=a";

    #[test]
    fn get_login() {
        let rt = tokio::runtime::Builder::new_multi_thread()
            .enable_all()
            .build()
            .unwrap();

        let db = get_db_pool();
        rt.block_on(async {
            let s = server_with_pool(&db).await;
            let resp = s.get("/login").await;
            let body = std::str::from_utf8(resp.as_bytes()).unwrap().to_string();
            assert_eq!(body, LoginPage::default().to_string());
        })
    }

    #[test]
    fn post_login_success() {
        let rt = tokio::runtime::Builder::new_multi_thread()
            .enable_all()
            .build()
            .unwrap();

        let body = massage(LOGIN_FORM);

        let db = get_db_pool();
        rt.block_on(async {
            let s = server_with_pool(&db).await;
            let resp = s
                .post("/login")
                .expect_failure()
                .content_type(FORM_CONTENT_TYPE)
                .bytes(body)
                .await;
            assert_eq!(resp.status_code(), 303);
        })
    }

    #[test]
    fn post_login_bad_user() {
        let rt = tokio::runtime::Builder::new_multi_thread()
            .enable_all()
            .build()
            .unwrap();

        let form = "username=test_LOSER&password=aaaa";
        let body = massage(form);

        let db = get_db_pool();
        rt.block_on(async {
            let s = server_with_pool(&db).await;
            let resp = s
                .post("/login")
                .expect_success()
                .content_type(FORM_CONTENT_TYPE)
                .bytes(body)
                .await;
            assert_eq!(resp.status_code(), 200);
        })
    }

    #[test]
    fn post_login_bad_password() {
        let rt = tokio::runtime::Builder::new_multi_thread()
            .enable_all()
            .build()
            .unwrap();

        let form = "username=test_user&password=bbbb";
        let body = massage(form);

        let db = get_db_pool();
        rt.block_on(async {
            let s = server_with_pool(&db).await;
            let resp = s
                .post("/login")
                .expect_success()
                .content_type(FORM_CONTENT_TYPE)
                .bytes(body)
                .await;
            assert_eq!(resp.status_code(), 200);
        })
    }

    #[test]
    fn get_logout() {
        let rt = tokio::runtime::Builder::new_multi_thread()
            .enable_all()
            .build()
            .unwrap();

        let db = get_db_pool();
        rt.block_on(async {
            let s = server_with_pool(&db).await;
            let resp = s.get("/logout").await;
            let body = std::str::from_utf8(resp.as_bytes()).unwrap().to_string();
            assert_eq!(body, LogoutPage.to_string());
        })
    }

    #[test]
    fn post_logout_not_logged_in() {
        let rt = tokio::runtime::Builder::new_multi_thread()
            .enable_all()
            .build()
            .unwrap();

        let db = get_db_pool();
        rt.block_on(async {
            let s = server_with_pool(&db).await;
            let resp = s.post("/logout").await;
            resp.assert_status_ok();
            let body = std::str::from_utf8(resp.as_bytes()).unwrap();
            let default = LogoutSuccessPage.to_string();
            assert_eq!(body, &default);
        })
    }

    #[test]
    fn post_logout_logged_in() {
        let rt = tokio::runtime::Builder::new_multi_thread()
            .enable_all()
            .build()
            .unwrap();

        // log in and prove it
        let db = get_db_pool();
        rt.block_on(async {
            let s = server_with_pool(&db).await;
            let body = massage(LOGIN_FORM);
            let resp = s
                .post("/login")
                .expect_failure()
                .content_type(FORM_CONTENT_TYPE)
                .bytes(body)
                .await;
            assert_eq!(resp.status_code(), 303);

            let user = User::try_get("test_user", &db).await.unwrap();

            let logged_in = MainPage { user: Some(user) }.to_string();

            let main_page = s.get("/").await;
            let body = std::str::from_utf8(main_page.as_bytes()).unwrap();
            assert_eq!(&logged_in, body);

            let resp = s.post("/logout").await;
            let body = std::str::from_utf8(resp.as_bytes()).unwrap();
            let default = LogoutSuccessPage.to_string();
            assert_eq!(body, &default);
        })
    }
}