From 6df9437531979203600f5301d72312333514fdbd Mon Sep 17 00:00:00 2001
From: Victor Mignot <root@victormignot.fr>
Date: Fri, 21 Oct 2022 19:59:52 -0400
Subject: [PATCH] Add log4rs config and fix main to use exit codes

---
 log4rs_config.yaml | 59 ++++++++++++++++++++++++++++++++++++++++++++++
 src/main.rs        | 50 +++++++++++++++++++++++++++++++--------
 2 files changed, 99 insertions(+), 10 deletions(-)
 create mode 100644 log4rs_config.yaml

diff --git a/log4rs_config.yaml b/log4rs_config.yaml
new file mode 100644
index 0000000..5b9ff44
--- /dev/null
+++ b/log4rs_config.yaml
@@ -0,0 +1,59 @@
+refresh_rate: 30 seconds
+
+appenders:
+
+  stdout:
+    kind: console
+    encoder:
+      pattern: "{d(%Y-%m-%d %H:%M:%S)} | {({l}):5.5} | {f}:{L} — {m}{n}"
+
+  rolling_logs:
+    kind: rolling_file
+    path: logs/yorokobot-latest.log
+    encoder:
+      pattern: "{d(%Y-%m-%d %H:%M:%S)} | {({l}):5.5} | {f}:{L} — {m}{n}"
+    policy:
+      trigger: # Specify when to switch to another log file 
+        kind: size
+        limit: 30 mb
+
+      roller: # Specify what to do with older logs files
+        kind: fixed_window
+        pattern: logs/yorokobot_{}.gz
+        count: 10
+
+  rolling_debug:
+    kind: rolling_file
+    path: logs/traces-latest.log
+    encoder:
+      pattern: "{d(%Y-%m-%d %H:%M:%S)} | {({l}):5.5} | {f}:{L} — {m}{n}"
+    policy:
+      trigger:
+        kind: size
+        limit: 30 mb
+        
+      roller:
+        kind: fixed_window
+        pattern: logs/yorokobot_traces_{}.gz
+        count: 20
+
+root:
+  level: warn
+  appenders:
+    - stdout
+
+loggers:
+  bot_infos:
+    level: info
+    appenders:
+      - rolling_debug
+
+  bot_warn_errors:
+    level: warn 
+    appenders:
+      - rolling_logs
+  
+  bot_traces:
+    level: debug
+    appenders:
+      - rolling_debug
diff --git a/src/main.rs b/src/main.rs
index 3b488d0..9267d17 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -1,3 +1,4 @@
+use log::error;
 use std::env;
 
 use yorokobot::{
@@ -6,22 +7,51 @@ use yorokobot::{
 };
 
 #[tokio::main]
-async fn main() {
-    let discord_token = env::var("DISCORD_TOKEN").expect("Cannot fetch Discord token");
+async fn main() -> std::process::ExitCode {
+    // Start the logger
+    log4rs::init_file("log4rs_config.yaml", Default::default()).unwrap();
 
-    let mongodb_uri = env::var("MONGODB_URI").expect("Cannot fetch Mongo URI");
+    let discord_token = match env::var("DISCORD_TOKEN") {
+        Ok(t) => t,
+        Err(_) => {
+            error!(target: "bot_warn_errors", "Could not find DISCORD_TOKEN environment variable.");
+            return std::process::ExitCode::FAILURE;
+        }
+    };
+
+    let mongodb_uri = match env::var("MONGODB_URI") {
+        Ok(u) => u,
+        Err(_) => {
+            error!(target: "bot_warn_errors", "Could not find MONGODB_URI environment variable.");
+            return std::process::ExitCode::FAILURE;
+        }
+    };
 
     let credentials = ClientCredentials {
         discord_token: &discord_token,
         mongo_uri: &mongodb_uri,
     };
 
-    let mut client = Client::new(credentials)
-        .await
-        .expect("Could not create client");
+    let mut client = match Client::new(credentials).await {
+        Ok(c) => c,
+        Err(_) => {
+            error!(target: "bot_warn_errors", "Could not instantiate the bot client.");
+            return std::process::ExitCode::FAILURE;
+        }
+    };
 
-    client.connect().await.unwrap_or_else(|error| match error {
-        ClientsError::Database(e) => panic!("Could not connect to database: {:?}", e),
-        ClientsError::Discord(e) => panic!("Could not connect to Discord: {:?}", e),
-    });
+    if let Err(error) = client.connect().await {
+        match error {
+            ClientsError::Database(e) => {
+                error!(target: "bot_warn_errors", "Could not connect to database: {:?}", e)
+            }
+            ClientsError::Discord(e) => {
+                error!(target: "bot_warn_errors", "Could not connect to Discord: {:?}", e)
+            }
+        };
+
+        return std::process::ExitCode::FAILURE;
+    }
+
+    std::process::ExitCode::SUCCESS
 }