bin/khatus

   1 #! /bin/bash
   2
   3 set -e
   4
   5 consume() {
   6     pipe="$1"
   7     debug="$2"
   8     bin="$3"
   9     prefixes_of_net_interfaces_to_show="$4"
  10     tail -f "$pipe" \
  11     | stdbuf -o L "$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 run_producer() {
  18     pipe="$1"
  19     bin="$2"
  20     cmd="$3"
  21     msg_head="$4"
  22     "$bin"/$cmd \
  23     2> >(
  24         while read line
  25         do
  26             echo "ERROR ${msg_head} $line" > "$pipe"
  27         done \
  28     ) \
  29     | while read line
  30         do
  31             echo "OK ${msg_head} $line" > "$pipe"
  32         done
  33 }
  34
  35 fork_watcher() {
  36     run_producer "$@" &
  37 }
  38
  39 fork_poller() {
  40     interval="$1"
  41     shift
  42     while :
  43     do
  44         run_producer "$@"
  45         sleep "$interval"
  46     done &
  47 }
  48
  49 main() {
  50     declare -A opts=(
  51         ["--debug"]=0
  52         ["--dir_bin"]="$HOME/bin"
  53         ["--file_pipe"]=$(mktemp)
  54         ["--weather_station_id"]='KJFK'
  55         ["--screen_brightness_device_name"]='acpi_video0'
  56         ["--prefixes_of_net_interfaces_to_show"]='w'  # comma-separated
  57         ["--wifi_interface"]=''
  58         ["--disk_space_device"]='/'
  59         ["--disk_io_device"]='sda'
  60         ["--thermal_zone"]=0
  61         ["--fan_path"]='/proc/acpi/ibm/fan'
  62         ["--interval_inp_datetime"]=1
  63         ["--interval_inp_brightness"]=1
  64         ["--interval_inp_weather"]=$(( 30 * 60))  # 30 minutes
  65         ["--interval_inp_mpd_state"]=1
  66         ["--interval_inp_mpd_song"]=1
  67         ["--interval_inp_volume"]=1
  68         ["--interval_inp_bluetooth"]=5
  69         ["--interval_inp_net_wifi"]=5
  70         ["--interval_inp_net_io"]=1
  71         ["--interval_inp_disk_space"]=5
  72         ["--interval_inp_disk_io"]=1
  73         ["--interval_inp_loadavg"]=1
  74         ["--interval_inp_temp"]=1
  75         ["--interval_inp_fan"]=1
  76         ["--interval_inp_mem"]=1
  77     )
  78     while :
  79     do
  80         key="$1"
  81         val="$2"
  82         case "$key" in
  83             '')
  84                 break
  85                 ;;
  86             * )
  87                 if [ -v opts["$key"] ]
  88                 then
  89                     if [ "$key" == "--debug" ]
  90                     then
  91                         opts["$key"]=1
  92                         shift
  93                     elif [ "$val" != "" ]
  94                     then
  95                         opts["$key"]="$val"
  96                         shift
  97                         shift
  98                     else
  99                         echo "Option $key requires an argument" >&2
 100                         exit 1
 101                     fi
 102                 else
 103                     echo "Unknown option: $key" >&2
 104                     exit 1
 105                 fi
 106         esac
 107     done
 108
 109     if [ "${opts['--wifi_interface']}" = ''  ]
 110     then
 111         echo 'Please provide the required parameter: --wifi_interface' >&2
 112         exit 1
 113     fi
 114
 115     (
 116         echo '=============================================='
 117         echo "Khatus starting with the following parameters:"
 118         echo '=============================================='
 119         for param in ${!opts[@]}
 120         do
 121             echo "$param := ${opts[$param]}"
 122         done \
 123         | column -ts: \
 124         | sort
 125         echo '----------------------------------------------'
 126     ) >&2
 127
 128     screen_brightness_device_path='/sys/class/backlight'
 129     screen_brightness_device_path+="/${opts['--screen_brightness_device_name']}"
 130
 131     # Just shorthand
 132     pipe="${opts['--file_pipe']}"
 133     bin="${opts['--dir_bin']}"
 134
 135     rm -f "$pipe"
 136     mkfifo "$pipe"
 137
 138     cmd_sens_screen_brightness='khatus_sensor_screen_brightness'
 139     cmd_sens_screen_brightness+=" $screen_brightness_device_path"
 140
 141     cmd_sens_weather="khatus_sensor_weather ${opts['--weather_station_id']}"
 142     cmd_sens_disk_space="khatus_sensor_disk_space ${opts['--disk_space_device']}"
 143     cmd_sens_disk_io="khatus_sensor_disk_io ${opts['--disk_io_device']}"
 144     cmd_sens_temperature="khatus_sensor_temperature ${opts['--thermal_zone']}"
 145     cmd_sens_fan="khatus_sensor_fan ${opts['--fan_path']}"
 146     cmd_sens_bluetooth="khatus_sensor_bluetooth_power $bin"
 147     cmd_sens_mpd_state="khatus_sensor_mpd_state $bin"
 148     cmd_sens_net_addr_io="khatus_sensor_net_addr_io $bin"
 149     cmd_sens_volume="khatus_sensor_volume $bin"
 150     cmd_sens_wifi="khatus_sensor_net_wifi_status $bin ${opts['--wifi_interface']}"
 151
 152     fork_watcher                                       "$pipe" "$bin" "khatus_sensor_energy $bin"     'in:ENERGY'
 153     fork_poller "${opts['--interval_inp_datetime']}"   "$pipe" "$bin" khatus_sensor_datetime          'in:DATE_TIME'
 154     fork_poller "${opts['--interval_inp_brightness']}" "$pipe" "$bin" "$cmd_sens_screen_brightness"   'in:SCREEN_BRIGHTNESS'
 155     fork_poller "${opts['--interval_inp_weather']}"    "$pipe" "$bin" "$cmd_sens_weather"             'in:WEATHER'
 156     fork_poller "${opts['--interval_inp_mpd_state']}"  "$pipe" "$bin" "$cmd_sens_mpd_state"           'in:MPD_STATE'
 157     fork_poller "${opts['--interval_inp_mpd_song']}"   "$pipe" "$bin" khatus_sensor_mpd_song          'in:MPD_SONG'
 158     fork_poller "${opts['--interval_inp_volume']}"     "$pipe" "$bin" "$cmd_sens_volume"              'in:VOLUME'
 159     fork_poller "${opts['--interval_inp_bluetooth']}"  "$pipe" "$bin" "$cmd_sens_bluetooth"           'in:BLUETOOTH_POWER'
 160     fork_poller "${opts['--interval_inp_net_wifi']}"   "$pipe" "$bin" "$cmd_sens_wifi"                'in:NET_WIFI_STATUS'
 161     fork_poller "${opts['--interval_inp_net_io']}"     "$pipe" "$bin" "$cmd_sens_net_addr_io"         'in:NET_ADDR_IO'
 162     fork_poller "${opts['--interval_inp_disk_space']}" "$pipe" "$bin" "$cmd_sens_disk_space"          'in:DISK_SPACE'
 163     fork_poller "${opts['--interval_inp_disk_io']}"    "$pipe" "$bin" "$cmd_sens_disk_io"             'in:DISK_IO'
 164     fork_poller "${opts['--interval_inp_loadavg']}"    "$pipe" "$bin" khatus_sensor_loadavg           'in:LOAD_AVG'
 165     fork_poller "${opts['--interval_inp_temp']}"       "$pipe" "$bin" "$cmd_sens_temperature"         'in:TEMPERATURE'
 166     fork_poller "${opts['--interval_inp_fan']}"        "$pipe" "$bin" "$cmd_sens_fan"                 'in:FAN'
 167     fork_poller "${opts['--interval_inp_mem']}"        "$pipe" "$bin" khatus_sensor_memory            'in:MEMORY'
 168
 169     consume \
 170       "$pipe" \
 171       "${opts['--debug']}" \
 172       "$bin" \
 173       "${opts['--prefixes_of_net_interfaces_to_show']}"
 174 }
 175
 176 main $@