From d79d657d70d54c030e391e8f03a4fe443539f70e Mon Sep 17 00:00:00 2001
From: Todor Andonov <todor.andonov@payhawk.com>
Date: Sat, 31 Aug 2024 23:51:21 +0300
Subject: [PATCH] implement

---
 src/files.rs |  4 ----
 src/main.rs  | 50 ++++++++++++++++++++++++++++++++++++++------------
 2 files changed, 38 insertions(+), 16 deletions(-)

diff --git a/src/files.rs b/src/files.rs
index 7294569..124b14a 100644
--- a/src/files.rs
+++ b/src/files.rs
@@ -1,13 +1,9 @@
 use anyhow::*;
 use glob::{MatchOptions, Pattern};
 use ignore::WalkBuilder;
-#[cfg(not(test))]
 use log::debug;
 use std::path::{Path, PathBuf};
 
-#[cfg(test)]
-use std::println as debug;
-
 #[derive(Debug, PartialEq)]
 pub enum Extension {
     Json,
diff --git a/src/main.rs b/src/main.rs
index cb8fc5f..e172355 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -1,7 +1,7 @@
 use clap::{CommandFactory, Parser, ValueEnum};
 use colored::*;
 use glob::GlobError;
-use log::{debug, error, info};
+use log::{debug, error, info, LevelFilter};
 use rayon::prelude::*;
 use serde::Serialize;
 use serde_json::Value;
@@ -43,6 +43,26 @@ pub enum IndentStyle {
     Spaces,
 }
 
+#[derive(Debug, Clone, ValueEnum)]
+pub enum LogLevel {
+    #[clap()]
+    Quiet,
+    #[clap()]
+    Default,
+    #[clap()]
+    Verbose,
+}
+
+impl LogLevel {
+    pub fn to_level_filter(&self) -> LevelFilter {
+        match self {
+            LogLevel::Quiet => LevelFilter::Off,
+            LogLevel::Default => LevelFilter::Error,
+            LogLevel::Verbose => LevelFilter::Debug,
+        }
+    }
+}
+
 type Result<T> = std::result::Result<T, CustomError>;
 
 #[derive(Parser, Debug)]
@@ -79,6 +99,10 @@ struct Args {
     /// Specify the desired indent style
     #[arg(long)]
     indent_style: Option<IndentStyle>,
+
+    /// Specify the log level
+    #[arg(value_enum, long, default_value_t=LogLevel::Default)]
+    log_level: LogLevel,
 }
 
 #[derive(Debug)]
@@ -94,36 +118,38 @@ struct Configuration {
 }
 
 fn print_error(err: &CustomError) {
-    eprintln!("{} {}", "Error:".red().bold(), err);
+    error!("{} {}", "Error:".red().bold(), err);
     match err {
         CustomError::Custom(msg) if msg == "No JSON files found matching the provided patterns" => {
-            eprintln!(
+            error!(
                 "{}",
                 "Make sure the file patterns are correct and the files exist.".yellow()
             );
             if let Err(err) = Args::command().print_help() {
-                eprintln!("Failed to print help message: {}", err);
+                error!("Failed to print help message: {}", err);
             }
         }
         _ => {
             if let Err(err) = Args::command().print_help() {
-                eprintln!("Failed to print help message: {}", err);
+                error!("Failed to print help message: {}", err);
             }
         }
     }
 }
 
 fn main() -> Result<()> {
-    env_logger::init();
-    if let Err(e) = run() {
+    let args = Args::parse();
+    env_logger::builder()
+        .filter_level(args.log_level.to_level_filter())
+        .init();
+    if let Err(e) = run(args) {
         print_error(&e);
         std::process::exit(1);
     } else {
         Ok(())
     }
 }
-fn run() -> Result<()> {
-    let args = Args::parse();
+fn run(args: Args) -> Result<()> {
     let start_time = Instant::now();
 
     if args.include.is_empty() {
@@ -170,7 +196,7 @@ fn run() -> Result<()> {
                 path.display().to_string().green(),
                 duration
             ),
-            Err(e) => eprintln!(
+            Err(e) => error!(
                 "{}: {} (in {:.2?})",
                 path.display().to_string().red(),
                 e,
@@ -181,7 +207,7 @@ fn run() -> Result<()> {
 
     let total_duration = start_time.elapsed();
 
-    println!(
+    info!(
         "{}",
         format!(
             "Processed {} file(s) in {:.2?}",
@@ -213,7 +239,7 @@ fn process_file(path: &PathBuf, cfg: &Configuration) -> Result<()> {
         info!("Sorted JSON written back to {:?}", path);
     } else {
         serde_json::to_string_pretty(&sorted_json)?;
-        info!("Sorted {:?}.\n", path);
+        info!("Sorted {:?}.", path);
     }
 
     Ok(())