FEATURE: format std log and add --short option (#9091)

# Description
- prettify formatting
- move message formatting to a private function
- allow short prefixes for loggers via `--short|-s` flag

# User-Facing Changes
- allow short prefixes for loggers via `--short|-s` flag

# Tests + Formatting
<!--
Don't forget to add tests that cover your changes.

Make sure you've run and fixed any issues with these commands:

- `cargo fmt --all -- --check` to check standard code formatting (`cargo
fmt --all` applies these changes)
- `cargo clippy --workspace -- -D warnings -D clippy::unwrap_used -A
clippy::needless_collect -A clippy::result_large_err` to check that
you're using the standard code style
- `cargo test --workspace` to check that all tests pass
- `cargo run -- crates/nu-std/tests/run.nu` to run the tests for the
standard library

> **Note**
> from `nushell` you can also use the `toolkit` as follows
> ```bash
> use toolkit.nu # or use an `env_change` hook to activate it
automatically
> toolkit check pr
> ```
-->

# After Submitting
<!-- If your PR had any user-facing changes, update [the
documentation](https://github.com/nushell/nushell.github.io) after the
PR is merged, if necessary. This will help us keep the docs up to date.
-->

---------

Co-authored-by: amtoine <stevan.antoine@gmail.com>
This commit is contained in:
Emily Grace Seville
2023-05-08 18:13:21 +10:00
committed by GitHub
parent fe60fb8679
commit a2dd948e71
2 changed files with 336 additions and 92 deletions

View File

@ -1,52 +1,131 @@
export def "log CRITICAL_LEVEL" [] { 50 }
export def "log ERROR_LEVEL" [] { 40 }
export def "log WARNING_LEVEL" [] { 30 }
export def "log INFO_LEVEL" [] { 20 }
export def "log DEBUG_LEVEL" [] { 10 }
export def "log CRITICAL_LEVEL_PREFIX" [] { "CRT" }
export def "log ERROR_LEVEL_PREFIX" [] { "ERR" }
export def "log WARNING_LEVEL_PREFIX" [] { "WRN" }
export def "log INFO_LEVEL_PREFIX" [] { "INF" }
export def "log DEBUG_LEVEL_PREFIX" [] { "DBG" }
def parse-string-level [level: string] {
(
if ($level in [(log CRITICAL_LEVEL_PREFIX) "CRIT" "CRITICAL"]) {
log CRITICAL_LEVEL
} else if ($level in [(log ERROR_LEVEL_PREFIX) "ERROR" ]) {
log ERROR_LEVEL
} else if ($level in [(log WARNING_LEVEL_PREFIX) "WARN" "WARNING"]) {
log WARNING_LEVEL
} else if ($level in [(log DEBUG_LEVEL_PREFIX) "DEBUG"]) {
log DEBUG_LEVEL
} else {
log INFO_LEVEL
}
)
export def "log CRITICAL_LEVEL" [] {
50
}
def parse-int-level [level: int] {
(
if ($level >= (log CRITICAL_LEVEL)) {
export def "log ERROR_LEVEL" [] {
40
}
export def "log WARNING_LEVEL" [] {
30
}
export def "log INFO_LEVEL" [] {
20
}
export def "log DEBUG_LEVEL" [] {
10
}
def parse-string-level [
level: string
] {
if $level in [(log CRITICAL_LEVEL_PREFIX) (log CRITICAL_LEVEL_PREFIX --short) "CRIT" "CRITICAL"] {
log CRITICAL_LEVEL
} else if $level in [(log ERROR_LEVEL_PREFIX) (log ERROR_LEVEL_PREFIX --short) "ERROR" ] {
log ERROR_LEVEL
} else if $level in [(log WARNING_LEVEL_PREFIX) (log WARNING_LEVEL_PREFIX --short) "WARN" "WARNING"] {
log WARNING_LEVEL
} else if $level in [(log DEBUG_LEVEL_PREFIX) (log DEBUG_LEVEL_PREFIX --short) "DEBUG"] {
log DEBUG_LEVEL
} else {
log INFO_LEVEL
}
}
export def "log CRITICAL_LEVEL_PREFIX" [
--short (-s)
] {
if $short {
"C"
} else {
"CRT"
}
}
export def "log ERROR_LEVEL_PREFIX" [
--short (-s)
] {
if $short {
"E"
} else {
"ERR"
}
}
export def "log WARNING_LEVEL_PREFIX" [
--short (-s)
] {
if $short {
"W"
} else {
"WRN"
}
}
export def "log INFO_LEVEL_PREFIX" [
--short (-s)
] {
if $short {
"I"
} else {
"INF"
}
}
export def "log DEBUG_LEVEL_PREFIX" [
--short (-s)
] {
if $short {
"D"
} else {
"DBG"
}
}
def parse-int-level [
level: int,
--short (-s)
] {
if $level >= (log CRITICAL_LEVEL) {
if $short {
log CRITICAL_LEVEL_PREFIX --short
} else {
log CRITICAL_LEVEL_PREFIX
} else if ($level >= (log ERROR_LEVEL)) {
}
} else if $level >= (log ERROR_LEVEL) {
if $short {
log ERROR_LEVEL_PREFIX --short
} else {
log ERROR_LEVEL_PREFIX
} else if ($level >= (log WARNING_LEVEL)) {
}
} else if $level >= (log WARNING_LEVEL) {
if $short {
log WARNING_LEVEL_PREFIX --short
} else {
log WARNING_LEVEL_PREFIX
} else if ($level >= (log INFO_LEVEL)) {
}
} else if $level >= (log INFO_LEVEL) {
if $short {
log INFO_LEVEL_PREFIX --short
} else {
log INFO_LEVEL_PREFIX
}
} else {
if $short {
log DEBUG_LEVEL_PREFIX --short
} else {
log DEBUG_LEVEL_PREFIX
}
)
}
}
def current-log-level [] {
let env_level = ($env | get -i NU_LOG_LEVEL | default (log INFO_LEVEL))
let env_level = ($env.NU_LOG_LEVEL? | default (log INFO_LEVEL))
try {
($env_level | into int)
$env_level | into int
} catch {
parse-string-level $env_level
}
@ -56,52 +135,129 @@ def now [] {
date now | date format "%Y-%m-%dT%H:%M:%S%.3f"
}
# Log critical message
export def "log critical" [message: string] {
if (current-log-level) > (log CRITICAL_LEVEL) { return }
print --stderr $"(ansi red_bold)(log CRITICAL_LEVEL_PREFIX)|(now)|($message)(ansi reset)"
}
# Log error message
export def "log error" [message: string] {
if (current-log-level) > (log ERROR_LEVEL) { return }
print --stderr $"(ansi red)(log ERROR_LEVEL_PREFIX)|(now)|($message)(ansi reset)"
}
# Log warning message
export def "log warning" [message: string] {
if (current-log-level) > (log WARNING_LEVEL) { return }
print --stderr $"(ansi yellow)(log WARNING_LEVEL_PREFIX)|(now)|($message)(ansi reset)"
}
# Log info message
export def "log info" [message: string] {
if (current-log-level) > (log INFO_LEVEL) { return }
print --stderr $"(ansi default)(log INFO_LEVEL_PREFIX)|(now)|($message)(ansi reset)"
}
# Log debug message
export def "log debug" [message: string] {
if (current-log-level) > (log DEBUG_LEVEL) { return }
print --stderr $"(ansi default_dimmed)(log DEBUG_LEVEL_PREFIX)|(now)|($message)(ansi reset)"
def log-formatted [
color: string,
prefix: string,
message: string
] {
print --stderr $"($color)($prefix)|(now)|(ansi u)($message)(ansi reset)"
}
# Log with custom message format and verbosity level
# Log a critical message
export def "log critical" [
message: string, # A message
--short (-s) # Whether to use a short prefix
] {
if (current-log-level) > (log CRITICAL_LEVEL) {
return
}
let prefix = if $short {
log CRITICAL_LEVEL_PREFIX --short
} else {
log CRITICAL_LEVEL_PREFIX
}
log-formatted (ansi red_bold) $prefix $message
}
# Log an error message
export def "log error" [
message: string, # A message
--short (-s) # Whether to use a short prefix
] {
if (current-log-level) > (log ERROR_LEVEL) {
return
}
let prefix = if $short {
log ERROR_LEVEL_PREFIX --short
} else {
log ERROR_LEVEL_PREFIX
}
log-formatted (ansi red) $prefix $message
}
# Log a warning message
export def "log warning" [
message: string, # A message
--short (-s) # Whether to use a short prefix
] {
if (current-log-level) > (log WARNING_LEVEL) {
return
}
let prefix = if $short {
log WARNING_LEVEL_PREFIX --short
} else {
log WARNING_LEVEL_PREFIX
}
log-formatted (ansi yellow) $prefix $message
}
# Log an info message
export def "log info" [
message: string, # A message
--short (-s) # Whether to use a short prefix
] {
if (current-log-level) > (log INFO_LEVEL) {
return
}
let prefix = if $short {
log INFO_LEVEL_PREFIX --short
} else {
log INFO_LEVEL_PREFIX
}
log-formatted (ansi default) $prefix $message
}
# Log a debug message
export def "log debug" [
message: string, # A message
--short (-s) # Whether to use a short prefix
] {
if (current-log-level) > (log DEBUG_LEVEL) {
return
}
let prefix = if $short {
log DEBUG_LEVEL_PREFIX --short
} else {
log DEBUG_LEVEL_PREFIX
}
log-formatted (ansi default_dimmed) $prefix $message
}
# Log a message with a specific format and verbosity level
#
# Format reference:
# > %MSG% will be replaced by $message
# > %DATE% will be replaced by the timestamp of log in standard Nushell's log format: "%Y-%m-%dT%H:%M:%S%.3f"
# > %LEVEL% will be replaced by the standard Nushell's log verbosity prefixes, e.g. "CRT"
# - %MSG% will be replaced by $message
# - %DATE% will be replaced by the timestamp of log in standard Nushell's log format: "%Y-%m-%dT%H:%M:%S%.3f"
# - %LEVEL% will be replaced by the standard Nushell's log verbosity prefixes, e.g. "CRT"
#
# Examples:
# > std log custom "my message" $"(ansi yellow)[%LEVEL%]MY MESSAGE: %MSG% [%DATE%](ansi reset)" (std log WARNING_LEVEL)
# - std log custom "my message" $"(ansi yellow)[%LEVEL%]MY MESSAGE: %MSG% [%DATE%](ansi reset)" (std log WARNING_LEVEL)
export def "log custom" [
message: string, # Message inserted into the log template
format: string, # A string to be printed into stderr.
log_level: int # Log's verbosity level
] {
if (current-log-level) > ($log_level) { return }
message: string, # A message
format: string, # A format
log_level: int # A log level
--short (-s) # Whether to use a short prefix
] {
if (current-log-level) > ($log_level) {
return
}
print --stderr ($format | str replace "%MSG%" $message | str replace "%DATE%" (now) | str replace "%LEVEL%" (parse-int-level $log_level | into string))
}
let level = ((if $short {
parse-int-level $log_level --short
} else {
parse-int-level $log_level
}) | into string)
print --stderr ([
["%MSG%" $message]
["%DATE%" (now)]
["%LEVEL%" $level]
] | reduce --fold $format {
|it, acc| $acc | str replace --all $it.0 $it.1
})
}