bin/khatus

   1 #! /bin/bash
   2
   3 set -e
   4
   5 consume() {
   6     pipe="$1"
   7     debug="$2"
   8     dir_bin="$3"
   9     prefixes_of_net_interfaces_to_show="$4"
  10     tail -f "$pipe" \
  11     | stdbuf -o L "$dir_bin"/khatus_controller \
  12         -v opt_debug="$debug" \
  13         -v opt_mpd_song_max_chars=10 \
  14         -v opt_prefixes_of_net_interfaces_to_show="$prefixes_of_net_interfaces_to_show"
  15 }
  16
  17 spawn_reactor() {
  18     pipe="$1"
  19     dir_bin="$2"
  20     log="$3"
  21     cmd="$4"
  22     msg_head="$5"
  23     "$dir_bin"/$cmd | while read line; do
  24         echo "${msg_head} $line" > "$pipe"
  25     done \
  26     2> "$log" \
  27     &
  28 }
  29
  30 spawn_poller() {
  31     pipe="$1"
  32     dir_bin="$2"
  33     log="$3"
  34     cmd="$4"
  35     msg_head="$5"
  36     interval="$6"
  37     while true; do
  38         "$dir_bin"/$cmd | while read line; do
  39             echo "${msg_head} $line" > "$pipe"
  40         done \
  41         2> "$log"
  42         sleep "$interval"
  43     done &
  44 }
  45
  46 main() {
  47     # Defaults
  48     debug=0
  49     dir_bin="$HOME/bin"
  50     dir_logs=$(mktemp -d)
  51     file_pipe=$(mktemp)
  52     weather_station_id='KJFK'
  53     screen_brightness_device_name='acpi_video0'
  54     prefixes_of_net_interfaces_to_show='w'  # comma-separated
  55     disk_space_device='/'
  56     disk_io_device='sda'
  57     thermal_zone=0
  58     fan_path='/proc/acpi/ibm/fan'
  59
  60     interval_inp_datetime=1
  61     interval_inp_brightness=1
  62     interval_inp_weather=$(( 30 * 60))  # 30 minutes
  63     interval_inp_mpd_state=1
  64     interval_inp_mpd_song=1
  65     interval_inp_volume=1
  66     interval_inp_bluetooth=5
  67     interval_inp_net_wifi=5
  68     interval_inp_net_io=1
  69     interval_inp_disk_space=5
  70     interval_inp_disk_io=1
  71     interval_inp_loadavg=1
  72     interval_inp_temp=1
  73     interval_inp_fan=1
  74     interval_inp_mem=1
  75
  76     # User-overrides
  77     # ---------------------------------------------------------------------
  78     # IMPORTANT:
  79     # In order for automatic value reporting, at startup, to work - ensure that
  80     # long option names match corresponding variable names!
  81     # ---------------------------------------------------------------------
  82     long_options=''
  83     long_options+='debug'
  84     long_options+=',dir_bin:'
  85     long_options+=',dir_logs:'
  86     long_options+=',file_pipe:'
  87     long_options+=',weather_station_id:'
  88     long_options+=',screen_brightness_device_name:'
  89     long_options+=',prefixes_of_net_interfaces_to_show:'
  90     long_options+=',disk_space_device:'
  91     long_options+=',disk_io_device:'
  92     long_options+=',thermal_zone:'
  93     long_options+=',fan_path:'
  94     long_options+=',interval_inp_datetime:'
  95     long_options+=',interval_inp_brightness:'
  96     long_options+=',interval_inp_weather:'
  97     long_options+=',interval_inp_mpd_state:'
  98     long_options+=',interval_inp_mpd_song:'
  99     long_options+=',interval_inp_volume:'
 100     long_options+=',interval_inp_bluetooth:'
 101     long_options+=',interval_inp_net_wifi:'
 102     long_options+=',interval_inp_net_io:'
 103     long_options+=',interval_inp_disk_space:'
 104     long_options+=',interval_inp_disk_io:'
 105     long_options+=',interval_inp_loadavg:'
 106     long_options+=',interval_inp_temp:'
 107     long_options+=',interval_inp_fan:'
 108     long_options+=',interval_inp_mem:'
 109     OPTS=$(
 110         getopt \
 111             -o 'd' \
 112             -l $long_options \
 113             -- "$@"
 114     )
 115     eval set -- "$OPTS"
 116     while true
 117     do
 118         case "$1" in
 119             -d|--debug)
 120                 debug=1
 121                 shift
 122                 ;;
 123             --dir_bin)
 124                 dir_bin="$2"
 125                 shift 2
 126                 ;;
 127             --dir_logs)
 128                 dir_logs="$2"
 129                 shift 2
 130                 ;;
 131             --file_pipe)
 132                 file_pipe="$2"
 133                 shift 2
 134                 ;;
 135             --weather_station_id)
 136                 weather_station_id="$2"
 137                 shift 2
 138                 ;;
 139             --screen_brightness_device_name)
 140                 screen_brightness_device_name="$2"
 141                 shift 2
 142                 ;;
 143             --prefixes_of_net_interfaces_to_show)
 144                 prefixes_of_net_interfaces_to_show="$2"
 145                 shift 2
 146                 ;;
 147             --disk_space_device)
 148                 disk_space_device="$2"
 149                 shift 2
 150                 ;;
 151             --disk_io_device)
 152                 disk_io_device="$2"
 153                 shift 2
 154                 ;;
 155             --thermal_zone)
 156                 thermal_zone="$2"
 157                 shift 2
 158                 ;;
 159             --fan_path)
 160                 fan_path="$2"
 161                 shift 2
 162                 ;;
 163             --interval_inp_datetime)
 164                 interval_inp_datetime="$2"
 165                 shift 2
 166                 ;;
 167             --interval_inp_brightness)
 168                 interval_inp_brightness="$2"
 169                 shift 2
 170                 ;;
 171             --interval_inp_weather)
 172                 interval_inp_weather="$2"
 173                 shift 2
 174                 ;;
 175             --interval_inp_mpd_state)
 176                 interval_inp_mpd_state="$2"
 177                 shift 2
 178                 ;;
 179             --interval_inp_mpd_song)
 180                 interval_inp_mpd_song="$2"
 181                 shift 2
 182                 ;;
 183             --interval_inp_volume)
 184                 interval_inp_volume="$2"
 185                 shift 2
 186                 ;;
 187             --interval_inp_bluetooth)
 188                 interval_inp_bluetooth="$2"
 189                 shift 2
 190                 ;;
 191             --interval_inp_net_wifi)
 192                 interval_inp_net_wifi="$2"
 193                 shift 2
 194                 ;;
 195             --interval_inp_net_io)
 196                 interval_inp_net_io="$2"
 197                 shift 2
 198                 ;;
 199             --interval_inp_disk_space)
 200                 interval_inp_disk_space="$2"
 201                 shift 2
 202                 ;;
 203             --interval_inp_disk_io)
 204                 interval_inp_disk_io="$2"
 205                 shift 2
 206                 ;;
 207             --interval_inp_loadavg)
 208                 interval_inp_loadavg="$2"
 209                 shift 2
 210                 ;;
 211             --interval_inp_temp)
 212                 interval_inp_temp="$2"
 213                 shift 2
 214                 ;;
 215             --interval_inp_fan)
 216                 interval_inp_fan="$2"
 217                 shift 2
 218                 ;;
 219             --interval_inp_mem)
 220                 interval_inp_mem="$2"
 221                 shift 2
 222                 ;;
 223             --)
 224                 shift
 225                 break
 226                 ;;
 227         esac
 228     done
 229
 230     screen_brightness_device_path='/sys/class/backlight'
 231     screen_brightness_device_path+="/$screen_brightness_device_name"
 232
 233     (
 234         echo '=============================================='
 235         echo "Khatus starting with the following parameters:"
 236         echo '=============================================='
 237         for param in \
 238             $(echo -n "$long_options" \
 239             | awk -v RS=, '{sub(":$", ""); print $0}'
 240             )
 241         do
 242             echo "$param := ${!param}"
 243         done \
 244         | column -ts:
 245         echo '----------------------------------------------'
 246     ) >&2
 247
 248     rm -f "$file_pipe"
 249     mkfifo "$file_pipe"
 250
 251     cmd_sens_screen_brightness='khatus_sensor_screen_brightness'
 252     cmd_sens_screen_brightness+=" $screen_brightness_device_path"
 253
 254     cmd_sens_weather="khatus_sensor_weather $weather_station_id"
 255
 256     cmd_sens_disk_space="khatus_sensor_disk_space $disk_space_device"
 257
 258     cmd_sens_disk_io="khatus_sensor_disk_io $disk_io_device"
 259
 260     cmd_sens_temperature="khatus_sensor_temperature $thermal_zone"
 261
 262     cmd_sens_fan="khatus_sensor_fan $fan_path"
 263
 264     # TODO: Redirect each worker's stderr to a dedicated log file
 265     pipe="$file_pipe"
 266     log="$dir_logs"/khatus_sensors.log
 267
 268     spawn_reactor "$pipe" "$dir_bin" "$log" "khatus_sensor_energy $dir_bin" 'in:ENERGY'
 269
 270     spawn_poller "$pipe" "$dir_bin" "$log" khatus_sensor_datetime        'in:DATE_TIME'         $interval_inp_datetime
 271     spawn_poller "$pipe" "$dir_bin" "$log" "$cmd_sens_screen_brightness" 'in:SCREEN_BRIGHTNESS' $interval_inp_brightness
 272     spawn_poller "$pipe" "$dir_bin" "$log" "$cmd_sens_weather"           'in:WEATHER'           $interval_inp_weather
 273     spawn_poller "$pipe" "$dir_bin" "$log" khatus_sensor_mpd_state       'in:MPD_STATE'         $interval_inp_mpd_state
 274     spawn_poller "$pipe" "$dir_bin" "$log" khatus_sensor_mpd_song        'in:MPD_SONG'          $interval_inp_mpd_song
 275     spawn_poller "$pipe" "$dir_bin" "$log" khatus_sensor_volume          'in:VOLUME'            $interval_inp_volume
 276     spawn_poller "$pipe" "$dir_bin" "$log" khatus_sensor_bluetooth_power 'in:BLUETOOTH_POWER'   $interval_inp_bluetooth
 277     spawn_poller "$pipe" "$dir_bin" "$log" khatus_sensor_net_wifi_status 'in:NET_WIFI_STATUS'   $interval_inp_net_wifi
 278     spawn_poller "$pipe" "$dir_bin" "$log" khatus_sensor_net_addr_io     'in:NET_ADDR_IO'       $interval_inp_net_io
 279     spawn_poller "$pipe" "$dir_bin" "$log" "$cmd_sens_disk_space"        'in:DISK_SPACE'        $interval_inp_disk_space
 280     spawn_poller "$pipe" "$dir_bin" "$log" "$cmd_sens_disk_io"           'in:DISK_IO'           $interval_inp_disk_io
 281     spawn_poller "$pipe" "$dir_bin" "$log" khatus_sensor_loadavg         'in:LOAD_AVG'          $interval_inp_loadavg
 282     spawn_poller "$pipe" "$dir_bin" "$log" "$cmd_sens_temperature"       'in:TEMPERATURE'       $interval_inp_temp
 283     spawn_poller "$pipe" "$dir_bin" "$log" "$cmd_sens_fan"               'in:FAN'               $interval_inp_fan
 284     spawn_poller "$pipe" "$dir_bin" "$log" khatus_sensor_memory          'in:MEMORY'            $interval_inp_mem
 285
 286     consume \
 287       "$pipe" \
 288       "$debug" \
 289       "$dir_bin" \
 290       "$prefixes_of_net_interfaces_to_show"
 291 }
 292
 293 main $@