7d3e63a160
The function, get_ip_from_net_and_fourth, is useful not only in guest-side scripts. This patch moves it from lib/functions.guest to lib/functions.sh. Change-Id: I069fc3e1c1f95937b30aad3809251b68ac0b6185
104 lines
2.4 KiB
Bash
104 lines
2.4 KiB
Bash
# This file contains bash functions that may be used by both guest and host
|
|
# systems.
|
|
|
|
# Non-recursive removal of all files except README.*
|
|
function clean_dir {
|
|
local target_dir=$1
|
|
if [ ! -e "$target_dir" ]; then
|
|
mkdir -pv "$target_dir"
|
|
elif [ ! -d "$target_dir" ]; then
|
|
echo >&2 "Not a directory: $target_dir"
|
|
return 1
|
|
fi
|
|
shopt -s nullglob
|
|
local entries=("$target_dir"/*)
|
|
if [ -n "${entries[0]-}" ]; then
|
|
for f in "${entries[@]}"; do
|
|
# Skip directories
|
|
if [ ! -f "$f" ]; then
|
|
continue
|
|
fi
|
|
|
|
# Skip README.*
|
|
if [[ $f =~ /README\. ]]; then
|
|
continue
|
|
fi
|
|
|
|
rm -f "$f"
|
|
done
|
|
fi
|
|
}
|
|
|
|
function is_root {
|
|
if [ $EUID -eq 0 ]; then
|
|
return 0
|
|
else
|
|
return 1
|
|
fi
|
|
}
|
|
|
|
function yes_or_no {
|
|
local prompt=$1
|
|
local input=""
|
|
while [ : ]; do
|
|
read -p "$prompt (Y/n): " input
|
|
case "$input" in
|
|
N|n)
|
|
return 1
|
|
;;
|
|
""|Y|y)
|
|
return 0
|
|
;;
|
|
*)
|
|
echo -e "${CError:-}Invalid input: ${CData:-}$input${CReset:-}"
|
|
;;
|
|
esac
|
|
done
|
|
}
|
|
|
|
#-------------------------------------------------------------------------------
|
|
# Network helpers
|
|
#-------------------------------------------------------------------------------
|
|
|
|
function get_ip_from_net_and_fourth {
|
|
local net_name=$1
|
|
local net="${!net_name}"
|
|
local fourth_octet=$2
|
|
|
|
echo "${net%.*}.$fourth_octet"
|
|
}
|
|
|
|
#-------------------------------------------------------------------------------
|
|
# Helpers to incrementally number files via name prefixes
|
|
#-------------------------------------------------------------------------------
|
|
|
|
function get_next_file_number {
|
|
local dir=$1
|
|
local ext=${2:-""}
|
|
|
|
# Get number of *.log files in directory
|
|
shopt -s nullglob
|
|
if [ -n "$ext" ]; then
|
|
# Count files with specific extension
|
|
local files=("$dir/"*".$ext")
|
|
else
|
|
# Count all files
|
|
local files=("$dir/"*)
|
|
fi
|
|
echo "${#files[*]}"
|
|
}
|
|
|
|
function get_next_prefix {
|
|
local dir=$1
|
|
local ext=$2
|
|
# Number of digits in prefix string (default 3)
|
|
local digits=${3:-3}
|
|
|
|
# Get number of *.$ext files in $dir
|
|
local cnt=$(get_next_file_number "$dir" "$ext")
|
|
|
|
printf "%0${digits}d" "$cnt"
|
|
}
|
|
|
|
# vim: set ai ts=4 sw=4 et ft=sh:
|