akulij/gongbotrs
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

move out admin commands to admin file

Browse Source
This commit is contained in:
Akulij 2025-04-03 04:53:16 +09:00
parent 87601f172a
commit 9e95935ddc
2 changed files with 47 additions and 40 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

44
src/admin.rs Normal file
View File

@ -0,0 +1,44 @@
use teloxide::{dispatching::dialogue::GetChatId, payloads::SendMessageSetters, prelude::*, types::InputFile, utils::{command::BotCommands, render::RenderMessageTextHelper}};
use crate::db::DB;
use crate::LogMsg;
#[derive(BotCommands, Clone)]
#[command(rename_rule = "lowercase")]
pub enum AdminCommands {
/// Shows your ID.
MyId,
/// Pin replied message
Pin,
/// Removes your admin privileges
Deop,
}
pub async fn admin_command_handler(
mut db: DB,
bot: Bot,
msg: Message,
cmd: AdminCommands,
) -> Result<(), teloxide::RequestError> {
let tguser = msg.from.clone().unwrap();
println!("MSG: {}", msg.html_text().unwrap());
match cmd {
AdminCommands::MyId => {
bot.send_message(msg.chat.id, format!("Your ID is: {}", tguser.id)).log().await?;
Ok(())
},
AdminCommands::Pin => {
if let Some(msg_to_pin) = msg.reply_to_message() {
bot.pin_chat_message(msg.chat.id, msg_to_pin.id).await?;
} else {
bot.send_message(msg.chat.id, "you need to reply to some message with this command").log().await?;
}
Ok(())
},
AdminCommands::Deop => {
db.set_admin(tguser.id.0 as i64, false).await;
bot.send_message(msg.chat.id, "You are not an admin anymore").await?;
Ok(())
}
}
}

43
src/main.rs
View File

@ -1,5 +1,8 @@
pub mod db;
pub mod admin;
use crate::db::DB;
use crate::admin::{AdminCommands, admin_command_handler};
use teloxide::{dispatching::dialogue::GetChatId, payloads::SendMessageSetters, prelude::*, types::InputFile, utils::{command::BotCommands, render::RenderMessageTextHelper}};
use envconfig::Envconfig;
@ -33,17 +36,6 @@ enum SecretCommands {
},
}
#[derive(BotCommands, Clone)]
#[command(rename_rule = "lowercase")]
enum AdminCommands {
/// Shows your ID.
MyId,
/// Pin replied message
Pin,
/// Removes your admin privileges
Deop,
}
trait LogMsg {
fn log(self) -> Self;
}
@ -147,35 +139,6 @@ async fn secret_command_handler(
}
}
async fn admin_command_handler(
mut db: DB,
bot: Bot,
msg: Message,
cmd: AdminCommands,
) -> Result<(), teloxide::RequestError> {
let tguser = msg.from.clone().unwrap();
println!("MSG: {}", msg.html_text().unwrap());
match cmd {
AdminCommands::MyId => {
bot.send_message(msg.chat.id, format!("Your ID is: {}", tguser.id)).log().await?;
Ok(())
},
AdminCommands::Pin => {
if let Some(msg_to_pin) = msg.reply_to_message() {
bot.pin_chat_message(msg.chat.id, msg_to_pin.id).await?;
} else {
bot.send_message(msg.chat.id, "you need to reply to some message with this command").log().await?;
}
Ok(())
},
AdminCommands::Deop => {
db.set_admin(tguser.id.0 as i64, false).await;
bot.send_message(msg.chat.id, "You are not an admin anymore").await?;
Ok(())
}
}
}
async fn echo(
bot: Bot,
msg: Message,