You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
2b-rs/src/commands/weeb/mod.rs

39 lines
1.0 KiB
Rust

use serenity::framework::standard::macros::group;
mod korone;
mod matsuri;
mod pekofy;
mod rushia;
mod sauce;
use crate::utils::context_data::get_database_from_context;
use crate::utils::error::BotError;
use rand::prelude::IteratorRandom;
use serenity::client::Context;
use serenity::framework::standard::CommandResult;
use serenity::model::channel::Message;
use korone::KORONE_COMMAND;
use matsuri::MATSURI_COMMAND;
use pekofy::PEKOFY_COMMAND;
use rushia::RUSHIA_COMMAND;
use sauce::SAUCE_COMMAND;
#[group]
#[commands(pekofy, sauce, matsuri, korone, rushia)]
pub struct Weeb;
/// Posts a random media entry with the given category
async fn post_random_media(ctx: &Context, msg: &Message, category: &str) -> CommandResult {
let database = get_database_from_context(ctx).await;
let media = database.get_media_by_category(category).await?;
let gif = media
.into_iter()
.choose(&mut rand::thread_rng())
.ok_or(BotError::from("No media found."))?;
msg.channel_id.say(ctx, gif.url).await?;
Ok(())
}