crush-cli 0.2.1

Command-line interface for the Crush compression library
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
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67

use std::fs::File;
use std::path::Path;
use tracing::Level;
use tracing_subscriber::fmt;
use tracing_subscriber::EnvFilter;

/// Map verbose count to log level
/// - 0 (no -v flags) = INFO level
/// - 1 (-v) = DEBUG level
/// - 2+ (-vv) = TRACE level
pub fn verbose_to_level(verbose: u8) -> &'static str {
    match verbose {
        0 => "info",
        1 => "debug",
        _ => "trace", // 2 or more
    }
}

/// Initialize logging with the given level and format
pub fn init_logging(level: &str, format: &str, log_file: Option<&Path>) {
    // Parse log level
    let level = match level {
        "error" => Level::ERROR,
        "warn" => Level::WARN,
        "info" => Level::INFO,
        "debug" => Level::DEBUG,
        "trace" => Level::TRACE,
        _ => Level::INFO,
    };

    // Create env filter
    let env_filter = EnvFilter::try_from_default_env()
        .or_else(|_| EnvFilter::try_new(level.as_str()))
        .unwrap_or_else(|_| EnvFilter::new("info"));

    // Set up subscriber based on format and output
    match (format, log_file) {
        ("json", Some(path)) => {
            let file = File::create(path).expect("Failed to create log file");
            fmt()
                .json()
                .with_env_filter(env_filter)
                .with_writer(move || file.try_clone().expect("Failed to clone file"))
                .init();
        }
        ("json", None) => {
            fmt()
                .json()
                .with_env_filter(env_filter)
                .with_writer(std::io::stderr)
                .init();
        }
        (_, Some(path)) => {
            let file = File::create(path).expect("Failed to create log file");
            fmt()
                .with_env_filter(env_filter)
                .with_writer(move || file.try_clone().expect("Failed to clone file"))
                .init();
        }
        _ => {
            fmt()
                .with_env_filter(env_filter)
                .with_writer(std::io::stderr)
                .init();
        }
    }
}