handle handlers

src/main.rs

@@ -1,31 +1,55 @@
 mod handler;
+mod lawsuit;
+mod model;
 
 use std::env;
 
 use color_eyre::{eyre::WrapErr, Result};
 use serenity::{model::prelude::*, prelude::*};
+use tracing::info;
+use tracing_subscriber::EnvFilter;
 
-use crate::handler::Handler;
+use crate::{handler::Handler, model::Mongo};
 
 #[tokio::main]
-fn main() -> Result<()> {
+async fn main() -> Result<()> {
     color_eyre::install()?;
 
     let _ = dotenv::dotenv();
 
+    tracing_subscriber::fmt()
+        .with_env_filter(EnvFilter::from_default_env())
+        .init();
+
+    info!("Starting up...");
+
+    let mongo_uri = env::var("MONGO_URI").wrap_err("MONGO_URI not found in the environment")?;
+    let db_name = env::var("DB_NAME").unwrap_or_else(|_| "court-bot".to_string());
+
+    let mongo = Mongo::connect(&mongo_uri, &db_name).await?;
+
+    info!("Connected to mongodb");
+
     let token = env::var("DISCORD_TOKEN").wrap_err("DISCORD_TOKEN not found in environment")?;
-    let guild_id = if let Ok(_) = env::var("DEV") {
-           SOme( GuildId(
-                env::var("GUILD_ID")
-                    .wrap_err("GUILD_ID not found in environment, must be set when DEV is set")?
-                    .parse()
-                    .wrap_err("GUILD_ID must be an integer")?,
-            ))
-        })
-} else {None};
+    let dev_guild_id = if env::var("DEV").is_ok() {
+        Some(GuildId(
+            env::var("GUILD_ID")
+                .wrap_err("GUILD_ID not found in environment, must be set when DEV is set")?
+                .parse()
+                .wrap_err("GUILD_ID must be an integer")?,
+        ))
+    } else {
+        None
+    };
+
+    let set_global_commands = env::var("SET_GLOBAL").is_ok();
 
     let mut client = Client::builder(token, GatewayIntents::empty())
-        .event_handler(Handler { dev_guild_id })
+        .event_handler(Handler {
+            dev_guild_id,
+            set_global_commands,
+            mongo,
+        })
         .await
         .wrap_err("failed to create discord client")?;