mermaid_cli/utils/
logger.rs

1use std::fs::OpenOptions;
2use std::path::PathBuf;
3use tracing::{debug, error, info, warn};
4use tracing_subscriber::{EnvFilter, layer::SubscriberExt, util::SubscriberInitExt};
5
6/// Get the log file path (~/.mermaid/mermaid.log)
7fn get_log_file_path() -> Option<PathBuf> {
8    std::env::var("HOME")
9        .ok()
10        .map(|home| PathBuf::from(home).join(".mermaid").join("mermaid.log"))
11}
12
13/// Initialize the logging system with tracing
14pub fn init_logger(verbose: bool) {
15    // If --verbose flag is set, override to debug level
16    // Otherwise use RUST_LOG environment variable, default to warn level (quieter)
17    let filter = if verbose {
18        EnvFilter::new("debug,mermaid=debug")
19    } else {
20        EnvFilter::try_from_default_env().unwrap_or_else(|_| EnvFilter::new("warn,mermaid=info"))
21    };
22
23    // Try to write logs to a file to avoid corrupting the TUI
24    // Falls back to no logging if file creation fails (TUI takes priority)
25    if let Some(log_path) = get_log_file_path() {
26        // Ensure parent directory exists
27        if let Some(parent) = log_path.parent() {
28            let _ = std::fs::create_dir_all(parent);
29        }
30
31        // Open log file for appending
32        if let Ok(file) = OpenOptions::new().create(true).append(true).open(&log_path) {
33            let fmt_layer = tracing_subscriber::fmt::layer()
34                .with_writer(file)
35                .with_target(false)
36                .with_thread_ids(false)
37                .with_thread_names(false)
38                .with_ansi(false) // No ANSI colors in file
39                .compact();
40
41            tracing_subscriber::registry()
42                .with(filter)
43                .with(fmt_layer)
44                .init();
45            return;
46        }
47    }
48
49    // Fallback: no logging if file creation fails (don't corrupt TUI)
50    tracing_subscriber::registry().with(filter).init();
51}
52
53/// Log an info message with category prefix (backward compatible)
54pub fn log_info(category: &str, message: impl std::fmt::Display) {
55    info!(category = %category, "{}", message);
56}
57
58/// Log a warning message with category prefix (backward compatible)
59pub fn log_warn(category: &str, message: impl std::fmt::Display) {
60    warn!(category = %category, "{}", message);
61}
62
63/// Log an error message with category prefix (backward compatible)
64pub fn log_error(category: &str, message: impl std::fmt::Display) {
65    error!(category = %category, "{}", message);
66}
67
68/// Log a debug message (backward compatible)
69pub fn log_debug(message: impl std::fmt::Display) {
70    debug!("{}", message);
71}
72
73/// Progress indicator for startup sequence
74pub fn log_progress(step: usize, total: usize, message: impl std::fmt::Display) {
75    info!(step = step, total = total, "{}", message);
76}
mermaid_cli/utils/logger.rs

mermaid_cli/utils/
logger.rs
