From 1d14f5deb142eaf8dac110581959ef2eb3d88b60 Mon Sep 17 00:00:00 2001
From: Gabriel Simmer <g@gmem.ca>
Date: Fri, 17 Nov 2023 11:29:15 +0000
Subject: [PATCH] Logging, default to log level info

---
 src/main.rs | 13 +++++++++----
 1 file changed, 9 insertions(+), 4 deletions(-)

diff --git a/src/main.rs b/src/main.rs
index 12e274b..ed58270 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -7,6 +7,8 @@ use axum::{
     routing::get,
     Router,
 };
+use env_logger::Env;
+use log::{debug, error, info};
 use prometheus::{
     register_gauge_vec, register_int_counter_vec, Encoder, GaugeVec, IntCounterVec, TextEncoder,
 };
@@ -160,6 +162,9 @@ fn register_metrics() {
 
 #[tokio::main]
 async fn main() -> Result<(), ()> {
+    let env = Env::default().filter_or("LOG_LEVEL", "info");
+
+    env_logger::init_from_env(env);
     let content = fs::read_to_string("config.toml").unwrap();
     let mut config: Config = toml::from_str(&content).unwrap();
     config.vrchat_token = Some(env::var("VRCHAT_AUTH_TOKEN").unwrap());
@@ -170,7 +175,7 @@ async fn main() -> Result<(), ()> {
         .route("/metrics", get(metrics_handler))
         .with_state(config);
 
-    println!("running at 0.0.0.0:6534");
+    info!("running at 0.0.0.0:6534");
     let _ = axum::Server::bind(&"0.0.0.0:6534".parse().unwrap())
         .serve(app.into_make_service())
         .await;
@@ -185,7 +190,7 @@ async fn metrics_handler(State(config): State<Config>) -> Result<impl IntoRespon
             .body(Full::from(b))
             .unwrap()),
         Err(e) => {
-            println!("{}", e);
+            error!("{}", e);
             Err(StatusCode::INTERNAL_SERVER_ERROR)
         }
     }
@@ -205,13 +210,13 @@ async fn metrics(config: Config) -> Result<Vec<u8>, WsError> {
 
     if config.groups.is_some() {
         for (name, group) in config.groups.unwrap() {
-            println!("scanning group {}", &name);
+            debug!("scanning group {}", &name);
             group_metrics(&client, &auth_cookie, name, group).await?;
         }
     }
     if config.worlds.is_some() {
         for (name, id) in config.worlds.unwrap() {
-            println!("scanning world {}", &name);
+            debug!("scanning world {}", &name);
             world_metrics(&client, &auth_cookie, name, id).await?;
         }
     }