2b-rs/bot-database/src/database/gifs.rs

use diesel::insert_into;
use diesel::prelude::*;
use tokio_diesel::*;

use crate::error::DatabaseResult;
use crate::models::*;
use crate::schema::*;
use crate::Database;

impl Database {
    /// Returns a list of all gifs in the database
    pub async fn get_all_gifs(&self) -> DatabaseResult<Vec<Gif>> {
        use gifs::dsl;
        log::debug!("Loading all gifs from the database");

        let gifs: Vec<Gif> = dsl::gifs.load_async::<Gif>(&self.pool).await?;
        Ok(gifs)
    }

    /// Returns a list of gifs by assigned category
    pub async fn get_gifs_by_category(&self, category: &str) -> DatabaseResult<Vec<Gif>> {
        use gifs::dsl;
        log::debug!("Searching for gifs in category '{}'", category);

        let gifs: Vec<Gif> = dsl::gifs
            .filter(dsl::category.eq(category))
            .load_async::<Gif>(&self.pool)
            .await?;
        Ok(gifs)
    }

    /// Adds a gif to the database
    pub async fn add_gif(
        &self,
        url: &str,
        category: Option<String>,
        name: Option<String>,
    ) -> DatabaseResult<()> {
        use gifs::dsl;
        log::debug!(
            "Inserting gif with url '{}' and name {:?} and category {:?}",
            url,
            name,
            category
        );
        insert_into(dsl::gifs)
            .values(GifInsert {
                url: url.to_string(),
                name,
                category,
            })
            .execute_async(&self.pool)
            .await?;

        Ok(())
    }
}