#!/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
}