#! /usr/bin/env bash
#
# Copyright The Mbed TLS Contributors
# SPDX-License-Identifier: Apache-2.0 OR GPL-2.0-or-later
#
# This swallows the output of the wrapped tool, unless there is an error.
# This helps reduce excess logging in the CI.

# If you are debugging a build / CI issue, you can get complete unsilenced logs
# by un-commenting the following line (or setting VERBOSE_LOGS in your environment):
# VERBOSE_LOGS=1

# don't silence invocations containing these arguments
NO_SILENCE=" --version | test "

TOOL=$(basename "$0")

# Locate original tool
ORIGINAL_TOOL=$(type -ap ${TOOL} | grep -v "$0" | head -n1 )

quote_args() {
    # similar to printf '%q' "$@"
    # but produce more human-readable results for common/simple cases like "a b"
    local args=("$@")
    s=""
    for a in "${args[@]}"; do
        simple_pattern='^[[:alnum:] _=+-]*$'
        if [[ $a =~ ' ' && $a =~ $simple_pattern ]]; then
            # a has spaces, but no other special characters that need escaping
            # (quoting after removing spaces yields no backslashes)
            # simplify quoted form to "$a" - e.g. yield "a b c" instead of a\ b\ c
            q="\"$a\""
        else
            # get bash to do the quoting
            q=$(printf '%q' "$a")
        fi
        s="$s $q"
    done
    echo $s
}

if [[ ! " $@ " =~ " --version " ]]; then
    # Display the command being invoked - if it succeeds, this is all that will
    # be displayed. Don't do this for invocations with --version, because
    # this output is often parsed by scripts, so we don't want to modify it.
    echo "${TOOL} $(quote_args "$@")"
fi

if [[ " $@ " =~ $NO_SILENCE || -n "${VERBOSE_LOGS}" ]]; then
    # Run original command with no output supression
    ${ORIGINAL_TOOL} "$@"
    EXIT_STATUS=$?
else
    # Run original command and capture output & exit status
    TMPFILE=$(mktemp /tmp/quiet-${TOOL}.XXXXXX)
    ${ORIGINAL_TOOL} "$@" > ${TMPFILE} 2>&1
    EXIT_STATUS=$?

    if [[ $EXIT_STATUS -ne 0 ]]; then
        # On error, display the full output
        cat ${TMPFILE}
    fi

    # Remove tmpfile
    rm ${TMPFILE}
fi

# Propagate the exit status
exit $EXIT_STATUS