feat(logging): add logging.jl for custom logging utilities

Author: amaurigmartins

src/logging.jl

@@ -0,0 +1,50 @@
+using Logging
+using Logging: AbstractLogger, LogLevel, Info, global_logger
+using LoggingExtras: TeeLogger, FileLogger
+using Dates
+using Printf
+
+struct TimestampLogger <: AbstractLogger
+    logger::AbstractLogger
+end
+
+Logging.min_enabled_level(logger::TimestampLogger) = Logging.min_enabled_level(logger.logger)
+Logging.shouldlog(logger::TimestampLogger, level, _module, group, id) =
+    Logging.shouldlog(logger.logger, level, _module, group, id)
+
+function Logging.handle_message(logger::TimestampLogger, level, message, _module, group, id,
+    filepath, line; kwargs...)
+    timestamp = Dates.format(now(), "yyyy-mm-dd HH:MM:SS")
+    new_message = "[$timestamp] $message"
+    Logging.handle_message(logger.logger, level, new_message, _module, group, id,
+        filepath, line; kwargs...)
+end
+
+function setup_logging!(verbosity::Int, logfile::Union{String,Nothing}=nothing)
+    level = verbosity >= 2 ? Logging.Debug :
+            verbosity == 1 ? Logging.Info : Logging.Warn
+
+    # Create console logger
+    console_logger = ConsoleLogger(stderr, level)
+
+    if isnothing(logfile)
+        # Log to console only
+        global_logger(TimestampLogger(console_logger))
+    else
+        # Try to set up file logging with fallback to console-only
+        try
+            file_logger = FileLogger(logfile, level)
+            combined_logger = TeeLogger(console_logger, file_logger)
+            global_logger(TimestampLogger(combined_logger))
+        catch e
+            @warn "Failed to set up file logging to $(_display_path(logfile)): $e"
+
+            global_logger(TimestampLogger(console_logger))
+        end
+    end
+end
+
+function __init__()
+    # Set a default logging level when the package is loaded at runtime.
+    setup_logging!(0)
+end