fixup! Logger: Add color support for different log levels

This commit is contained in:
Fijxu 2024-10-08 18:36:50 -03:00
parent 17b525f2a6
commit d2edd4b63f
No known key found for this signature in database
GPG Key ID: 32C1DDF333EDA6A4
2 changed files with 5 additions and 3 deletions

View File

@ -224,7 +224,7 @@ https_only: false
##
## Enables colors in logs. Useful for debugging purposes
## This is overridden if "-k" or "--colorize"
## This is overridden if "-k" or "--colorize"
## are passed on the command line.
## Colors are also disabled if the environment variable
## NO_COLOR is present and has any value

View File

@ -12,7 +12,9 @@ enum LogLevel
end
class Invidious::LogHandler < Kemal::BaseLogHandler
def initialize(@io : IO = STDOUT, @level = LogLevel::Debug, @use_color : Bool = true)
def initialize(@io : IO = STDOUT, @level = LogLevel::Debug, use_color : Bool = true)
Colorize.enabled = use_color
Colorize.on_tty_only!
end
def call(context : HTTP::Server::Context)
@ -56,7 +58,7 @@ class Invidious::LogHandler < Kemal::BaseLogHandler
{% for level in %w(trace debug info warn error fatal) %}
def {{level.id}}(message : String)
if LogLevel::{{level.id.capitalize}} >= @level
puts("#{Time.utc} [{{level.id}}] #{message}".colorize(color(LogLevel::{{level.id.capitalize}})).toggle(@use_color))
puts("#{Time.utc} [{{level.id}}] #{message}".colorize(color(LogLevel::{{level.id.capitalize}})))
end
end
{% end %}