tenebrous-dicebot/src/commands/management.rs

use super::{Command, Execution, ExecutionResult};
use crate::context::Context;
use crate::db::Users;
use crate::error::BotError::{AccountDoesNotExist, AuthenticationError, PasswordCreationError};
use crate::logic::hash_password;
use crate::models::{AccountStatus, User};
use async_trait::async_trait;

pub struct RegisterCommand(pub String);

#[async_trait]
impl Command for RegisterCommand {
    fn name(&self) -> &'static str {
        "register user account"
    }

    fn is_secure(&self) -> bool {
        true
    }

    async fn execute(&self, ctx: &Context<'_>) -> ExecutionResult {
        let pw_hash = hash_password(&self.0).map_err(|e| PasswordCreationError(e))?;
        let user = User {
            username: ctx.username.to_owned(),
            password: Some(pw_hash),
            account_status: AccountStatus::Registered,
            ..Default::default()
        };

        ctx.db.upsert_user(&user).await?;
        Execution::success(format!(
            "User account registered/updated. Please log in to external applications \
             with username {} and the password you set.",
            ctx.username
        ))
    }
}

pub struct CheckCommand(pub String);

#[async_trait]
impl Command for CheckCommand {
    fn name(&self) -> &'static str {
        "check user password"
    }

    fn is_secure(&self) -> bool {
        true
    }

    async fn execute(&self, ctx: &Context<'_>) -> ExecutionResult {
        let user = ctx.db.authenticate_user(&ctx.username, &self.0).await?;

        match user {
            Some(_) => Execution::success("Password is correct!".to_string()),
            None => Err(AuthenticationError.into()),
        }
    }
}

pub struct UnregisterCommand;

#[async_trait]
impl Command for UnregisterCommand {
    fn name(&self) -> &'static str {
        "unregister user account"
    }

    fn is_secure(&self) -> bool {
        true
    }

    async fn execute(&self, ctx: &Context<'_>) -> ExecutionResult {
        let user = ctx.db.get_user(&ctx.username).await?;

        match user {
            Some(_) => {
                ctx.db.delete_user(&ctx.username).await?;
                Execution::success("Your user account has been removed.".to_string())
            }
            None => Err(AccountDoesNotExist.into()),
        }
    }
}