reminder-bot/src/main.rs

78 lines
1.5 KiB
Rust
Raw Normal View History

2020-08-06 14:22:13 +00:00
mod framework;
use serenity::{
client::{
bridge::gateway::GatewayIntents,
Client, Context,
},
model::{
channel::{
Message,
},
},
framework::standard::{
Args, CommandResult,
},
prelude::TypeMapKey,
};
use regex_command_attr::command;
2020-08-06 14:22:13 +00:00
use sqlx::{
Pool,
mysql::{
MySqlConnection,
}
};
use dotenv::dotenv;
use std::{
sync::Arc,
env,
};
use crate::framework::RegexFramework;
2020-08-06 14:22:13 +00:00
struct SQLPool;
impl TypeMapKey for SQLPool {
type Value = Pool<MySqlConnection>;
}
struct ReqwestClient;
impl TypeMapKey for ReqwestClient {
type Value = Arc<reqwest::Client>;
}
static THEME_COLOR: u32 = 0x00e0f3;
#[tokio::main]
async fn main() -> Result<(), Box<dyn std::error::Error + Send + Sync>> {
dotenv()?;
let framework = RegexFramework::new()
.ignore_bots(true)
.default_prefix("$")
.add_command("help".to_string(), &HELP_COMMAND)
.add_command("h".to_string(), &HELP_COMMAND)
2020-08-06 14:22:13 +00:00
.build();
let mut client = Client::new(&env::var("DISCORD_TOKEN").expect("Missing DISCORD_TOKEN from environment"))
.intents(GatewayIntents::GUILD_MESSAGES | GatewayIntents::GUILDS | GatewayIntents::DIRECT_MESSAGES)
.framework(framework)
.await.expect("Error occurred creating client");
2020-08-06 14:22:13 +00:00
client.start_autosharded().await?;
Ok(())
}
#[command]
async fn help(_ctx: &Context, _msg: &Message, _args: Args) -> CommandResult {
2020-08-06 14:22:13 +00:00
println!("Help command called");
Ok(())
}