Commit | Line | Data |
---|---|---|
4aba78da LM |
1 | #!/bin/bash |
2 | ||
3 | # default values | |
4 | PROGNAME=$(basename $0) | |
5 | FULLPROG=$0 | |
6 | ||
7 | # usage | |
8 | function usage () { | |
9 | echo "usage: $PROGNAME [-h] [-r] cmd" | |
10 | echo " -h: help message" | |
11 | echo " -s: set stats" | |
12 | echo " -r: reset stats" | |
13 | echo "cmd: shell command" | |
14 | } | |
15 | ||
16 | # reset | |
17 | function reset () { | |
18 | echo "reseting stats" | |
19 | [ -f $resource ] && rm $resource | |
20 | } | |
21 | ||
22 | # variables | |
23 | set= | |
24 | cmd= | |
25 | resource=$HOME/.$PROGNAME | |
26 | ||
27 | # parameter processing | |
28 | while [ $# -gt 0 ]; do | |
29 | if [ "$cmd" ]; then | |
30 | cmd="$cmd $1" | |
31 | else | |
32 | case "$1" in | |
33 | -h) usage; exit 0;; | |
34 | -s) set="yes";; | |
35 | -r) reset; exit 0;; | |
36 | *) cmd="$1";; | |
37 | esac | |
38 | fi | |
39 | shift | |
40 | done | |
41 | ||
42 | # check arguments | |
43 | [ "$cmd" ] || { echo "no command specified"; exit 1; } | |
44 | [ "$(which ${cmd/ *})" ] || { echo "command not found '$cmd'"; exit 1; } | |
45 | ||
46 | # read old values | |
47 | [ -f $resource ] && source $resource | |
48 | ||
49 | # main | |
50 | var=$(echo _$cmd | sed 's/[^A-Za-z0-9]/_/g;s/__*/_/g') | |
51 | newvar=($(eval $cmd | awk '{ for (i = 1; i <= NF; i++) if ($(i) ~ /^[-+0-9][0-9]*$/) printf " %s", $(i) }')) | |
52 | eval oldvar=(\${$var[@]}) | |
53 | ||
54 | # compute diff | |
55 | for ((i=0; i<${#newvar[@]}; i++)); do | |
56 | [ "${oldvar[$i]}" ] && v[$i]=$((newvar[i] - oldvar[i])) || v[$i]=${newvar[$i]} | |
57 | done | |
58 | ||
59 | # ouput | |
60 | if [ "$set" = "yes" ]; then | |
61 | echo "$var=(${newvar[@]})" >>$resource | |
62 | awk -F= '{ a[$1]=$2 } END { for (i in a) print i "=" a[i] }' $resource > $resource.n | |
63 | mv -f $resource.n $resource | |
64 | else | |
65 | str="" | |
66 | for ((i=0; i<${#v[@]}; i++)); do | |
67 | str="$str; v[$i]=${v[$i]}" | |
68 | done | |
69 | eval $cmd | gawk 'BEGIN { k=0 '"$str"' } { n = 1; for (i=1; i<=NF; i++) if ($(i) ~ /^[-+0-9][0-9]*$/) { if ((i == 1) && ! ($0 ~ /^ /)) $(i) = v[k]; else $0 = gensub(/ [-+0-9][0-9]*/, " " v[k], n); k++; n++ } print }' | |
70 | fi |