Content Comparison

#!/bin/bash
# -----------------------------------------------------------------------------------------------------------------------------
# Script: castor-change-log-level.sh
# -----------------------------------------------------------------------------------------------------------------------------
# Description:
# This script changes the log level for the Castor cluster or node(s) using the Swarm API.
# The script supports changing the log level for the entire cluster or individual node(s).
# The script can run in a detachable session using 'screen' or 'tmux'.
# -----------------------------------------------------------------------------------------------------------------------------
# Written by Milton Suen (milton.suen@datacore.com) Oct 31, 2024
# Revision History:
# v1.0.0 - Update to support running the script in a detachable session using screen or tmux.
# v1.1.0 - 2025-02-20 Add support node(s) level log level change.
# v1.2.0 - 2025-02-26 SUPSCR-208:
#        - Enforced proper credential formatting: credentials must be in the username:password format.
#        - Fixed help message shows script name without hard code it.
# v1.2.1 - 2025-02-26 SUPSCR-209: Auto detect CSN or SCS to adjust castor.log file path.
# v1.2.2 - 2025-02-27 Address the issue of the script not display correct when the castor.log file is rotated.
# v1.2.3 - 2025-02-27 Address the issue of log level not display correct within detach session.
# v1.2.4 - 2025-02-27 Bug fix: The default node-level log level is 0 (unset), which differs from the cluster-level default of 30.
# v1.2.5 - 2025-02-28 SUPSCR-208:
#        --------------------------- Credentials validation and password prompt enhancements.
#        - screen or tmux required with detachable mode, script will stopped with '-D' option if neither is installed.
#        - Fixed an issue where users were prompted to enter password multiple times issue.
#        - Passwords are now hidden in debug output messages.
# v1.2.6 - 2025-02-28 SUPSCR-208:
#        - Disabled credential display on the screen.
#        - Bug fix: Resolved the "debug: command not found" error when removing the -d [IP address] option.
# v1.3.0 - 2025-02-28 minor bug fix and enhancement.
# v1.3.1 - 2025-03-01 SUPSCR-208:
#        - Fixed an issue where credentials enclosed in single quotes (') were not processed correctly.
# -----------------------------------------------------------------------------------------------------------------------------
# Current Version: 1.23.31
# -----------------------------------------------------------------------------------------------------------------------------
# KB: https://perifery.atlassian.net/wiki/spaces/public/pages/3872161835/Setting+and+Managing+Swarm+Log+Levels+with+script
# -----------------------------------------------------------------------------------------------------------------------------

# Define colors
RED='\033[0;31m'
BOLD_RED='\033[1;31m'
GREEN='\033[0;32m'
BOLD_GREEN='\033[1;32m'
UNDERLINE_BOLD_GREEN='\033[4;32m'
YELLOW='\033[0;33m'
BOLD_YELLOW='\033[1;33m'
BLUE='\033[0;34m'
BOLD_BLUE='\033[1;34m'
MAGENTA='\033[0;35m'
BOLD_MAGENTA='\033[1;35m'
CYAN='\033[0;36m'
BOLD_CYAN='\033[1;36m'
RESET='\033[0m' # Reset color to default

SCRIPT_NAME=$(basename "$0")

# Function to display usage information
usage() {
    echo "Usage: ./$SCRIPT_NAME -d swarm_ip -p admin:password [-i new_log_level | -L new_node_log_level] [-t duration_in_seconds] [-D]"
    echo "  -d, --swarm_ip           IP address of the Swarm API endpoint. Supports single or multiple IPs separated by \",\", \";\" or \" \"."
    echo "                           If multiple IPs are provided, the script will update the log level for all nodes."
    echo "  

# Define colors
RED='\033[0;31m'
BOLD_RED='\033[1;31m'
GREEN='\033[0;32m'
BOLD_GREEN='\033[1;32m'
UNDERLINE_BOLD_GREEN='\033[4;32m'
YELLOW='\033[0;33m'
BOLD_YELLOW='\033[1;33m'
BLUE='\033[0;34m'
BOLD_BLUE='\033[1;34m'
MAGENTA='\033[0;35m'
BOLD_MAGENTA='\033[1;35m'
CYAN='\033[0;36m'
BOLD_CYAN='\033[1;36m'
RESET='\033[0m' # Reset color to default

SCRIPT_NAME=$(basename "$0")

# Function to display usage information
usage() {
    echo ""
    echo "Usage: ./$SCRIPT_NAME -d swarm_ip -p admin:password [-i new_log_level | -L new_node_log_level] [-t duration_in_seconds] [-D]"
    echo "  -d, --swarm_ip           IP address of the Swarm API endpoint. Supports single or multiple IPs separated by \",\", \";\" or \" \"."
    echo "                           If multiple IPs are provided, the script will update the log level for all nodes."
    echo "                           (Alternatively, set the SCSP_HOST environment variable to the Swarm IP.)"
    echo "  -p, --credentials        Credentials in the format admin:password"
    echo "  -i, --log.level          New cluster log level to set (5, 10, 15, 20, 30, 40, 50, chatter, debug, announce, info, error, critical, default)"
    echo "  -L, --node.log.level     New node log level to set (0, 5, 10, 15, 20, 30, 40, 50, chatter, debug, announce, info, error, critical, default)"
    echo "                           **Either -i or -L must be specified, but not both.**"
    echo "  -t, --time               (Optional) Duration in seconds to keep the new log level (must be greater than 0)"
    echo "  -D, --detach             (Optional) Detach the script from the current terminal and run in a detachable session using screen or tmux"
    echo ""
    exit 1
}

# Default options
detachable=false
debug=false
output_log="castor-change-log-level_output.log"		# Log file for capturing detachable session output
log_level_type="cluster"							# Default log level type
default_log_level=30								# Default log level
log_file="/var/log/datacore/castor.log"		        # Default log file location
SCRIPTDIR=$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )
JQLOCATION=$SCRIPTDIR/jq
MAX_RETRIES=3                                       # Maximum number of password retries attempts
username="admin"                                    # Default username
password=""                                         # Default password

# Global associative array for log levels
declare -A log_levels=(
    [5]="chatter"
    [10]="debug"
    [15]="audit"
    [20]="info"
    [30]="warning"
    [40]="error"
    [45]="defect"
    [50]="critical"
    [60]="announce"
    [30]="default"
)

# Global associative array for log level names
declare -A log_level_names=(
    ["chatter"]=5
    ["debug"]=10
    ["audit"]=15
    ["info"]=20
    ["warning"]=30
    ["error"]=40
    ["defect"]=45
    ["critical"]=50
    ["announce"]=60
    ["default"]=30
)

# Function to get the current timestamp
timestamp() {
    date +"%Y-%m-%d_%H:%M:%S.%4N"
}

# Function to display debug messages if debug mode is enabled
debug_msg() {
    if $debug; then
        local caller_line=${BASH_LINENO[0]:-unknown}  # Fallback to "unknown" if empty
        echo -e "$(timestamp) [DEBUG] (Line $caller_line) $*"
    fi
}

# Function to check if either 'screen' or 'tmux' is installed
check_screen_or_tmux() {
    if ! command -v screen &>/dev/null && ! command -v tmux &>/dev/null; then
        echo -e ""
        echo -e "---------------------------------------------------------------------------------------"
        echo -e "  ${YELLOW}Warning${RESET}: Neither '${BOLD_GREEN}screen${RESET}' nor '${BOLD_GREEN}tmux${RESET}' is installed. Cannot run in detachable mode."
        echo -e "  Please install either '${BOLD_GREEN}screen${RESET}' or '${BOLD_GREEN}tmux${RESET}' to run the script in a detachable session."
        echo -e "---------------------------------------------------------------------------------------"
        echo -e ""
        detachable=false  # Disable detachable session
        debug_msg "detachable=${YELLOW}false${RESET}"
        debug_msg "Stopping the script..."
        exit 1
    fi
}

# Function to format file size
format_size() {
    local size=$1
    if (( size >= 1073741824 )); then
        echo "$(awk "BEGIN {printf \"%.1fGB\", $size/1073741824}")"
    elif (( size >= 1048576 )); then
        echo "$(awk "BEGIN {printf \"%.1fMB\", $size/1048576}")"
    elif (( size >= 1024 )); then
        echo "$(awk "BEGIN {printf \"%.1fKB\", $size/1024}")"
    else
        echo "${size}B"
    fi
}

# Function to format duration
format_duration() {
    local duration=$1
    local hours=$((duration / 3600))
    local minutes=$(( (duration % 3600) / 60 ))
    local seconds=$((duration % 60))
    printf "%02d:%02d:%02d" $hours $minutes $seconds
}

# Function to check if jq is available and set up JSON parsing method
check_jq() {
    if [[ -x "/usr/local/bin/jq" ]]; then
        echo "/usr/local/bin/jq"
    elif [[ -x "$(pwd)/jq" ]]; then
        echo "$(pwd)/jq"
    elif command -v jq &>/dev/null; then
        echo "jq"
    else
        echo "grep"
    fi
}

if [[ -f "$JQLOCATION" ]]; then
    jq_or_grep=$JQLOCATION
else
    jq_or_grep=$(check_jq)
fi
debug_msg "jq_or_grep: $jq_or_grep"
#jq_or_grep=$(check_jq)

# Function to determine the log file path
determine_log_file() {
    if [[ -f "/var/log/datacore/castor.log" ]]; then
        echo "/var/log/datacore/castor.log"
    elif [[ -f "/var/log/caringo/castor.log" ]]; then
    (Alternatively, set the SCSP_HOST environment variable to the Swarm IP.)echo "/var/log/caringo/castor.log"
    echoelse
"  -p, --credentials        Credentials in the format admin:password"
    echo "  -i, --log.levelecho "Error: Log file not found in /var/log/datacore/castor.log or /var/log/caringo/castor.log"
        exit New1
cluster log level to set (5, 10, 15,fi
20,}
30, 40, 50, chatter, debug, announce, info, error, critical, default)"
    echo "  -L, --node.log.level
log_file=$(determine_log_file)

# Function to check if credentials are valid
check_credentials() {
    local CREDENTIALS="$1"
    New node log level to set (0, 5, 10, 15, 20, 30, 40, 50, chatter, debug, announce, info, error, critical, default)"
    echo "    local SWARM_IP="$2"
    debug_msg "Credentials: [hidden for security]"
    debug_msg "Swarm IP: $SWARM_IP"

    # API endpoint for validating user credentials
    local VALIDATE_URL="http://${SWARM_IP}:91/api/validateUser"
    debug_msg "Validate URL: $VALIDATE_URL"

    # Make the **Either -i or -L must be specified, but not both.**API request
    debug_msg "Validating user credentials..."
    echodebug_msg " curl -t,s --time               (Optional) Duration in seconds to keep the new log level (must be greater than 0)"
    echo "  -D, --detach             (Optional) Detach the script from the current terminal and run in a detachable session using screen or tmux"
    exit 1
}

# Default options
detachable=false
debug=false
output_log="castor-change-log-level_output.log"		# Log file for capturing detachable session output
log_level_type="cluster"							# Default log level type
default_log_level=30								# Default log level
log_file="/var/log/datacore/castor.log"		        # Default log file location
SCRIPTDIR=$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )
JQLOCATION=$SCRIPTDIR/jq

# Global associative array for log levels
declare -A log_levels=(
    [5]="chatter"
    [10]="debug"
    [15]="audit"
    [20]="info"
    [30]="warning"
    [40]="error"
    [45]="defect"
    [50]="critical"
    [60]="announce"
    [30]="default"
)

# Global associative array for log level names
declare -A log_level_names=(
    ["chatter"]=5
    ["debug"]=10
    ["audit"]=15
    ["info"]=20
    ["warning"]=30
    ["error"]=40
    ["defect"]=45
    ["critical"]=50
    ["announce"]=60
    ["default"]=30
)

# Function to get the current timestamp
timestamp() {
    date +"%Y-%m-%d_%H:%M:%S.%4N"
}

# Function to display debug messages if debug mode is enabled
debug_msg() {
    if $debug; thenu \"********\" -X GET \"$VALIDATE_URL\" -H 'Content-Type: application/json'"
    RESPONSE=$(curl -s -u "$CREDENTIALS" -X GET "$VALIDATE_URL" -H 'Content-Type: application/json')
    debug_msg "Validate User Response: $RESPONSE"

    # Check if the response contains "isValid": true
    if echo "$RESPONSE" | "$jq_or_grep" -e '.isValid == true' > /dev/null 2>&1; then
        debug_msg "Authentication successful for user '${CREDENTIALS%%:*}'"
        return 0  # Success
    elif echo "$RESPONSE" | "$jq_or_grep" -e '.isValid == false' > /dev/null 2>&1; then
        debug_msg "Authentication failed for user '${CREDENTIALS%%:*}'"
        return 1  # Failure
    else
        debug_msg "Error: Unable to validate credentials. Please check your inputs."
        return 1  # Failure
    fi
}

# Function to print credentials - hide password
print_credentials() {
    local CREDENTIALS="$1"
    local USERNAME="${CREDENTIALS%%:*}"
    echo -e "${GREEN}$USERNAME${RESET}":"${GREEN}********${RESET}"
}

# Parse input arguments
while [[ "$#" -gt 0 ]]; do
    case $1 in
        -d|--swarm_ip) swarm_ip="$2"; debug_msg "Set swarm_ip to $swarm_ip"; shift 2 ;;
        -p|--credentials)
            local caller_line=${BASH_LINENO[0]:-unknown}  # Fallback to "unknown" if emptycredentials="$2"
            shift 2 echo;;
-e "$(timestamp) [DEBUG] (Line $caller_line) $*"   -i|--log.level)
 fi }  # Function to check if either 'screen' or 'tmux' is installed
check_screen_or_tmux() {if [[ -n "$new_log_level" ]]; then
     if ! command -v screen &>/dev/null && ! command -v tmux &>/dev/null; then
        echo "Error: Neither 'screen' nor 'tmux' is installed. Cannot run in detachable mode." echo "Error: Options -i (cluster log leve) and -L (node log level) cannot be used together."
               detachable=false usage
# Disable detachable session         debug_msg "detachable=${YELLOW}false${RESET}"fi
     fi }  # Function to format fileif size
format_size() {[[ ${log_level_names[$2]} ]]; then
       local size=$1     if (( size >= 1073741824 )); thennew_log_level=${log_level_names[$2]}
            elif echo[[ "$(awk "BEGIN {printf \"%.1fGB\", $size/1073741824}")"${log_levels[$2]} ]]; then
      elif (( size >= 1048576 )); then    new_log_level=$2
    echo "$(awk "BEGIN {printf \"%.1fMB\", $size/1048576}")"   else
 elif (( size >= 1024 )); then         echo "$(awk "BEGIN {printf \"%.1fKB\", $size/1024}")"Invalid log level: $2"
       else         echo "${size}B"exit 1
    fi }  # Function to format duration format_duration()fi
{     local duration=$1     local hours=$((duration / 3600)) new_log_level_name=${log_levels[$new_log_level]}
           local minutes=$(( (duration % 3600) / 60 ))log_level_type="cluster"
           local seconds=$((duration % 60))default_log_level=30
      printf "%02d:%02d:%02d" $hours $minutes $seconds } debug_msg # Function"Set new_log_level to check if jq is available and set up JSON parsing method
check_jq() {$new_log_level ($new_log_level_name)"
            shift 2
     if [[ -x "/usr/local/bin/jq" ]]; then  ;;
      echo "/usr/local/bin/jq"
    elif [[ -x "$(pwd)/jq" ]]; then -L|--node.log.level)
            if [[ echo-n "$(pwd)/jq$new_log_level" ]]; then
  elif command -v jq &>/dev/null; then         echo "jq"Error: Options -i (cluster log elseleve) and -L (node log level) cannot be used echo together."grep"
    fi }  if [[ -f "$JQLOCATION" ]]; then     jq_or_grep=$JQLOCATION
elseusage
     jq_or_grep=$(check_jq)
fi
debug_msg "jq_or_grep: $jq_or_grep"
#jq_or_grep=$(check_jq)

# Function to determine the log file path
determine_log_file() {       fi
            if [[ -f "/var/log/datacore/castor.log"${log_level_names[$2]} ]]; then
        echo "/var/log/datacore/castor.log"       elif [[ -f "/var/log/caringo/castor.log" ]]; then new_log_level=${log_level_names[$2]}
            elif echo "/var/log/caringo/castor.log"
    else
[[ ${log_levels[$2]} || $2 -eq 0 ]]; then
       echo "Error: Log file not found in /var/log/datacore/castor.log or /var/log/caringo/castor.log"new_log_level=$2
        exit 1     fi } 
new_log_level_file=$(determine_log_file)

# Function to check if credentials are valid
check_credentials() {name="default"
            else
    local CREDENTIALS="$1"     local SWARM_IP="$2"     debug_msgecho "CredentialsInvalid log level: $CREDENTIALS$2"

   debug_msg "Swarm IP: $SWARM_IP"      # API endpoint for validatingexit user1
credentials     local VALIDATE_URL="http://${SWARM_IP}:91/api/validateUser"

       fi
   # Make the API request     RESPONSE=$(curl -s -u "$CREDENTIALS" -X GET "$VALIDATE_URL" -H 'Content-Type: application/json')
    debug_msg "Validate User Response: $RESPONSE"new_log_level_name=${log_levels[$new_log_level]}
            log_level_type="node"
            # Check if the response contains "isValid": truedefault_log_level=0
           if echodebug_msg "$RESPONSE" | "$jq_or_grep" -e '.isValid == true' > /dev/null 2>&1; then
Set new_log_level to $new_log_level ($new_log_level_name)"
            shift 2
       return 0  # Success ;;
   elif echo "$RESPONSE" | "$jq_or_grep" -e '.isValid == false' > /dev/null 2>&1; thent|--time)
            if [[  debug_msg "Authentication failed for user '${CREDENTIALS%%:*}'"
-n "$2" && "$2" != -* && "$2" -gt 0 ]]; then
       return 1  # Failure     elseduration="$2"
        debug_msg "Error: Unable to validate credentials. Please check yourdebug_msg inputs."
Set duration to $duration"
    return 1  # Failure     fi }  #shift Parse2
input arguments while [[ "$#" -gt 0 ]]; do    else
case $1 in         -d|--swarm_ip) swarm_ip="$2"; debug_msg "Set swarm_ip toecho $swarm_ip";Error: shiftDuration 2must ;;be a number greater than 0."
   -p|--credentials)             credentials="$2"exit 1
           if ! [[ "$credentials" =~ ^[^:]+:[^:]+$ ]]; thenfi
            ;;
       echo -e ""
        D|--detach) detachable=true; debug_msg "Set detachable to true"; shift ;;
       echo -e "Error: Credentials must be in the format ${BOLD_GREEN}username:password$-debug) debug=true; debug_msg "Set debug to true with ${YELLOW}--debug${RESET}"; shift ;;
        *) usage ;;
   exit 1esac
done

debug_msg "Set log_level_type to ${YELLOW}$log_level_type${RESET}"

# Set default values fiif not provided
debug_msg "Checking for default log level values..."

# Change ifthe [[ "$credentials" =~ [^a-zA-Z0-9:] ]]; then
                echo -e ""
 default log level based on Log Level Type
debug_msg "Changing default log level based on Log Level Type: ${YELLOW}$log_level_type${RESET}"
if [[ -n "$new_log_level" ]]; then
    if [[ "$log_level_type"        echo -e "Warning: Password contains special characters. Please enclose the credentials with single quotes (${BOLD_YELLOW}'${BOLD_GREEN}username:password${RESET}${BOLD_YELLOW}'${RESET})."== "node" ]]; then
        # new_log_level=0
        debug_msg        exit 1
  "Checing node default log level"
        default_log_level=0
fi        if [[ $new_log_level ==  debug_msg "Set credentials"; shift 2 ;;30 ]]; then
            -i|--log.level)new_log_level=$default_log_level
            if [[ -n "$newnew_log_level_name="default"
]]; then       fi
         echodebug_msg "Error:node Options -i (cluster log leve) and -L (node log level) cannot be used together."new_log_level: to $new_log_level"
        debug_msg "node - new_log_level_name: to $new_log_level_name"
    elif  usage
   [[ "$log_level_type" == "cluster" ]]; then
        fidefault_log_level=30
    fi
    debug_msg "New log iflevel: [[ ${GREEN}$new_log_level_names[$2]} ]]; thenlevel${RESET}"
    debug_msg "New log level name: ${GREEN}$new_log_level_name${RESET}"
    debug_msg "Set   newdefault_log_level=${ to: $default_log_level_names[$2]}
"
fi

# Check if 'screen' or 'tmux' is installed
if [[ elif [[ ${log_levels[$2]}"$detachable" == true ]]; then
    check_screen_or_tmux
fi

# If swarm_ip is not provided, try using  new_log_level=$2
       SCSP_HOST environment variable
if [[ -z "$swarm_ip" ]]; then
    elseif [[ -n "$SCSP_HOST" ]]; then
        swarm_ip="$SCSP_HOST"
   echo   "Invalid log level:debug_msg $2"Using Swarm IP from SCSP_HOST: $swarm_ip"
    else
       exit 1echo "Error: swarm_ip not provided and SCSP_HOST is not set."
   fi     usage
    fi
fi

new_log_level_name=${log_levels[$new_log_level]}
            log_level_type="cluster"
 # Check if required arguments are provided
if [[ -z "$credentials" || -z "$new_log_level" ]]; then
    usage
fi

# Split the default_log_level=30
            debug_msg "Set new_log_level to $new_log_level ($new_log_level_name)"
            shift 2
            ;;
        -L|--node.log.level)
            if [[ -n "$new_log_level" ]]; then
                echo "Error: Options -i (cluster log leve) and -L (node log level) cannot be used together."
  swarm_ip into an array of IP addresses if it contains delimiters
IFS=';, ' read -r -a ip_array <<< "$swarm_ip"

# Validate credentials before proceeding
debug_msg "Validating credentials..."
#if [[ "$credentials" =~ ^[^:]+$ ]]; then
if [[ -n "$credentials" ]]; then
    CREDENTIALS="$credentials"
    ATTEMPT=$MAX_RETRIES
    USERNAME=""
    PASSWORD=""
    debug_msg "CREDENTIALS: $(print_credentials "$CREDENTIALS")"
    debug_msg "ATTEMPT: $ATTEMPT"

    # Check if credentials contain a colon (username:password format)
    debug_msg "Checking for colon in credentials..."
    if [[ "$CREDENTIALS" == *":"* ]]; then
        debug_msg "Credentials contain colon"
 usage       USERNAME="${CREDENTIALS%%:*}"
     fi   PASSWORD="${CREDENTIALS#*:}"

        if# [[ ${log_level_names[$2]} ]]; then
  # Check if credentials contains special characters
        # debug_msg "Checking for  new_log_level=${log_level_names[$2]}
 special characters in credentials..."
        # if elif [[ ${log_levels[$2]} || $2 -eq 0"$PASSWORD" == *['!@#\$%^\&*()_+''-']* ]]; then
        #        new_log_level=$2
  debug_msg "Credentials contain special characters"
        #     new_log_level_name="defaultdebug_msg "Username: $USERNAME"
        #    else debug_msg "Password: $PASSWORD"
        #     echo -e "Invalid"
log level: $2"      #     echo -e "${YELLOW}Warning${RESET}: Password contains special exitcharacters. 1Please enclose the credentials in single quotes. ('username:password')"
     fi   #     echo -e    new_log_level_name=${log_levels[$new_log_level]}""
        #     log_level_type="node"
exit 1
        # fi
 default_log_level=0       # Validate credentials
   debug_msg  "Set new_log_level to $new_log_level ($new_log_level_name)check_credentials "$CREDENTIALS" "${ip_array[0]}"
        if [[ $? -ne shift0 2]]; then
           ;; debug_msg "Invalid credentials. Please check your username  -t|--time)and password."
            if [[echo -ne "$2"
&& "$2" != -* && "$2" -gt 0 ]]; then
                 echo -e "${RED}Error${RESET}: Invalid credentials. Please check your username and password."
          duration="$2"  echo -e ""
            debug_msg "Set duration to $duration"exit 1
        fi
        debug_msg "Validate shiftcredentials"
2        credentials=$CREDENTIALS
    else
        debug_msg "Credentials do not contain password"
  echo "Error: Duration must be a number greater than 0.debug_msg "Username: $(print_credentials "$CREDENTIALS")"
        USERNAME="$CREDENTIALS"
       exit 1# Prompt for password
        while fi[[ $ATTEMPTS -lt $MAX_RETRIES ]]; do
       ;;     echo    -D|--detach) detachable=true; debug_msg "Set detachable to true"; shift ;;""
            read -sp "Enter password --debug) debug=true; debug_msg "Set debug to true with ${YELLOW}--debug${RESET}"; shift ;;for user $USERNAME: " PASSWORD
            echo ""
*) usage ;;     esac done  debug_msg "Set log_level_type to ${YELLOW}$log_level_type${RESET}"

# Set default values if not provided
if [[ -z "$new_log_level" ]]; then echo ""
            CREDENTIALS="$USERNAME:$PASSWORD"
          if [[ "$log_level_type" == "cluster" ]]; thendebug_msg "CREDENTIALS: $(print_credentials "$CREDENTIALS"))"
            new_log_level=$default_log_levelcheck_credentials "$CREDENTIALS" "${ip_array[0]}"
       else     if [[ $?  new_log_level=-eq 0 ]]; then
  fi     new_log_level_name="default"     new_log_level_name=${log_levels[$new_log_level]}     debug_msg "Set default new_log_level to $new_log_level ($new_log_level_nameCREDENTIALS: $(print_credentials "$CREDENTIALS"))"
fi  # Check if 'screen' or 'tmux' is installed check_screen_or_tmux  # If swarm_ip is not provided, try using SCSP_HOST environment variable
if [[ -z "$swarm_ip" ]]; thendebug_msg "Valid credentials"
            if [[ -n "$SCSP_HOST" ]]; then credentials=$CREDENTIALS
         swarm_ip="$SCSP_HOST"       break
 debug "Using Swarm IP from SCSP_HOST: $swarm_ip"     else
         echo "Error: swarm_ip not provided and SCSP_HOST isecho not-e set.""
        usage     fi fi  #echo Check if required arguments are provided
if [[ -z "$credentials" || -z "$new_log_level" ]]; then-e "${RED}Error${RESET}: Invalid credentials. Please check your username and password."
           usage fi
 # Split the swarm_ip into an array of IP addresses if it contains delimiters
IFS=';, ' read -r -a ip_array <<< "$swarm_ip"

# Validate credentials before proceeding
debug_msg "Validating credentials..." ((ATTEMPTS++))
        done
    fi
else
    if ! check_credentials "$credentials" "${ip_array[0]}"; then
        echo -e ""
        echo -e "${YELLOW}Warning${RESET}: ${RED}Invalid credentials${RESET}."
Please check your username and password in the format ${GREEN}username:password${RESET}."
    echo -e ""
    usage     exit 1
    fi
fi

# Retrieve cluster name and handle JSON parsing using the first IP address
debug_msg "Retrieving the cluster name from Swarm API using IP: ${ip_array[0]}"
if [[ "$jq_or_grep" == "grep" ]]; then
    clusterName=$(curl --user "$credentials" -sS "http://${ip_array[0]}:91/api/storage/clusters" | grep -oP '"name":\s*"\K[^"]+')
else
    clusterName=$(curl --user "$credentials" -sS "http://${ip_array[0]}:91/api/storage/clusters" | "$jq_or_grep" -r '._embedded.clusters[0].name')
fi

if [[ -z "$clusterName" ]]; then
    echo "Failed to retrieve the cluster name. Please check your inputs."
    exit 1
fi
debug_msg "Cluster Name: $clusterName"

# Main logic function to run the script tasks
main_script() {
    local swarm_ip="$1"
    local credentials="$2"
    local new_log_level="$3"
    local new_log_level_name="$4"
    local duration="$5"
    local log_level_type="$6"
    #local log_file="/var/log/datacore/castor.log"
    if [[ -z "$log_file" ]]; then
        log_file=$(determine_log_file)
    fi
    local initial_size=$(stat -c%s "$log_file" 2>/dev/null || echo 0)
    local current_log_level
    local clusterName="$7"
    local jq_or_grep="$8"
    local detachable="$9"
    if [[ "$detachable" ]]; then
        local debug="${10}"
        eval "$(echo "${11}" | sed 's/declare -A/declare -A/')"
        eval "$(echo "${12}" | sed 's/declare -A/declare -A/')"
    fi
    debug_msg "******************************************************************"
    debug_msg "local variables"
    debug_msg "Log Level Type: ${GREEN}$log_level_type${RESET}"
    debug_msg "local variablesDefault Log Level: ${GREEN}$default_log_level${RESET}"
    debug_msg "Swarm IP: $swarm_ip"
    debug_msg "Credentials: $(print_credentials "$credentials")"
    debug_msg "New Log Level: $new_log_level"
    debug_msg "New log Level Name: $new_log_level_name"
    debug_msg "Duration: $duration"
    debug_msg "Log Level Type: $log_level_type"
    debug_msg "Log File: $log_file"
    debug_msg "Initial Log File Size: $initial_size"
    debug_msg "Current Log Level: $current_log_level"
    debug_msg "Cluster Name: $clusterName"
    debug_msg "jq_or_grep: $jq_or_grep"
    debug_msg "Detach: $detachable"
    debug_msg "Debug: $debug"
    debug_msg "**********************************************************"

    # Split the swarm_ip into an array of IP addresses
    IFS=';, ' read -r -a ip_array <<< "$swarm_ip"
    debug_msg "IP Array: ${ip_array[*]}"

    # Display initial information
    if [[ "$log_level_type" == "cluster" ]]; then
        echo -e "Swarm IP: ${GREEN}${ip_array[0]}${RESET}"
    else
        echo -e "Swarm IPs: ${GREEN}${ip_array[*]}${RESET}"
    fi
    # echo -e "Swarm IP: ${GREEN}$swarm_ip${RESET}"
    echo -e debug_msg "Credentials: ${GREEN}[hidden for security]${RESET}"
    echo -e "Cluster Name: ${GREEN}$clusterName${RESET}"

    debug_msg "Starting main_script function..."
    debug_msg "Log level type: $log_level_type"

    # Store the original log levels
    declare -A original_log_levels
    declare -A original_log_level_names

    if [[ $log_level_type == "cluster" ]]; then
        debug_msg "Setting cluster log level to $new_log_level ($new_log_level_name)"
        # Retrieve current log level
        if [[ "$jq_or_grep" == "grep" ]]; then
            current_log_level=$(curl --user "$credentials" -sS "http://${ip_array[0]}:91/api/storage/clusters/$clusterName/settings/log.level" | grep -oP '"value":\s*\K[0-9]+')
        else
            current_log_level=$(curl --user "$credentials" -sS "http://${ip_array[0]}:91/api/storage/clusters/$clusterName/settings/log.level" | "$jq_or_grep" -r '.value')
        fi
        current_log_level_name=${log_levels[$current_log_level]}
        debug_msg "Current cluster log level: ${BOLD_GREEN}$current_log_level${RESET}"
        debug_msg "Current log level name: ${BOLD_GREEN}$current_log_level_name${RESET}"
        echo -e ""
        echo -e "New cluster log level: ${BOLD_GREEN}$new_log_level_name${RESET}"
        echo -e "Current cluster log level is ${GREEN}$current_log_level_name.${RESET}"

        # Skip update if new level matches the current level
        if [[ "$current_log_level" -eq "$new_log_level" ]]; then
            echo ""
            echo -e "Cluster log level is already set to ${BOLD_GREEN}$new_log_level_name${RESET}. No changes made."
            return
        fi

        # Update the cluster log level using PUT
        debug_msg "Updating cluster log level to $new_log_level_name"
        debug_msg "Cluster Name: $clusterName"
    # Update the cluster log level using PUTdebug_msg "Credentials: $(print_credentials "$credentials")"
        debug_msg "Updatingcurl cluster log level to $new_log_level_name"
        debug_msg "Cluster Name: $clusterName--user \"$(print_credentials "$credentials")\" -sS -X PUT -H \"Content-Type: application/json\" \"http://${ip_array[0]}:91/api/storage/clusters/$clusterName/settings/log.level\" -d \"{\\\"value\\\": $new_log_level}\""
        response=$(curl --user "$credentials" -sS -X PUT -H "Content-Type: application/json" \
            "http://${ip_array[0]}:91/api/storage/clusters/$clusterName/settings/log.level" \
            -d "{\"value\": $new_log_level}")
        debug_msg "Response: $response"

        if [[ "$jq_or_grep" == "grep" ]]; then
            updated_log_level=$(echo "$response" | grep -oP '"value":\s*\K[0-9]+')
        else
            updated_log_level=$(echo "$response" | "$jq_or_grep" -r '.value')
        fi
        debug_msg "Updated cluster log level: $updated_log_level"

        if [[ "$updated_log_level" -eq "$new_log_level" ]]; then
            echo -e "Log level changed successfully from ${GREEN}$current_log_level${RESET} →-> ${BOLD_GREEN}$new_log_level${RESET}."
        else
            echo -e "Failed to update log level. Response: ${RED}$response${RESET}"
            exit 1
        fi

        # Countdown and revert log level
        if [[ -n "$duration" && "$duration" -gt 0 ]]; then
            echo -e "Keeping log level at ${YELLOW}$new_log_level_name${RESET} for ${YELLOW}$duration${RESET} second(s)seconds (${YELLOW}$(format_duration $duration)${RESET}) ..."
            echo -e ""
            for ((i=duration; i>0; i--)); do
                printf -v countdown "%02d:%02d:%02d" $((i/3600)) $(( (i%3600) / 60 )) $((i%60))
                echo -ne "Countdown: ${YELLOW}$countdown${RESET} remaining...\r"
                sleep 1
            done
            echo -e "\n\nTime's up! Reverting log level back to ${GREEN}$current_log_level${RESET}..."

            # Revert the log level back to the original value
            response=$(curl --user "$credentials" -sS -X PUT -H "Content-Type: application/json" \
                "http://${ip_array[0]}:91/api/storage/clusters/$clusterName/settings/log.level" \
                -d "{\"value\": $current_log_level}")
            debug_msg "Response: $response"

            if [[ "$jq_or_grep" == "grep" ]]; then
                reverted_log_level=$(echo "$response" | grep -oP '"value":\s*\K[0-9]+')
            else
                reverted_log_level=$(echo "$response" | "$jq_or_grep" -r '.value')
            fi
            debug_msg "Reverted cluster log level: $reverted_log_level"

            final_size=$(stat -c%s "$log_file" 2>/dev/null || echo 0)
            debug_msg "Initial log file size: $initial_size"
            debug_msg "Final log file size: $final_size"
            size_diff=$(( final_size - initial_size ))
            size_diff_formatted=$(format_size "$size_diff")
            duration_formatted=$(format_duration "$duration")
	    if (( size_diff < 0 )); then
		echo -e "castor.log file was rotated."
	    else
            echo -e ""
            echo -e "Approximate ${BOLD_GREEN}$size_diff_formatted${RESET} new logs were generated at log level ${BOLD_GREEN}$new_log_level${RESET}. Current castor.log size is ${BOLD_GREEN}$(format_size "$final_size")${RESET} after ${YELLOW}$duration${RESET} seconds (${YELLOW}$duration_formattedformatted${RESET})."
            echo -e ""
	    fi

            if [[ "$reverted_log_level" -eq "$current_log_level" ]]; then
                echo -e "Log level reverted successfully back to ${BOLD_GREEN}$current_log_level${RESET}."
                echo -e ""
            else
                echo -e "Failed to revert log level. Response: ${RED}$response${RESET}. Response: ${RED}$response${RESET}"
                echo -e ""
                exit 1
            fi
        else
            echo "Log level change is permanent until manually modified."
        fi
    elif [[ "$log_level_type" == "node" ]]; then
         # First loop: Change the node log level
        local same_log_level=false
        for ip in "${ip_array[@]}"; do
            # Retrieve current node log level
            debug_msg "Retrieving current node log level for IP: $ip"
            debug_msg "curl -s -u \"$(print_credentials "$credentials")\" \"http://$ip:91/api/storage/nodes/_self/settings/log.nodeLogLevel\""
            current_log_level=$(curl -s -u "$credentials" "http://$ip:91/api/storage/nodes/_self/settings/log.nodeLogLevel" | $jq_or_grep -r '.value')
            debug_msg "Current log level for IP $ip: $current_log_level"
            original_log_levels["$ip"]=$current_log_level
            debug_msg "Original log level for IP $ip: ${original_log_levels["$ip"]}"
            if [[ "$current_log_level" == "0" ]]; then
                current_log_level_name="default"
            else
                current_log_level_name=${log_levels[$current_log_level]}
            fi
            original_log_level_names["$ip"]=$current_log_level_name
            debug_msg "Original log level name: ${original_log_level_names["$ip"]}"
            if [[ "$new_log_level" == "0" ]]; then
                new_log_level_name="default"
            else
                new_log_level_name=${log_levels[$new_log_level]}
            fi
            debug_msg "Current log level for IP $ip: $current_log_level ($current_log_level_name)"
            echo ""
            echo -e "New node log level: ${BOLD_GREEN}$new_log_level_name${RESET}"
            echo -e "Current node log level for IP $ip is ${GREEN}$current_log_level_name.${RESET}"

            # Skip update if new level matches the current level
            if [[ "$current_log_level" -eq "$new_log_level" ]]; then
                same_log_level=true
                echo ""
                echo -e "Node log level for IP $ip is already set to ${BOLD_GREEN}$new_log_level_name${RESET}. No changes made."
                continue
            else
                same_log_level=false
            fi

            if [[ "$same_log_level" == false ]]; then
  fi              # Update the node log level using PUT
                debug_msg "Updating node log level to $new_log_level_name for IP $ip..."
                debug_msg "curl --user \"$(print_credentials "$credentials")\" -sS -X PUT -H \"Content-Type: application/json\" \"http://$ip:91/api/storage/nodes/_self/settings/log.nodeLogLevel?value=$new_log_level\""
                response=$(curl --user "$credentials" -sS -X PUT -H "Content-Type: application/json" \
                    "http://$ip:91/api/storage/nodes/_self/settings/log.nodeLogLevel?value=$new_log_level")
                debug_msg "Response: $response"

                if [[ "$jq_or_grep" == "grep" ]]; then
                    updated_log_level=$(echo "$response" | grep -oP '"value":\s*\K[0-9]+')
                else
                    updated_log_level=$(echo "$response" | "$jq_or_grep" -r '.value')
                fi
                debug_msg "Updated log level for IP $ip: $updated_log_level"

                if [[ "$updated_log_level" -eq "$new_log_level" ]]; then
                    echo -e "Node log level for IP $ip changed successfully from ${GREEN}$current_log_level_name${RESET} →-> ${BOLD_GREEN}$new_log_level_name${RESET}."
                else
                    echo -e "Failed to update node log level for IP $ip. Response: ${RED}$response${RESET}"
                    exit 1
                fi
            fi
        done

        # Second loop: Countdown if duration is provided
        if [[ "$same_log_level" == false ]]; then\
            if [[ -n "$duration" && "$duration" -gt 0 ]]; then; then
                echo -e ""
                echo -e "Keeping node(s) log level at ${YELLOW}$new_log_level_name${RESET} for ${YELLOW}$duration${RESET} second(s) seconds (${YELLOW}$(format_duration $duration)${RESET}) ..."
                echo -e ""
                for ((i=duration; i>0; i--)); do
                    printf -v countdown "%02d:%02d:%02d" $((i/3600)) $(( (i%3600) / 60 )) $((i%60))
                    echo -ne "Countdown: ${YELLOW}$countdown${RESET} remaining...\r"
                    sleep 1
                done
                echo -e "\n\nTime's up! Reverting node log level back to original levels..."

                # Third loop: Revert the node log level
                for ip in "${ip_array[@]}"; do
                    current_log_level=${original_log_levels["$ip"]}
                    current_log_level_name=${original_log_level_names["$ip"]}
                    debug_msg "Reverting node log level back to $current_log_level_name for IP $ip..."
                    debug_msg "curl --user \"$(print_credentials "$credentials")\" -sS -X PUT -H \"Content-Type: application/json\" \"http://$ip:91/api/storage/nodes/_self/settings/log.nodeLogLevel?value=$current_log_level\""
                    response=$(curl --user "$credentials" -sS -X PUT -H "Content-Type: application/json" \
                        "http://$ip:91/api/storage/nodes/_self/settings/log.nodeLogLevel?value=$current_log_level")
                    debug_msg "Response: $response"

                    if [[ "$jq_or_grep" == "grep" ]]; then
                        reverted_log_level=$(echo "$response" | grep -oP '"value":\s*\K[0-9]+')
                    else
                        reverted_log_level=$(echo "$response" | "$jq_or_grep" -r '.value')
                    fi

                    final_size=$(stat -c%s "$log_file" 2>/dev/null || echo 0)
                    size_diff=$(( final_size - initial_size ))
                    debug_msg "Initial log file size: $initial_size"
                    debug_msg "Final log file size: $final_size"
                    size_diff_formatted=$(format_size "$size_diff")
                    duration_formatted=$(format_duration "$duration")

                    if [[ "$reverted_log_level" -eq "$current_log_level" ]]; then
                        echo -e "Node log level for IP $ip reverted successfully back to ${BOLD_GREEN}${current_log_level_name}${RESET}."
                    else
                        echo -e "Failed to revert node log level for IP $ip. Response: ${RED}$response${RESET}"
                        exit 1
                    fi
                done

                # Combine the log output for multiple IP addresses into a single summary
	            if (( size_diff < 0 )); then
		
            echo -e "castor.log file was rotated."
	            else
                    echo -e ""
                    echo -e "Approximate ${BOLD_GREEN}$size_diff_formatted${RESET} new logs were generated at log level ${BOLD_GREEN}$new_log_level_name${RESET} for IP ${ip_array[*]}. Current castor.log size is ${BOLD_GREEN}$(format_size "$final_size")${RESET} after $duration_formatted."
	 ${YELLOW}$duration${RESET} seconds (${YELLOW}$duration_formatted${RESET})."
                    echo -e ""
            fi
   fi         else
                echo "Log level change is permanent until manually modified."
            fi
        fi
    fi
}

# Run in detachable or directly
if $detachable; then
    # Pass the main_script function to the screen session and store the output in a file
    debug_msg "**********************************************************" | tee -a "$output_log"
    debug_msg "Detach mode - Parameters passed to main_script:" | tee -a "$output_log"
    debug_msg "     Swarm IP: $swarm_ip" | tee -a "$output_log"
    debug_msg "     Credentials: $(print_credentials "$credentials")"  | tee -a "$output_log"
    debug_msg "     New Log Level: $new_log_level" | tee -a "$output_log"
    debug_msg "     New log Level Name: $new_log_level_name" | tee -a "$output_log"
    debug_msg "     Duration: $duration" | tee -a "$output_log"
    debug_msg "     Log Level Type: $log_level_type" | tee -a "$output_log"
    debug_msg "     Log File: $log_file" | tee -a "$output_log"
    debug_msg "     Initial Log File Size: $initial_size" | tee -a "$output_log"
    debug_msg "     Current Log Level: $current_log_level" | tee -a "$output_log"
    debug_msg "     Cluster Name: $clusterName" | tee -a "$output_log"
    debug_msg "     jq_or_grep: $jq_or_grep" | tee -a "$output_log"
    debug_msg "     Detach: $detachable" | tee -a "$output_log"
    debug_msg "     Debug: $debug" | tee -a "$output_log"
    debug_msg "**********************************************************" | tee -a "$output_log"

        # Convert associative arrays to strings and pass them to the screen session
    log_levels_string=$(declare -p log_levels)
    log_level_names_string=$(declare -p log_level_names)
    debug_msg "log_levels_string: $log_levels_string" | tee -a "$output_log"
    debug_msg "log_level_names_string: $log_level_names_string" | tee -a "$output_log"

    if command -v screen &>/dev/null; then
        echo -e "Running in ${YELLOW}screen${RESET} detachable mode..." | tee -a "$output_log"
        # if [[ "$debug" ]]; then
        #     echo -e "Debug mode enabled. Check the output log for debug messages."
        #     sleep 5
        # fi "$output_log"
        screen -dmS indexercastor_log_script bash -c "$(declare -f main_script timestamp debug_msg format_size format_duration check_jq determine_log_file print_credentials); main_script \"$swarm_ip\" \"$credentials\" \"$new_log_level\" \"$new_log_level_name\" \"$duration\" \"$log_level_type\" \"$clusterName\" \"$jq_or_grep\" \"$detachable\" \"$debug\" \"${log_levels_string}\" \"${log_level_names_string}\" | tee \"$output_log\""
        screen -r indexercastor_log_script
        # Wait for the screen session to complete and then display the output log
        sleep 1
        while screen -list | grep -q "indexercastor_log_script"; do
            sleep 1
        done
    elif command -v tmux &>/dev/null; then
        echo -e "Running in ${YELLOW}tmux${RESET} detachable mode..." | tee -a "$output_log"
        # if [[ "$debug" ]]; then
        #     echo -e "Debug mode enabled. Check the output log for debug messages."
        #     sleep 5
        # fi
        tmux new-session -d -s indexercastor_log_script "$(declare -f main_script timestamp debug_msg format_size format_duration check_jq determine_log_file print_credentials); main_script \"$swarm_ip\" \"$credentials\" \"$new_log_level\" \"$new_log_level_name\" \"$duration\" \"$log_level_type\" \"$clusterName\" \"$jq_or_grep\" \"$detachable\" \"$debug\" \"${log_levels_string}\" \"${log_level_names_string}\" | tee \"$output_log\""
        tmux attach-session -t indexercastor_log_script
    else
        echo "Error: Neither screen nor tmux available. Run without --detachable."
        exit 1
    fi

    echo ""
    cat "$output_log"
else
    # main_script "$swarm_ip" "$credentials" "$new_log_level" "$duration" "$clusterName" "$jq_or_grep" | tee "$output_log"
    debug_msg "**********************************************************"
    debug_msg "Parameters passed to main_script:"
    debug_msg "Swarm IP: $swarm_ip"
    debug_msg "Credentials: $(print_credentials "$credentials")"
    debug_msg "New Log Level: $new_log_level"
    debug_msg "Duration: $duration"
    debug_msg "Cluster Name: $clusterName"
    debug_msg "jq_or_grep: $jq_or_grep"
    debug_msg "**********************************************************"
    debug_msg "Running main_script function..."
    main_script "$swarm_ip" "$credentials" "$new_log_level" "$new_log_level_name" "$duration" "$log_level_type" "$clusterName" "$jq_or_grep" | tee "$output_log"
fi