mini_admin_bot/src/lib.rs

use chrono::Local;
use std::sync::Arc;
use teloxide::{prelude::*, utils::command::BotCommands};
pub mod ban_stiker;
pub mod check_permissions;
mod database;
mod filter_files;
mod pole_dialogue;
pub mod rewrite_links;
mod spoiler_mangas;
mod telegram_utils;

#[derive(BotCommands, Clone)]
#[command(
    rename_rule = "lowercase",
    description = "These commands are supported:"
)]
enum Command {
    #[command(description = "ban a stiker")]
    Torquemada,
    #[command(description = "display this text.")]
    Help,
    #[command(description = "list pole points")]
    Top,
    #[command(description = "get the server time")]
    Time,
    #[command(description = "get the bot version")]
    Version,
}

/*#[tokio::main]
async fn main() {
    run().await;
}*/

pub async fn run() {
    //teloxide::enable_logging!();
    pretty_env_logger::init();
    log::info!("Starting bot");
    let bot = Bot::from_env();
    let permissions = Arc::new(check_permissions::GroupPermissions::new());
    let p1 = Arc::clone(&permissions);
    let p2 = Arc::clone(&permissions);
    let handler = Update::filter_message()
        .branch(
            dptree::filter(move |msg: Message| !p1.compare(&msg.chat.id.to_string())).endpoint(
                |msg: Message, bot: Bot| async move {
                    if !msg.chat.is_private() {
                        println!("{}", msg.chat.id.0);
                        bot.leave_chat(msg.chat.id).await?;
                    }
                    Ok(())
                },
            ),
        )
        .branch(
            dptree::filter(|msg: Message| is_channel_user(msg)).endpoint(
                |msg: Message, bot: Bot| async move {
                    bot.delete_message(msg.chat.id, msg.id).await?;
                    Ok(())
                },
            ),
        )
        .branch(
            Update::filter_message()
                .filter_command::<Command>()
                .endpoint(|msg: Message, bot: Bot, command: Command| answer(bot, msg, command)),
        )
        .branch(
            dptree::filter(|msg: Message| is_media(msg.clone()) && ban_stiker::check_media(msg))
                .endpoint(|msg: Message, bot: Bot| async move {
                    bot.delete_message(msg.chat.id, msg.id).await?;
                    Ok(())
                }),
        )
        .branch(
            dptree::filter(move |msg: Message| {
                is_photo(msg.clone()) && p2.compar_manga_filtered(&msg.chat.id.to_string())
            })
            .endpoint(|msg: Message, bot: Bot| spoiler_mangas::check_image(msg, bot)),
        )
        .branch(
            dptree::filter(move |msg: Message| filter_files::analyce_name::check_file(msg.clone()))
                .endpoint(|msg: Message, bot: Bot| filter_files::action::take_actions(msg, bot)),
        )
        .branch(
            dptree::filter(move |msg: Message| {
                rewrite_links::check_contain_links::contain_links(msg.clone())
            })
            .endpoint(|msg: Message, bot: Bot| {
                rewrite_links::check_contain_links::fix_links(msg, bot)
            }),
        )
        .branch(
            dptree::filter(move |msg: Message| permissions.compar_party(&msg.chat.id.to_string()))
                .endpoint(|msg: Message, bot: Bot| pole_dialogue::exe_pole(msg, bot)),
        );
    Dispatcher::builder(bot, handler).build().dispatch().await;
}

async fn answer(bot: Bot, msg: Message, command: Command) -> anyhow::Result<()> {
    match command {
        Command::Torquemada => match ban_stiker::ban_media(msg.clone(), bot.clone()).await {
            Ok(_o) => {
                bot.send_message(msg.chat.id, "Otro fichero que se va a la hoguera")
                    .await?
            }
            Err(e) => bot.send_message(msg.chat.id, e.to_string()).await?,
        },
        Command::Help => {
            bot.send_message(msg.chat.id, Command::descriptions().to_string())
                .await?
        }
        Command::Top => pole_dialogue::get_top(msg, bot).await?,
        Command::Time => {
            bot.send_message(
                msg.chat.id,
                Local::now().format("%Y-%m-%d:%H:%M:%S").to_string(),
            )
            .await?
        }
        Command::Version => {
            bot.send_message(msg.chat.id, env!("CARGO_PKG_VERSION"))
                .await?
        }
    };

    Ok(())
}

fn is_media(msg: Message) -> bool {
    is_stiker(msg.clone()) || is_gif(msg.clone()) || is_photo(msg.clone())
}

fn is_stiker(msg: Message) -> bool {
    match msg.sticker() {
        Some(_) => true,
        None => false,
    }
}

fn is_gif(msg: Message) -> bool {
    match msg.animation() {
        Some(_) => true,
        None => false,
    }
}

fn is_photo(msg: Message) -> bool {
    match msg.photo() {
        Some(_) => true,
        None => false,
    }
}

fn is_channel_user(msg: Message) -> bool {
    match msg.from {
        Some(u) => u.is_channel(),
        None => false,
    }
}