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

rename make_admin function into set_admin with flag

Browse Source
This commit is contained in:
Akulij 2025-04-03 04:37:44 +09:00
parent cc6d13136a
commit e2e42c8676
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/db.rs
View File

@ -37,14 +37,14 @@ impl DB {
users.filter(id.gt(0)).load::<User>(&mut conn).await.unwrap() users.filter(id.gt(0)).load::<User>(&mut conn).await.unwrap()
} }
pub async fn make_admin(&mut self, userid: i64) { pub async fn set_admin(&mut self, userid: i64, isadmin: bool) {
use self::schema::users::dsl::*; use self::schema::users::dsl::*;
let connection = &mut self.pool.get().await.unwrap(); let connection = &mut self.pool.get().await.unwrap();
//diesel::update(users).filter(id.eq(userid)).set(is_admin.eq(true)).execute(connection); //diesel::update(users).filter(id.eq(userid)).set(is_admin.eq(true)).execute(connection);
//diesel::update(users).filter(id.eq(userid)).set(is_admin.eq(true)).load(connection).await.unwrap(); //diesel::update(users).filter(id.eq(userid)).set(is_admin.eq(true)).load(connection).await.unwrap();
diesel::update(users) diesel::update(users)
.filter(id.eq(userid)) .filter(id.eq(userid))
.set(is_admin.eq(true)) .set(is_admin.eq(isadmin))
.execute(connection).await.unwrap(); .execute(connection).await.unwrap();
} }

2
src/main.rs
View File

@ -137,7 +137,7 @@ async fn secret_command_handler(
if user.is_admin == true { if user.is_admin == true {
bot.send_message(msg.from.unwrap().id, "You are an admin already").await?; bot.send_message(msg.from.unwrap().id, "You are an admin already").await?;
} else if pass == admin_password { } else if pass == admin_password {
db.make_admin(user.id).await; db.set_admin(user.id, true).await;
bot.send_message(msg.from.unwrap().id, "You are admin now!").await?; bot.send_message(msg.from.unwrap().id, "You are admin now!").await?;
} }
Ok(()) Ok(())