From e2e2f2238f5743340718ed049999dba5f4e4b4f7 Mon Sep 17 00:00:00 2001
From: Keuin <keuinx@gmail.com>
Date: Mon, 28 Mar 2022 20:55:19 +0800
Subject: Make some fields in config file optional. Especially, if log_file is
 not specified, will not write logs to file.

---
 src/main.rs | 45 +++++++++++++++++++++++++++------------------
 1 file changed, 27 insertions(+), 18 deletions(-)

(limited to 'src/main.rs')

diff --git a/src/main.rs b/src/main.rs
index 2c4263d..e85b74e 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -7,6 +7,7 @@ use std::str::FromStr;
 use teloxide::prelude2::*;
 use tracing::{debug, info, Level};
 use tracing::instrument;
+use tracing_appender::non_blocking::WorkerGuard;
 use tracing_subscriber::fmt;
 use tracing_subscriber::layer::SubscriberExt;
 use warp::Filter;
@@ -23,8 +24,6 @@ mod token;
 mod config;
 
 const CONFIG_FILE_NAME: &str = "kimikuri.json";
-const MAX_BODY_LENGTH: u64 = 1024 * 16;
-const DEFAULT_LOG_LEVEL: Level = Level::DEBUG;
 
 fn with_db(db_pool: DbPool) -> impl Filter<Extract=(DbPool, ), Error=Infallible> + Clone {
     warp::any().map(move || db_pool.clone())
@@ -47,27 +46,37 @@ async fn main() {
         Ok(l) => l,
         Err(_) => {
             eprintln!("Invalid log level: {}. Use {:?} instead.",
-                      config.log_level, DEFAULT_LOG_LEVEL);
-            DEFAULT_LOG_LEVEL
+                      config.log_level, config::DEFAULT_LOG_LEVEL);
+            Level::from_str(config::DEFAULT_LOG_LEVEL).unwrap()
         }
     };
     eprintln!("Configuration is loaded. Set log level to {:?}.", log_level);
-    let log_file_path = path::Path::new(&config.log_file);
-    let parent = log_file_path.parent().expect("Cannot extract parent.");
-    let filename = log_file_path.file_name().expect("Cannot extract file name.");
-    let (nb_file_appender, _guard) =
-        tracing_appender::non_blocking(
-            tracing_appender::rolling::never(parent, filename));
+    let _guard: WorkerGuard;
     let subscriber = fmt::Subscriber::builder()
         .with_max_level(log_level)
         .with_writer(io::stderr) // log to stderr
-        .finish()
-        .with(fmt::Layer::default()
-            .with_writer(nb_file_appender) // log to file
-            .with_ansi(false)); // remove color control characters from log file
-
-    tracing::subscriber::set_global_default(subscriber)
-        .expect("Failed to set default subscriber");
+        .finish();
+    if !config.log_file.is_empty() {
+        let log_file_path = path::Path::new(&config.log_file);
+        let parent = log_file_path.parent()
+            .expect("Invalid log_file: Cannot extract parent.");
+        let filename = log_file_path.file_name()
+            .expect("Invalid log_file: Cannot extract file name.");
+        let (nb_file_appender, guard) = tracing_appender::non_blocking(
+            tracing_appender::rolling::never(parent, filename));
+        _guard = guard;
+        let subscriber = subscriber.with(
+            fmt::Layer::default()
+                .with_writer(nb_file_appender) // log to file
+                .with_ansi(false) // remove color control characters from log file
+        );
+        tracing::subscriber::set_global_default(subscriber)
+            .expect("Failed to set default subscriber");
+    } else {
+        // log file is not specified, do not write logs to file
+        tracing::subscriber::set_global_default(subscriber)
+            .expect("Failed to set default subscriber");
+    }
 
     let db = config.db_file.as_str();
     info!(db, "Opening database...");
@@ -80,7 +89,7 @@ async fn main() {
 
     info!("Initializing HTTP routes...");
     let route_post = warp::post()
-        .and(warp::body::content_length_limit(MAX_BODY_LENGTH))
+        .and(warp::body::content_length_limit(config.max_body_size))
         .and(warp::body::json())
         .and(with_db(db.clone()))
         .and(with_bot(bot.clone()))
-- 
cgit v1.2.3