mirror of
https://github.com/versity/versitygw.git
synced 2025-12-23 05:05:16 +00:00
70 lines
1.6 KiB
Bash
70 lines
1.6 KiB
Bash
#!/usr/bin/env bash
|
|
|
|
# levels: 1 - crit, 2 - err, 3 - warn, 4 - info, 5 - debug, 6 - trace
|
|
|
|
log() {
|
|
if [[ $# -ne 2 ]]; then
|
|
echo "log function requires level, message"
|
|
return 1
|
|
fi
|
|
# shellcheck disable=SC2153
|
|
if [[ $1 -gt $LOG_LEVEL ]]; then
|
|
return 0
|
|
fi
|
|
log_level=""
|
|
case "$1" in
|
|
1) log_level="CRIT";;
|
|
2) log_level="ERROR";;
|
|
3) log_level="WARN";;
|
|
4) log_level="INFO";;
|
|
5) log_level="DEBUG";;
|
|
6) log_level="TRACE";;
|
|
*) echo "invalid log level $1"; return 1
|
|
esac
|
|
if [[ "$2" == *"--secret"* ]]; then
|
|
log_mask "$log_level" "$2"
|
|
return 0
|
|
fi
|
|
log_message "$log_level" "$2"
|
|
}
|
|
|
|
log_mask() {
|
|
if [[ $# -ne 2 ]]; then
|
|
echo "mask and log requires level, string"
|
|
return 1
|
|
fi
|
|
local masked_args=() # Initialize an array to hold the masked arguments
|
|
|
|
IFS=' ' read -r -a array <<< "$2"
|
|
|
|
mask_next=false
|
|
for arg in "${array[@]}"; do
|
|
if [[ $mask_next == true ]]; then
|
|
masked_args+=("********")
|
|
mask_next=false
|
|
elif [[ "$arg" == --secret_key=* ]]; then
|
|
masked_args+=("--secret_key=********")
|
|
elif [[ "$arg" == --secret=* ]]; then
|
|
masked_args+=("--secret=********")
|
|
else
|
|
if [[ "$arg" == "--secret_key" ]] || [[ "$arg" == "--secret" ]]; then
|
|
mask_next=true
|
|
fi
|
|
masked_args+=("$arg")
|
|
fi
|
|
done
|
|
log_message "$log_level" "${masked_args[*]}"
|
|
}
|
|
|
|
log_message() {
|
|
if [[ $# -ne 2 ]]; then
|
|
echo "log message requires level, message"
|
|
return 1
|
|
fi
|
|
now="$(date "+%Y-%m-%d %H:%M:%S")"
|
|
echo "$now $1 $2"
|
|
if [[ -n "$TEST_LOG_FILE" ]]; then
|
|
echo "$now $1 $2" >> "$TEST_LOG_FILE"
|
|
fi
|
|
}
|