mirror of
https://github.com/natelandau/shell-scripting-templates.git
synced 2025-11-11 22:43:47 -05:00
commit 61bf734812cb62ba6e0ec224bc15f7928705a8a2 Author: Nathaniel Landau <nate@natelandau.com> Date: Thu Oct 21 15:44:21 2021 -0400 Major overhaul continued - rename templates - add checks utilities - add new array utilities - rename files - add assorted utilities - improve documentation commit 546178fff3b526f492eb0eeffc63f79537e75de3 Author: Nathaniel Landau <nate@natelandau.com> Date: Wed Oct 20 16:31:14 2021 -0400 Update conventions commit f6d0642f85518efda9c5d8472b99d1c14163e381 Author: Nathaniel Landau <nate@natelandau.com> Date: Wed Oct 20 09:47:09 2021 -0400 minor formatting changes commit 2217612b55e3f9faf803a2d0c937ea2261206505 Author: Nathaniel Landau <nate@natelandau.com> Date: Tue Oct 19 17:59:09 2021 -0400 add new functions commit 347ba7aa738dcd6a5ad9d70886b38da3a17dc89e Author: Nathaniel Landau <nate@natelandau.com> Date: Tue Oct 19 12:06:44 2021 -0400 major overhaul - Add standaloneTemplate.sh - Rework README - Refactor inline documentation - Enforce coding standards - Remove CSV utilities - Add new array utilities - add _useGNUutils_ - more ... commit cd8e0d49aef25eeaf6b3e71a3c9e1f29ab9b06f5 Author: Nathaniel Landau <nate@natelandau.com> Date: Sun Oct 17 09:56:08 2021 -0400 Add debug functions commit f7c5c0a3d19815dcc6ba80b5f5a2ebb77ef88b07 Author: Nathaniel Landau <nate@natelandau.com> Date: Sat Oct 16 21:10:01 2021 -0400 add new array functions _joinArray_, _isEmptyArray_, _sortArray_, _reverseSortArray_, and _mergearrays_ commit d8bc3d8cabdbcee3c479f97b43a45bdfe3bdafe0 Author: Nathaniel Landau <nate@natelandau.com> Date: Fri Oct 15 17:27:12 2021 -0400 add _columnize_ commit 2fd2ae9435f476bc3968c3eb0d793db4bf1d9eaf Author: Nathaniel Landau <nate@natelandau.com> Date: Mon Oct 11 22:17:45 2021 -0400 _progressBar_: Fix unbound variable commit e8933d15fc955a1acc665e9a081f131e681855d5 Author: Nathaniel Landau <nate@natelandau.com> Date: Sun Oct 10 11:50:42 2021 -0400 _alert_: header now underlined commit c9ce894361dec7d3513c038794a155519baf26bc Author: Nathaniel Landau <nate@natelandau.com> Date: Tue Oct 5 09:49:42 2021 -0400 _alert_: line numbers to gray commit 4aaddd336ce613f629a7e6a62ef3b27ffc24d22d Author: Nathaniel Landau <nate@natelandau.com> Date: Fri Oct 8 15:05:20 2021 -0400 _usage_ to stdout commit e2372fc3122ec1f20acc27f04d29b3785f014e25 Author: Nathaniel Landau <nate@natelandau.com> Date: Tue Oct 5 09:38:26 2021 -0400 _setPATH_: remove unneeded logic commit e60c75b6c954ac4bd146e2758252168027b9a43d Author: Nathaniel Landau <nate@natelandau.com> Date: Tue Oct 5 09:25:38 2021 -0400 _findSource_: bugfix commit 0e84912e1ccd7203e5beff9f8737f8374f4aa5d8 Author: Nathaniel Landau <nate@natelandau.com> Date: Thu Sep 30 16:29:25 2021 -0400 add requirements to documentation commit 2c24843e3ada591e1868a94416e40b5ac0aa4994 Author: Nathaniel Landau <nate@natelandau.com> Date: Thu Sep 30 15:34:10 2021 -0400 _uniqueFilename_: improve extension handling commit 08bc2dfdcc8632efee9179e9c960a574fc17cf0c Author: Nathaniel Landau <nate@natelandau.com> Date: Mon Sep 27 15:13:53 2021 -0400 improve hooks script commit 641918f1559d3b3aa38a9bbdf418938b2b81c176 Author: Nathaniel Landau <nate@natelandau.com> Date: Fri Sep 24 08:16:52 2021 -0400 _inArry_: case insensitivity commit eae10f170680540fdb4a1222add7e54f8785ea63 Author: Nathaniel Landau <nate@natelandau.com> Date: Mon Sep 20 18:31:44 2021 -0400 clean up alerting commit 700acd56f57fd57db84ef0e232ef41cdd7aee43c Author: Nathaniel Landau <nate@natelandau.com> Date: Mon Sep 20 18:22:11 2021 -0400 refactor _execute_ commit d893f86900a9fed9d91a0c9cc06c13b6b34d9926 Author: Nathaniel Landau <nate@natelandau.com> Date: Mon Sep 20 18:19:18 2021 -0400 'fatal' replaces 'die' commit 3326857bf127bef36cd9982246aa5b826d796d0a Author: Nathaniel Landau <nate@natelandau.com> Date: Fri Sep 17 08:29:50 2021 -0400 _execute_: ensure quiet and verbose work together
696 lines
23 KiB
Bash
Executable File
696 lines
23 KiB
Bash
Executable File
#!/usr/bin/env bash
|
|
|
|
_mainScript_() {
|
|
|
|
GITROOT=$(git rev-parse --show-toplevel 2>/dev/null)
|
|
_setPATH_ "/usr/local/opt/grep/libexec/gnubin" "/usr/local/bin" "${HOME}/bin"
|
|
|
|
_gitStopWords_() {
|
|
# DESC: Check if any specified words are found in the current diff. If found, the pre-commit fails
|
|
# ARGS: $1 (Required) - File to check
|
|
# OUTS: None
|
|
# USAGE: Call the function
|
|
# NOTE: Requires a file located at `~/.git_stop_words` containing one word per line.
|
|
|
|
# Fail if any matching words are present in the diff
|
|
|
|
STOP_WORD_FILE="${HOME}/.git_stop_words"
|
|
GIT_DIFF_TEMP="${TMP_DIR}/diff.txt"
|
|
|
|
if [ -f "${STOP_WORD_FILE}" ]; then
|
|
|
|
if [[ $(basename "${STOP_WORD_FILE}") == "$(basename "${1}")" ]]; then
|
|
debug "Don't check stop words file for stop words. Skipping $(basename "${1}")"
|
|
return 0
|
|
fi
|
|
debug "Checking for stop words"
|
|
|
|
# remove blank lines from stopwords file
|
|
cat "${STOP_WORD_FILE}" | sed '/^$/d' >"${TMP_DIR}/pattern_file.txt"
|
|
|
|
# Add diff to a temporary file
|
|
git diff --cached -- "${1}" | grep '^+' >"${GIT_DIFF_TEMP}"
|
|
|
|
if grep --file="${TMP_DIR}/pattern_file.txt" "${GIT_DIFF_TEMP}"; then
|
|
error "Found git stop word in '$(basename "${1}")'"
|
|
_safeExit_ 1
|
|
fi
|
|
else
|
|
debug "Could not find git stopwords file expected at '${STOP_WORD_FILE}'. Continuing..."
|
|
fi
|
|
}
|
|
|
|
_ignoreSymlinks_() {
|
|
# Ensure that no symlinks have been added to the repository.
|
|
|
|
local gitIgnore="${GITROOT}/.gitignore"
|
|
local havesymlink=false
|
|
local f
|
|
|
|
# Work on files not yet staged
|
|
for f in $(git status --porcelain | grep '^??' | sed 's/^?? //'); do
|
|
if [ -L "${f}" ]; then
|
|
if ! grep "${f}" "${gitIgnore}"; then
|
|
if echo -e "\n${f}" >>"${gitIgnore}"; then
|
|
info "Added symlink '${f}' to .gitignore"
|
|
else
|
|
fatal "Could not add symlink '${f}' to .gitignore"
|
|
fi
|
|
fi
|
|
havesymlink=true
|
|
fi
|
|
done
|
|
|
|
# Work on files that were mistakenly staged
|
|
for f in $(git status --porcelain | grep '^A' | sed 's/^A //'); do
|
|
if [ -L "${f}" ]; then
|
|
if ! grep "${f}" "${gitIgnore}"; then
|
|
if echo -e "\n${f}" >>"${gitIgnore}"; then
|
|
info "Added symlink '${f}' to .gitignore"
|
|
else
|
|
fatal "Could not add symlink '${f}' to .gitignore"
|
|
fi
|
|
fi
|
|
havesymlink=true
|
|
fi
|
|
done
|
|
|
|
if ${havesymlink}; then
|
|
error "At least one symlink was added to the repo."
|
|
error "Commit aborted..."
|
|
_safeExit_ 1
|
|
fi
|
|
}
|
|
|
|
_lintYAML_() {
|
|
if command -v yaml-lint >/dev/null; then
|
|
debug "Linting YAML File"
|
|
if ! yaml-lint "${1}"; then
|
|
error "Error in ${1}"
|
|
_safeExit_ 1
|
|
else
|
|
success "yaml-lint passed: '${1}'"
|
|
fi
|
|
elif command -v yamllint >/dev/null; then
|
|
debug "Linting YAML File"
|
|
if [ -f "$(git rev-parse --show-toplevel)/.yamllint.yml" ]; then
|
|
if ! yamllint -c "$(git rev-parse --show-toplevel)/.yamllint.yml" "${1}"; then
|
|
error "YAML Error in ${1}"
|
|
_safeExit_ 1
|
|
else
|
|
success "yamllint passed: '${1}'"
|
|
fi
|
|
else
|
|
if ! yamllint "${1}"; then
|
|
error "YAML Error in ${1}"
|
|
_safeExit_ 1
|
|
else
|
|
success "yamllint passed: '${1}'"
|
|
fi
|
|
fi
|
|
else
|
|
notice "No YAML linter installed. Continuiing..."
|
|
fi
|
|
}
|
|
|
|
_lintShellscripts_() {
|
|
if command -v shellcheck >/dev/null; then
|
|
debug "Linting shellscript: ${1}"
|
|
if ! shellcheck --exclude=2016,2059,2001,2002,2148,1090,2162,2005,2034,2154,2086,2155,2181,2164,2120,2119,1083,1117,2207,1091 "${1}"; then
|
|
error "Error in ${1}"
|
|
_safeExit_ 1
|
|
else
|
|
success "shellcheck passed: '${1}'"
|
|
fi
|
|
else
|
|
notice "Shellcheck not installed. Continuing..."
|
|
fi
|
|
}
|
|
|
|
_BATS_() {
|
|
local filename
|
|
|
|
# Run BATS unit tests on individual files
|
|
filename="$(basename "${1}")"
|
|
filename="${filename%.*}"
|
|
[ -f "${GITROOT}/test/${filename}.bats" ] \
|
|
&& {
|
|
notice "Running ${filename}.bats"
|
|
if ! "${GITROOT}/test/${filename}.bats" -t; then
|
|
error "Error in ${1}"
|
|
_safeExit_ 1
|
|
fi
|
|
}
|
|
unset filename
|
|
}
|
|
|
|
_lintAnsible_() {
|
|
|
|
if ! command -v ansible-lint >/dev/null; then
|
|
notice "Found Ansible files but ansible-lint is not available. Continuing..."
|
|
return 0
|
|
elif [[ "$(basename ${1})" =~ (^\.|^requirements|j2|vault\.yml|variables|meta|defaults?|inventory) ]]; then
|
|
# Don't lint files that are not Ansible playbooks
|
|
debug "won't ansible lint: ${1}"
|
|
return 0
|
|
elif [[ ${1} =~ /(handlers|vars/|defaults/|meta/|molecule/|templates/|files/)/ ]]; then
|
|
# Don't lint in directory names that are not likely to contain Ansible playbooks
|
|
debug "Won't ansible lint: ${1}"
|
|
return 0
|
|
fi
|
|
|
|
ANSIBLE_COMMAND="ansible-lint -vv --parseable-severity"
|
|
if [ -f "$(git rev-parse --show-toplevel)/.ansible-lint.yml" ]; then
|
|
ANSIBLE_COMMAND="ansible-lint -p -c $(git rev-parse --show-toplevel)/.ansible-lint.yml"
|
|
fi
|
|
|
|
debug "Linting ansible file: ${1}"
|
|
if ! ${ANSIBLE_COMMAND} "${1}"; then
|
|
error "Ansible-lint error"
|
|
_safeExit_ 1
|
|
else
|
|
success "ansible-lint passed: ${1}"
|
|
fi
|
|
}
|
|
|
|
# RUN SCRIPT LOGIC
|
|
|
|
# Attempt to discern if we are working on an repo that contains ansible files
|
|
IS_ANSIBLE_REPO=false
|
|
if find "$(git rev-parse --show-toplevel)" -type f -mindepth 1 -maxdepth 1 \
|
|
-name "inventory.yml" \
|
|
-o -name "ansible.cfg" \
|
|
-o -name ".ansible-lint.yml" &>/dev/null; then
|
|
|
|
IS_ANSIBLE_REPO=true
|
|
fi
|
|
|
|
_ignoreSymlinks_
|
|
|
|
while read -r STAGED_FILE; do
|
|
debug "Working on: ${STAGED_FILE}"
|
|
if [ -f "${STAGED_FILE}" ]; then
|
|
|
|
_gitStopWords_ "${STAGED_FILE}"
|
|
|
|
if [[ ${STAGED_FILE} =~ \.(yaml|yml)$ ]]; then
|
|
_lintYAML_ "${STAGED_FILE}"
|
|
if [ "${IS_ANSIBLE_REPO}" = true ]; then
|
|
_lintAnsible_ "${STAGED_FILE}"
|
|
fi
|
|
fi
|
|
if [[ ${STAGED_FILE} =~ \.(bash|sh)$ || "$(head -n 1 "${STAGED_FILE}")" =~ ^#!.*bash$ ]]; then
|
|
_lintShellscripts_ "${STAGED_FILE}"
|
|
fi
|
|
if [[ ${STAGED_FILE} =~ \.(sh|bash|bats|zsh)$ || "$(head -n 1 "${STAGED_FILE}")" =~ ^#!.*bash$ ]]; then
|
|
_BATS_ "${STAGED_FILE}"
|
|
fi
|
|
fi
|
|
|
|
done < <(git diff --cached --name-only --line-prefix="$(git rev-parse --show-toplevel)/")
|
|
|
|
} # end _mainScript_
|
|
|
|
# ################################## Flags and defaults
|
|
# Script specific
|
|
|
|
# Common
|
|
LOGFILE="${HOME}/logs/$(basename "$0").log"
|
|
QUIET=false
|
|
LOGLEVEL=ERROR
|
|
VERBOSE=false
|
|
FORCE=false
|
|
DRYRUN=false
|
|
declare -a ARGS=()
|
|
NOW=$(LC_ALL=C date +"%m-%d-%Y %r") # Returns: 06-14-2015 10:34:40 PM
|
|
DATESTAMP=$(LC_ALL=C date +%Y-%m-%d) # Returns: 2015-06-14
|
|
HOURSTAMP=$(LC_ALL=C date +%r) # Returns: 10:34:40 PM
|
|
TIMESTAMP=$(LC_ALL=C date +%Y%m%d_%H%M%S) # Returns: 20150614_223440
|
|
LONGDATE=$(LC_ALL=C date +"%a, %d %b %Y %H:%M:%S %z") # Returns: Sun, 10 Jan 2016 20:47:53 -0500
|
|
GMTDATE=$(LC_ALL=C date -u -R | sed 's/\+0000/GMT/') # Returns: Wed, 13 Jan 2016 15:55:29 GMT
|
|
|
|
# ################################## Custom utility functions
|
|
_setPATH_() {
|
|
# DESC: Add directories to $PATH so script can find executables
|
|
# ARGS: $@ - One or more paths
|
|
# OUTS: $PATH
|
|
# USAGE: _setPATH_ "/usr/local/bin" "${HOME}/bin" "$(npm bin)"
|
|
local NEWPATH NEWPATHS USERPATH
|
|
|
|
for USERPATH in "$@"; do
|
|
NEWPATHS+=("$USERPATH")
|
|
done
|
|
|
|
for NEWPATH in "${NEWPATHS[@]}"; do
|
|
if [ -d "${NEWPATH}" ]; then
|
|
if ! echo "${PATH}" | grep -Eq "(^|:)${NEWPATH}($|:)"; then
|
|
PATH="${NEWPATH}:${PATH}"
|
|
debug "Added '${NEWPATH}' to PATH"
|
|
else
|
|
debug "_setPATH_: '${NEWPATH}' already exists in PATH"
|
|
fi
|
|
else
|
|
debug "_setPATH_: can not find: ${NEWPATH}"
|
|
fi
|
|
done
|
|
}
|
|
# ################################## Common Functions for script template
|
|
_setColors_() {
|
|
# DESC: Sets colors use for alerts.
|
|
# ARGS: None
|
|
# OUTS: None
|
|
# USAGE: echo "${blue}Some text${reset}"
|
|
|
|
if tput setaf 1 &>/dev/null; then
|
|
bold=$(tput bold)
|
|
underline=$(tput smul)
|
|
reverse=$(tput rev)
|
|
reset=$(tput sgr0)
|
|
|
|
if [[ $(tput colors) -ge 256 ]] 2>/dev/null; then
|
|
white=$(tput setaf 231)
|
|
blue=$(tput setaf 38)
|
|
yellow=$(tput setaf 11)
|
|
tan=$(tput setaf 3)
|
|
green=$(tput setaf 82)
|
|
red=$(tput setaf 1)
|
|
purple=$(tput setaf 171)
|
|
gray=$(tput setaf 250)
|
|
else
|
|
white=$(tput setaf 7)
|
|
blue=$(tput setaf 38)
|
|
yellow=$(tput setaf 3)
|
|
tan=$(tput setaf 3)
|
|
green=$(tput setaf 2)
|
|
red=$(tput setaf 1)
|
|
purple=$(tput setaf 13)
|
|
gray=$(tput setaf 7)
|
|
fi
|
|
else
|
|
bold="\033[4;37m"
|
|
reset="\033[0m"
|
|
underline="\033[4;37m"
|
|
reverse=""
|
|
white="\033[0;37m"
|
|
blue="\033[0;34m"
|
|
yellow="\033[0;33m"
|
|
tan="\033[0;33m"
|
|
green="\033[1;32m"
|
|
red="\033[0;31m"
|
|
purple="\033[0;35m"
|
|
gray="\033[0;37m"
|
|
fi
|
|
}
|
|
|
|
_alert_() {
|
|
# DESC: Controls all printing of messages to log files and stdout.
|
|
# ARGS: $1 (required) - The type of alert to print
|
|
# (success, header, notice, dryrun, debug, warning, error,
|
|
# fatal, info, input)
|
|
# $2 (required) - The message to be printed to stdout and/or a log file
|
|
# $3 (optional) - Pass '${LINENO}' to print the line number where the _alert_ was triggered
|
|
# OUTS: None
|
|
# USAGE: [ALERTTYPE] "[MESSAGE]" "${LINENO}"
|
|
# NOTES: The colors of each alert type are set in this function
|
|
# For specified alert types, the funcstac will be printed
|
|
|
|
local function_name color
|
|
local alertType="${1}"
|
|
local message="${2}"
|
|
local line="${3:-}" # Optional line number
|
|
|
|
if [[ -n ${line} && ${alertType} =~ ^(fatal|error) && ${FUNCNAME[2]} != "_trapCleanup_" ]]; then
|
|
message="${message} (line: ${line}) $(_functionStack_)"
|
|
elif [[ -n ${line} && ${FUNCNAME[2]} != "_trapCleanup_" ]]; then
|
|
message="${message} (line: ${line})"
|
|
elif [[ -z ${line} && ${alertType} =~ ^(fatal|error) && ${FUNCNAME[2]} != "_trapCleanup_" ]]; then
|
|
message="${message} $(_functionStack_)"
|
|
fi
|
|
|
|
if [[ ${alertType} =~ ^(error|fatal) ]]; then
|
|
color="${bold}${red}"
|
|
elif [ "${alertType}" == "info" ]; then
|
|
color="${gray}"
|
|
elif [ "${alertType}" == "warning" ]; then
|
|
color="${red}"
|
|
elif [ "${alertType}" == "success" ]; then
|
|
color="${green}"
|
|
elif [ "${alertType}" == "debug" ]; then
|
|
color="${purple}"
|
|
elif [ "${alertType}" == "header" ]; then
|
|
color="${bold}${tan}"
|
|
elif [ ${alertType} == "notice" ]; then
|
|
color="${bold}"
|
|
elif [ ${alertType} == "input" ]; then
|
|
color="${bold}${underline}"
|
|
elif [ "${alertType}" = "dryrun" ]; then
|
|
color="${blue}"
|
|
else
|
|
color=""
|
|
fi
|
|
|
|
_writeToScreen_() {
|
|
|
|
("${QUIET}") && return 0 # Print to console when script is not 'quiet'
|
|
[[ ${VERBOSE} == false && ${alertType} =~ ^(debug|verbose) ]] && return 0
|
|
|
|
if ! [[ -t 1 ]]; then # Don't use colors on non-recognized terminals
|
|
color=""
|
|
reset=""
|
|
fi
|
|
|
|
echo -e "$(date +"%r") ${color}$(printf "[%7s]" "${alertType}") ${message}${reset}"
|
|
}
|
|
_writeToScreen_
|
|
|
|
_writeToLog_() {
|
|
[[ ${alertType} == "input" ]] && return 0
|
|
[[ ${LOGLEVEL} =~ (off|OFF|Off) ]] && return 0
|
|
if [ -z "${LOGFILE:-}" ]; then
|
|
LOGFILE="$(pwd)/$(basename "$0").log"
|
|
fi
|
|
[ ! -d "$(dirname "${LOGFILE}")" ] && mkdir -p "$(dirname "${LOGFILE}")"
|
|
[[ ! -f ${LOGFILE} ]] && touch "${LOGFILE}"
|
|
|
|
# Don't use colors in logs
|
|
if command -v gsed &>/dev/null; then
|
|
local cleanmessage="$(echo "${message}" | gsed -E 's/(\x1b)?\[(([0-9]{1,2})(;[0-9]{1,3}){0,2})?[mGK]//g')"
|
|
else
|
|
local cleanmessage="$(echo "${message}" | sed -E 's/(\x1b)?\[(([0-9]{1,2})(;[0-9]{1,3}){0,2})?[mGK]//g')"
|
|
fi
|
|
echo -e "$(date +"%b %d %R:%S") $(printf "[%7s]" "${alertType}") [$(/bin/hostname)] ${cleanmessage}" >>"${LOGFILE}"
|
|
}
|
|
|
|
# Write specified log level data to logfile
|
|
case "${LOGLEVEL:-ERROR}" in
|
|
ALL | all | All)
|
|
_writeToLog_
|
|
;;
|
|
DEBUG | debug | Debug)
|
|
_writeToLog_
|
|
;;
|
|
INFO | info | Info)
|
|
if [[ ${alertType} =~ ^(error|fatal|warning|info|notice|success) ]]; then
|
|
_writeToLog_
|
|
fi
|
|
;;
|
|
NOTICE | notice | Notice)
|
|
if [[ ${alertType} =~ ^(error|fatal|warning|notice|success) ]]; then
|
|
_writeToLog_
|
|
fi
|
|
;;
|
|
WARN | warn | Warn)
|
|
if [[ ${alertType} =~ ^(error|fatal|warning) ]]; then
|
|
_writeToLog_
|
|
fi
|
|
;;
|
|
ERROR | error | Error)
|
|
if [[ ${alertType} =~ ^(error|fatal) ]]; then
|
|
_writeToLog_
|
|
fi
|
|
;;
|
|
FATAL | fatal | Fatal)
|
|
if [[ ${alertType} =~ ^fatal ]]; then
|
|
_writeToLog_
|
|
fi
|
|
;;
|
|
OFF | off)
|
|
return 0
|
|
;;
|
|
*)
|
|
if [[ ${alertType} =~ ^(error|fatal) ]]; then
|
|
_writeToLog_
|
|
fi
|
|
;;
|
|
esac
|
|
|
|
} # /_alert_
|
|
|
|
error() { _alert_ error "${1}" "${2:-}"; }
|
|
warning() { _alert_ warning "${1}" "${2:-}"; }
|
|
notice() { _alert_ notice "${1}" "${2:-}"; }
|
|
info() { _alert_ info "${1}" "${2:-}"; }
|
|
success() { _alert_ success "${1}" "${2:-}"; }
|
|
dryrun() { _alert_ dryrun "${1}" "${2:-}"; }
|
|
input() { _alert_ input "${1}" "${2:-}"; }
|
|
header() { _alert_ header "== ${1} ==" "${2:-}"; }
|
|
debug() { _alert_ debug "${1}" "${2:-}"; }
|
|
fatal() {
|
|
_alert_ fatal "${1}" "${2:-}"
|
|
_safeExit_ "1"
|
|
}
|
|
|
|
_functionStack_() {
|
|
# DESC: Prints the function stack in use
|
|
# ARGS: None
|
|
# OUTS: Prints [function]:[file]:[line]
|
|
# NOTE: Does not print functions from the alert class
|
|
local _i
|
|
funcStackResponse=()
|
|
for ((_i = 1; _i < ${#BASH_SOURCE[@]}; _i++)); do
|
|
case "${FUNCNAME[$_i]}" in "_alert_" | "_trapCleanup_" | fatal | error | warning | notice | info | debug | dryrun | header | success) continue ;; esac
|
|
funcStackResponse+=("${FUNCNAME[$_i]}:$(basename ${BASH_SOURCE[$_i]}):${BASH_LINENO[_i - 1]}")
|
|
done
|
|
printf "( "
|
|
printf %s "${funcStackResponse[0]}"
|
|
printf ' < %s' "${funcStackResponse[@]:1}"
|
|
printf ' )\n'
|
|
}
|
|
|
|
_safeExit_() {
|
|
# DESC: Cleanup and exit from a script
|
|
# ARGS: $1 (optional) - Exit code (defaults to 0)
|
|
# OUTS: None
|
|
|
|
if [[ -d ${SCRIPT_LOCK:-} ]]; then
|
|
if command rm -rf "${SCRIPT_LOCK}"; then
|
|
debug "Removing script lock"
|
|
else
|
|
warning "Script lock could not be removed. Try manually deleting ${tan}'${LOCK_DIR}'${red}"
|
|
fi
|
|
fi
|
|
|
|
if [[ -n ${TMP_DIR:-} && -d ${TMP_DIR:-} ]]; then
|
|
if [[ ${1:-} == 1 && -n "$(ls "${TMP_DIR}")" ]]; then
|
|
# Do something here to save TMP_DIR on a non-zero script exit for debugging
|
|
command rm -r "${TMP_DIR}"
|
|
debug "Removing temp directory"
|
|
else
|
|
command rm -r "${TMP_DIR}"
|
|
debug "Removing temp directory"
|
|
fi
|
|
fi
|
|
|
|
trap - INT TERM EXIT
|
|
exit ${1:-0}
|
|
}
|
|
|
|
_trapCleanup_() {
|
|
# DESC: Log errors and cleanup from script when an error is trapped
|
|
# ARGS: $1 - Line number where error was trapped
|
|
# $2 - Line number in function
|
|
# $3 - Command executing at the time of the trap
|
|
# $4 - Names of all shell functions currently in the execution call stack
|
|
# $5 - Scriptname
|
|
# $6 - $BASH_SOURCE
|
|
# OUTS: None
|
|
|
|
local line=${1:-} # LINENO
|
|
local linecallfunc=${2:-}
|
|
local command="${3:-}"
|
|
local funcstack="${4:-}"
|
|
local script="${5:-}"
|
|
local sourced="${6:-}"
|
|
|
|
funcstack="'$(echo "$funcstack" | sed -E 's/ / < /g')'"
|
|
|
|
if [[ ${script##*/} == "${sourced##*/}" ]]; then
|
|
fatal "${7:-} command: '${command}' (line: ${line}) [func: $(_functionStack_)]"
|
|
else
|
|
fatal "${7:-} command: '${command}' (func: ${funcstack} called at line ${linecallfunc} of '${script##*/}') (line: $line of '${sourced##*/}') "
|
|
fi
|
|
|
|
_safeExit_ "1"
|
|
}
|
|
|
|
_makeTempDir_() {
|
|
# DESC: Creates a temp directory to house temporary files
|
|
# ARGS: $1 (Optional) - First characters/word of directory name
|
|
# OUTS: $TMP_DIR - Temporary directory
|
|
# USAGE: _makeTempDir_ "$(basename "$0")"
|
|
|
|
[ -d "${TMP_DIR:-}" ] && return 0
|
|
|
|
if [ -n "${1:-}" ]; then
|
|
TMP_DIR="${TMPDIR:-/tmp/}${1}.$RANDOM.$RANDOM.$$"
|
|
else
|
|
TMP_DIR="${TMPDIR:-/tmp/}$(basename "$0").$RANDOM.$RANDOM.$RANDOM.$$"
|
|
fi
|
|
(umask 077 && mkdir "${TMP_DIR}") || {
|
|
fatal "Could not create temporary directory! Exiting."
|
|
}
|
|
debug "\$TMP_DIR=${TMP_DIR}"
|
|
}
|
|
|
|
_acquireScriptLock_() {
|
|
# DESC: Acquire script lock
|
|
# ARGS: $1 (optional) - Scope of script execution lock (system or user)
|
|
# OUTS: $SCRIPT_LOCK - Path to the directory indicating we have the script lock
|
|
# NOTE: This lock implementation is extremely simple but should be reliable
|
|
# across all platforms. It does *not* support locking a script with
|
|
# symlinks or multiple hardlinks as there's no portable way of doing so.
|
|
# If the lock was acquired it's automatically released in _safeExit_()
|
|
|
|
local LOCK_DIR
|
|
if [[ ${1:-} == 'system' ]]; then
|
|
LOCK_DIR="${TMPDIR:-/tmp/}$(basename "$0").lock"
|
|
else
|
|
LOCK_DIR="${TMPDIR:-/tmp/}$(basename "$0").$UID.lock"
|
|
fi
|
|
|
|
if command mkdir "${LOCK_DIR}" 2>/dev/null; then
|
|
readonly SCRIPT_LOCK="${LOCK_DIR}"
|
|
debug "Acquired script lock: ${tan}${SCRIPT_LOCK}${purple}"
|
|
else
|
|
error "Unable to acquire script lock: ${tan}${LOCK_DIR}${red}"
|
|
fatal "If you trust the script isn't running, delete the lock dir"
|
|
fi
|
|
}
|
|
|
|
_parseOptions_() {
|
|
# Iterate over options
|
|
# breaking -ab into -a -b when needed and --foo=bar into --foo bar
|
|
optstring=h
|
|
unset options
|
|
while (($#)); do
|
|
case $1 in
|
|
# If option is of type -ab
|
|
-[!-]?*)
|
|
# Loop over each character starting with the second
|
|
for ((i = 1; i < ${#1}; i++)); do
|
|
c=${1:i:1}
|
|
options+=("-$c") # Add current char to options
|
|
# If option takes a required argument, and it's not the last char make
|
|
# the rest of the string its argument
|
|
if [[ $optstring == *"$c:"* && ${1:i+1} ]]; then
|
|
options+=("${1:i+1}")
|
|
break
|
|
fi
|
|
done
|
|
;;
|
|
# If option is of type --foo=bar
|
|
--?*=*) options+=("${1%%=*}" "${1#*=}") ;;
|
|
# add --endopts for --
|
|
--) options+=(--endopts) ;;
|
|
# Otherwise, nothing special
|
|
*) options+=("$1") ;;
|
|
esac
|
|
shift
|
|
done
|
|
set -- "${options[@]:-}"
|
|
unset options
|
|
|
|
# Read the options and set stuff
|
|
while [[ ${1:-} == -?* ]]; do
|
|
case $1 in
|
|
# Custom options
|
|
|
|
# Common options
|
|
-h | --help)
|
|
_usage_ >&2
|
|
_safeExit_
|
|
;;
|
|
--loglevel)
|
|
shift
|
|
LOGLEVEL=${1}
|
|
;;
|
|
--logfile)
|
|
shift
|
|
LOGFILE="${1}"
|
|
;;
|
|
-n | --dryrun) DRYRUN=true ;;
|
|
-v | --verbose) VERBOSE=true ;;
|
|
-q | --quiet) QUIET=true ;;
|
|
--force) FORCE=true ;;
|
|
--endopts)
|
|
shift
|
|
break
|
|
;;
|
|
*) fatal "invalid option: '$1'." ;;
|
|
esac
|
|
shift
|
|
done
|
|
ARGS+=("$@") # Store the remaining user input as arguments.
|
|
}
|
|
|
|
_usage_() {
|
|
cat <<EOF
|
|
|
|
${bold}$(basename "$0") [OPTION]...${reset}
|
|
|
|
This script runs a number of automated tests on files that are staged in Git prior to
|
|
allowing them to be committed
|
|
|
|
${bold}Options:${reset}
|
|
-h, --help Display this help and exit
|
|
--loglevel [LEVEL] One of: FATAL, ERROR, WARN, INFO, NOTICE, DEBUG, ALL, OFF
|
|
(Default is 'ERROR')
|
|
--logfile [FILE] Full PATH to logfile. (Default is '${HOME}/logs/$(basename "$0").log')
|
|
-n, --dryrun Non-destructive. Makes no permanent changes.
|
|
-q, --quiet Quiet (no output)
|
|
-v, --verbose Output more information. (Items echoed to 'verbose')
|
|
--force Skip all user interaction. Implied 'Yes' to all actions.
|
|
|
|
${bold}Example Usage:${reset}
|
|
|
|
$ $(basename "$0") -vn --logfile "/path/to/file.log" --loglevel 'WARN'
|
|
EOF
|
|
}
|
|
|
|
# ################################## INITIALIZE AND RUN THE SCRIPT
|
|
# (Comment or uncomment the lines below to customize script behavior)
|
|
|
|
trap '_trapCleanup_ ${LINENO} ${BASH_LINENO} "${BASH_COMMAND}" "${FUNCNAME[*]}" "${0}" "${BASH_SOURCE[0]}"' EXIT INT TERM SIGINT SIGQUIT
|
|
|
|
# Trap errors in subshells and functions
|
|
set -o errtrace
|
|
|
|
# Exit on error. Append '||true' if you expect an error
|
|
set -o errexit
|
|
|
|
# Use last non-zero exit code in a pipeline
|
|
set -o pipefail
|
|
|
|
# Make `for f in *.txt` work when `*.txt` matches zero files
|
|
# shopt -s nullglob globstar
|
|
|
|
# Set IFS to preferred implementation
|
|
IFS=$' \n\t'
|
|
|
|
# Run in debug mode
|
|
# set -o xtrace
|
|
|
|
# Initialize color constants
|
|
_setColors_
|
|
|
|
# Disallow expansion of unset variables
|
|
set -o nounset
|
|
|
|
# Force arguments when invoking the script
|
|
# [[ $# -eq 0 ]] && _parseOptions_ "-h"
|
|
|
|
# Parse arguments passed to script
|
|
_parseOptions_ "$@"
|
|
|
|
# Create a temp directory '$TMP_DIR'
|
|
_makeTempDir_ "$(basename "$0")"
|
|
|
|
# Acquire script lock
|
|
# _acquireScriptLock_
|
|
|
|
# Run the main logic script
|
|
_mainScript_
|
|
|
|
# Exit cleanly
|
|
_safeExit_
|