use std::io::Write;
use std::sync::Arc;
use std::sync::OnceLock;
use arc_swap::ArcSwap;
use deno_runtime::deno_telemetry;
use deno_runtime::deno_telemetry::OtelConfig;
use deno_runtime::deno_telemetry::OtelConsoleConfig;
struct CliLoggerInner {
otel_console_config: OtelConsoleConfig,
logger: env_logger::Logger,
}
struct CliLogger {
inner: ArcSwap<CliLoggerInner>,
on_log_start: fn(),
on_log_end: fn(),
}
impl CliLogger {
pub fn filter(&self) -> log::LevelFilter {
self.inner.load().logger.filter()
}
}
impl log::Log for CliLogger {
fn enabled(&self, metadata: &log::Metadata) -> bool {
self.inner.load().logger.enabled(metadata)
}
fn log(&self, record: &log::Record) {
if self.enabled(record.metadata()) {
(self.on_log_start)();
match self.inner.load().otel_console_config {
OtelConsoleConfig::Ignore => {
self.inner.load().logger.log(record);
}
OtelConsoleConfig::Capture => {
self.inner.load().logger.log(record);
deno_telemetry::handle_log(record);
}
OtelConsoleConfig::Replace => {
deno_telemetry::handle_log(record);
}
}
(self.on_log_end)();
}
}
fn flush(&self) {
self.inner.load().logger.flush();
}
}
pub struct InitLoggingOptions {
pub on_log_start: fn(),
pub on_log_end: fn(),
pub maybe_level: Option<log::Level>,
pub otel_config: Option<OtelConfig>,
}
static LOGGER: OnceLock<CliLogger> = OnceLock::new();
pub fn init(options: InitLoggingOptions) {
let log_level = options.maybe_level.unwrap_or(log::Level::Info);
let logger = env_logger::Builder::from_env(
env_logger::Env::new()
.filter_or("DENO_LOG", log_level.to_level_filter().to_string())
.write_style("DENO_LOG_STYLE"),
)
.filter_module("rustyline", log::LevelFilter::Off)
.filter_module("wgpu", log::LevelFilter::Error)
.filter_module("gfx", log::LevelFilter::Error)
.filter_module("globset", log::LevelFilter::Error)
.filter_module("deno::lsp::performance", log::LevelFilter::Debug)
.filter_module("rustls", log::LevelFilter::Off)
.filter_module("swc_ecma_codegen", log::LevelFilter::Off)
.filter_module("swc_common::source_map", log::LevelFilter::Off)
.filter_module("swc_ecma_transforms_optimization", log::LevelFilter::Off)
.filter_module("swc_ecma_parser", log::LevelFilter::Error)
.filter_module("swc_ecma_lexer", log::LevelFilter::Error)
.filter_module("tracing::span", log::LevelFilter::Off)
.filter_module("tower_lsp", log::LevelFilter::Trace)
.filter_module("opentelemetry_sdk", log::LevelFilter::Off)
.filter_module("editpe", log::LevelFilter::Error)
.filter_module("cranelift_codegen", log::LevelFilter::Off)
.write_style(if deno_terminal::colors::use_color() {
env_logger::WriteStyle::Always
} else {
env_logger::WriteStyle::Never
})
.format(|buf, record| {
let mut target = record.target().to_string();
if let Some(line_no) = record.line() {
target.push(':');
target.push_str(&line_no.to_string());
}
if record.level() <= log::Level::Info
|| (record.target() == "deno::lsp::performance"
&& record.level() == log::Level::Debug)
{
writeln!(buf, "{}", record.args())
} else {
writeln!(
buf,
"{} RS - {} - {}",
record.level(),
target,
record.args()
)
}
})
.build();
let otel_console_config = options
.otel_config
.map(|c| c.console)
.unwrap_or(OtelConsoleConfig::Ignore);
let cli_logger = LOGGER.get_or_init(move || CliLogger {
on_log_start: options.on_log_start,
on_log_end: options.on_log_end,
inner: ArcSwap::new(Arc::new(CliLoggerInner {
logger: env_logger::Builder::new().build(),
otel_console_config,
})),
});
cli_logger.inner.swap(Arc::new(CliLoggerInner {
logger,
otel_console_config,
}));
let _ = log::set_logger(cli_logger);
log::set_max_level(cli_logger.filter());
}