what2watch/src/login.rs

use argon2::{
    password_hash::{PasswordHash, PasswordVerifier},
    Argon2,
};
use axum::{
    extract::State,
    http::StatusCode,
    response::{IntoResponse, Redirect, Response},
    Form,
};
use serde::{Deserialize, Serialize};
use sqlx::SqlitePool;

use crate::{AuthContext, LoginPage, LogoutPage, LogoutSuccessPage, User};

//-************************************************************************
// 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(),
            _ => (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,
}

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

/// Handle login queries
#[axum::debug_handler]
pub async fn post_login(
    mut auth: AuthContext,
    State(pool): State<SqlitePool>,
    Form(login): Form<LoginPostForm>,
) -> Result<impl IntoResponse, LoginError> {
    let username = &login.username;
    let username = username.trim();

    let pw = &login.password;
    let pw = pw.trim();

    let user = User::try_get(username, &pool).await.map_err(|e| {
        tracing::debug!("{e}");
        LoginErrorKind::Unknown
    })?;

    let verifier = Argon2::default();
    let hash = PasswordHash::new(&user.pwhash).map_err(|_| LoginErrorKind::Internal)?;
    match verifier.verify_password(pw.as_bytes(), &hash) {
        Ok(_) => {
            // 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: AuthContext) -> impl IntoResponse {
    if auth.current_user.is_some() {
        auth.logout().await;
    }
    LogoutSuccessPage
}

//-************************************************************************
// 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.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.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.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.bytes()).unwrap();
            assert_eq!(&logged_in, body);

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