rename make_admin function into set_admin with flag

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

View File

@ -37,14 +37,14 @@ impl DB {
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::*;
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)).load(connection).await.unwrap();
diesel::update(users)
.filter(id.eq(userid))
.set(is_admin.eq(true))
.set(is_admin.eq(isadmin))
.execute(connection).await.unwrap();
}

View File

@ -137,7 +137,7 @@ async fn secret_command_handler(
if user.is_admin == true {
bot.send_message(msg.from.unwrap().id, "You are an admin already").await?;
} 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?;
}
Ok(())