155 lines
4.8 KiB
Bash
155 lines
4.8 KiB
Bash
#!/bin/bash
|
|
|
|
# Color variables
|
|
COLOR_RESET="\033[0m"
|
|
COLOR_BLUE="\033[1;34m"
|
|
COLOR_YELLOW="\033[1;33m"
|
|
COLOR_GREEN="\033[1;32m"
|
|
COLOR_RED="\033[1;31m"
|
|
|
|
# Symbols
|
|
CHECK_MARK="\033[1;32m✔\033[0m"
|
|
CROSS_MARK="\033[1;31m✘\033[0m"
|
|
|
|
|
|
# Function to center text
|
|
center_text() {
|
|
local text="$1"
|
|
local colorless_text=$(echo -e "$text" | sed 's/\x1b\[[0-9;]*m//g')
|
|
local width=$(tput cols)
|
|
local text_length=${#colorless_text}
|
|
local padding=$(( (width - text_length) / 2 ))
|
|
printf "%${padding}s%s\n" "" "$text"
|
|
}
|
|
|
|
# Function to print header with script name
|
|
print_header() {
|
|
local script_name=$1
|
|
local script_link=$2
|
|
clear
|
|
echo -e "${COLOR_GREEN}"
|
|
center_text " ______ _ ______ _ _ "
|
|
center_text "| ___ \ | | | ___| | | | | "
|
|
center_text "| |_/ / __ _ ___| |__ | |_ ___ _ __ _ __ ___ __ _| |_| |_ ___ _ __ "
|
|
center_text "| ___ \/ _\` / __| '_ \ | _/ _ \| '__| '_ \` _ \ / _\` | __| __/ _ \ '__|"
|
|
center_text "| |_/ / (_| \__ \ | | | | || (_) | | | | | | | | (_| | |_| || __/ | "
|
|
center_text "\____/ \__,_|___/_| |_| \_| \___/|_| |_| |_| |_|\__,_|\__|\__\___|_| "
|
|
center_text " "
|
|
echo -e "${COLOR_YELLOW}\n"
|
|
center_text "${script_name}"
|
|
echo -e "\n"
|
|
center_text "${script_link}"
|
|
echo -e "${COLOR_RESET}\n"
|
|
echo -e "\n\n"
|
|
}
|
|
|
|
|
|
|
|
# Function to display a spinner
|
|
spinner() {
|
|
local pid=$1
|
|
local func_name=$2
|
|
local delay=0.1
|
|
local spinstr=("⠋" "⠙" "⠹" "⠸" "⠼" "⠴" "⠦" "⠧" "⠇" "⠏")
|
|
local temp
|
|
|
|
# Hide cursor
|
|
tput civis
|
|
|
|
while kill -0 "$pid" 2>/dev/null; do
|
|
for temp in "${spinstr[@]}"; do
|
|
printf "\r${COLOR_BLUE}Function: %s${COLOR_RESET} - ${COLOR_YELLOW}Status: Running... %s${COLOR_RESET}" "$func_name" "$temp"
|
|
sleep $delay
|
|
done
|
|
done
|
|
|
|
# Show cursor
|
|
tput cnorm
|
|
}
|
|
|
|
# Function to handle cleanup on exit
|
|
cleanup() {
|
|
local exit_status=$?
|
|
tput cnorm
|
|
if [[ -f "$temp_file" ]]; then
|
|
rm -f "$temp_file"
|
|
fi
|
|
if [[ $exit_status -ne 0 ]]; then
|
|
printf "\n${COLOR_RED}Script interrupted or an error occurred${COLOR_RESET}\n"
|
|
fi
|
|
exit $exit_status
|
|
}
|
|
|
|
# Set trap for cleanup on exit or interrupt
|
|
trap cleanup EXIT
|
|
trap 'exit 130' INT
|
|
|
|
# Function to format the output of another function
|
|
format_output() {
|
|
local func_name=$1
|
|
local display_name=${2:-$func_name}
|
|
temp_file=$(mktemp)
|
|
|
|
# Run the function in the background and capture its output
|
|
( $func_name >"$temp_file" 2>&1 ) &
|
|
local pid=$!
|
|
spinner $pid "$display_name"
|
|
wait $pid
|
|
local exit_status=$?
|
|
|
|
if [ $exit_status -eq 0 ]; then
|
|
printf "\r${COLOR_BLUE}Function: %s${COLOR_RESET} - ${COLOR_GREEN}Status: Finished ${CHECK_MARK}${COLOR_RESET} \n" "$display_name"
|
|
else
|
|
printf "\r${COLOR_BLUE}Function: %s${COLOR_RESET} - ${COLOR_RED}Status: Error ${CROSS_MARK}${COLOR_RESET} \n" "$display_name"
|
|
fi
|
|
|
|
echo -e "${COLOR_BLUE}Output:${COLOR_RESET}"
|
|
cat "$temp_file"
|
|
echo "" # Ensure a new line after the function output
|
|
rm -f "$temp_file"
|
|
return $exit_status
|
|
}
|
|
|
|
# Function to format the output of another function that requires user input
|
|
format_output_with_input() {
|
|
local func_name=$1
|
|
local display_name=${2:-$func_name}
|
|
temp_file=$(mktemp)
|
|
|
|
echo -e "${COLOR_BLUE}Function: $display_name${COLOR_RESET} - ${COLOR_YELLOW}Status: Running...${COLOR_RESET}"
|
|
|
|
# Run the function in the foreground to handle interactive input
|
|
$func_name 2>&1 | tee "$temp_file"
|
|
local exit_status=${PIPESTATUS[0]}
|
|
|
|
if [ $exit_status -eq 0 ]; then
|
|
printf "\r${COLOR_BLUE}Function: %s${COLOR_RESET} - ${COLOR_GREEN}Status: Finished ${CHECK_MARK}${COLOR_RESET} \n" "$display_name"
|
|
else
|
|
printf "\r${COLOR_BLUE}Function: %s${COLOR_RESET} - ${COLOR_RED}Status: Error ${CROSS_MARK}${COLOR_RESET} \n" "$display_name"
|
|
fi
|
|
|
|
echo "" # Ensure a new line after the function output
|
|
rm -f "$temp_file"
|
|
return $exit_status
|
|
}
|
|
|
|
# Function to print final message
|
|
final_message() {
|
|
local script_name=$1
|
|
local success=$2
|
|
if [ -z "$success" ]; then
|
|
success=1
|
|
fi
|
|
if [ "$success" -eq 0 ]; then
|
|
echo -e "\n${COLOR_GREEN}${script_name} completed successfully!${COLOR_RESET}\n"
|
|
else
|
|
echo -e "\n${COLOR_RED}${script_name} encountered errors!${COLOR_RESET}\n"
|
|
fi
|
|
}
|
|
|
|
# Export the functions for use in other scripts
|
|
export -f print_header
|
|
export -f format_output
|
|
export -f format_output_with_input
|
|
export -f final_message
|
|
export COLOR_RESET COLOR_BLUE COLOR_YELLOW COLOR_GREEN COLOR_RED CHECK_MARK CROSS_MARK
|