2021-02-19 17:00:45 +11:00
|
|
|
use anyhow::Result;
|
2021-03-04 11:28:58 +11:00
|
|
|
use tracing_subscriber::filter::LevelFilter;
|
|
|
|
use tracing_subscriber::FmtSubscriber;
|
2020-10-16 09:14:39 +11:00
|
|
|
|
2021-04-08 13:09:52 +10:00
|
|
|
pub fn init(level: LevelFilter) -> Result<()> {
|
2021-02-22 12:41:04 +11:00
|
|
|
if level == LevelFilter::OFF {
|
2020-10-16 09:14:39 +11:00
|
|
|
return Ok(());
|
|
|
|
}
|
|
|
|
|
2020-12-04 16:27:17 +11:00
|
|
|
let is_terminal = atty::is(atty::Stream::Stderr);
|
2021-03-17 11:41:28 +11:00
|
|
|
|
|
|
|
let builder = FmtSubscriber::builder()
|
|
|
|
.with_env_filter(format!("asb={},swap={}", level, level))
|
2020-12-04 16:27:17 +11:00
|
|
|
.with_writer(std::io::stderr)
|
2020-10-16 09:14:39 +11:00
|
|
|
.with_ansi(is_terminal)
|
2021-03-17 11:41:28 +11:00
|
|
|
.with_target(false);
|
|
|
|
|
|
|
|
if !is_terminal {
|
|
|
|
builder.without_time().init();
|
|
|
|
} else {
|
|
|
|
builder.init();
|
|
|
|
}
|
2020-10-16 09:14:39 +11:00
|
|
|
|
2021-05-05 13:49:11 +10:00
|
|
|
tracing::info!(%level, "Initialized tracing");
|
2020-10-16 09:14:39 +11:00
|
|
|
|
|
|
|
Ok(())
|
|
|
|
}
|