feat: configurable logging destination

Author: onbjerg

crates/cli/src/utils.rs

@@ -1,5 +1,7 @@
 //! Utility functions used by the Solar CLI.
 
+use std::io::{self};
+
 use solar_interface::diagnostics::DiagCtxt;
 
 #[cfg(feature = "mimalloc")]
@@ -43,9 +45,28 @@ pub const fn new_allocator() -> Allocator {
     new_wrapped_allocator()
 }
 
+#[derive(Default)]
+pub enum LogDestination {
+    #[default]
+    Stdout,
+    Stderr,
+}
+
+#[cfg(feature = "tracing")]
+impl<'a> tracing_subscriber::fmt::MakeWriter<'a> for LogDestination {
+    type Writer = Box<dyn io::Write>;
+
+    fn make_writer(&'a self) -> Self::Writer {
+        match self {
+            Self::Stdout => Box::new(std::io::stdout().lock()),
+            Self::Stderr => Box::new(std::io::stderr().lock()),
+        }
+    }
+}
+
 /// Initialize the tracing logger.
 #[must_use]
-pub fn init_logger() -> impl Sized {
+pub fn init_logger(dst: LogDestination) -> impl Sized {
     #[cfg(not(feature = "tracing"))]
     {
         if std::env::var_os("RUST_LOG").is_some() {
@@ -60,14 +81,14 @@ pub fn init_logger() -> impl Sized {
     }
 
     #[cfg(feature = "tracing")]
-    match try_init_logger() {
+    match try_init_logger(dst) {
         Ok(guard) => guard,
         Err(e) => DiagCtxt::new_early().fatal(e).emit(),
     }
 }
 
 #[cfg(feature = "tracing")]
-fn try_init_logger() -> Result<impl Sized, String> {
+fn try_init_logger(dst: LogDestination) -> Result<impl Sized, String> {
     use tracing_subscriber::prelude::*;
 
     let (profile_layer, guard) = match std::env::var("SOLAR_PROFILE").as_deref() {
@@ -90,7 +111,7 @@ fn try_init_logger() -> Result<impl Sized, String> {
     tracing_subscriber::Registry::default()
         .with(tracing_subscriber::EnvFilter::from_default_env())
         .with(profile_layer)
-        .with(tracing_subscriber::fmt::layer())
+        .with(tracing_subscriber::fmt::layer().with_writer(dst))
         .try_init()
         .map(|()| guard)
         .map_err(|e| e.to_string())

crates/solar/src/main.rs

@@ -12,7 +12,7 @@ static ALLOC: utils::Allocator = utils::new_allocator();
 fn main() -> ExitCode {
     signal_handler::install();
     panic_hook::install();
-    let _guard = utils::init_logger();
+    let _guard = utils::init_logger(Default::default());
     let args = match parse_args(std::env::args_os()) {
         Ok(args) => args,
         Err(e) => e.exit(),