p4bl0t/src/discord/mod.rs

mod commands;
mod events;
pub mod utils;
pub mod error;

use poise::FrameworkBuilder;
use utils::serenity;

use commands::logging;
use utils::{BotData, Context, Error};

use self::events::event_handler;

pub type Result = ::std::result::Result<(), Error>;

/// Bootstraps the Discord bot.
///
/// # Panics
///
/// Panics if the environment `DISCORD_TOKEN` is unavailable.
pub fn make_bot() -> FrameworkBuilder<BotData, Error> {
    poise::Framework::builder()
        .options(poise::FrameworkOptions {
            commands: vec![logging()],
            event_handler: |ctx, event, framework, data| {
                Box::pin(event_handler(ctx, event, framework, data))
            },
            ..Default::default()
        })
        .token(std::env::var("DISCORD_TOKEN").expect("missing DISCORD_TOKEN"))
        .intents(serenity::GatewayIntents::non_privileged())
        .setup(|ctx, _ready, framework| {
            Box::pin(async move {
                poise::builtins::register_globally(
                    ctx,
                    &framework.options().commands,
                )
                .await?;
                Ok(BotData::new().await?)
            })
        })
}