| 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 |