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.
67 lines
1.8 KiB
Rust
67 lines
1.8 KiB
Rust
use crate::commands::*;
|
|
use crate::database::{get_database, Database};
|
|
use crate::utils::error::{BotError, BotResult};
|
|
use crate::utils::store::{Store, StoreData};
|
|
use serenity::async_trait;
|
|
use serenity::client::{Context, EventHandler};
|
|
use serenity::framework::standard::macros::hook;
|
|
use serenity::framework::standard::CommandResult;
|
|
use serenity::framework::StandardFramework;
|
|
use serenity::model::channel::Message;
|
|
use serenity::Client;
|
|
use songbird::SerenityInit;
|
|
|
|
struct Handler;
|
|
|
|
#[async_trait]
|
|
impl EventHandler for Handler {}
|
|
|
|
pub async fn get_client() -> BotResult<Client> {
|
|
let token = dotenv::var("BOT_TOKEN").map_err(|_| BotError::MissingToken)?;
|
|
let database = get_database()?;
|
|
|
|
let client = Client::builder(token)
|
|
.framework(get_framework())
|
|
.register_songbird()
|
|
.await?;
|
|
{
|
|
let mut data = client.data.write().await;
|
|
data.insert::<Store>(StoreData::new(database))
|
|
}
|
|
|
|
Ok(client)
|
|
}
|
|
|
|
pub fn get_framework() -> StandardFramework {
|
|
StandardFramework::default()
|
|
.configure(|c| {
|
|
c.prefix(
|
|
dotenv::var("BOT_PREFIX")
|
|
.unwrap_or("~!".to_string())
|
|
.as_str(),
|
|
)
|
|
.allow_dm(true)
|
|
.ignore_bots(true)
|
|
})
|
|
.group(&MINECRAFT_GROUP)
|
|
.group(&MISC_GROUP)
|
|
.group(&MUSIC_GROUP)
|
|
.after(after_hook)
|
|
.before(before_hook)
|
|
}
|
|
|
|
#[hook]
|
|
async fn after_hook(ctx: &Context, msg: &Message, cmd_name: &str, error: CommandResult) {
|
|
// Print out an error if it happened
|
|
if let Err(why) = error {
|
|
let _ = msg.channel_id.say(&ctx, format!("{}", why)).await;
|
|
println!("Error in {}: {:?}", cmd_name, why);
|
|
}
|
|
}
|
|
|
|
#[hook]
|
|
async fn before_hook(ctx: &Context, msg: &Message, _: &str) -> bool {
|
|
let _ = msg.channel_id.broadcast_typing(ctx).await;
|
|
true
|
|
}
|