raphael/tty
0
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

feat(database): adding the anyhow result return for bot function

Browse source
This commit is contained in:
Raphael 2026-02-25 21:54:08 +01:00
parent b82199dab1
commit aa499ead1c
No known key found for this signature in database
1 changed files with 6 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

11
src/database/bot.rs
View file

@ -1,16 +1,17 @@
use sqlx::{PgPool, query, query_as};
use crate::models::bot::{DbBot, BotPresence, BotActivity};
use anyhow::Result;
const BOT_ID: i32 = 1;
pub async fn init(db: &PgPool) -> Result<(), sqlx::Error> {
pub async fn init(db: &PgPool) -> Result<()> {
query!("INSERT INTO bots (id) VALUES ($1) ON CONFLICT DO NOTHING", BOT_ID)
.execute(db)
.await?;
Ok(())
}
pub async fn get(db: &PgPool) -> Result<Option<DbBot>, sqlx::Error> {
pub async fn get(db: &PgPool) -> Result<Option<DbBot>> {
let bot: Option<DbBot> = query_as!(
DbBot,
r#"SELECT status, activity_type as "activity_type: BotActivity", presence as "presence: BotPresence" FROM bots WHERE id = $1"#,
@ -21,21 +22,21 @@ pub async fn get(db: &PgPool) -> Result<Option<DbBot>, sqlx::Error> {
Ok(bot)
}
pub async fn set_status(db: &PgPool, status: &str) -> Result<(), sqlx::Error> {
pub async fn set_status(db: &PgPool, status: &str) -> Result<()> {
query!("UPDATE bots SET status = $1 WHERE id = $2", status, BOT_ID)
.execute(db)
.await?;
Ok(())
}
pub async fn set_activity(db: &PgPool, activity: BotActivity) -> Result<(), sqlx::Error> {
pub async fn set_activity(db: &PgPool, activity: BotActivity) -> Result<()> {
query!("UPDATE bots SET activity_type = $1::bot_activity WHERE id = $2", activity as BotActivity, BOT_ID)
.execute(db)
.await?;
Ok(())
}
pub async fn set_presence(db: &PgPool, presence: BotPresence) -> Result<(), sqlx::Error> {
pub async fn set_presence(db: &PgPool, presence: BotPresence) -> Result<()> {
query!("UPDATE bots SET presence = $1::bot_presence WHERE id = $2", presence as BotPresence, BOT_ID)
.execute(db)
.await?;