summaryrefslogtreecommitdiff
path: root/rola-cli/src/output/setup.rs
blob: 824348be82d46c64506cdf2a1fdc0c339a3b980a (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
use mingling::{Program, macros::program_setup};
use shared_functions::info;

use crate::{
    ThisProgram,
    output::{
        ansi_control::disable_ansi,
        env_logger::{EnvLogger, init_envlogger},
    },
};

#[program_setup]
pub fn color_output_setup(program: &mut Program<ThisProgram>) {
    program.global_flag("--no-color", |_| {
        disable_ansi();
    });
}

#[program_setup]
pub fn env_logger_setup(program: &mut Program<ThisProgram>) {
    program.stdout_setting.verbose = program.pick_global_flag(["-V", "--verbose"]);
    let log_show_time = program.pick_global_flag("--log-time");
    let log_level = program
        .pick_global_argument("--log-level")
        .unwrap_or("info".to_string());
    if program.stdout_setting.verbose {
        init_envlogger(EnvLogger {
            show_time: log_show_time,
            show_level: log_level.as_str() != "disable",
            level: match log_level.as_str() {
                "trace" => log::Level::Trace,
                "debug" => log::Level::Debug,
                "warn" => log::Level::Warn,
                "error" => log::Level::Error,
                _ => log::Level::Info,
            },
        });
    }

    info!("Verbose mode enabled!");
}