consume() {
pipe="$1"
debug="$2"
- dir_bin="$3"
+ bin="$3"
prefixes_of_net_interfaces_to_show="$4"
tail -f "$pipe" \
- | stdbuf -o L "$dir_bin"/khatus_controller \
+ | stdbuf -o L "$bin"/khatus_controller \
-v opt_debug="$debug" \
-v opt_mpd_song_max_chars=10 \
-v opt_prefixes_of_net_interfaces_to_show="$prefixes_of_net_interfaces_to_show"
}
-spawn() {
+run_producer() {
pipe="$1"
- dir_bin="$2"
- log="$3"
- cmd="$4"
- msg_head="$5"
- interval="$6"
- while true; do
- "$dir_bin"/$cmd | while read line; do
- echo "${msg_head} $line" > "$pipe"
+ bin="$2"
+ cmd="$3"
+ msg_head="$4"
+ "$bin"/$cmd \
+ 2> >(
+ while read line
+ do
+ echo "ERROR ${msg_head} $line" > "$pipe"
done \
- 2> "$log"
+ ) \
+ | while read line
+ do
+ echo "OK ${msg_head} $line" > "$pipe"
+ done
+}
+
+fork_watcher() {
+ run_producer "$@" &
+}
+
+fork_poller() {
+ interval="$1"
+ shift
+ while :
+ do
+ run_producer "$@"
sleep "$interval"
done &
}
main() {
- # Defaults
- debug=0
- dir_bin="$HOME/bin"
- dir_data=$(mktemp -d)
- weather_station_id='KJFK'
- screen_brightness_device_name='acpi_video0'
- prefixes_of_net_interfaces_to_show='w' # comma-separated
- disk_space_device='/'
- disk_io_device='sda'
- thermal_zone=0
- fan_path='/proc/acpi/ibm/fan'
-
- interval_inp_datetime=1
- interval_inp_brightness=1
- interval_inp_weather=$(( 30 * 60)) # 30 minutes
- interval_inp_mpd_state=1
- interval_inp_mpd_song=1
- interval_inp_volume=1
- interval_inp_bluetooth=5
- interval_inp_net_wifi=5
- interval_inp_net_io=1
- interval_inp_disk_space=5
- interval_inp_disk_io=1
- interval_inp_loadavg=1
- interval_inp_temp=1
- interval_inp_fan=1
- interval_inp_mem=1
- interval_inp_energy=1
-
- # User-overrides
- # ---------------------------------------------------------------------
- # IMPORTANT:
- # In order for automatic value reporting, at startup, to work - ensure that
- # long option names match corresponding variable names!
- # ---------------------------------------------------------------------
- long_options=''
- long_options+='debug'
- long_options+=',dir_bin:'
- long_options+=',dir_data:'
- long_options+=',weather_station_id:'
- long_options+=',screen_brightness_device_name:'
- long_options+=',prefixes_of_net_interfaces_to_show:'
- long_options+=',disk_space_device:'
- long_options+=',disk_io_device:'
- long_options+=',thermal_zone:'
- long_options+=',fan_path:'
- long_options+=',interval_inp_datetime:'
- long_options+=',interval_inp_brightness:'
- long_options+=',interval_inp_weather:'
- long_options+=',interval_inp_mpd_state:'
- long_options+=',interval_inp_mpd_song:'
- long_options+=',interval_inp_volume:'
- long_options+=',interval_inp_bluetooth:'
- long_options+=',interval_inp_net_wifi:'
- long_options+=',interval_inp_net_io:'
- long_options+=',interval_inp_disk_space:'
- long_options+=',interval_inp_disk_io:'
- long_options+=',interval_inp_loadavg:'
- long_options+=',interval_inp_temp:'
- long_options+=',interval_inp_fan:'
- long_options+=',interval_inp_mem:'
- long_options+=',interval_inp_energy:'
- OPTS=$(
- getopt \
- -o 'd' \
- -l $long_options \
- -- "$@"
+ declare -A opts=(
+ ["--debug"]=0
+ ["--dir_bin"]="$HOME/bin"
+ ["--file_pipe"]=$(mktemp)
+ ["--weather_station_id"]='KJFK'
+ ["--screen_brightness_device_name"]='acpi_video0'
+ ["--prefixes_of_net_interfaces_to_show"]='w' # comma-separated
+ ["--wifi_interface"]=''
+ ["--disk_space_device"]='/'
+ ["--disk_io_device"]='sda'
+ ["--thermal_zone"]=0
+ ["--fan_path"]='/proc/acpi/ibm/fan'
+ ["--interval_inp_datetime"]=1
+ ["--interval_inp_brightness"]=1
+ ["--interval_inp_weather"]=$(( 30 * 60)) # 30 minutes
+ ["--interval_inp_mpd_state"]=1
+ ["--interval_inp_mpd_song"]=1
+ ["--interval_inp_volume"]=1
+ ["--interval_inp_bluetooth"]=5
+ ["--interval_inp_net_wifi"]=5
+ ["--interval_inp_net_io"]=1
+ ["--interval_inp_disk_space"]=5
+ ["--interval_inp_disk_io"]=1
+ ["--interval_inp_loadavg"]=1
+ ["--interval_inp_temp"]=1
+ ["--interval_inp_fan"]=1
+ ["--interval_inp_mem"]=1
)
- eval set -- "$OPTS"
- while true
+ while :
do
- case "$1" in
- -d|--debug)
- debug=1
- shift
- ;;
- --dir_bin)
- dir_bin="$2"
- shift 2
- ;;
- --dir_data)
- dir_data="$2"
- shift 2
- ;;
- --weather_station_id)
- weather_station_id="$2"
- shift 2
- ;;
- --screen_brightness_device_name)
- screen_brightness_device_name="$2"
- shift 2
- ;;
- --prefixes_of_net_interfaces_to_show)
- prefixes_of_net_interfaces_to_show="$2"
- shift 2
- ;;
- --disk_space_device)
- disk_space_device="$2"
- shift 2
- ;;
- --disk_io_device)
- disk_io_device="$2"
- shift 2
- ;;
- --thermal_zone)
- thermal_zone="$2"
- shift 2
- ;;
- --fan_path)
- fan_path="$2"
- shift 2
- ;;
- --interval_inp_datetime)
- interval_inp_datetime="$2"
- shift 2
- ;;
- --interval_inp_brightness)
- interval_inp_brightness="$2"
- shift 2
- ;;
- --interval_inp_weather)
- interval_inp_weather="$2"
- shift 2
- ;;
- --interval_inp_mpd_state)
- interval_inp_mpd_state="$2"
- shift 2
- ;;
- --interval_inp_mpd_song)
- interval_inp_mpd_song="$2"
- shift 2
- ;;
- --interval_inp_volume)
- interval_inp_volume="$2"
- shift 2
- ;;
- --interval_inp_bluetooth)
- interval_inp_bluetooth="$2"
- shift 2
- ;;
- --interval_inp_net_wifi)
- interval_inp_net_wifi="$2"
- shift 2
- ;;
- --interval_inp_net_io)
- interval_inp_net_io="$2"
- shift 2
- ;;
- --interval_inp_disk_space)
- interval_inp_disk_space="$2"
- shift 2
- ;;
- --interval_inp_disk_io)
- interval_inp_disk_io="$2"
- shift 2
- ;;
- --interval_inp_loadavg)
- interval_inp_loadavg="$2"
- shift 2
- ;;
- --interval_inp_temp)
- interval_inp_temp="$2"
- shift 2
- ;;
- --interval_inp_fan)
- interval_inp_fan="$2"
- shift 2
- ;;
- --interval_inp_mem)
- interval_inp_mem="$2"
- shift 2
- ;;
- --interval_inp_energy)
- interval_inp_energy="$2"
- shift 2
- ;;
- --)
- shift
+ key="$1"
+ val="$2"
+ case "$key" in
+ '')
break
;;
+ * )
+ if [ -v opts["$key"] ]
+ then
+ if [ "$key" == "--debug" ]
+ then
+ opts["$key"]=1
+ shift
+ elif [ "$val" != "" ]
+ then
+ opts["$key"]="$val"
+ shift
+ shift
+ else
+ echo "Option $key requires an argument" >&2
+ exit 1
+ fi
+ else
+ echo "Unknown option: $key" >&2
+ exit 1
+ fi
esac
done
- pipe="$dir_data/khatus_data_pipe"
- screen_brightness_device_path='/sys/class/backlight'
- screen_brightness_device_path+="/$screen_brightness_device_name"
+ if [ "${opts['--wifi_interface']}" = '' ]
+ then
+ echo 'Please provide the required parameter: --wifi_interface' >&2
+ exit 1
+ fi
(
echo '=============================================='
echo "Khatus starting with the following parameters:"
echo '=============================================='
- for param in \
- $(echo -n "$long_options" \
- | awk -v RS=, '{sub(":$", ""); print $0}'
- )
+ for param in ${!opts[@]}
do
- echo "$param := ${!param}"
+ echo "$param := ${opts[$param]}"
done \
- | column -ts:
+ | column -ts: \
+ | sort
echo '----------------------------------------------'
) >&2
- mkdir -p "$dir_data"
+ screen_brightness_device_path='/sys/class/backlight'
+ screen_brightness_device_path+="/${opts['--screen_brightness_device_name']}"
+
+ # Just shorthand
+ pipe="${opts['--file_pipe']}"
+ bin="${opts['--dir_bin']}"
+
rm -f "$pipe"
mkfifo "$pipe"
cmd_sens_screen_brightness='khatus_sensor_screen_brightness'
cmd_sens_screen_brightness+=" $screen_brightness_device_path"
- cmd_sens_weather="khatus_sensor_weather $weather_station_id"
-
- cmd_sens_disk_space="khatus_sensor_disk_space $disk_space_device"
-
- cmd_sens_disk_io="khatus_sensor_disk_io $disk_io_device"
-
- cmd_sens_temperature="khatus_sensor_temperature $thermal_zone"
-
- cmd_sens_fan="khatus_sensor_fan $fan_path"
-
- # TODO: Redirect each worker's stderr to a dedicated log file
- log="$dir_data"/sensors.log
- spawn "$pipe" "$dir_bin" "$log" khatus_sensor_datetime 'in:DATE_TIME' $interval_inp_datetime
- spawn "$pipe" "$dir_bin" "$log" "$cmd_sens_screen_brightness" 'in:SCREEN_BRIGHTNESS' $interval_inp_brightness
- spawn "$pipe" "$dir_bin" "$log" "$cmd_sens_weather" 'in:WEATHER' $interval_inp_weather
- spawn "$pipe" "$dir_bin" "$log" khatus_sensor_mpd_state 'in:MPD_STATE' $interval_inp_mpd_state
- spawn "$pipe" "$dir_bin" "$log" khatus_sensor_mpd_song 'in:MPD_SONG' $interval_inp_mpd_song
- spawn "$pipe" "$dir_bin" "$log" khatus_sensor_volume 'in:VOLUME' $interval_inp_volume
- spawn "$pipe" "$dir_bin" "$log" khatus_sensor_bluetooth_power 'in:BLUETOOTH_POWER' $interval_inp_bluetooth
- spawn "$pipe" "$dir_bin" "$log" khatus_sensor_net_wifi_status 'in:NET_WIFI_STATUS' $interval_inp_net_wifi
- spawn "$pipe" "$dir_bin" "$log" khatus_sensor_net_addr_io 'in:NET_ADDR_IO' $interval_inp_net_io
- spawn "$pipe" "$dir_bin" "$log" "$cmd_sens_disk_space" 'in:DISK_SPACE' $interval_inp_disk_space
- spawn "$pipe" "$dir_bin" "$log" "$cmd_sens_disk_io" 'in:DISK_IO' $interval_inp_disk_io
- spawn "$pipe" "$dir_bin" "$log" khatus_sensor_loadavg 'in:LOAD_AVG' $interval_inp_loadavg
- spawn "$pipe" "$dir_bin" "$log" "$cmd_sens_temperature" 'in:TEMPERATURE' $interval_inp_temp
- spawn "$pipe" "$dir_bin" "$log" "$cmd_sens_fan" 'in:FAN' $interval_inp_fan
- spawn "$pipe" "$dir_bin" "$log" khatus_sensor_memory 'in:MEMORY' $interval_inp_mem
- spawn "$pipe" "$dir_bin" "$log" khatus_sensor_energy 'in:ENERGY' $interval_inp_energy
+ cmd_sens_weather="khatus_sensor_weather ${opts['--weather_station_id']}"
+ cmd_sens_disk_space="khatus_sensor_disk_space ${opts['--disk_space_device']}"
+ cmd_sens_disk_io="khatus_sensor_disk_io ${opts['--disk_io_device']}"
+ cmd_sens_temperature="khatus_sensor_temperature ${opts['--thermal_zone']}"
+ cmd_sens_fan="khatus_sensor_fan ${opts['--fan_path']}"
+ cmd_sens_bluetooth="khatus_sensor_bluetooth_power $bin"
+ cmd_sens_mpd_state="khatus_sensor_mpd_state $bin"
+ cmd_sens_net_addr_io="khatus_sensor_net_addr_io $bin"
+ cmd_sens_volume="khatus_sensor_volume $bin"
+ cmd_sens_wifi="khatus_sensor_net_wifi_status $bin ${opts['--wifi_interface']}"
+
+ fork_watcher "$pipe" "$bin" "khatus_sensor_energy $bin" 'in:ENERGY'
+ fork_poller "${opts['--interval_inp_datetime']}" "$pipe" "$bin" khatus_sensor_datetime 'in:DATE_TIME'
+ fork_poller "${opts['--interval_inp_brightness']}" "$pipe" "$bin" "$cmd_sens_screen_brightness" 'in:SCREEN_BRIGHTNESS'
+ fork_poller "${opts['--interval_inp_weather']}" "$pipe" "$bin" "$cmd_sens_weather" 'in:WEATHER'
+ fork_poller "${opts['--interval_inp_mpd_state']}" "$pipe" "$bin" "$cmd_sens_mpd_state" 'in:MPD_STATE'
+ fork_poller "${opts['--interval_inp_mpd_song']}" "$pipe" "$bin" khatus_sensor_mpd_song 'in:MPD_SONG'
+ fork_poller "${opts['--interval_inp_volume']}" "$pipe" "$bin" "$cmd_sens_volume" 'in:VOLUME'
+ fork_poller "${opts['--interval_inp_bluetooth']}" "$pipe" "$bin" "$cmd_sens_bluetooth" 'in:BLUETOOTH_POWER'
+ fork_poller "${opts['--interval_inp_net_wifi']}" "$pipe" "$bin" "$cmd_sens_wifi" 'in:NET_WIFI_STATUS'
+ fork_poller "${opts['--interval_inp_net_io']}" "$pipe" "$bin" "$cmd_sens_net_addr_io" 'in:NET_ADDR_IO'
+ fork_poller "${opts['--interval_inp_disk_space']}" "$pipe" "$bin" "$cmd_sens_disk_space" 'in:DISK_SPACE'
+ fork_poller "${opts['--interval_inp_disk_io']}" "$pipe" "$bin" "$cmd_sens_disk_io" 'in:DISK_IO'
+ fork_poller "${opts['--interval_inp_loadavg']}" "$pipe" "$bin" khatus_sensor_loadavg 'in:LOAD_AVG'
+ fork_poller "${opts['--interval_inp_temp']}" "$pipe" "$bin" "$cmd_sens_temperature" 'in:TEMPERATURE'
+ fork_poller "${opts['--interval_inp_fan']}" "$pipe" "$bin" "$cmd_sens_fan" 'in:FAN'
+ fork_poller "${opts['--interval_inp_mem']}" "$pipe" "$bin" khatus_sensor_memory 'in:MEMORY'
consume \
"$pipe" \
- "$debug" \
- "$dir_bin" \
- "$prefixes_of_net_interfaces_to_show"
+ "${opts['--debug']}" \
+ "$bin" \
+ "${opts['--prefixes_of_net_interfaces_to_show']}"
}
main $@
home / code / khatus.git / blobdiff