use crate::prelude::*;
use std::sync::Arc;
use teloxide::{dispatching::DpHandlerDescription, dptree::Handler};

/// A wierd filter that checks for the existance of a master password.
///
/// # Returns
///
/// Returns None if account exists, Some(None) if there's an account and Some(Some(String)) if an error occures.
/// The String represents the error that occured
#[inline]
async fn master_pass_exists(
    msg: Message,
    db: DatabaseConnection,
) -> Option<Option<Arc<dyn std::error::Error + Send + Sync>>> {
    let user_id = match msg.from() {
        Some(user) => user.id.0,
        None => return Some(Some(Arc::new(NoUserInfo))),
    };
    match MasterPass::exists(user_id, &db).await {
        Ok(true) => None,
        Ok(false) => Some(None),
        Err(err) => Some(Some(Arc::new(err))),
    }
}

#[inline]
async fn notify_about_no_master_pass(
    bot: Throttle<Bot>,
    result: Option<Arc<dyn std::error::Error + Send + Sync>>,
    msg: Message,
) -> crate::Result<()> {
    if let Some(error) = result {
        return Err(error.into());
    }
    bot.send_message(
        msg.chat.id,
        "No master password set. Use /set_master_pass to set it",
    )
    .reply_markup(deletion_markup())
    .await?;
    Ok(())
}

/// Gets a handler that filters out the messages of users that don't have a master password set
#[inline]
pub fn get_handler() -> Handler<'static, DependencyMap, crate::Result<()>, DpHandlerDescription> {
    dptree::filter_map_async(master_pass_exists).endpoint(notify_about_no_master_pass)
}