several changes and updates

src/bin/spamcontest.rs

@@ -1,3 +1,4 @@
+use log::{debug, error, LevelFilter};
 use serenity::prelude::*;
 use spamcontest::Handler;
 use std::{env, process};
@@ -6,11 +7,16 @@ const TOKEN_VAR_KEY: &str = "DISCORD_TOKEN";
 
 #[tokio::main]
 async fn main() {
+    env_logger::builder()
+        .filter_module(module_path!(), LevelFilter::Info)
+        .parse_default_env()
+        .init();
+
     let token = match env::var(TOKEN_VAR_KEY) {
         Ok(token) => token,
         Err(err) => {
-            eprintln!("Unable to get {}: {}", TOKEN_VAR_KEY, err);
-            process::exit(1)
+            error!("Unable to get {}: {}", TOKEN_VAR_KEY, err);
+            process::exit(1);
         }
     };
 
@@ -21,12 +27,20 @@ async fn main() {
     {
         Ok(client) => client,
         Err(err) => {
-            eprintln!("Unable to start client: {}", err);
-            process::exit(2)
+            error!("Unable to start client: {:?}", err);
+            process::exit(2);
         }
     };
 
+    let shard_manager = client.shard_manager.clone();
+    tokio::spawn(async move {
+        tokio::signal::ctrl_c().await.expect("ctrc+c handler error");
+        debug!("Shutting down...");
+        shard_manager.lock().await.shutdown_all().await;
+    });
+
     if let Err(err) = client.start().await {
-        eprintln!("An Error occurred while running the client: {:?}", err)
+        error!("An Error occurred while running the client: {:?}", err);
+        process::exit(2);
     }
 }