lightspeed_auth 0.66.0

use crate::model::auth_account::{AuthAccountData, AuthAccountModel, AuthAccountStatus};
use crate::repository::AuthAccountRepository;
use c3p0::sqlx::*;
use c3p0::*;
use lightspeed_core::error::{ErrorCodes, LsError};

#[derive(Clone)]
pub struct MySqlAuthAccountRepository {}

impl Default for MySqlAuthAccountRepository {
    fn default() -> Self {
        Self::new()
    }
}

impl MySqlAuthAccountRepository {
    pub fn new() -> Self {
        Self {}
    }
}

impl AuthAccountRepository for MySqlAuthAccountRepository {
    type DB = MySql;

    async fn fetch_all_by_status(
        &self,
        tx: &mut MySqlConnection,
        status: AuthAccountStatus,
        start_user_id: i64,
        limit: u32,
    ) -> Result<Vec<AuthAccountModel>, LsError> {
        Ok(AuthAccountModel::query_with_tail(
            r#"
            where id >= ? and JSON_VALUE(DATA, '$.status' RETURNING CHAR(255)) = ?
            order by id asc
            limit ?
        "#,
        )
        .bind(start_user_id)
        .bind(status.as_ref())
        .bind(limit as i64)
        .fetch_all(tx)
        .await?)
    }

    async fn fetch_by_id(&self, tx: &mut MySqlConnection, user_id: i64) -> Result<AuthAccountModel, LsError> {
        Ok(tx.fetch_one_by_id::<AuthAccountData>(user_id).await?)
    }

    async fn fetch_by_username(&self, tx: &mut MySqlConnection, username: &str) -> Result<AuthAccountModel, LsError> {
        self.fetch_by_username_optional(tx, username).await?.ok_or_else(|| LsError::BadRequest {
            message: format!("No user found with username [{username}]"),
            code: ErrorCodes::NOT_FOUND,
        })
    }

    async fn fetch_by_username_optional(
        &self,
        tx: &mut MySqlConnection,
        username: &str,
    ) -> Result<Option<AuthAccountModel>, LsError> {
        Ok(AuthAccountModel::query_with_tail(
            r#"
            where JSON_VALUE(DATA, '$.username' RETURNING CHAR(255)) = ?
            limit 1
        "#,
        )
        .bind(username)
        .fetch_optional(tx)
        .await?)
    }

    async fn fetch_by_email_optional(
        &self,
        tx: &mut MySqlConnection,
        email: &str,
    ) -> Result<Option<AuthAccountModel>, LsError> {
        Ok(AuthAccountModel::query_with_tail(
            r#"
            where JSON_VALUE(DATA, '$.email' RETURNING CHAR(255)) = ?
            limit 1
        "#,
        )
        .bind(email)
        .fetch_optional(tx)
        .await?)
    }

    async fn save(
        &self,
        tx: &mut MySqlConnection,
        model: NewRecord<AuthAccountData>,
    ) -> Result<AuthAccountModel, LsError> {
        Ok(tx.save(model).await?)
    }

    async fn update(&self, tx: &mut MySqlConnection, model: AuthAccountModel) -> Result<AuthAccountModel, LsError> {
        Ok(tx.update(model).await?)
    }

    async fn delete(&self, tx: &mut MySqlConnection, model: AuthAccountModel) -> Result<AuthAccountModel, LsError> {
        Ok(tx.delete(model).await?)
    }

    async fn delete_by_id(&self, tx: &mut MySqlConnection, user_id: i64) -> Result<u64, LsError> {
        Ok(tx.delete_by_id::<AuthAccountData>(user_id).await?)
    }
}