You've already forked mariadb-columnstore-engine
mirror of
https://github.com/mariadb-corporation/mariadb-columnstore-engine.git
synced 2025-07-29 08:21:15 +03:00
603 lines
16 KiB
Bash
603 lines
16 KiB
Bash
#!/bin/bash
|
|
|
|
if [[ -n "$TERM" && "$TERM" != "dumb" && $(command -v tput) ]]; then
|
|
TPUT_AVAILABLE=true
|
|
else
|
|
TPUT_AVAILABLE=false
|
|
fi
|
|
|
|
if [[ $TPUT_AVAILABLE == true ]]; then
|
|
color_normal=$(tput sgr0)
|
|
color_bold=$(tput bold)
|
|
color_red="$color_bold$(tput setaf 1)"
|
|
color_green="$color_bold$(tput setaf 2)"
|
|
color_fawn=$(tput setaf 3)
|
|
color_beige="$color_fawn"
|
|
color_yellow="$color_bold$color_fawn"
|
|
color_darkblue=$(tput setaf 4)
|
|
|
|
color_blue="$color_bold$color_darkblue"
|
|
color_purple=$(tput setaf 5)
|
|
color_magenta="$color_purple"
|
|
color_pink="$color_bold$color_purple"
|
|
color_darkcyan=$(tput setaf 6)
|
|
color_cyan="$color_bold$color_darkcyan"
|
|
color_gray=$(tput setaf 7)
|
|
color_darkgray="$color_bold"$(tput setaf 0)
|
|
color_white="$color_bold$color_gray"
|
|
|
|
if [[ $(tput colors) == '256' ]]; then
|
|
color_red=$(tput setaf 196)
|
|
color_yellow=$(tput setaf 228)
|
|
color_cyan=$(tput setaf 87)
|
|
color_green=$(tput setaf 156)
|
|
color_darkgray=$(tput setaf 59)
|
|
fi
|
|
|
|
else
|
|
# Basic attributes
|
|
color_normal="\e[0m" # sgr0
|
|
color_bold="\e[1m" # bold
|
|
|
|
# Standard 8-color palette
|
|
color_red="\e[1;31m" # bold + setaf 1
|
|
color_green="\e[1;32m" # bold + setaf 2
|
|
color_fawn="\e[33m" # setaf 3
|
|
color_beige="$color_fawn" # alias
|
|
color_yellow="\e[1;33m" # bold + setaf 3
|
|
color_darkblue="\e[34m" # setaf 4
|
|
color_blue="\e[1;34m" # bold + setaf 4
|
|
color_purple="\e[35m" # setaf 5
|
|
color_magenta="$color_purple" # alias
|
|
color_pink="\e[1;35m" # bold + setaf 5
|
|
color_darkcyan="\e[36m" # setaf 6
|
|
color_cyan="\e[1;36m" # bold + setaf 6
|
|
color_gray="\e[37m" # setaf 7
|
|
color_darkgray="\e[1;30m" # bold + setaf 0
|
|
color_white="\e[1;37m" # bold + setaf 7
|
|
if [ "$TERM" = "xterm-256color" ] || [ "$COLORTERM" = "truecolor" ] || [ "$COLORTERM" = "24bit" ]; then
|
|
# Only set 256-color codes if actually in a 256-color terminal
|
|
color_red="\e[91m" # bright red
|
|
color_yellow="\e[93m" # light yellow
|
|
color_cyan="\e[96m" # bright cyan
|
|
color_green="\e[92m" # light green
|
|
color_darkgray="\e[90m" # dark gray
|
|
fi
|
|
fi
|
|
|
|
message() {
|
|
echo -e $color_cyan ・ $@$color_normal
|
|
}
|
|
|
|
warn() {
|
|
echo -e $color_yellow ・ $@$color_normal
|
|
}
|
|
|
|
error() {
|
|
echo -e $color_red ・ $@$color_normal
|
|
}
|
|
|
|
message_split() {
|
|
echo -e $color_darkgray ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ $color_normal
|
|
}
|
|
|
|
message_splitted() {
|
|
message_split
|
|
if [[ $TPUT_AVAILABLE == true ]]; then
|
|
echo $color_green ・ $@$color_normal
|
|
else
|
|
echo "$@"
|
|
fi
|
|
message_split
|
|
}
|
|
|
|
colorify_array() {
|
|
PROMT=""
|
|
for a in "$@"; do
|
|
if [[ $TPUT_AVAILABLE == true ]]; then
|
|
i=$((((i + 1) % (123 - 106)) + 106))
|
|
if [[ $(tput colors) == '256' ]]; then
|
|
PROMT="$PROMT $(tput setaf $i)$a$color_normal"
|
|
else
|
|
PROMT="$PROMT $a"
|
|
fi
|
|
else
|
|
PROMT="$PROMT $a"
|
|
fi
|
|
done
|
|
echo $PROMT
|
|
}
|
|
|
|
newline_array() {
|
|
PROMT=""
|
|
for a in "$@"; do
|
|
PROMT="$PROMT$a\n"
|
|
done
|
|
echo -e $PROMT
|
|
}
|
|
|
|
function spinner() {
|
|
freq=${1:-10}
|
|
points=(⣾ ⣽ ⣻ ⢿ ⡿ ⣟ ⣯ ⣷)
|
|
colored_points=($(colorify_array ${points[@]}))
|
|
len=${#points[@]}
|
|
point_num=0
|
|
line_num=0
|
|
while read data; do
|
|
line_num=$((line_num + 1))
|
|
if [[ $((line_num % freq)) = 0 ]]; then
|
|
point_num=$(((point_num + 1) % len))
|
|
if [[ $TPUT_AVAILABLE == true ]]; then
|
|
echo -ne "\r${colored_points[point_num]}"
|
|
else
|
|
echo -ne "\r${points[point_num]}"
|
|
fi
|
|
fi
|
|
done
|
|
echo
|
|
}
|
|
|
|
function onelinearizator() {
|
|
while read data; do
|
|
echo -ne "\r\e[K$data"
|
|
done
|
|
echo
|
|
}
|
|
|
|
detect_distro() {
|
|
if [ -f /etc/os-release ]; then
|
|
. /etc/os-release
|
|
export OS=$NAME
|
|
export OS_VERSION=$VERSION_ID
|
|
elif type lsb_release >/dev/null 2>&1; then
|
|
# linuxbase.org
|
|
export OS=$(lsb_release -si)
|
|
export OS_VERSION=$(lsb_release -sr)
|
|
elif [ -f /etc/lsb-release ]; then
|
|
# For some versions of Debian/Ubuntu without lsb_release command
|
|
. /etc/lsb-release
|
|
export OS=$DISTRIB_ID
|
|
OS_VERSION=$DISTRIB_RELEASE
|
|
elif [ -f /etc/debian_version ]; then
|
|
# Older Debian/Ubuntu/etc.
|
|
OS=Debian
|
|
OS_VERSION=$(cat /etc/debian_version)
|
|
else
|
|
# Fall back to uname, e.g. "Linux <version>", also works for BSD, etc.
|
|
OS=$(uname -s)
|
|
OS_VERSION=$(uname -r)
|
|
fi
|
|
OS=$(echo $OS | cut -f 1 -d " " | tr '[:upper:]' '[:lower:]')":"$OS_VERSION
|
|
message "Detected $color_yellow$OS $OS_VERSION$color_normal"
|
|
}
|
|
|
|
select_pkg_format() {
|
|
local distro="$1"
|
|
|
|
if [[ "$distro" == *rocky* ]]; then
|
|
export PKG_FORMAT="rpm"
|
|
else
|
|
export PKG_FORMAT="deb"
|
|
fi
|
|
}
|
|
|
|
menuStr=""
|
|
|
|
function hideCursor() {
|
|
printf "\e[?25l"
|
|
|
|
# capture CTRL+C so cursor can be reset
|
|
trap "showCursor && exit 0" 2
|
|
}
|
|
|
|
function showCursor() {
|
|
printf "\e[?25h"
|
|
}
|
|
|
|
function clearLastMenu() {
|
|
local msgLineCount=$(printf "$menuStr" | wc -l)
|
|
# moves the curser up N lines so the output overwrites it
|
|
echo -en "\e[${msgLineCount}A"
|
|
|
|
# clear to end of screen to ensure there's no text left behind from previous input
|
|
[ $1 ] && tput ed
|
|
}
|
|
|
|
function renderMenu() {
|
|
local start=0
|
|
local selector=""
|
|
local instruction="$1"
|
|
local selectedIndex=$2
|
|
local listLength=$itemsLength
|
|
local longest=0
|
|
local spaces=""
|
|
menuStr="\n $instruction\n"
|
|
|
|
# Get the longest item from the list so that we know how many spaces to add
|
|
# to ensure there's no overlap from longer items when a list is scrolling up or down.
|
|
for ((i = 0; i < $itemsLength; i++)); do
|
|
if ((${#menuItems[i]} > longest)); then
|
|
longest=${#menuItems[i]}
|
|
fi
|
|
done
|
|
spaces=$(printf ' %.0s' $(eval "echo {1.."$(($longest))"}"))
|
|
|
|
if [ $3 -ne 0 ]; then
|
|
listLength=$3
|
|
|
|
if [ $selectedIndex -ge $listLength ]; then
|
|
start=$(($selectedIndex + 1 - $listLength))
|
|
listLength=$(($selectedIndex + 1))
|
|
fi
|
|
fi
|
|
|
|
for ((i = $start; i < $listLength; i++)); do
|
|
local currItem="${menuItems[i]}"
|
|
currItemLength=${#currItem}
|
|
|
|
if [[ $i = $selectedIndex ]]; then
|
|
selectedChoice="${currItem}"
|
|
selector="${color_green}ᐅ${color_normal}"
|
|
currItem="${color_green}${currItem}${color_normal}"
|
|
else
|
|
selector=" "
|
|
fi
|
|
|
|
currItem="${spaces:0:0}${currItem}${spaces:currItemLength}"
|
|
|
|
menuStr="${menuStr}\n ${selector} ${currItem}"
|
|
done
|
|
|
|
menuStr="${menuStr}\n"
|
|
|
|
# whether or not to overwrite the previous menu output
|
|
[ $4 ] && clearLastMenu
|
|
|
|
printf "${menuStr}"
|
|
}
|
|
|
|
function getChoice() {
|
|
local KEY__ARROW_UP=$(echo -e "\e[A")
|
|
local KEY__ARROW_DOWN=$(echo -e "\e[B")
|
|
local KEY__ENTER=$(echo -e "\n")
|
|
local captureInput=true
|
|
local displayHelp=false
|
|
local maxViewable=0
|
|
local instruction="Select an item from the list:"
|
|
local selectedIndex=0
|
|
|
|
remainingArgs=()
|
|
while [[ $# -gt 0 ]]; do
|
|
key="$1"
|
|
|
|
case $key in
|
|
-h | --help)
|
|
displayHelp=true
|
|
shift
|
|
;;
|
|
-i | --index)
|
|
selectedIndex=$2
|
|
shift 2
|
|
;;
|
|
-m | --max)
|
|
maxViewable=$2
|
|
shift 2
|
|
;;
|
|
-o | --options)
|
|
menuItems=$2[@]
|
|
menuItems=("${!menuItems}")
|
|
shift 2
|
|
;;
|
|
-q | --query)
|
|
instruction="$2"
|
|
shift 2
|
|
;;
|
|
*)
|
|
remainingArgs+=("$1")
|
|
shift
|
|
;;
|
|
esac
|
|
done
|
|
|
|
# just display help
|
|
if $displayHelp; then
|
|
echo
|
|
echo "Usage: getChoice [OPTION]..."
|
|
echo "Renders a keyboard navigable menu with a visual indicator of what's selected."
|
|
echo
|
|
echo " -h, --help Displays this message"
|
|
echo " -i, --index The initially selected index for the options"
|
|
echo " -m, --max Limit how many options are displayed"
|
|
echo " -o, --options An Array of options for a User to choose from"
|
|
echo " -q, --query Question or statement presented to the User"
|
|
echo
|
|
echo "Example:"
|
|
echo " foodOptions=(\"pizza\" \"burgers\" \"chinese\" \"sushi\" \"thai\" \"italian\" \"shit\")"
|
|
echo
|
|
echo " getChoice -q \"What do you feel like eating?\" -o foodOptions -i \$((\${#foodOptions[@]}-1)) -m 4"
|
|
echo " printf \"\\n First choice is '\${selectedChoice}'\\n\""
|
|
echo
|
|
echo " getChoice -q \"Select another option in case the first isn't available\" -o foodOptions"
|
|
echo " printf \"\\n Second choice is '\${selectedChoice}'\\n\""
|
|
echo
|
|
|
|
return 0
|
|
fi
|
|
|
|
set -- "${remainingArgs[@]}"
|
|
local itemsLength=${#menuItems[@]}
|
|
|
|
# no menu items, at least 1 required
|
|
if [[ $itemsLength -lt 1 ]]; then
|
|
printf "\n [ERROR] No menu items provided\n"
|
|
exit 1
|
|
fi
|
|
|
|
renderMenu "$instruction" $selectedIndex $maxViewable
|
|
hideCursor
|
|
|
|
while $captureInput; do
|
|
read -rsn3 key # `3` captures the escape (\e'), bracket ([), & type (A) characters.
|
|
|
|
case "$key" in
|
|
"$KEY__ARROW_UP")
|
|
selectedIndex=$((selectedIndex - 1))
|
|
(($selectedIndex < 0)) && selectedIndex=$((itemsLength - 1))
|
|
|
|
renderMenu "$instruction" $selectedIndex $maxViewable true
|
|
;;
|
|
|
|
"$KEY__ARROW_DOWN")
|
|
selectedIndex=$((selectedIndex + 1))
|
|
(($selectedIndex == $itemsLength)) && selectedIndex=0
|
|
|
|
renderMenu "$instruction" $selectedIndex $maxViewable true
|
|
;;
|
|
|
|
"$KEY__ENTER")
|
|
clearLastMenu true
|
|
showCursor
|
|
captureInput=false
|
|
;;
|
|
esac
|
|
done
|
|
}
|
|
|
|
function optparse.throw_error() {
|
|
local message="$1"
|
|
error "OPTPARSE: ERROR: $message"
|
|
exit 1
|
|
}
|
|
|
|
# -----------------------------------------------------------------------------------------------------------------------------
|
|
function optparse.define() {
|
|
if [ $# -lt 3 ]; then
|
|
optparse.throw_error "optparse.define <short> <long> <variable> [<desc>] [<default>] [<value>]"
|
|
fi
|
|
for option_id in $(seq 1 $#); do
|
|
local option="$(eval "echo \$$option_id")"
|
|
local key="$(echo $option | awk -F "=" '{print $1}')"
|
|
local value="$(echo $option | awk -F "=" '{print $2}')"
|
|
|
|
#essentials: shortname, longname, description
|
|
if [ "$key" = "short" ]; then
|
|
local shortname="$value"
|
|
if [ ${#shortname} -ne 1 ]; then
|
|
optparse.throw_error "short name expected to be one character long"
|
|
fi
|
|
local short="-${shortname}"
|
|
elif [ "$key" = "long" ]; then
|
|
local longname="$value"
|
|
if [ ${#longname} -lt 2 ]; then
|
|
optparse.throw_error "long name expected to be atleast one character long"
|
|
fi
|
|
local long="--${longname}"
|
|
elif [ "$key" = "desc" ]; then
|
|
local desc="$value"
|
|
elif [ "$key" = "default" ]; then
|
|
local default="$value"
|
|
elif [ "$key" = "variable" ]; then
|
|
local variable="$value"
|
|
elif [ "$key" = "value" ]; then
|
|
local val="$value"
|
|
fi
|
|
done
|
|
|
|
if [ "$variable" = "" ]; then
|
|
optparse.throw_error "You must give a variable for option: ($short/$long)"
|
|
fi
|
|
|
|
if [ "$val" = "" ]; then
|
|
val="\$OPTARG"
|
|
fi
|
|
|
|
# build OPTIONS and help
|
|
optparse_usage="${optparse_usage}#NL#TB${short} $(printf "%-25s %s" "${long}:" "${desc}")"
|
|
if [ "$default" != "" ]; then
|
|
optparse_usage="${optparse_usage} [default:$default]"
|
|
fi
|
|
optparse_contractions="${optparse_contractions}#NL#TB#TB${long})#NL#TB#TB#TBparams=\"\$params ${short}\";;"
|
|
if [ "$default" != "" ]; then
|
|
optparse_defaults="${optparse_defaults}#NL${variable}=${default}"
|
|
fi
|
|
optparse_arguments_string="${optparse_arguments_string}${shortname}"
|
|
if [ "$val" = "\$OPTARG" ]; then
|
|
optparse_arguments_string="${optparse_arguments_string}:"
|
|
fi
|
|
optparse_process="${optparse_process}#NL#TB#TB${shortname})#NL#TB#TB#TB${variable}=\"$val\";;"
|
|
}
|
|
|
|
# -----------------------------------------------------------------------------------------------------------------------------
|
|
function optparse.build() {
|
|
local build_file="$(mktemp "${TMPDIR:-/tmp}/optparse-XXXXXX")"
|
|
|
|
# Building getopts header here
|
|
|
|
# Function usage
|
|
cat <<EOF >$build_file
|
|
function usage(){
|
|
cat << XXX
|
|
usage: \$0 [OPTIONS]
|
|
OPTIONS:
|
|
$optparse_usage
|
|
-? --help : usage
|
|
XXX
|
|
}
|
|
# Contract long options into short options
|
|
params=""
|
|
while [ \$# -ne 0 ]; do
|
|
param="\$1"
|
|
shift
|
|
case "\$param" in
|
|
$optparse_contractions
|
|
"-?"|--help)
|
|
usage
|
|
exit 0;;
|
|
*)
|
|
if [[ "\$param" == --* ]]; then
|
|
echo -e "Unrecognized long option: \$param"
|
|
usage
|
|
exit 1
|
|
fi
|
|
params="\$params \"\$param\"";;
|
|
esac
|
|
done
|
|
eval set -- "\$params"
|
|
# Set default variable values
|
|
$optparse_defaults
|
|
# Process using getopts
|
|
while getopts "$optparse_arguments_string" option; do
|
|
case \$option in
|
|
# Substitute actions for different variables
|
|
$optparse_process
|
|
:)
|
|
echo "Option - \$OPTARG requires an argument"
|
|
exit 1;;
|
|
*)
|
|
usage
|
|
exit 1;;
|
|
esac
|
|
done
|
|
# Clean up after self
|
|
rm $build_file
|
|
EOF
|
|
|
|
local -A o=(['#NL']='\n' ['#TB']='\t')
|
|
|
|
for i in "${!o[@]}"; do
|
|
sed -i "s/${i}/${o[$i]}/g" $build_file
|
|
done
|
|
|
|
# Unset global variables
|
|
unset optparse_usage
|
|
unset optparse_process
|
|
unset optparse_arguments_string
|
|
unset optparse_defaults
|
|
unset optparse_contractions
|
|
|
|
# Return file name to parent
|
|
echo "$build_file"
|
|
}
|
|
|
|
function retry_eval() {
|
|
if [ "$#" -lt 2 ]; then
|
|
error "Usage: retry_eval <max_retries> <command...>"
|
|
return 1
|
|
fi
|
|
|
|
local max_retries=$1
|
|
shift # Remove max_retries from arguments
|
|
local attempt=1
|
|
local initial_delay=1
|
|
|
|
while [ "$attempt" -le "$max_retries" ]; do
|
|
message_split
|
|
message "Attempt $attempt of $max_retries: $*"
|
|
if eval "$@"; then
|
|
message "Command '$@' done"
|
|
message_split
|
|
return 0
|
|
fi
|
|
if [ "$attempt" -lt "$max_retries" ]; then
|
|
delay=$((initial_delay * 2 ** (attempt - 1)))
|
|
warn "Retrying command "$@" in $delay seconds..."
|
|
message_split
|
|
sleep "$delay"
|
|
fi
|
|
((attempt++))
|
|
done
|
|
|
|
error "Max retries reached for command: $*"
|
|
message_split
|
|
exit 13
|
|
}
|
|
|
|
function execInnerDocker() {
|
|
local container_name=$1
|
|
shift 1 # Remove first arg (container_name)
|
|
|
|
docker exec -t "$container_name" bash -c "$@"
|
|
local dockerCommandExitCode=$?
|
|
|
|
if [[ $dockerCommandExitCode -ne 0 ]]; then
|
|
error "Command \"$@\" failed in container \"$container_name\""
|
|
return $dockerCommandExitCode
|
|
fi
|
|
}
|
|
|
|
function execInnerDockerNoTTY() {
|
|
local container_name=$1
|
|
shift 1
|
|
|
|
docker exec "$container_name" bash -c "$@"
|
|
local dockerCommandExitCode=$?
|
|
|
|
if [[ $dockerCommandExitCode -ne 0 ]]; then
|
|
error "Command \"$@\" failed in container \"$container_name\""
|
|
return $dockerCommandExitCode
|
|
fi
|
|
}
|
|
|
|
function execInnerDockerWithRetry() {
|
|
local max_retries=5
|
|
local container_name=$1
|
|
shift 1
|
|
|
|
local cmd=("$@")
|
|
local attempt=1
|
|
local dockerCommandExitCode=0
|
|
|
|
local docker_funcs=$(declare -f retry_eval color_normal color_cyan color_yellow color_red error warn message message_split)
|
|
|
|
# Build the full command to execute in docker
|
|
local full_command="$docker_funcs; retry_eval $max_retries \"${cmd[*]}\""
|
|
|
|
# Execute the command in docker
|
|
docker exec -t "$container_name" bash -c "$full_command"
|
|
dockerCommandExitCode=$?
|
|
|
|
if [[ $dockerCommandExitCode -ne 0 ]]; then
|
|
error "Command \"${cmd[*]}\" failed in container \"$container_name\" after $max_retries attempts"
|
|
return $dockerCommandExitCode
|
|
fi
|
|
|
|
return 0
|
|
}
|
|
|
|
function change_ubuntu_mirror() {
|
|
local region="$1"
|
|
message "Changing Ubuntu mirror to $region"
|
|
sed -i "s|//\(${region}\.\)\?archive\.ubuntu\.com|//${region}.archive.ubuntu.com|g" /etc/apt/sources.list 2>/dev/null || true
|
|
sed -i "s|//\(${region}\.\)\?archive\.ubuntu\.com|//${region}.archive.ubuntu.com|g" /etc/apt/sources.list.d/ubuntu.sources 2>/dev/null || true
|
|
cat /etc/apt/sources.list.d/ubuntu.sources /etc/apt/sources.list 2>/dev/null | grep archive || true
|
|
message_split
|
|
}
|
|
|
|
change_ubuntu_mirror_in_docker() {
|
|
local container_name=$1
|
|
local region=$2
|
|
local docker_funcs=$(declare -f color_normal color_cyan color_yellow color_red error warn message message_split change_ubuntu_mirror)
|
|
|
|
execInnerDocker "$container_name" "$docker_funcs; change_ubuntu_mirror ${region}"
|
|
}
|