113 lines
3.0 KiB
Smarty
113 lines
3.0 KiB
Smarty
{{- define "divingbell.shcommon" -}}
|
|
#!/bin/bash
|
|
|
|
set -o errtrace
|
|
set -o pipefail
|
|
|
|
declare -Ax __log_types=(
|
|
{{- if .Values.conf.log_colors }}
|
|
[ERROR]='fd=2, color=\e[01;31m'
|
|
[TRACE]='fd=2, color=\e[01;31m'
|
|
[WARN]='fd=1, color=\e[01;93m'
|
|
[INFO]='fd=1, color=\e[01;37m'
|
|
[DEBUG]='fd=1, color=\e[01;90m'
|
|
{{- else }}
|
|
[ERROR]='fd=2,'
|
|
[TRACE]='fd=2,'
|
|
[WARN]='fd=1,'
|
|
[INFO]='fd=1,'
|
|
[DEBUG]='fd=1,'
|
|
{{- end }}
|
|
)
|
|
for __log_type in "${!__log_types[@]}"; do
|
|
alias log.${__log_type}="echo ${__log_type}"
|
|
done
|
|
shopt -s expand_aliases
|
|
|
|
__text_formatter(){
|
|
local log_prefix='None'
|
|
local default_log_type='INFO'
|
|
local default_xtrace_type='DEBUG'
|
|
local log_type
|
|
local color_prefix
|
|
local fd
|
|
for log_type in "${!__log_types[@]}"; do
|
|
if [[ ${1} == ${log_type}* ]]; then
|
|
log_prefix=''
|
|
color_prefix="$(echo ${__log_types["${log_type}"]} |
|
|
cut -d',' -f2 | cut -d'=' -f2)"
|
|
fd="$(echo ${__log_types["${log_type}"]} |
|
|
cut -d',' -f1 | cut -d'=' -f2)"
|
|
break
|
|
fi
|
|
done
|
|
if [ "${log_prefix}" = "None" ]; then
|
|
# xtrace output usually begins with "+" or "'", mark as debug
|
|
if [[ ${1} = '+'* ]] || [[ ${1} = \'* ]]; then
|
|
log_prefix="${default_xtrace_type} "
|
|
log_type="${default_xtrace_type}"
|
|
else
|
|
log_prefix="${default_log_type} "
|
|
log_type="${default_log_type}"
|
|
fi
|
|
color_prefix="$(echo ${__log_types["${log_type}"]} |
|
|
cut -d',' -f2 | cut -d'=' -f2)"
|
|
fd="$(echo ${__log_types["${log_type}"]} |
|
|
cut -d',' -f1 | cut -d'=' -f2)"
|
|
fi
|
|
local color_suffix=''
|
|
if [ -n "${color_prefix}" ]; then
|
|
color_suffix='\e[0m'
|
|
fi
|
|
timestamp=$(date "+%m-%d-%y %H:%M:%S")
|
|
echo -e "${color_prefix}${timestamp} ${log_prefix}${1}${color_suffix}" >&${fd}
|
|
}
|
|
# Due to this unresolved issue: http://bit.ly/2xPmOY9 we choose preservation of
|
|
# message ordering at the expense of applying appropriate tags to stderr. As a
|
|
# result, stderr from subprocesses will still display as INFO level messages.
|
|
# However we can still log ERROR messages using the aliased log handlers.
|
|
exec >& >(while read line; do
|
|
if [ "${line}" = '__EXIT_MARKER__' ]; then
|
|
break
|
|
else
|
|
__text_formatter "${line}"
|
|
fi
|
|
done)
|
|
|
|
die(){
|
|
# write to stderr any passed error message
|
|
if [[ $@ = *[!\ ]* ]]; then
|
|
log.ERROR "$@"
|
|
fi
|
|
log.TRACE "Backtrace:"
|
|
for ((i=0;i<${#FUNCNAME[@]}-1;i++)); do
|
|
log.TRACE $(caller $i)
|
|
done
|
|
echo __EXIT_MARKER__
|
|
log.ERROR "Exiting with unspecified error."
|
|
set +x
|
|
# Exit after pipe closes to ensure all output is flushed first
|
|
while : ; do
|
|
echo "Waiting on exit..." || exit 1
|
|
done
|
|
}
|
|
export -f die
|
|
trap 'die' ERR
|
|
set -x
|
|
|
|
write_test(){
|
|
local write_test_file
|
|
write_test_file=$(mktemp -p "${1}/" -t __write_test.XXXXXXXX) &&
|
|
rm ${write_test_file} ||
|
|
die "Write test to ${1} failed."
|
|
}
|
|
|
|
die_if_null(){
|
|
local var="${1}"
|
|
shift
|
|
[ -n "${var}" ] || die "Null variable exception $@"
|
|
}
|
|
|
|
###############################################################################
|
|
{{- end -}}
|