112 lines
2.9 KiB
Rust
112 lines
2.9 KiB
Rust
#[macro_use]
|
|
extern crate lazy_static;
|
|
|
|
mod models;
|
|
mod framework;
|
|
mod commands;
|
|
mod time_parser;
|
|
|
|
use serenity::{
|
|
client::{
|
|
bridge::gateway::GatewayIntents,
|
|
Client,
|
|
},
|
|
framework::Framework,
|
|
prelude::TypeMapKey,
|
|
};
|
|
|
|
use sqlx::{
|
|
Pool,
|
|
mysql::{
|
|
MySqlPool,
|
|
MySqlConnection,
|
|
}
|
|
};
|
|
|
|
use dotenv::dotenv;
|
|
|
|
use std::{
|
|
sync::Arc,
|
|
env,
|
|
};
|
|
|
|
use crate::framework::RegexFramework;
|
|
use crate::commands::{
|
|
info_cmds,
|
|
reminder_cmds,
|
|
todo_cmds,
|
|
moderation_cmds,
|
|
};
|
|
|
|
struct SQLPool;
|
|
|
|
impl TypeMapKey for SQLPool {
|
|
type Value = Pool<MySqlConnection>;
|
|
}
|
|
|
|
struct ReqwestClient;
|
|
|
|
impl TypeMapKey for ReqwestClient {
|
|
type Value = Arc<reqwest::Client>;
|
|
}
|
|
|
|
struct FrameworkCtx;
|
|
|
|
impl TypeMapKey for FrameworkCtx {
|
|
type Value = Arc<Box<dyn Framework + Send + Sync>>;
|
|
}
|
|
|
|
static THEME_COLOR: u32 = 0x8fb677;
|
|
|
|
#[tokio::main]
|
|
async fn main() -> Result<(), Box<dyn std::error::Error + Send + Sync>> {
|
|
dotenv()?;
|
|
|
|
let framework = RegexFramework::new(env::var("CLIENT_ID").expect("Missing CLIENT_ID from environment").parse()?)
|
|
.ignore_bots(true)
|
|
.default_prefix("$")
|
|
|
|
.add_command("ping", &info_cmds::PING_COMMAND)
|
|
|
|
.add_command("help", &info_cmds::HELP_COMMAND)
|
|
.add_command("info", &info_cmds::INFO_COMMAND)
|
|
.add_command("donate", &info_cmds::DONATE_COMMAND)
|
|
.add_command("dashboard", &info_cmds::DASHBOARD_COMMAND)
|
|
.add_command("clock", &info_cmds::CLOCK_COMMAND)
|
|
|
|
.add_command("todo", &todo_cmds::TODO_PARSE_COMMAND)
|
|
|
|
.add_command("blacklist", &moderation_cmds::BLACKLIST_COMMAND)
|
|
.add_command("restrict", &moderation_cmds::RESTRICT_COMMAND)
|
|
.add_command("timezone", &moderation_cmds::TIMEZONE_COMMAND)
|
|
.add_command("prefix", &moderation_cmds::PREFIX_COMMAND)
|
|
.add_command("lang", &moderation_cmds::LANGUAGE_COMMAND)
|
|
|
|
.add_command("pause", &reminder_cmds::PAUSE_COMMAND)
|
|
.add_command("offset", &reminder_cmds::OFFSET_COMMAND)
|
|
.add_command("nudge", &reminder_cmds::NUDGE_COMMAND)
|
|
|
|
.build();
|
|
|
|
let framework_arc = Arc::new(Box::new(framework) as Box<dyn Framework + Send + Sync>);
|
|
|
|
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_arc(framework_arc.clone())
|
|
.await.expect("Error occurred creating client");
|
|
|
|
{
|
|
let pool = MySqlPool::new(&env::var("DATABASE_URL").expect("Missing DATABASE_URL from environment")).await.unwrap();
|
|
|
|
let mut data = client.data.write().await;
|
|
|
|
data.insert::<SQLPool>(pool);
|
|
data.insert::<ReqwestClient>(Arc::new(reqwest::Client::new()));
|
|
data.insert::<FrameworkCtx>(framework_arc);
|
|
}
|
|
|
|
client.start_autosharded().await?;
|
|
|
|
Ok(())
|
|
}
|