use crate::errors::NoUserInfo;
use crate::markups::deletion_markup;
use entity::prelude::*;
use sea_orm::DatabaseConnection;
use std::sync::Arc;
use teloxide::{adaptors::Throttle, dispatching::DpHandlerDescription, prelude::*};

/// 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
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))),
    }
}

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")
        .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)
}