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 minorMinor 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.
# v1.3.2 - 2025-03-02 minorMinor bug fix and enhancement.
# v1.3.3 - 2025----------03-02 Fixed tmux command to long issue.
# v1.3.4 - Enhanced password handling to better support password with special characters.
# -----------------------------------------------------------------------------------------------------------------------------
# Current Version: 1.3.24
# -----------------------------------------------------------------------------------------------------------------------------
# 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 -e ""
    echo -e "${BOLD_GREEN}Usage:${RESET} ./$SCRIPT_NAME -d swarm_ip -p admin:password [-i new_log_level | -L new_node_log_level] [-t duration_in_seconds] [-D | --detach]"
    echo -e "  -d, --swarm_ip           IP address of the Swarm API endpoint. Supports single or multiple IPs separated by \",\", \";\" or \" \"."
    echo -e "                           If multiple IPs are provided, the script will update the log level for all nodes."
    echo -e "                           (Alternatively, set the SCSP_HOST environment variable to the Swarm IP.)"
    echo -e "  -p, --credentials        Credentials in the format admin:password"
    echo -e "  -i, --log.level          Set cluster log level to set (5, 10, 15, 20, 30, 40, 50)"
    echo -e "--

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

# 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
)

# Separate default values for cluster and node levels
declare -A default_log_levels=(
    ["cluster"]=30
    ["node"]=0
)

# Separate default values for cluster and node level names
declare -A default_log_level_names=(
    ["cluster"]="default"
    ["node"]="default"
)

# Function to get the current timestamp
timestamp() {
    date -u +"%Y-%m-%dT%H:%M:%S.%3NZ"
}

# Function to display debug messages if debug mode is enabled
debug_msg() {
    if $debug; then
        local caller_func=${FUNCNAME[1]:-main}                  # Fallback to "unknown" if empty
    Aliases:  chatter, debug, announce, info, error, critical, default."local caller_line=${BASH_LINENO[0]:-unknown}        echo -e "  -L, --node.log.level     Set node log level to set (0, 5, 10, 15, 20, 30, 40, 50)# Fallback to "unknown" if empty
        echo -e "$(timestamp) [DEBUG] ($caller_func:$caller_line) $*"
    echofi
-e}
"
# Function to check if either 'screen' or 'tmux' is installed
check_screen_or_tmux() {
    if ! command -v screen &>/dev/null && ! command Aliases: chatter, debug, announce, info, error, critical, default."-v tmux &>/dev/null; then
        echo -e "" 
        echo -e "---------------------------------------------------------------------------------------"
        echo -e "  ${YELLOW}Warning${RESET}:  **Either -i or -L must be specified, but not both.**"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 -t, --time     run the script in a detachable session."
         (Optional) Duration in seconds to keep the new log level (must be greater than 0)"echo -e "---------------------------------------------------------------------------------------"
        detachable=false  # Disable detachable session
    echo -e "  -D, --detachusage
    fi
}

# Function to format file size
format_size(Optional) Detach the script from the current terminal and run in a detachable session using screen or tmux" {
    local size=$1
    if (( size >= 1073741824 )); then
        echo -e " "$(awk "BEGIN {printf \"%.1fGB\", $size/1073741824}")"
    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=3elif (( 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))
 # Maximum number of password retries attempts
username="admin"      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 DefaultJSON username
password=""  parsing method
check_jq() {
    for jq_path in "/usr/local/bin/jq" "$(pwd)/jq"; do
        [[ -x "$jq_path" ]] &&       echo "$jq_path" && return
    done
   # Defaultcommand password-v 
# 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"
)

# 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
)

# Separate default values for cluster and node levels
declare -A default_log_levels=(
    ["cluster"]=30
    ["node"]=0
)

# Separate default values for cluster and node level names
declare -A default_log_level_names=(
    ["cluster"]="default"
    ["node"]="default"
)

# Function to get the current timestamp
timestamp() {
    date -u +"%Y-%m-%dT%H:%M:%S.%3NZ"
}

# Function to display debug messages if debug mode is enabled
debug_msg() {
    if $debug; then
        local caller_func=${FUNCNAME[1]:-main}                  # Fallback to "unknown" if empty
        local caller_line=${BASH_LINENO[0]:-unknown}            # Fallback to "unknown" if empty
        echo -e "$(timestamp) [DEBUG] ($caller_func:$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 modejq &>/dev/null && echo "jq" || echo "grep"
}

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() {
    for log in "/var/log/datacore/castor.log" "/var/log/caringo/castor.log"; do
        if [[ -f "$log" ]]; then
            echo "$log"
            return
        fi
    done
    echo -e "${RED}Error: Log file not found.${RESET}"
    exit 1
}

log_file=$(determine_log_file)

# Function to check if credentials are valid
check_credentials() {
    local CREDENTIALS="$1"
    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 API request
    debug_msg "Validating user credentials..."
    debug_msg "curl -s -u \"********\" -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."
        echo -e "return 1  # Failure
  Please install either '${BOLD_GREEN}screen${RESET}' or '${BOLD_GREEN}tmux${RESET}' to run the script in a detachable session."
   fi
}

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

# Parse input arguments
while [[ "$#" -gt 0 ]]; do
    case detachable=false$1 in
# Disable detachable session     -d|--swarm_ip) swarm_ip="$2"; debug_msg "Set if [[ "$detachable" == false ]]; thenswarm_ip to $swarm_ip"; shift 2 ;;
        -p|--credentials)
    debug_msg "detachable=${YELLOW}false${RESET}"       credentials="$2"
     debug_msg "Stopping the script..."
          shift 2 ;;
        exit 1-i|--log.level)
         fi   if [[ fi
}

# Function to format file size
format_size() {-n "$new_log_level" ]]; then
             local size=$1  echo "Error: Options if-i (( size >= 1073741824 )); then
  cluster log leve) and -L (node log level) cannot be used together."
     echo "$(awk "BEGIN {printf \"%.1fGB\", $size/1073741824}")"     elif ((usage
size >= 1048576 )); then        fi
echo "$(awk "BEGIN {printf \"%.1fMB\", $size/1048576}")"     elif (( size >= 1024 )); then log_level_type="cluster"
           echo "$(awk "BEGIN {printf \"%.1fKB\", $size/1024}")"if [[ ${log_level_names[$2]} ]]; then
      else         echo "${size}B" new_log_level=${log_level_names[$2]}
      fi }  # Function to formatelif duration
format_duration() {
    local duration=$1[[ "$2" == "default" ]]; then
       local hours=$((duration / 3600))     local minutes=$(( (duration % 3600) / 60 ))new_log_level=${default_log_levels[$log_level_type]}
               local seconds=$((duration % 60)) new_log_level_name="default"
            elif printf "%02d:%02d:%02d" $hours $minutes $seconds
}

# Function to check if jq is available and set up JSON parsing method
check_jq() {[[ ${log_levels[$2]} ]]; then
                new_log_level=$2
     for jq_path in "/usr/local/bin/jq" "$(pwd)/jq"; do  else
      [[ -x "$jq_path" ]] && echo "$jq_path" && return  echo "Invalid log donelevel: $2"
   command -v jq &>/dev/null && echo "jq" || echo "grep" }  if [[exit -f "$JQLOCATION" ]]; then1
     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
determinenew_log_file() {level_name=${log_levels[$new_log_level]}
    for log in "/var/log/datacore/castor.log" "/var/log/caringo/castor.log"; do   default_log_level=30
     if [[ -f "$log" ]]; then  debug_msg "Set new_log_level to $new_log_level ($new_log_level_name)"
     echo "$log"      shift 2
     return       ;;
 fi     done  -L|--node.log.level)
  echo -e "${RED}Error: Log file not found.${RESET}"    if exit[[ 1
}

log_file=$(determine_log_file)

# Function to check if credentials are valid
check_credentials() {-n "$new_log_level" ]]; then
                localecho CREDENTIALS="$1"Error: Options -i (cluster log local SWARM_IP="$2"
    debug_msg "Credentials: [hidden for security]leve) and -L (node log level) cannot be used together."
    debug_msg "Swarm IP: $SWARM_IP"      # API endpoint forusage
validating user credentials     local VALIDATE_URL="http://${SWARM_IP}:91/api/validateUser"    fi
debug_msg  "Validate URL: $VALIDATE_URL"      # Make the API request log_level_type="node"
      debug_msg "Validating user credentials..."   if  debug_msg "curl -s -u \"********\" -X GET \"$VALIDATE_URL\" -H 'Content-Type: application/json'"[[ ${log_level_names[$2]} ]]; then
             RESPONSE=$(curl -s -u "$CREDENTIALS" -X GET "$VALIDATE_URL" -H 'Content-Type: application/json') new_log_level=${log_level_names[$2]}
          debug_msg "Validate Userelif Response: $RESPONSE[[ "$2" == "default" ]]; then
 # Check if the response contains "isValid": true     if echo "$RESPONSE" | "$jq_or_grep" -e '.isValid == true' > /dev/null 2>&1; then new_log_level=${default_log_levels[$log_level_type]}
                debug_msg "Authentication successful for user '${CREDENTIALS%%:*}'"new_log_level_name="default"
            returnelif 0  # Success
    elif echo "$RESPONSE" | "$jq_or_grep" -e '.isValid == false' > /dev/null 2>&1; then
 [[ ${log_levels[$2]} || $2 -eq 0 ]]; then
                new_log_level=$2
      debug_msg "Authentication failed for user '${CREDENTIALS%%:*}'"     new_log_level_name="default"
   return 1  # Failure     else
        debug_msg "Error: Unable to validate credentials. Please check yourecho inputs."Invalid log level: $2"
     return 1  # Failure     fi } exit #1
Function to print credentials - hide password print_credentials() {    fi
local CREDENTIALS="$1"     local USERNAME="${CREDENTIALS%%:*}"     echo -e "${GREEN}$USERNAME${RESET}":"${GREEN}********${RESET}"
}

# Parse input arguments
while [[ "$#" -gt 0 ]]; donew_log_level_name=${log_levels[$new_log_level]}
            default_log_level=0
    case $1 in         -d|--swarm_ip) swarm_ip="$2"; debug_msg "Set swarmnew_log_iplevel to $swarm_ip";$new_log_level ($new_log_level_name)"
            shift 2
            ;;
        -pt|--credentialstime)
            credentials=if [[ -n "$2" && "$2" != -* && "$2" -gt 0 ]]; then
  shift 2 ;;         -i|--log.level)   duration="$2"
         if [[ -n "$new_log_level" ]]; then  debug_msg "Set duration to $duration"
          echo "Error: Options -i (cluster log leve)shift and2
-L (node log level) cannot be used together."     else
           usage     echo -e ""
     fi           echo  log_level_type="cluster"-e "---------------------------------------------------------------------------------------"
            if [[ ${log_level_names[$2]} ]]; thenecho -e "    ${RED}Error${RESET}: Duration must be a number greater     new_log_level=than ${log_level_names[$2]}BOLD_GREEN}0${RESET}."
            elif [[ "$2" == "default" ]]; thenecho -e "    Please specify the duration using the ${BOLD_GREEN}-t${RESET} ${GREEN}<seconds>${RESET}      new_log_level=or ${default_log_levels[$log_level_type]}
      BOLD_GREEN}--time${RESET} ${GREEN}<seconds>${RESET} option."
         new_log_level_name="default"       echo -e "---------------------------------------------------------------------------------------"
   elif [[ ${log_levels[$2]} ]]; then         usage
       new_log_level=$2     fi
       else     ;;
        -D|--detach) detachable=true; debug_msg echo "Invalid log level: $2"
   Set detachable to true"; shift ;;
        --debug) debug=true; debug_msg "Set exitdebug 1to true with ${YELLOW}--debug${RESET}"; shift ;;
       fi *) usage ;;
    esac
done

debug_msg "Set new_log_level_name=type to ${log_levels[$new_log_level]}
            default_log_level=30
  YELLOW}$log_level_type${RESET}"

# Check if 'screen' or 'tmux' is installed
if [[ "$detachable" == true ]]; then
    if     debug_msg "Set new_log_level to $new_log_level ($new_log_level_name)"
            shift 2[[ -z "$duration" ]]; then
        echo -e ""
     ;;   echo -e "---------------------------------------------------------------------------------------"
   -L|--node.log.level)     echo -e "    ${RED}Error${RESET}: Duration ifmust [[be -n "$new_log_level" ]]; then
 specified when running in detachable mode."
        echo -e "    echoPlease "Error:specify Optionsthe -iduration (clusterusing log leve) and -L (node log level) cannot be used together."
       the ${BOLD_GREEN}-t${RESET} or ${BOLD_GREEN}--time${RESET} option."
        echo -e "---------------------------------------------------------------------------------------"
        usage
    fi
      check_screen_or_tmux
fi

# If swarm_ip is not provided, try using    log_level_type="node"
            SCSP_HOST environment variable
if [[ ${log_level_names[$2]} -z "$swarm_ip" ]]; then
                new_log_level=${log_level_names[$2]}
            elif [[ "$2" == "defaultif [[ -n "$SCSP_HOST" ]]; then
                new_log_level=${default_log_levels[$log_level_type]}swarm_ip="$SCSP_HOST"
        debug_msg "Using Swarm IP from    new_log_level_name="defaultSCSP_HOST: $swarm_ip"
    else
       elif [[ ${log_levels[$2]} || $2 -eq 0 ]]; then
 echo "Error: swarm_ip not provided and SCSP_HOST is not set."
        usage
    fi
fi
new_log_level=$2
# Check if required arguments are provided
if [[ -z "$credentials" ||     new-z "$new_log_level_name="default" ]]; then
    usage
fi

# Split the elseswarm_ip into an array of IP addresses if it contains delimiters
IFS=';, ' read -r -a ip_array echo<<< "Invalid log level: $2$swarm_ip"

# Validate credentials before proceeding
debug_msg "Validating credentials..."
#if [[ "$credentials" =~ ^[^:]+$ ]]; then
exitif 1[[ -n "$credentials" ]]; then
    CREDENTIALS="$credentials"
   fi ATTEMPT=$MAX_RETRIES
    USERNAME=""
      new_log_level_name=${log_levels[$new_log_level]}PASSWORD=""
    debug_msg "CREDENTIALS: $(print_credentials "$CREDENTIALS")"
    default_log_level=0
 debug_msg "ATTEMPT: $ATTEMPT"

    # # Capture the original debug_msg "Set new_log_level to $new_log_level ($new_log_level_name)"
command as executed by the user
    # ORIGINAL_INPUT=$(ps -o args= -p $$)
 shift 2  # debug_msg "ORIGINAL_INPUT: $ORIGINAL_INPUT"

    # Check ;;if credentials contain a colon (username:password format)
  -t|--time)  if [[ "$CREDENTIALS" == *":"* ]]; then
    if [[ -n "$2" && debug_msg "$2"Checking !=for -*colon && "$2in credentials..."
-gt 0 ]]; then     debug_msg "Credentials contain colon"
        durationUSERNAME="$2${CREDENTIALS%%:*}"
        PASSWORD="${CREDENTIALS#*:}"
        debug_msg "Username: "Set duration to $duration"
$USERNAME"
        debug_msg "Password: $PASSWORD"

        # Define special characters that may shiftcause 2issues
        debug_msg "Defining special characters..."
else        SPECIAL_CHARS="!@#\$%^&*()_+{}|:<>?~\`-=[]\\;',./\""
        echodebug_msg "ErrorSpecial characters: Duration$SPECIAL_CHARS"
must
be a number greater than 0."   # Debug: Show the final regex pattern
       exit 1debug_msg "Special characters regex: $SPECIAL_REGEX"

       fi # Check if password start with '$'
     ;;   #   When credentials  -D|--detach) detachable=true; debug_msg "Set detachable to true"; shift ;;contains ":" and password is empty
        # --debug) debug=true; debug_msg "Set debug to true with ${YELLOW}--debug${RESET}"; shift ;; it will be treated as environment variable 
        # *) usage ;;Bash will try to expand esacthe donevariable and debug_msg "Set log_level_type to ${YELLOW}$log_level_type${RESET}"

# Check if 'screen' or 'tmux' is installed
if [[ "$detachable" == true if it starts with '$'
        if [[ -z "$PASSWORD" ]]; then
    check_screen_or_tmux
fi

# If swarm_ip is not provided, try using SCSP_HOST environment variable
if [[ -z "$swarm_ip" ]]; then        debug_msg "Password is empty, bash may interpret the password as an environment variable."
            echo -e ""
           if [[echo -n "$SCSP_HOST" ]]; thene "${RED}Error${RESET}: Password cannot start with '${BOLD_YELLOW}\$'${RESET}."
            swarm_ip="$SCSP_HOST"
        debug_msg "Using Swarm IP from SCSP_HOST: $swarm_ip"echo -e "Bash may interpret the password as an environment variable."
         else   echo -e "To prevent issues, enclose echothe "Error: swarm_ip not provided and SCSP_HOST is not set."password in single quotes (${BOLD_YELLOW}'${RESET})."
            usageecho -e ""
  fi fi  # Check if required arguments are provided ifecho [[ -ze "$credentials" || -z "$new_log_level" ]]; then${GREEN}Example${RESET}:"
          usage fi echo #-e 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..."
#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)-p ${BOLD_YELLOW}'${RESET}${GREEN}admin:\$password123${RESET}${BOLD_YELLOW}'${RESET}"
            echo -e ""
            exit 1
        fi
        
        # Validate credentials
        check_credentials "$CREDENTIALS" "${ip_array[0]}"
        if [[ $? -ne 0 ]]; then
            debug_msg "Checking for colon in credentials..."
 Invalid credentials. Please check your username and password."
  if [[ "$CREDENTIALS" == *":"* ]]; then    echo -e ""
  debug_msg "Credentials contain colon"       echo -e USERNAME="${CREDENTIALS%%:*}"
        PASSWORD="${CREDENTIALS#*:}"RED}Error${RESET}: Invalid credentials. Please check your username and password."
            #echo Validate-e credentials"If your password contains special characters, enclose it in check_credentials "$CREDENTIALS" "single quotes (${ip_array[0]}"
     BOLD_YELLOW}'${RESET}) to prevent misinterpretation by the system."
  if [[ $? -ne 0 ]]; then    echo -e "${GREEN}Special characters include${RESET}: ${YELLOW}$SPECIAL_CHARS${RESET}"
      debug_msg "Invalid credentials. Please check your usernameecho and-e password.""
            echo -e "${GREEN}Example${RESET}:"
            echo -e "    -p ${RED}Error$BOLD_YELLOW}'${RESET}${GREEN}: Invalid credentials. Please check your username and password.admin:password!@#${RESET}${BOLD_YELLOW}'${RESET}"
            echo -e ""
            exit 1
        fi
        debug_msg "Validate credentials"
        credentials=$CREDENTIALS
    else
        debug_msg "Credentials do not contain password"
        debug_msg "Username: $(print_credentials "$CREDENTIALS")"
        USERNAME="$CREDENTIALS"
        # Prompt for password
        while [[ $ATTEMPTS -lt $MAX_RETRIES ]]; do
            echo ""
            read -sp "Enter password for user $USERNAME: " PASSWORD
            echo ""
            echo ""
            if [[ -z "$PASSWORD" ]]; then
                echo -e "${RED}Error${RESET}: Password cannot be empty."
                ((ATTEMPTS++))
                continue
            fi
            CREDENTIALS="$USERNAME:$PASSWORD"
            debug_msg "CREDENTIALS: $(print_credentials "$CREDENTIALS"))"
            check_credentials "$CREDENTIALS" "${ip_array[0]}"
            if [[ $? -eq 0 ]]; then
                debug_msg "CREDENTIALS: $(print_credentials "$CREDENTIALS"))"
                debug_msg "Valid credentials"
                credentials=$CREDENTIALS
                break
            else
                echo -e ""
                echo -e "${RED}Error${RESET}: Invalid credentials. Please check your username and password."
                echo -e ""
            fi
            ((ATTEMPTS++))
        done
        if [[ $ATTEMPTS -ge $MAX_RETRIES ]]; then
            echo -e ""
            echo -e "${RED}Error${RESET}: Maximum number of password attempts reached. Exiting script."
            exit 1
        fi
    fi
else
    if ! check_credentials "$credentials" "${ip_array[0]}"; then
        echo -e ""
        echo -e "${YELLOW}Warning${RESET}: ${RED}Invalid credentials${RESET}."
        echo -e ""
        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
    local default_log_levels="$13"
    local default_log_levels_name="$14"
    debug_msg "**********************************************************"
    debug_msg "local variables"
    debug_msg "Log Level Type: ${GREEN}$log_level_type${RESET}"
    debug_msg "Default 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 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}"
    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

    default_log_level=${default_log_levels[$log_level_type]}
    default_log_level_name=${default_log_level_names[$log_level_type]}
    debug_msg "Default log level: $default_log_level"
    debug_msg "Default log level name: $default_log_level_name"

    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]}
        if [[ $current_log_level == 30 ]]; then
            current_log_level_name="default"
        fi
        if [[ $new_log_level == 30 ]]; then
            new_log_level_name="default"
        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} (${BOLD_GREEN}$new_log_level${RESET})"
        echo -e "Current cluster log level is ${BOLD_GREEN}$current_log_level_name.$name${RESET} (${BOLD_GREEN}$current_log_level${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} (${BOLD_GREEN}$new_log_level${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"
        debug_msg "Credentials: $(print_credentials "$credentials")"
        debug_msg "curl --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 "${GREEN}$(timestamp)${RESET} Log level changed successfully from ${BOLD_GREEN}$current_log_level_name${RESET} (${BOLD_GREEN}$current_log_level${RESET}) -> ${BOLD_GREEN}$new_log_level_name${RESET} (${BOLD_GREEN}$new_log_level${RESET})."
        else
            echo -e "${GREEN}$(timestamp)${RESET} 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} (${BOLD_GREENYELLOW}$new_log_level${RESET}) for ${YELLOW}$duration${RESET} 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_name${RESET} (${BOLD_GREEN}$current_log_level${RESET}) ..."

            # Revert the log level back to the original value
            # echo -e "Level log revert on $(timestamp)"
            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"
           fi size_diff=$(( final_size - initial_size ))
       debug_msg "Reverted cluster log level: $reverted_log_level" size_diff_formatted=$(format_size "$size_diff")
             finalduration_sizeformatted=$(stat -c%s "$log_file" 2>/dev/null || echo 0)
format_duration "$duration")
	    if (( size_diff < 0 )); then
 debug_msg		echo -e "Initial castor.log file size: $initial_sizewas rotated."
	    else
       debug_msg "Final log file size: $final_size     echo -e ""
            echo -e "Approximate size_diff=$(( final_size - initial_size ))
            size_diff_formatted=${BOLD_GREEN}$size_diff_formatted${RESET} new logs were generated at log level ${BOLD_GREEN}$new_log_level_name${RESET} (${BOLD_GREEN}$new_log_level${RESET}). Current castor.log size is ${BOLD_GREEN}$(format_size "$size$final_diffsize")${RESET} after ${YELLOW}$duration${RESET}           duration_formatted=$(format_duration "$duration")
	seconds (${YELLOW}$duration_formatted${RESET})."
     if (( size_diff < 0 )); then 		echo -e "castor.log file was rotated."
	    elsefi
        if    echo[[ "$reverted_log_level" -eeq "$current_log_level" ]]; then
            echo -e "Approximate ${BOLD_GREEN}$size_diff_formatted$$(timestamp)${RESET} newLog logslevel werereverted generatedsuccessfully atback logto level ${BOLD_GREEN}$new$current_log_level_name${RESET} (${BOLD_GREEN}$new$current_log_level${RESET}). Current castor.log size is ${BOLD_GREEN}$(format_size "$final_size""
            echo -e ""
        else
            echo -e "${GREEN}$(timestamp)${RESET} after ${YELLOW}$duration${RESET} seconds (${YELLOW}$duration_formatted${RESET})."
  Failed to revert log level. Response: ${RED}$response${RESET}"
            echo -e ""
         echo -e "" 	exit 1
  fi      fi
  if [[ "$reverted_log_level" -eq "$current_log_level" ]]; thenelse
            echo -e "${GREEN}$(timestamp)${RESET} Log level revertedchange successfullyis backpermanent to ${BOLD_GREEN}$current_log_level_name${RESET} (${BOLD_GREEN}$current_log_level${RESET})until manually modified."
        fi
   echo -e elif [[ "$log_level_type" == "node" ]]; then
    else    # First loop: Change the node log level
 echo   -e "${GREEN}$(timestamp)${RESET} Failed to revertlocal same_log level. Response: ${RED}$response${RESET}"
     _level=false
        for ip in "${ip_array[@]}"; do
      echo -e ""    # Retrieve current node log level
   exit 1        debug_msg fi"Retrieving current node log level for IP: $ip"
 else           debug_msg "curl echo-s -eu \"${GREEN}$(timestamp)${RESET} Log level change is permanent until manually modified."(print_credentials "$credentials")\" \"http://$ip:91/api/storage/nodes/_self/settings/log.nodeLogLevel\""
            current_log_level=$(curl -s -u fi
    elif [[ "$log_level_type" == "node" ]]; then"$credentials" "http://$ip:91/api/storage/nodes/_self/settings/log.nodeLogLevel" | $jq_or_grep -r '.value')
          # First loop: Change the node debug_msg "Current log level for IP       local same$ip: $current_log_level=false"
         for ip in "${ip_array[@]}"; do
            # Retrieve current node log  original_log_levels["$ip"]=$current_log_level
            debug_msg "RetrievingOriginal current node log level for IP $ip: ${original_log_levels["$ip"]}"
            debug_msg "curl -s -u \"$(print_credentials "$credentials")\" \"http://$ip:91/api/storage/nodes/_self/settings/log.nodeLogLevel\""if [[ "$current_log_level" == "0" ]]; then
                current_log_level=$(curl -s -u "$credentials" "http://$ip:91/api/storage/nodes/_self/settings/log.nodeLogLevel" | $jq_or_grep -r '.value')
  _name="default"
            else
         debug_msg "Current log level for IP $ip:  current_log_level_name=${log_levels[$current_log_level"]}
            fi
            original_log_level_levelsnames["$ip"]=$current_log_level_name
            debug_msg "Original log level for IP $ipname: ${original_log_levelslevel_names["$ip"]}"
            if [[ "$current$new_log_level" == "0" ]]; then
                currentnew_log_level_name="default"
            else
                currentnew_log_level_name=${log_levels[$current$new_log_level]}
            fi
            original_log_level_names["$ip"]=$current_log_level_name
            debug_msg "Original log level name: ${originaldebug_msg "Current log level for IP $ip: $current_log_level ($current_log_level_names["$ip"]}"name)"
            ifecho [[ "$new_log_level"
== "0" ]]; then         echo -e "New node log    newlevel: ${BOLD_GREEN}$new_log_level_name="default"name${RESET} (${BOLD_GREEN}$new_log_level${RESET})"
            elseecho -e "Current node log level for           newIP ${BOLD_GREEN}$ip${RESET} is ${BOLD_GREEN}$current_log_level_name=name${RESET} (${log_levels[$newBOLD_GREEN}$current_log_level]}level${RESET})."

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

            if [[ "$current$same_log_level" -eq "$new_log_level"== false ]]; then
                # Update the node log level using PUT
                debug_msg "Updating node log level to $new_log_level_name for IP $ip..."
            same_log_level=true    debug_msg "curl --user \"$(print_credentials "$credentials")\" -sS -X PUT -H \"Content-Type:    echo application/json\" \"http://$ip:91/api/storage/nodes/_self/settings/log.nodeLogLevel?value=$new_log_level\""
                echoresponse=$(curl -e-user "$credentials"Node log-sS level-X forPUT IP ${BOLD_GREEN}$ip${RESET} is already set to ${BOLD_GREEN}$new_log_level_name${RESET} (${BOLD_GREEN}$new_log_level${RESET}). No changes made."-H "Content-Type: application/json" \
                    "http://$ip:91/api/storage/nodes/_self/settings/log.nodeLogLevel?value=$new_log_level")
    continue             else
debug_msg "Response: $response"

              same_log_level=false  if [[ "$jq_or_grep" == "grep" ]]; then
    fi              if [[ "$sameupdated_log_level=$(echo "$response" == false ]]; then| grep -oP '"value":\s*\K[0-9]+')
                else
      # Update the node log level using PUT       updated_log_level=$(echo "$response" | "$jq_or_grep" -r '.value')
    debug_msg "Updating node log level to $new_log_level_name for IP $ip..."   fi
             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\"" debug_msg "Updated log level for IP $ip: $updated_log_level"

                response=$(curl --user "$credentials" -sS -X PUT -H "Content-Type: application/json" \
if [[ "$updated_log_level" -eq "$new_log_level" ]]; then
                   "http://$ip:91/api/storage/nodes/_self/settings/log.nodeLogLevel?value=$new_log_level")
                debug_msg "Response: $response echo -e "${GREEN}$(timestamp)${RESET} Node log level for IP ${BOLD_GREEN}$ip${RESET} changed successfully from ${BOLD_GREEN}$current_log_level_name${RESET} (${GREEN}$current_log_level${RESET}) -> ${BOLD_GREEN}$new_log_level_name${RESET} (${BOLD_GREEN}$new_log_level${RESET})."
                else
if [[ "$jq_or_grep" == "grep" ]]; then                     updated_log_level=$(echo "$response" | grep -oPe '"value":\s*\K[0-9]+')
          "${GREEN}$(timestamp)${RESET} Failed to update node log level for IP ${BOLD_GREEN}$ip${RESET}. Response: ${RED}$response${RESET}"
     else               exit 1
    updated_log_level=$(echo "$response" | "$jq_or_grep" -r '.value')          fi
      fi      fi
        done

debug_msg "Updated log level for IP $ip: $updated_log_level"  # Second loop: Countdown if duration is provided
        if [[ "$updated$same_log_level" -eq "$new_log_level"== false ]]; then\
                    echoif [[ -en "${GREEN}$(timestamp)${RESET} Node log level for IP ${BOLD_GREEN}$ip${RESET} changed successfully from ${GREEN}$current_log_level_name${RESET} (${GREEN}$current_log_level${RESET}) -> ${BOLD_GREEN}$new_log_level_name${RESET} (${BOLD_GREEN}$new_log_level${RESET})."
 "$duration" && "$duration" -gt 0 ]]; then
              else  echo -e ""
                echo -e "${GREEN}$(timestamp)${RESET} Failed to update node log level for IP ${BOLD_GREEN}$ip${RESET}. Response: ${RED}$response${RESET}"
   Keeping node(s) log level at ${YELLOW}$new_log_level_name${RESET} (${YELLOW}$new_log_level${RESET}) for ${YELLOW}$duration${RESET} seconds (${YELLOW}$(format_duration $duration)${RESET}) ..."
                exitecho 1-e ""
               fi for ((i=duration; i>0;          fi
        done

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

               for ((i=duration; i>0; i--)); do
    # Third loop: Revert the node log level
                printffor -vip countdownin "%02d:%02d:%02d" $((i/3600)) $(( (i%3600) / 60 )) $((i%60))
  ${ip_array[@]}"; do
                    current_log_level=${original_log_levels["$ip"]}
                 echo -ne "Countdown: ${YELLOW}$countdown${RESET} remaining...\r" current_log_level_name=${original_log_level_names["$ip"]}
                    debug_msg sleep"Reverting 1node log level back to $current_log_level_name for IP $ip..."
        done            debug_msg "curl --user \"$(print_credentials "$credentials")\" -sS -X echoPUT -e "\n\nTime's up! Reverting node log level back to original levels..."H \"Content-Type: application/json\" \"http://$ip:91/api/storage/nodes/_self/settings/log.nodeLogLevel?value=$current_log_level\""
                    # echo -e "Node level log revert on $(timestamp) for IP $ip"
# Third loop: Revert the node log level             response=$(curl --user "$credentials" -sS for-X ipPUT in "${ip_array[@]}"; do-H "Content-Type: application/json" \
                        current_log_level=${original_log_levels["$ip"]}"http://$ip:91/api/storage/nodes/_self/settings/log.nodeLogLevel?value=$current_log_level")
                    current_log_level_name=${original_log_level_names["$ip"]}debug_msg "Response: $response"

                   debug_msg "Reverting node log level backif to[[ $current"$jq_log_level_name for IP $ip..."or_grep" == "grep" ]]; then
                     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\"" reverted_log_level=$(echo "$response" | grep -oP '"value":\s*\K[0-9]+')
                    else
        # echo -e "Node level log revert on $(timestamp) for IP $ip"     reverted_log_level=$(echo "$response" | "$jq_or_grep" -r '.value')
          response=$(curl --user "$credentials" -sS -X PUT -H "Content-Type: application/json" \ fi

                    final_size=$(stat -c%s "http://$ip:91/api/storage/nodes/_self/settings/log.nodeLogLevel?value=$current_log_level"$log_file" 2>/dev/null || echo 0)
                    debug_msg "Response: $response"

size_diff=$(( final_size - initial_size ))
                   if [[ "$jq_or_grep" == "grep" ]]; then
   debug_msg "Initial log file size: $initial_size"
                    reverted_log_level=$(echodebug_msg "$response"Final |log grepfile -oP '"value":\s*\K[0-9]+')size: $final_size"
                    else
size_diff_formatted=$(format_size "$size_diff")
                       reverted_log_levelduration_formatted=$(echoformat_duration "$response" | "$jq_or_grep" -r '.value'$duration")

                   fi if [[ "$reverted_log_level" -eq "$current_log_level" ]]; then
              final_size=$(stat -c%s "$log_file" 2>/dev/null || echo 0)       echo -e "${GREEN}$(timestamp)${RESET} Node log level for IP ${BOLD_GREEN}$ip${RESET} reverted successfully back   size_diff=$(( final_size - initial_size ))to ${BOLD_GREEN}${current_log_level_name}${RESET} (${BOLD_GREEN}$current_log_level${RESET})."
                    debug_msgelse
"Initial log file size: $initial_size"                    echo debug_msg "Final log file size: $final_size"
            -e "${GREEN}$(timestamp)${RESET} Failed to revert node log level for IP ${BOLD_GREEN}$ip${RESET}. Response: ${RED}$response${RESET}"
       size_diff_formatted=$(format_size "$size_diff")                exit 1
   duration_formatted=$(format_duration "$duration")                fi
     if [[ "$reverted_log_level" -eq "$current_log_level" ]]; then     done

                # Combine echothe -e "${GREEN}$(timestamp)${RESET} Node log level for IP ${BOLD_GREEN}$ip${RESET} reverted successfully back to ${BOLD_GREEN}${current_log_level_name}${RESET} (${BOLD_GREEN}$current_log_level${RESET})."log output for multiple IP addresses into a single summary
            if (( size_diff < 0 )); then
     else       echo -e "castor.log file was rotated."
            echoelse
-e "${GREEN}$(timestamp)${RESET} Failed to revert node log level for IP ${BOLD_GREEN}$ip${RESET}. Response: ${RED}$response${RESET}"        echo -e ""
              exit 1     echo -e "Approximate ${BOLD_GREEN}$size_diff_formatted${RESET} new logs were generated at log level ${BOLD_GREEN}$new_log_level_name${RESET} (${BOLD_GREEN}$new_log_level${RESET}) for IP  fi
         ${ip_array[*]}. Current castor.log size is ${BOLD_GREEN}$(format_size "$final_size")${RESET} after ${YELLOW}$duration${RESET} seconds (${YELLOW}$duration_formatted${RESET})."
      done              echo -e ""
 # Combine the log output for multiple IP addresses into a singlefi
summary            else
if (( size_diff < 0 )); then             echo -e "castor.log file was rotated."
    ${GREEN}$(timestamp)${RESET} Log level change is permanent until manually modified."
       else     fi
        fi
    fi
}
echo
-e# ""Run in detachable or directly
if $detachable; then
    # Pass the main_script function to    echo -e "Approximate ${BOLD_GREEN}$size_diff_formatted${RESET} new logs were generated at log level ${BOLD_GREEN}$new_log_level_name${RESET} (${BOLD_GREEN}$new_log_level${RESET}) for IP ${ip_array[*]}. Current castor.log size is ${BOLD_GREEN}$(format_size "$final_size")${RESET} after ${YELLOW}$duration${RESET} seconds (${YELLOW}$duration_formatted${RESET})."
                    echo -e ""
            fi
            else
                echo -e "${GREEN}$(timestamp)${RESET} 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 "**********************************************************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 "Detach   mode - ParametersCurrent passedLog to main_script:Level: $current_log_level" | tee -a "$output_log"
    debug_msg "     SwarmCluster IPName: $swarm_ip$clusterName" | tee -a "$output_log"
    debug_msg "     Credentialsjq_or_grep: $(print_credentials "$credentials")"  $jq_or_grep" | tee -a "$output_log"
    debug_msg "     New Log LevelDetach: $new_log_level$detachable" | tee -a "$output_log"
    debug_msg "     Debug: $debug" | tee New-a log Level Name: $new_log_level_name"$output_log"
    debug_msg "**********************************************************" | tee -a "$output_log"

   debug_msg "# Convert associative arrays to Duration:strings $duration"and |pass teethem -a "$output_log"
 to the screen session
  debug_msg "     Log Level Type: $log_level_type" | tee -a "$output_log" log_levels_string=$(declare -p log_levels)
    debug_msg "     Log File: $log_file" | tee -a "$output_log"log_level_names_string=$(declare -p log_level_names)
    debug_msg "     Initial Log File Size: $initial_sizelog_levels_string: $log_levels_string" | tee -a "$output_log"
    debug_msg "     Current Log Level: $current_log_levellog_level_names_string: $log_level_names_string" | tee -a "$output_log"

    if command -v screen &>/dev/null; then
   debug_msg "     Cluster Name: $clusterNameecho -e "Running in ${YELLOW}screen${RESET} detachable mode..." | tee -a "$output_log"
    debug_msg "   screen -dmS jqcastor_orlog_grep: $jq_or_grep" | tee -a "$output_log"
   script bash -c "$(declare -f main_script timestamp 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 toformat_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}\" \"${default_log_levels}\" \"${default_log_levels_name}\" | tee \"$output_log\""
        screen -r castor_log_script
        # Wait for the screen session to complete and then display log_levels_string=$(declare -p log_levels)
    log_level_names_string=$(declare -p log_level_names)the output log
        sleep 1
        while screen -list | grep -q "castor_log_script"; do
    debug_msg "log_levels_string: $log_levels_string" | tee -a "$output_log"  sleep 1
 debug_msg "log_level_names_string: $log_level_names_string" | tee -a "$output_log" done
    ifelif command -v screentmux &>/dev/null; then
        echo -e "Running in ${YELLOW}screen$tmux${RESET} detachable mode..." | tee -a "$output_log" detachable mode..." > "$output_log"  # Truncate log file to remove old entries

        # Create a temp script file
        temp_script="$SCRIPTDIR/castor_log_script.sh"

        # Write the script to a file to avoid printing function definitions
        cat screen<<EOF -dmS> castor"$temp_log_script"
bash -c "#!/bin/bash
$(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}\" \"${default_log_levels}\" \"${default_log_levels_name}\" | tee \-a "$output_log\""
rm -f "$temp_script"         screen -r castor_log_script         # WaitRemove for the screenscript sessionfile
totmux complete and then display the output logkill-session -t castor_log_script
EOF

        # Ensure sleepthe 1script is executable
      while screen -list | grep -qchmod +x "castor$temp_log_script";
do
        # Start tmux  sleep 1session and execute the script inside
   done     eliftmux commandnew-session -vd tmux &>/dev/null; then
-s castor_log_script "bash $temp_script"

      echo -e "Running# in ${YELLOW}tmux${RESET} detachable mode..." | tee -a "$output_log"Attach session only if it's still running
        while tmux newhas-session -d -st castor_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}\" \"${default_log_levels}\" \"${default_log_levels_name}\" | tee \"$output_log\""
    2>/dev/null; do
            tmux attach-session -t castor_log_script
        done
        # if tmux has-session -t castor_log_script 2>/dev/null; then
        #     tmux attach-session -t castor_log_script
        # fi
    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