295 lines
8.4 KiB
Bash
295 lines
8.4 KiB
Bash
#!/usr/bin/env bash
|
|
|
|
# sf -- script framework (https://github.com/Deleh/sf)
|
|
|
|
# Text formatting variables
|
|
sftrs=$'\e[0m'
|
|
sftbf=$'\e[1m'
|
|
sftdim=$'\e[2m'
|
|
sftul=$'\e[4m'
|
|
sftblk=$'\e[5m'
|
|
sftinv=$'\e[7m'
|
|
sfthd=$'\e[8m'
|
|
sftclr=$'\e[1A\e[K'
|
|
sftk=$'\e[30m'
|
|
sftr=$'\e[31m'
|
|
sftg=$'\e[32m'
|
|
sfty=$'\e[33m'
|
|
sftb=$'\e[34m'
|
|
sftm=$'\e[35m'
|
|
sftc=$'\e[36m'
|
|
sftw=$'\e[97m'
|
|
|
|
# Public output functions
|
|
function sferr {
|
|
echo "${sftbf}${sftr}ERROR${sftrs} $1"
|
|
[ -z "$2" ] && exit 1
|
|
}
|
|
|
|
function sfwarn {
|
|
echo "${sftbf}${sfty}WARNING${sftrs} $1"
|
|
}
|
|
|
|
# Public input functions
|
|
function sfask {
|
|
if [ -n "$2" ]; then
|
|
read -r -p "$1? [${sftbf}y${sftrs}/${sftbf}N${sftrs}] " sfin
|
|
[[ "$sfin" =~ n|N|^$ ]] && sfin=false || sfin=true
|
|
else
|
|
read -r -p "$1? [${sftbf}Y${sftrs}/${sftbf}n${sftrs}] " sfin
|
|
[[ "$sfin" =~ y|Y|^$ ]] && sfin=true || sfin=false
|
|
fi
|
|
}
|
|
|
|
function sfget {
|
|
if [ -n "$2" ]; then
|
|
read -r -p "$1 [${sftbf}$2${sftrs}]: " sfin
|
|
else
|
|
read -r -p "$1: " sfin
|
|
fi
|
|
[ "$sfin" == "" ] && [ "$2" != "" ] && sfin="$2"
|
|
}
|
|
|
|
# Internal error function
|
|
function _sferr {
|
|
echo "${sftbf}${sftr}SF PARSE ERROR${sftrs} $1"
|
|
exit 1
|
|
}
|
|
|
|
# Declare variables for parsing
|
|
OLDIFS=$IFS
|
|
IFS=";"
|
|
_sfpargs=()
|
|
_sfpheads=()
|
|
_sfpoffset=0
|
|
_sfptails=()
|
|
_sfpusage=""
|
|
_sfoheads=()
|
|
_sfooffset=0
|
|
_sfotails=()
|
|
declare -A _sfflags
|
|
declare -A _sfargs
|
|
|
|
# Parse sf arguments
|
|
for a in "${sfargs[@]}"; do
|
|
|
|
# Get amount of ;
|
|
_sfsubst=${a//";"}
|
|
_sfcount="$(((${#a} - ${#_sfsubst})))"
|
|
|
|
if [ "$_sfcount" -eq 1 ]; then
|
|
|
|
# Read positional argument declaration
|
|
read -r -a _sfparsearr <<< "${a}"
|
|
|
|
# Add to positional argument arry
|
|
[[ " ${_sfpargs[*]} " =~ " ${_sfparsearr[0]} " ]] && _sferr "'${_sfparsearr[0]}' is already set: $a"
|
|
_sfpargs+=("${_sfparsearr[0]}")
|
|
|
|
# Set usage header and append description arrays
|
|
_sfpusage="$_sfpusage ${_sfparsearr[0]}"
|
|
_sfphead="${_sfparsearr[0]}"
|
|
[ "${#_sfphead}" -gt "${_sfpoffset}" ] && _sfpoffset="${#_sfphead}"
|
|
_sfpheads+=("$_sfphead")
|
|
_sfptails+=("${_sfparsearr[1]}")
|
|
|
|
# Flags
|
|
elif [ "$_sfcount" -eq 2 ]; then
|
|
|
|
# Read flag declaration
|
|
read -r -a _sfparsearr <<< "${a}"
|
|
|
|
# Set mappings
|
|
[ -n "${_sfflags["--${_sfparsearr[0]}"]}" ] && _sferr "'${_sfparsearr[0]}' is already set: $a"
|
|
_sfflags["--${_sfparsearr[0]}"]="${_sfparsearr[0]}"
|
|
[ -n "${_sfflags["-${_sfparsearr[1]}"]}" ] && _sferr "'${_sfparsearr[1]}' is already set: $a"
|
|
_sfflags["-${_sfparsearr[1]}"]="${_sfparsearr[0]}"
|
|
|
|
# Set default value
|
|
declare "${_sfparsearr[0]//-/_}"=false
|
|
|
|
# Append description arrays
|
|
_sfohead="-${_sfparsearr[1]}, --${_sfparsearr[0]}"
|
|
[ "${#_sfohead}" -gt "${_sfooffset}" ] && _sfooffset="${#_sfohead}"
|
|
_sfoheads+=("$_sfohead")
|
|
_sfotails+=("${_sfparsearr[2]}")
|
|
|
|
# Arguments
|
|
elif [ "$_sfcount" -eq 4 ]; then
|
|
|
|
# Read argument declaration
|
|
read -r -a _sfparsearr <<< "${a}"
|
|
|
|
# Set mappings
|
|
[ -n "${_sfargs["--${_sfparsearr[0]}"]}" ] && _sferr "'${_sfparsearr[0]}' is already set: $a"
|
|
_sfargs["--${_sfparsearr[0]}"]="${_sfparsearr[0]}"
|
|
[ -n "${_sfargs["-${_sfparsearr[1]}"]}" ] && _sferr "'${_sfparsearr[1]}' is already set: $a"
|
|
_sfargs["-${_sfparsearr[1]}"]="${_sfparsearr[0]}"
|
|
|
|
# Set default value
|
|
declare "${_sfparsearr[0]//-/_}"="${_sfparsearr[3]}"
|
|
|
|
# Append description arrays
|
|
_sfohead="-${_sfparsearr[1]}, --${_sfparsearr[0]} ${_sfparsearr[2]}"
|
|
[ "${#_sfohead}" -gt "${_sfooffset}" ] && _sfooffset="${#_sfohead}"
|
|
_sfoheads+=("$_sfohead")
|
|
_sfotails+=("${_sfparsearr[4]} (default: ${_sfparsearr[3]})")
|
|
|
|
else
|
|
_sferr "Wrong argument declaration: $a"
|
|
fi
|
|
done
|
|
|
|
# Set _sfwidth to current terminal size
|
|
_sfwidth=$(stty size | cut -d ' ' -f 2)
|
|
|
|
# Create positional argument description with correct line breaks
|
|
_sfpoffset=$(( "_sfpoffset" + 3 ))
|
|
_sfpdesc=""
|
|
for i in "${!_sfptails[@]}"; do
|
|
_sfptail="${_sfptails[$i]}"
|
|
if [ $(( "${#_sfptail}" + "$_sfpoffset" )) -gt "$_sfwidth" ]; then
|
|
_sfptail=$(echo "$_sfptail" | fold -s -w "$(( _sfwidth - _sfpoffset ))")
|
|
_sfptail="${_sfptail//$' \n'/$'\n;'}"
|
|
fi
|
|
_sfpdesc="${_sfpdesc} ${_sfpheads[$i]};${_sfptail}\n"
|
|
done
|
|
|
|
# Create option description with correct line breaks
|
|
_sfooffset=$(( "_sfooffset" + 3 ))
|
|
_sfodesc=" -h, --help;Show this help message\n"
|
|
for i in "${!_sfotails[@]}"; do
|
|
_sfotail="${_sfotails[$i]}"
|
|
if [ $(( "${#_sfotail}" + "$_sfooffset" )) -gt "$_sfwidth" ]; then
|
|
_sfotail=$(echo "$_sfotail" | fold -s -w "$(( _sfwidth - _sfooffset ))")
|
|
_sfotail="${_sfotail//$' \n'/$'\n;'}"
|
|
fi
|
|
_sfodesc="${_sfodesc} ${_sfoheads[$i]};${_sfotail}\n"
|
|
done
|
|
|
|
# Check if at least one positional argument is set if 'sfparr' is used
|
|
[ "$sfparr" == true ] && [ "${#_sfpargs[@]}" == 0 ] && _sferr "At least one positional argument must be used with 'sfparr'"
|
|
|
|
# Parse examples
|
|
_sfeheads=()
|
|
_sfetails=()
|
|
_sfeoffset=0
|
|
for e in "${sfexamples[@]}"; do
|
|
|
|
# Get amount of ;
|
|
_sfsubst=${e//";"}
|
|
_sfcount="$(((${#e} - ${#_sfsubst})))"
|
|
|
|
if [ "$_sfcount" -eq 1 ]; then
|
|
|
|
# Read example
|
|
read -r -a _sfparsearr <<< "${e}"
|
|
|
|
# Append example arrays
|
|
_sfehead="${_sfparsearr[0]}"
|
|
[ "${#_sfehead}" -gt "${_sfeoffset}" ] && _sfeoffset="${#_sfehead}"
|
|
_sfeheads+=("$_sfehead")
|
|
_sfetails+=("${_sfparsearr[1]}")
|
|
|
|
else
|
|
_sferr "Wrong example declaration: $e"
|
|
fi
|
|
done
|
|
|
|
# Create examples description with correct line breaks
|
|
_sfeoffset=$(( "_sfeoffset" + 3 ))
|
|
_sfexamples=""
|
|
for i in "${!_sfetails[@]}"; do
|
|
_sfetail="${_sfetails[$i]}"
|
|
if [ $(( "${#_sfetail}" + "$_sfeoffset" )) -gt "$_sfwidth" ]; then
|
|
_sfetail=$(echo "$_sfetail" | fold -s -w "$(( _sfwidth - _sfeoffset ))")
|
|
_sfetail="${_sfetail//$' \n'/$'\n;'}"
|
|
fi
|
|
_sfexamples="${_sfexamples} ${_sfeheads[$i]};${_sfetail}\n"
|
|
done
|
|
|
|
IFS=$OLDIFS
|
|
|
|
# Usage function
|
|
function _sfusage {
|
|
echo -n "Usage: $(basename "$0") [OPTIONS]"
|
|
echo -ne "$_sfpusage"
|
|
[ "$sfparr" == true ] && echo -n " ..."
|
|
echo
|
|
[ -n "${sfdesc}" ] && echo -e "\n$sfdesc"
|
|
if [ "$_sfpdesc" != "" ]; then
|
|
echo -e "\nPOSITIONAL ARGUMENTS"
|
|
echo -e "$_sfpdesc" | column -s ";" -t -W 2
|
|
fi
|
|
if [ "$_sfodesc" != "" ]; then
|
|
echo -e "\nOPTIONS"
|
|
echo -e "$_sfodesc" | column -s ";" -t -W 2
|
|
fi
|
|
if [ "$_sfexamples" != "" ]; then
|
|
echo -e "\nEXAMPLES"
|
|
echo -e "$_sfexamples" | column -s ";" -t -W 2
|
|
fi
|
|
if [ -n "${sfextra}" ]; then
|
|
echo -e "\n$sfextra"
|
|
fi
|
|
exit 0
|
|
}
|
|
|
|
# Check for help flag
|
|
for a in "$@"; do
|
|
# Check if help flag ist set
|
|
[ "$a" == "-h" ] || [ "$a" == "--help" ] && _sfusage
|
|
done
|
|
|
|
# Parse arguments
|
|
while (( "$#" )); do
|
|
|
|
# Check if flag
|
|
if [ -n "${_sfflags["$1"]}" ]; then
|
|
declare "${_sfflags["$1"]//-/_}"=true
|
|
|
|
# Check if argument
|
|
elif [ -n "${_sfargs["$1"]}" ]; then
|
|
|
|
# Check if argument has value
|
|
if [ -n "$2" ] && [ "${2:0:1}" != "-" ]; then
|
|
declare "${_sfargs["$1"]//-/_}"="$2"
|
|
shift
|
|
else
|
|
sferr "Argument for '$1' missing"
|
|
fi
|
|
|
|
# Handle positional arguments and wrong arguments/flags
|
|
else
|
|
# Check if arg starts with -
|
|
if [ "${1:0:1}" == "-" ]; then
|
|
sferr "Unsupported argument/flag: $1"
|
|
else
|
|
# Set positional argument
|
|
if [ "${#_sfpargs[@]}" != 0 ]; then
|
|
declare "${_sfpargs[0]//-/_}"="$1"
|
|
[ "$sfparr" == true ] && _sfplast="${_sfpargs[0]//-/_}" && _sfparr=("$1")
|
|
_sfpargs=("${_sfpargs[@]:1}")
|
|
elif [ "$sfparr" == true ]; then
|
|
_sfparr+=("$1")
|
|
else
|
|
sferr "Too many positional arguments"
|
|
fi
|
|
fi
|
|
fi
|
|
shift
|
|
done
|
|
|
|
# Parse additional arguments if 'sfparr' is set
|
|
[ "$sfparr" == true ] && [ "${#_sfparr[@]}" -gt 0 ] && read -r -a "${_sfplast?}" <<< "${_sfparr[@]}"
|
|
|
|
# Check if positional arguments left
|
|
if [ "${#_sfpargs[@]}" -gt 0 ]; then
|
|
for p in "${_sfpargs[@]}"; do
|
|
sferr "Positional argument '$p' missing" 0
|
|
done
|
|
exit 1
|
|
fi
|
|
|
|
# Unset all internal variables and functions
|
|
unset a e i _sfargs _sfehead _sfeheads _sfeoffset _sferr _sfetails _sfexamples _sfflags _sfodesc _sfohead _sfoheads _sfooffset _sfotails _sfpargs _sfparr _sfpdesc _sfphead _sfpheads _sfplast _sfpoffset _sfptails _sfpusage _sfusage _sfwidth
|