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

follow new db function declarations

Browse Source
This commit is contained in:
Akulij 2025-04-28 22:43:02 +03:00
parent c11e671b15
commit 1c9ab867ed
1 changed files with 26 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

38
src/admin.rs
View File

@ -3,8 +3,11 @@ use teloxide::{
utils::{command::BotCommands, render::RenderMessageTextHelper}, utils::{command::BotCommands, render::RenderMessageTextHelper},
}; };
use crate::db::{CallDB, DB};
use crate::LogMsg; use crate::LogMsg;
use crate::{
db::{CallDB, DB},
BotResult,
};
// These are should not appear in /help // These are should not appear in /help
#[derive(BotCommands, Clone)] #[derive(BotCommands, Clone)]
@ -30,9 +33,15 @@ pub async fn admin_command_handler(
bot: Bot, bot: Bot,
msg: Message, msg: Message,
cmd: AdminCommands, cmd: AdminCommands,
) -> Result<(), teloxide::RequestError> { ) -> BotResult<()> {
let tguser = msg.from.clone().unwrap(); let tguser = match msg.from.clone() {
println!("MSG: {}", msg.html_text().unwrap()); Some(user) => user,
None => return Ok(()), // do nothing, cause its not usecase of function
};
println!(
"MSG: {}",
msg.html_text().unwrap_or("|EMPTY_MESSAGE|".into())
);
match cmd { match cmd {
AdminCommands::MyId => { AdminCommands::MyId => {
bot.send_message(msg.chat.id, format!("Your ID is: {}", tguser.id)) bot.send_message(msg.chat.id, format!("Your ID is: {}", tguser.id))
@ -69,22 +78,27 @@ pub async fn secret_command_handler(
msg: Message, msg: Message,
cmd: SecretCommands, cmd: SecretCommands,
admin_password: String, admin_password: String,
) -> Result<(), teloxide::RequestError> { ) -> BotResult<()> {
println!("Admin Pass: {}", admin_password); println!("Admin Pass: {}", admin_password);
let tguser = msg.from.clone().unwrap(); let tguser = match msg.from.clone() {
Some(user) => user,
None => return Ok(()), // do nothing, cause its not usecase of function
};
let user = db let user = db
.get_or_init_user(tguser.id.0 as i64, &tguser.first_name) .get_or_init_user(tguser.id.0 as i64, &tguser.first_name)
.await; .await?;
println!("MSG: {}", msg.html_text().unwrap()); println!(
"MSG: {}",
msg.html_text().unwrap_or("|EMPTY_MESSAGE|".into())
);
match cmd { match cmd {
SecretCommands::Secret { pass } => { SecretCommands::Secret { pass } => {
if user.is_admin { if user.is_admin {
bot.send_message(msg.from.unwrap().id, "You are an admin already") bot.send_message(tguser.id, "You are an admin already")
.await?; .await?;
} else if pass == admin_password { } else if pass == admin_password {
db.set_admin(user.id, true).await; db.set_admin(user.id, true).await?;
bot.send_message(msg.from.unwrap().id, "You are an admin now!") bot.send_message(tguser.id, "You are an admin now!").await?;
.await?;
} }
Ok(()) Ok(())
} }