Allow setting log level

floriandejonckheere · Mar 7, 2024 · 80b5852 · 80b5852
1 parent 9da2058
commit 80b5852
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/lib/mosaik/logger.rb b/lib/mosaik/logger.rb
@@ -8,18 +8,22 @@
 module MOSAIK
   class Logger < SimpleDelegator
     def initialize
-      super(::Logger.new($stdout, formatter:))
+      super(::Logger.new($stdout, level:, formatter:))
+    end
+
+    def level
+      ENV.fetch("LOG_LEVEL", "info")
     end
 
     private
 
     def formatter
       proc do |severity, _time, _progname, msg|
-        abort("#{File.basename($PROGRAM_NAME)}: #{msg}".white.on_red) if severity == "FATAL"
+        abort("#{File.basename($PROGRAM_NAME)}: #{msg}".white.on_red) if severity == ::Logger::FATAL
 
         msg = "#{msg}\n"
-        msg = msg.yellow if severity == "DEBUG"
-        msg = msg.red if severity == "ERROR"
+        msg = msg.yellow if severity == ::Logger::DEBUG
+        msg = msg.red if severity == ::Logger::ERROR
 
         msg
       end