bin/khatus

   1 #! /bin/bash
   2
   3 MSG_FS='|'
   4
   5 set -e
   6
   7 executable_name_of_cmd() {
   8     basename "$(echo $1 | awk '{print $1; exit}')"
   9 }
  10
  11 run_producer() {
  12     pipe="$1"
  13     bin="$2"
  14     cmd="$3"
  15     executable_name="$4"
  16     perf_log="$5"
  17
  18     if [ ! "$perf_log" = '' ]
  19     then
  20         # %S system  time in seconds
  21         # %U user    time in seconds
  22         # %e elapsed time in seconds
  23         # %c context switches involuntary
  24         # %w context switches voluntary
  25         # %x exit code
  26         time_fmt='%S %U %e %c %w %x'
  27         time="/usr/bin/time -ao ${perf_log} -f "
  28         time_sep=' '
  29     else
  30         time_fmt=''
  31         time=''
  32         time_sep=''
  33     fi
  34
  35     ${time}"${time_fmt}"${time_sep}$bin/$cmd \
  36     2> >(
  37         while read line
  38         do
  39             echo "ERROR${MSG_FS}${executable_name}${MSG_FS}$line" > "$pipe"
  40         done \
  41     ) \
  42     | while read line
  43         do
  44             echo "OK${MSG_FS}${executable_name}${MSG_FS}$line" > "$pipe"
  45         done
  46     cmd_exit_code=${PIPESTATUS[0]}
  47     if [ "$cmd_exit_code" -ne 0 ]
  48     then
  49         echo
  50             "ERROR${MSG_FS}${executable_name}${MSG_FS}NON_ZERO_EXIT_CODE${MSG_FS}$cmd_exit_code" \
  51             > "$pipe"
  52     fi
  53 }
  54
  55 fork_watcher() {
  56     pipe="$1"
  57     bin="$2"
  58     cmd="$3"
  59     executable_name=$(executable_name_of_cmd "$cmd")
  60     run_producer "$pipe" "$bin" "$cmd" "$executable_name" &
  61 }
  62
  63 fork_poller() {
  64     interval="$1"
  65     perf_log_dir="$2"
  66     shift 2
  67     pipe="$1"
  68     bin="$2"
  69     cmd="$3"
  70
  71     executable_name=$(basename "$(echo $cmd | awk '{print $1; exit}')")
  72
  73     if [ ! "$perf_log_dir" = '' ]
  74     then
  75         cmd="$3"
  76         perf_log_file=${executable_name}.log
  77         mkdir -p "$perf_log_dir"
  78         perf_log_path="$perf_log_dir/$perf_log_file"
  79     fi
  80
  81     while :
  82     do
  83         run_producer "$pipe" "$bin" "$cmd" "$executable_name" "$perf_log_path"
  84         sleep "$interval"
  85     done &
  86 }
  87
  88 main() {
  89     declare -A opts=(
  90         ["--debug"]=0
  91         ["--dir_bin"]="$HOME/bin"
  92         ["--dir_perf_logs"]=''
  93         ["--file_pipe"]=$(mktemp)
  94         ["--weather_station_id"]='KJFK'
  95         ["--screen_brightness_device_name"]='acpi_video0'
  96         ["--net_interfaces_to_show"]=''  # comma-separated
  97         ["--wifi_interface"]=''
  98         ["--disk_space_device"]='/'
  99         ["--disk_io_device"]='sda'
 100         ["--thermal_zone"]=0
 101         ["--fan_path"]='/proc/acpi/ibm/fan'
 102         ["--pulseaudio_sink"]='0'
 103         ["--interval_inp_datetime"]=1
 104         ["--interval_inp_brightness"]=1
 105         ["--interval_inp_weather"]=$(( 30 * 60))  # 30 minutes
 106         ["--interval_inp_mpd"]=1
 107         ["--interval_inp_volume"]=1
 108         ["--interval_inp_bluetooth"]=1
 109         ["--interval_inp_net_wifi"]=1
 110         ["--interval_inp_net_io"]=1
 111         ["--interval_inp_disk_space"]=1
 112         ["--interval_inp_disk_io"]=1
 113         ["--interval_inp_loadavg"]=1
 114         ["--interval_inp_temp"]=1
 115         ["--interval_inp_fan"]=1
 116         ["--interval_inp_mem"]=1
 117     )
 118     while :
 119     do
 120         key="$1"
 121         val="$2"
 122         case "$key" in
 123             '')
 124                 break
 125                 ;;
 126             * )
 127                 if [ -v opts["$key"] ]
 128                 then
 129                     if [ "$key" == "--debug" ]
 130                     then
 131                         opts["$key"]=1
 132                         shift
 133                     elif [ "$val" != "" ]
 134                     then
 135                         opts["$key"]="$val"
 136                         shift
 137                         shift
 138                     else
 139                         echo "Option $key requires an argument" >&2
 140                         exit 1
 141                     fi
 142                 else
 143                     echo "Unknown option: $key" >&2
 144                     exit 1
 145                 fi
 146         esac
 147     done
 148
 149     if [ "${opts['--wifi_interface']}" = ''  ]
 150     then
 151         echo 'Please provide the required parameter: --wifi_interface' >&2
 152         exit 1
 153     fi
 154
 155     (
 156         echo '=============================================='
 157         echo "Khatus starting with the following parameters:"
 158         echo '=============================================='
 159         for param in ${!opts[@]}
 160         do
 161             echo "$param := ${opts[$param]}"
 162         done \
 163         | column -ts: \
 164         | sort
 165         echo '----------------------------------------------'
 166     ) >&2
 167
 168     screen_brightness_device_path='/sys/class/backlight'
 169     screen_brightness_device_path+="/${opts['--screen_brightness_device_name']}"
 170
 171     # Just shorthand
 172     pipe="${opts['--file_pipe']}"
 173     bin="${opts['--dir_bin']}"
 174     perf="${opts['--dir_perf_logs']}"
 175
 176     rm -f "$pipe"
 177     mkfifo "$pipe"
 178
 179     cmd_sens_screen_brightness='khatus_sensor_screen_brightness'
 180     cmd_sens_screen_brightness+=" $screen_brightness_device_path"
 181
 182     cmd_sens_weather="khatus_sensor_weather $bin ${opts['--weather_station_id']}"
 183     cmd_sens_disk_space="khatus_sensor_disk_space $bin ${opts['--disk_space_device']}"
 184     cmd_sens_disk_io="khatus_sensor_disk_io $bin ${opts['--disk_io_device']}"
 185     cmd_sens_temperature="khatus_sensor_temperature ${opts['--thermal_zone']}"
 186     cmd_sens_fan="khatus_sensor_fan $bin ${opts['--fan_path']}"
 187     cmd_sens_bluetooth="khatus_sensor_bluetooth_power $bin"
 188     cmd_sens_mpd="khatus_sensor_mpd $bin"
 189     cmd_sens_net_addr_io="khatus_sensor_net_addr_io $bin"
 190     cmd_sens_volume="khatus_sensor_volume $bin"
 191     cmd_sens_wifi="khatus_sensor_net_wifi_status $bin ${opts['--wifi_interface']}"
 192     cmd_sens_loadavg="khatus_sensor_loadavg $bin"
 193     cmd_sens_memory="khatus_sensor_memory $bin"
 194
 195     fork_watcher                                               "$pipe" "$bin" "khatus_sensor_energy $bin"
 196     fork_poller "${opts['--interval_inp_datetime']}"   "$perf" "$pipe" "$bin" khatus_sensor_datetime
 197     fork_poller "${opts['--interval_inp_brightness']}" "$perf" "$pipe" "$bin" "$cmd_sens_screen_brightness"
 198     fork_poller "${opts['--interval_inp_weather']}"    "$perf" "$pipe" "$bin" "$cmd_sens_weather"
 199     fork_poller "${opts['--interval_inp_mpd']}"        "$perf" "$pipe" "$bin" "$cmd_sens_mpd"
 200     fork_poller "${opts['--interval_inp_volume']}"     "$perf" "$pipe" "$bin" "$cmd_sens_volume"
 201     fork_poller "${opts['--interval_inp_bluetooth']}"  "$perf" "$pipe" "$bin" "$cmd_sens_bluetooth"
 202     fork_poller "${opts['--interval_inp_net_wifi']}"   "$perf" "$pipe" "$bin" "$cmd_sens_wifi"
 203     fork_poller "${opts['--interval_inp_net_io']}"     "$perf" "$pipe" "$bin" "$cmd_sens_net_addr_io"
 204     fork_poller "${opts['--interval_inp_disk_space']}" "$perf" "$pipe" "$bin" "$cmd_sens_disk_space"
 205     fork_poller "${opts['--interval_inp_disk_io']}"    "$perf" "$pipe" "$bin" "$cmd_sens_disk_io"
 206     fork_poller "${opts['--interval_inp_loadavg']}"    "$perf" "$pipe" "$bin" "$cmd_sens_loadavg"
 207     fork_poller "${opts['--interval_inp_temp']}"       "$perf" "$pipe" "$bin" "$cmd_sens_temperature"
 208     fork_poller "${opts['--interval_inp_fan']}"        "$perf" "$pipe" "$bin" "$cmd_sens_fan"
 209     fork_poller "${opts['--interval_inp_mem']}"        "$perf" "$pipe" "$bin" "$cmd_sens_memory"
 210
 211     stdbuf -o L tail -f "$pipe" \
 212     | stdbuf -o L "$bin"/khatus_bar \
 213         -F "$MSG_FS" \
 214         -v opt_debug=""${opts['--debug']}"" \
 215         -v opt_mpd_song_max_chars=10 \
 216         -v opt_net_interfaces_to_show="${opts['--net_interfaces_to_show']}" \
 217         -v opt_pulseaudio_sink="${opts['--pulseaudio_sink']}" \
 218     | stdbuf -o L tee >("$bin"/khatus_actuate_status_bar_to_xsetroot_name) \
 219     | stdbuf -o L "$bin"/khatus_monitor_energy \
 220     | stdbuf -o L "$bin"/khatus_monitor_errors \
 221     | stdbuf -o L tee >("$bin"/khatus_actuate_alert_to_notify_send) \
 222     > /dev/null
 223
 224 }
 225
 226 main $@