Simplified Handler::new
This commit is contained in:
		@@ -7,14 +7,13 @@ async fn get_master_pass(
 | 
			
		||||
    msg: Message,
 | 
			
		||||
    db: DatabaseConnection,
 | 
			
		||||
    dialogue: MainDialogue,
 | 
			
		||||
    master_password: String,
 | 
			
		||||
    master_pass: String,
 | 
			
		||||
) -> crate::Result<()> {
 | 
			
		||||
    dialogue.exit().await?;
 | 
			
		||||
    let user_id = msg.from().ok_or(NoUserInfo)?.id.0;
 | 
			
		||||
    let model = spawn_blocking(move || {
 | 
			
		||||
        master_pass::ActiveModel::from_unencrypted(user_id, &master_password)
 | 
			
		||||
    })
 | 
			
		||||
    .await?;
 | 
			
		||||
    let model =
 | 
			
		||||
        spawn_blocking(move || master_pass::ActiveModel::from_unencrypted(user_id, &master_pass))
 | 
			
		||||
            .await?;
 | 
			
		||||
    model.insert(&db).await?;
 | 
			
		||||
    bot.send_message(msg.chat.id, "Success")
 | 
			
		||||
        .reply_markup(deletion_markup())
 | 
			
		||||
@@ -41,7 +40,9 @@ pub async fn set_master_pass(
 | 
			
		||||
        .await?;
 | 
			
		||||
    dialogue
 | 
			
		||||
        .update(State::GetNewMasterPass(Handler::new(
 | 
			
		||||
            get_master_pass,
 | 
			
		||||
            |bot, msg, db, dialogue, master_pass| {
 | 
			
		||||
                Box::pin(get_master_pass(bot, msg, db, dialogue, master_pass))
 | 
			
		||||
            },
 | 
			
		||||
            &previous,
 | 
			
		||||
        )))
 | 
			
		||||
        .await?;
 | 
			
		||||
 
 | 
			
		||||
@@ -12,7 +12,7 @@ macro_rules! handler {
 | 
			
		||||
            let previous = bot.send_message(msg.chat.id, $message).await?;
 | 
			
		||||
            dialogue
 | 
			
		||||
                .update($next_state(Handler::new(
 | 
			
		||||
                    move |bot, msg, db, dialogue, param| $next_func(bot, msg, db, dialogue, $($param,)* param),
 | 
			
		||||
                    move |bot, msg, db, dialogue, param| Box::pin($next_func(bot, msg, db, dialogue, $($param,)* param)),
 | 
			
		||||
                    &previous,
 | 
			
		||||
                )))
 | 
			
		||||
                .await?;
 | 
			
		||||
@@ -45,7 +45,7 @@ macro_rules! ask_name_handler {
 | 
			
		||||
                .await?;
 | 
			
		||||
            dialogue
 | 
			
		||||
                .update(State::GetExistingName(Handler::new(
 | 
			
		||||
                    $next_func,
 | 
			
		||||
                    move |bot, msg, db, dialogue, param| Box::pin($next_func(bot, msg, db, dialogue, $($param,)* param)),
 | 
			
		||||
                    &previous,
 | 
			
		||||
                )))
 | 
			
		||||
                .await?;
 | 
			
		||||
 
 | 
			
		||||
@@ -1,6 +1,6 @@
 | 
			
		||||
use crate::prelude::*;
 | 
			
		||||
use futures::future::BoxFuture;
 | 
			
		||||
use std::{future::Future, sync::Arc};
 | 
			
		||||
use std::sync::Arc;
 | 
			
		||||
use teloxide::types::MessageId;
 | 
			
		||||
use tokio::sync::Mutex;
 | 
			
		||||
 | 
			
		||||
@@ -43,17 +43,20 @@ pub type PackagedHandler<T> = Arc<Mutex<Handler<T>>>;
 | 
			
		||||
impl<T> Handler<T> {
 | 
			
		||||
    /// Convinience method to convert a simple async function and a previous message into PackagedHandler
 | 
			
		||||
    #[inline]
 | 
			
		||||
    pub fn new<H, F>(f: H, previous: impl Into<MessageIds>) -> PackagedHandler<T>
 | 
			
		||||
    pub fn new<H>(f: H, previous: impl Into<MessageIds>) -> PackagedHandler<T>
 | 
			
		||||
    where
 | 
			
		||||
        H: FnOnce(Throttle<Bot>, Message, DatabaseConnection, MainDialogue, T) -> F
 | 
			
		||||
        H: FnOnce(
 | 
			
		||||
                Throttle<Bot>,
 | 
			
		||||
                Message,
 | 
			
		||||
                DatabaseConnection,
 | 
			
		||||
                MainDialogue,
 | 
			
		||||
                T,
 | 
			
		||||
            ) -> BoxFuture<'static, crate::Result<()>>
 | 
			
		||||
            + Send
 | 
			
		||||
            + 'static,
 | 
			
		||||
        F: Future<Output = crate::Result<()>> + Send + 'static,
 | 
			
		||||
    {
 | 
			
		||||
        let handler = Self {
 | 
			
		||||
            func: Some(Box::new(|bot, msg, db, dialogue, val| {
 | 
			
		||||
                Box::pin(f(bot, msg, db, dialogue, val))
 | 
			
		||||
            })),
 | 
			
		||||
            func: Some(Box::new(f)),
 | 
			
		||||
            previous: previous.into(),
 | 
			
		||||
        };
 | 
			
		||||
        Arc::new(Mutex::new(handler))
 | 
			
		||||
 
 | 
			
		||||
		Reference in New Issue
	
	Block a user