aboutsummaryrefslogtreecommitdiff
path: root/bin/muffle
blob: 6f9330609367b13efac644a6cbf471192ddb5f55 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
#!/bin/sh
set -eu

usage() {
	cat <<-'EOF'
		Usage:
		  muffle [--] COMMAND...
		  muffle -h
	EOF
}

help() {
	cat <<-'EOF'


		Options:
		  -h, --help    show this message

		  COMMAND       the COMMAND to be muffled


		Conditionally omit STDOUT and STDERR of COMMAND:
		if command exits with an error, print its original output to the
		corresponding streams, but when there is no error, throw the
		output away, "muffling" the COMMAND.


		Examples:

		  Only print anything if the test fails:

		    $ muffle sh tests/assert-makefile.sh
	EOF
}


for flag in "$@"; do
	case "$flag" in
		(--)
			break
			;;
		(--help)
			usage
			help
			exit
			;;
		(*)
			;;
	esac
done

while getopts 'h' flag; do
	case "$flag" in
		(h)
			usage
			help
			exit
			;;
		(*)
			usage >&2
			exit 2
			;;
	esac
done
shift $((OPTIND - 1))


OUT="$(mkstemp)"
ERR="$(mkstemp)"
STATUS_F="$(mkstemp)"
trap 'rm -f "$OUT" "$ERR" "$STATUS_F"' EXIT

timed() {
	ts -s -m '%.s'
}

echo 0 > "$STATUS_F"
{
	{ "$@" || echo $? > "$STATUS_F"; } | timed | pre out > "$OUT"
} 2>&1 | timed | pre err > "$ERR"

STATUS="$(cat "$STATUS_F")"
if [ "$STATUS" != 0 ]; then
	cat "$OUT" "$ERR" |
	sort -k2  | while read -r line; do
		if [ "$(echo "$line" | cut -d' ' -f1)" = 'out:' ]; then
			echo "$line" | cut -d' ' -f3- >&1
		else
			echo "$line" | cut -d' ' -f3- >&2
		fi
	done
	exit "$STATUS"
fi