From e2e42c8676d86063915f3e5dfd32c1c12a1a2b49 Mon Sep 17 00:00:00 2001 From: Akulij Date: Thu, 3 Apr 2025 04:37:44 +0900 Subject: [PATCH] rename make_admin function into set_admin with flag --- src/db.rs | 4 ++-- src/main.rs | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/src/db.rs b/src/db.rs index 486bbf3..3087891 100644 --- a/src/db.rs +++ b/src/db.rs @@ -37,14 +37,14 @@ impl DB { users.filter(id.gt(0)).load::(&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(); } diff --git a/src/main.rs b/src/main.rs index 131b4cf..4a696a4 100644 --- a/src/main.rs +++ b/src/main.rs @@ -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(())