62 lines
1.9 KiB
Rust
62 lines
1.9 KiB
Rust
use std::path::PathBuf;
|
|
use tracing_appender::rolling::{RollingFileAppender, Rotation};
|
|
use tracing_subscriber::{
|
|
EnvFilter, fmt, fmt::time::LocalTime, layer::SubscriberExt, util::SubscriberInitExt,
|
|
};
|
|
|
|
/// Get the default log directory path: ~/.picobot/logs
|
|
pub fn get_default_log_dir() -> PathBuf {
|
|
let home = dirs::home_dir().unwrap_or_else(|| PathBuf::from("."));
|
|
home.join(".picobot").join("logs")
|
|
}
|
|
|
|
/// Get the default config file path: ~/.picobot/config.json
|
|
pub fn get_default_config_path() -> PathBuf {
|
|
let home = dirs::home_dir().unwrap_or_else(|| PathBuf::from("."));
|
|
home.join(".picobot").join("config.json")
|
|
}
|
|
|
|
/// Initialize logging with file appender
|
|
/// Logs are written to ~/.picobot/logs/ with daily rotation
|
|
pub fn init_logging() {
|
|
let log_dir = get_default_log_dir();
|
|
|
|
// Create log directory if it doesn't exist
|
|
if !log_dir.exists()
|
|
&& let Err(e) = std::fs::create_dir_all(&log_dir)
|
|
{
|
|
eprintln!(
|
|
"Warning: Failed to create log directory {}: {}",
|
|
log_dir.display(),
|
|
e
|
|
);
|
|
}
|
|
|
|
// Create file appender with daily rotation
|
|
let file_appender = RollingFileAppender::new(Rotation::DAILY, &log_dir, "picobot.log");
|
|
|
|
// Build subscriber with both console and file output
|
|
let env_filter = EnvFilter::try_from_default_env().unwrap_or_else(|_| EnvFilter::new("info"));
|
|
|
|
let file_layer = fmt::layer()
|
|
.with_writer(file_appender)
|
|
.with_timer(LocalTime::rfc_3339())
|
|
.with_ansi(false)
|
|
.with_target(true)
|
|
.with_level(true)
|
|
.with_thread_ids(true);
|
|
|
|
let console_layer = fmt::layer()
|
|
.with_timer(LocalTime::rfc_3339())
|
|
.with_target(true)
|
|
.with_level(true);
|
|
|
|
tracing_subscriber::registry()
|
|
.with(env_filter)
|
|
.with(console_layer)
|
|
.with(file_layer)
|
|
.init();
|
|
|
|
tracing::info!("Logging initialized. Log directory: {}", log_dir.display());
|
|
}
|