re-factor hooks
* gracefully pass missing dependencies (grep and sed)
This commit is contained in:
parent
91266ca8eb
commit
e96345eb87
2 changed files with 123 additions and 67 deletions
|
@ -20,38 +20,63 @@
|
|||
YADM_DIRECTORY=$(dirname $YADM_HOOK_REPO)
|
||||
YADM_CHECKSUMS=$YADM_DIRECTORY/files.checksums
|
||||
|
||||
CHECKSUM_ALGORITHM="512256"
|
||||
CHECKSUM_ALGORITHM_NAME="SHA-512/256"
|
||||
WARNING_MESSAGE="No checksums were created."
|
||||
|
||||
# check if "shasum" exists
|
||||
if command -v shasum > /dev/null; then
|
||||
# check if "shasum" supports SHA-512/256 algorithm
|
||||
echo -n | shasum --algorithm 512256 2>&1 1> /dev/null
|
||||
|
||||
function print_warning_and_exit {
|
||||
MESSAGE=$1
|
||||
|
||||
# set output color to yellow
|
||||
echo -e "\033[1;33m"
|
||||
echo "WARNING: $MESSAGE $WARNING_MESSAGE"
|
||||
|
||||
# reset output color
|
||||
echo -e "\033[0m"
|
||||
|
||||
exit $YADM_HOOK_EXIT
|
||||
}
|
||||
|
||||
|
||||
function ensure_command {
|
||||
COMMAND_NAME=$1
|
||||
|
||||
# check if command exists
|
||||
if ! command -v "$COMMAND_NAME" > /dev/null; then
|
||||
print_warning_and_exit "command \"$COMMAND_NAME\" not found."
|
||||
fi
|
||||
}
|
||||
|
||||
|
||||
function ensure_algorithm {
|
||||
# check if "shasum" supports algorithm
|
||||
echo -n | shasum --algorithm "$CHECKSUM_ALGORITHM" &> /dev/null
|
||||
|
||||
if [ $? -ne 0 ]; then
|
||||
echo
|
||||
echo "WARNING: \"shasum\" does not support SHA-512/256. No checksums were created."
|
||||
exit $YADM_HOOK_EXIT
|
||||
print_warning_and_exit "\"shasum\" does not support $CHECKSUM_ALGORITHM_NAME."
|
||||
fi
|
||||
}
|
||||
|
||||
# empty checksum file
|
||||
echo -n > $YADM_CHECKSUMS
|
||||
|
||||
# calculate checksums for encrypted files
|
||||
for included in ${YADM_ENCRYPT_INCLUDE_FILES[@]}; do
|
||||
shasum --algorithm 512256 $included >> $YADM_CHECKSUMS
|
||||
ERROR_CODE=$?
|
||||
# check if "shasum" exists and supports algorithm
|
||||
ensure_command shasum
|
||||
ensure_algorithm
|
||||
|
||||
# signal errors
|
||||
if [ $ERROR_CODE -ne 0 ]; then
|
||||
exit $ERROR_CODE
|
||||
fi
|
||||
done
|
||||
# empty checksum file
|
||||
echo -n > $YADM_CHECKSUMS
|
||||
|
||||
echo "Wrote checksums: $YADM_CHECKSUMS (SHA-512/256)"
|
||||
else
|
||||
echo
|
||||
echo "WARNING: command \"shasum\" not found. No checksums were created."
|
||||
exit $YADM_HOOK_EXIT
|
||||
fi
|
||||
# calculate checksums for encrypted files
|
||||
for included in ${YADM_ENCRYPT_INCLUDE_FILES[@]}; do
|
||||
shasum --algorithm $CHECKSUM_ALGORITHM $included >> $YADM_CHECKSUMS
|
||||
|
||||
# signal errors
|
||||
if [ $? -ne 0 ]; then
|
||||
exit $?
|
||||
fi
|
||||
done
|
||||
|
||||
echo "Wrote checksums: $YADM_CHECKSUMS ($CHECKSUM_ALGORITHM_NAME)"
|
||||
|
||||
# return exit status of the yadm command
|
||||
exit $YADM_HOOK_EXIT
|
||||
|
|
|
@ -20,50 +20,81 @@
|
|||
YADM_DIRECTORY=$(dirname $YADM_HOOK_REPO)
|
||||
YADM_CHECKSUMS=$YADM_DIRECTORY/files.checksums
|
||||
|
||||
CHECKSUM_ALGORITHM="512256"
|
||||
CHECKSUM_ALGORITHM_NAME="SHA-512/256"
|
||||
WARNING_MESSAGE="Checksums were not verified."
|
||||
|
||||
|
||||
function print_warning_and_exit {
|
||||
MESSAGE=$1
|
||||
|
||||
# set output color to yellow
|
||||
echo -e "\033[1;33m"
|
||||
echo "WARNING: $MESSAGE $WARNING_MESSAGE"
|
||||
|
||||
# reset output color
|
||||
echo -e "\033[0m"
|
||||
|
||||
exit $YADM_HOOK_EXIT
|
||||
}
|
||||
|
||||
|
||||
function ensure_command {
|
||||
COMMAND_NAME=$1
|
||||
|
||||
# check if command exists
|
||||
if ! command -v "$COMMAND_NAME" > /dev/null; then
|
||||
print_warning_and_exit "command \"$COMMAND_NAME\" not found."
|
||||
fi
|
||||
}
|
||||
|
||||
|
||||
function ensure_algorithm {
|
||||
# check if "shasum" supports algorithm
|
||||
echo -n | shasum --algorithm "$CHECKSUM_ALGORITHM" &> /dev/null
|
||||
|
||||
if [ $? -ne 0 ]; then
|
||||
print_warning_and_exit "\"shasum\" does not support $CHECKSUM_ALGORITHM_NAME."
|
||||
fi
|
||||
}
|
||||
|
||||
|
||||
# check whether file with checksums exists
|
||||
if [ -f $YADM_CHECKSUMS ]; then
|
||||
# check if "shasum" exists
|
||||
if command -v shasum > /dev/null; then
|
||||
# check if "shasum" supports SHA-512/256 algorithm
|
||||
echo -n | shasum --algorithm 512256 2>&1 1> /dev/null
|
||||
|
||||
if [ $? -ne 0 ]; then
|
||||
echo
|
||||
echo "WARNING: \"shasum\" does not support SHA-512/256. Checksums were not verified."
|
||||
exit $YADM_HOOK_EXIT
|
||||
fi
|
||||
|
||||
# check encrypted files for differences and capture output
|
||||
YADM_CHECKSUM_OUTPUT=$(shasum --algorithm 512256 --check $YADM_CHECKSUMS 2> /dev/null)
|
||||
ERROR_CODE=$?
|
||||
|
||||
# some checksums do not match
|
||||
if [ $ERROR_CODE -ne 0 ]; then
|
||||
echo
|
||||
echo "Some SHA-512/256 sums do not match:"
|
||||
|
||||
# set output color to red
|
||||
echo -e "\033[0;31m"
|
||||
|
||||
# display mismatching files
|
||||
while IFS= read -r line; do
|
||||
echo $line | grep -iv "\sok$" | sed 's/^/ / ; s/: FAILED$//'
|
||||
done <<< "$YADM_CHECKSUM_OUTPUT"
|
||||
|
||||
# reset output color
|
||||
echo -e "\033[0m"
|
||||
echo "Consider running either \"yadm encrypt\" or \"yadm decrypt\"."
|
||||
|
||||
# signal error
|
||||
exit $ERROR_CODE
|
||||
fi
|
||||
else
|
||||
echo
|
||||
echo "WARNING: command \"shasum\" not found. Checksums were not verified."
|
||||
exit $YADM_HOOK_EXIT
|
||||
fi
|
||||
if [ ! -f $YADM_CHECKSUMS ]; then
|
||||
# return exit status of the yadm command
|
||||
exit $YADM_HOOK_EXIT
|
||||
fi
|
||||
|
||||
# return exit status of the yadm command
|
||||
exit $YADM_HOOK_EXIT
|
||||
# check if "shasum" exists and supports algorithm
|
||||
ensure_command shasum
|
||||
ensure_algorithm
|
||||
|
||||
# check encrypted files for differences and capture output
|
||||
YADM_CHECKSUM_OUTPUT=$(shasum --algorithm "$CHECKSUM_ALGORITHM" --check $YADM_CHECKSUMS 2> /dev/null)
|
||||
ERROR_CODE=$?
|
||||
|
||||
# some checksums do not match
|
||||
if [ $ERROR_CODE -ne 0 ]; then
|
||||
echo
|
||||
echo "Some $CHECKSUM_ALGORITHM_NAME sums do not match:"
|
||||
|
||||
# set output color to red
|
||||
echo -e "\033[0;31m"
|
||||
|
||||
# display mismatching files
|
||||
while IFS= read -r line; do
|
||||
# try to beautify output
|
||||
if command -v grep > /dev/null && command -v sed > /dev/null; then
|
||||
echo $line | grep -iv "\sok$" | sed 's/^/ / ; s/: FAILED$//'
|
||||
else
|
||||
echo $line
|
||||
fi
|
||||
done <<< "$YADM_CHECKSUM_OUTPUT"
|
||||
|
||||
# reset output color
|
||||
echo -e "\033[0m"
|
||||
echo "Consider running either \"yadm encrypt\" or \"yadm decrypt\"."
|
||||
|
||||
# signal error
|
||||
exit $ERROR_CODE
|
||||
fi
|
||||
|
|
Loading…
Reference in a new issue