#!/bin/bash
if ! type "include" > /dev/null 2>&1; then
  include() {
  . "$(cd "$(dirname "$1")" && pwd)/${2}";
  };
fi

include "${BASH_SOURCE[0]}" ../../pragma_once.bash
if ! pragma_once ${BASH_SOURCE[0]}; then return 0;
  source ../../pragma_once.bash
fi





# 这个文件是构建的基础文件 存放最顶层的函数
RESET='\033[0m'  # 重置颜色
BLACK='\033[0;30m'
RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[0;33m'
BLUE='\033[0;34m'
PURPLE='\033[0;35m'
CYAN='\033[0;36m'
WHITE='\033[0;37m'
BOLD_BLACK='\033[1;30m'
BOLD_RED='\033[1;31m'
BOLD_GREEN='\033[1;32m'
BOLD_YELLOW='\033[1;33m'
BOLD_BLUE='\033[1;34m'
BOLD_PURPLE='\033[1;35m'
BOLD_CYAN='\033[1;36m'
BOLD_WHITE='\033[1;37m'
export LOG_LEVEL="${LOG_LEVEL:-3}"               # 0=ERROR 1=WARN 2=INFO 3=DEBUG 4=TRACE
export LOG_PATH="${LOG_PATH:-}"                  # 空则不写文件
export LOG_STDOUT="${LOG_STDOUT:-1}"             # 1 输出终端
export LOG_COLOR_STDOUT="${LOG_COLOR_STDOUT:-1}" # 默认是否彩色
export LOG_INIT_MODE="${LOG_INIT_MODE:-rotate}" # truncate|append|rotate
export LOG_INITIALIZED=0


log_init() {
  # 如果没有设置 LOG_PATH，直接返回
  [[ -z "$LOG_PATH" ]] && return 0

  # 如果已经初始化过，直接返回
  (( LOG_INITIALIZED )) && return 0

  # 获取日志文件的目录
  local log_dir
  log_dir=$(dirname "$LOG_PATH")

  # 确保日志文件的目录存在，如果不存在则创建
  if [[ ! -d "$log_dir" ]]; then
    mkdir -p "$log_dir" || { echo "ERROR: Failed to create log directory: $log_dir"; exit 1; }
  fi

  # 确保日志文件存在，如果不存在则创建
  if [[ ! -f "$LOG_PATH" ]]; then
    touch "$LOG_PATH" || { echo "ERROR: Failed to create log file: $LOG_PATH"; exit 1; }
    chmod 777 $LOG_PATH
  fi

  # 根据初始化模式进行操作
  case "$LOG_INIT_MODE" in
  truncate)
    : > "$LOG_PATH"  # 清空日志文件
    ;;
  append)
  # 保持原样，直接追加
    ;;
  rotate)
  # 如果日志文件存在，重命名为带时间戳的文件
    if [[ -f "$LOG_PATH" ]]; then
      mv "$LOG_PATH" "$LOG_PATH.$(date +%Y%m%d_%H%M%S)"
    fi
    ;;
  *)
    echo "Invalid LOG_INIT_MODE: $LOG_INIT_MODE"
    exit 1
    ;;
  esac

  # 标记为已初始化
  LOG_INITIALIZED=1
}
log() {

  log_init

  local level_name="$1"
  local level_num="$2"
  local use_color="$3"
  shift 3

  (( level_num > LOG_LEVEL )) && return 0

  local fmt="$1"
  shift

  local body

  printf -v body -- "$fmt" "$@"
  local timestamp
  #timestamp="$(date '+%Y-%m-%d %H:%M:%S')"

  timestamp="$(TZ=Asia/Shanghai date '+%Y-%m-%d %H:%M:%S')"

  local msg
  printf -v msg "[%s] [%s] %s" \
    "$timestamp" \
    "$level_name" \
    "$body"

  # -------- stdout --------
  if (( LOG_STDOUT )); then

    local enable_color

    if [[ -z "$use_color" ]]; then
      enable_color="$LOG_COLOR_STDOUT"
    else
      enable_color="$use_color"
    fi

    if (( enable_color )) && [[ -t 1 ]]; then

      local color=""
      case "$level_name" in
      ERROR) color="$BOLD_RED" ;;
      WARN)  color="$BOLD_YELLOW" ;;
      INFO)  color="$GREEN" ;;
      DEBUG) color="$CYAN" ;;
      TRACE) color="$PURPLE" ;;
      *)     color="$WHITE" ;;
      esac

      printf "%b%s%b\n" "$color" "$msg" "$RESET"
    else
      printf "%s\n" "$msg"
    fi
  fi

  # -------- file --------
  if [[ -n "$LOG_PATH" ]]; then
    printf "%s\n" "$msg" >> "$LOG_PATH"
  fi
}


log_error() { log ERROR 0 1 "$@"; }
log_warn()  { log WARN  1 1 "$@"; }
log_info()  { log INFO  2 1 "$@"; }
log_debug() { log DEBUG 3 1 "$@"; }
log_trace() { log TRACE 4 1 "$@"; }