test
This commit is contained in:
ai-dev
171
sambanas/rootfs/etc/s6-overlay/s6-rc.d/mqtt-disk-handler/run
Normal file
171
sambanas/rootfs/etc/s6-overlay/s6-rc.d/mqtt-disk-handler/run
Normal file
@@ -0,0 +1,171 @@
|
||||
#!/command/with-contenv bashio
|
||||
# shellcheck shell=bash
|
||||
# ==============================================================================
|
||||
# Start mqtt service for disk only
|
||||
# ==============================================================================
|
||||
|
||||
readonly MAX_TIMEFRAME=60
|
||||
readonly MIN_TIMEFRAME=10
|
||||
readonly AVG_TIMEFRAME=$(((MAX_TIMEFRAME + MIN_TIMEFRAME) / 2))
|
||||
|
||||
if bashio::config.true "mqtt_nexgen_entities"; then
|
||||
exec sleep infinity
|
||||
fi
|
||||
|
||||
if [ -f /root/.config/mosquitto_pub ]; then
|
||||
bashio::log.info "Starting the MQTT daemon for disks info..."
|
||||
|
||||
# Send autodiscovery entities
|
||||
topic=$(bashio::config 'mqtt_topic')
|
||||
if [ "$topic" = "null" ]; then topic="sambanas"; fi
|
||||
|
||||
# Send discovery messages.
|
||||
if ! bashio::config.true "autodiscovery.disable_persistent"; then prs="-r"; fi
|
||||
|
||||
jdevice=$(jq -r -c -n --arg topic "$topic" --arg smbv "$(smbd -V | sed s/Version\ //)" --arg addon "$(bashio::addon.version)" '
|
||||
{device:{
|
||||
identifiers:[],
|
||||
name: "SambaNas Physical Disk ",
|
||||
hw_version: $addon,
|
||||
sw_version: $smbv,
|
||||
model: "SambaNas",
|
||||
manufacturer: "@Dianlight",
|
||||
via_device: $topic
|
||||
}}')
|
||||
|
||||
device_scan=$(smartctl --scan-open)
|
||||
|
||||
if [ -z "$device_scan" ]; then
|
||||
bashio::log.warning "No disk with S.M.A.R.T support found. Disable sensor report"
|
||||
exec sleep infinity
|
||||
fi
|
||||
|
||||
while read -r -a device; do
|
||||
|
||||
row=$(smartctl -A -l error -l selftest "${device[0]}" -j | jq '. * (.ata_smart_attributes.table // [] | INDEX(.name)) | del(.ata_smart_attributes.table)')
|
||||
#bashio::log.info "2.2 ${row}"
|
||||
for entity in device.name device.type device.protocol power_on_time.hours power_cycle_count temperature.current \
|
||||
Raw_Read_Error_Rate.raw.value Reallocated_Sector_Ct.raw.value Wear_Leveling_Count.raw.value UDMA_CRC_Error_Count.raw.value \
|
||||
ata_smart_error_log.summary.count ata_smart_self_test_log.standard.count; do
|
||||
|
||||
if [[ -z $(jq ".$entity // empty" <<<"$row") ]]; then
|
||||
# bashio::log.info "2.2 Missing ${entity} in ${row}"
|
||||
continue
|
||||
fi
|
||||
|
||||
exmsg={}
|
||||
etype=""
|
||||
|
||||
base=$(jq --arg topic "$topic" --arg entity "$entity" '
|
||||
{
|
||||
name:($topic+" "+$entity +" "+.device.info_name),
|
||||
unique_id:((.device.name | explode | join("")) + "-" + ($entity|explode|join(""))),
|
||||
value_template:("{{ value_json." +$entity+ "}}"),
|
||||
state_topic:($topic + "/" + (.device.name | gsub("[^A-z]";"")) + "/state" ),
|
||||
oth:{
|
||||
uuid:.device.name | explode | join(""),
|
||||
label:.device.name
|
||||
}
|
||||
}' <<<"$row")
|
||||
|
||||
case "$entity" in
|
||||
device.name | device.type | device.protocol) #TEXT
|
||||
etype="sensor"
|
||||
exmsg=$(jq --arg topic "$topic" --arg entity "$entity" '
|
||||
{
|
||||
mode: "text",
|
||||
icon:"mdi:harddisk"
|
||||
}' <<<"$row")
|
||||
;;
|
||||
power_on_time.hours) #Time
|
||||
etype="sensor"
|
||||
exmsg=$(jq --arg topic "$topic" --arg entity "$entity" '
|
||||
{
|
||||
unit_of_measurement: "h",
|
||||
device_class: "duration"
|
||||
}' <<<"$row")
|
||||
;;
|
||||
power_cycle_count) #number
|
||||
etype="sensor"
|
||||
exmsg=$(jq --arg topic "$topic" --arg entity "$entity" '
|
||||
{
|
||||
_unit_of_measurement: "",
|
||||
icon:"mdi:power-cycle"
|
||||
}' <<<"$row")
|
||||
;;
|
||||
Raw_Read_Error_Rate.raw.value | Reallocated_Sector_Ct.raw.value | Wear_Leveling_Count.raw.value | UDMA_CRC_Error_Count.raw.value | ata_smart_error_log.summary.count | ata_smart_self_test_log.standard.count) #number
|
||||
etype="sensor"
|
||||
exmsg=$(jq --arg topic "$topic" --arg entity "$entity" '
|
||||
{
|
||||
_unit_of_measurement: "",
|
||||
icon:"mdi:chart-box-outline"
|
||||
}' <<<"$row")
|
||||
;;
|
||||
temperature.current) #temperature
|
||||
etype="sensor"
|
||||
exmsg=$(jq --arg topic "$topic" --arg entity "$entity" '
|
||||
{
|
||||
unit_of_measurement: "°C",
|
||||
device_class: "temperature",
|
||||
_icon:"mdi:thermometer"
|
||||
}' <<<"$row")
|
||||
;;
|
||||
*)
|
||||
bashio::log.warning "Autodiscovery for $entity missing!"
|
||||
;;
|
||||
esac
|
||||
#bashio::log.info "2.4 $base $exmsg"
|
||||
msg=$(echo "$base" "$jdevice" "$exmsg" | jq -s 'add|.device.identifiers[.device.identifiers|length]=.oth.uuid|.device.name=(.device.name + .oth.label)|del(.oth)')
|
||||
#bashio::log.debug "$msg"
|
||||
mosquitto_pub "${prs}" -t "homeassistant/${etype}/${topic}/$(jq -r '.device.name | explode | join("")' <<<$row)-${entity//[\.\/]/-}/config" -m "$msg"
|
||||
done
|
||||
|
||||
# bashio::log.info "2.3"
|
||||
|
||||
done <<<"${device_scan}"
|
||||
|
||||
while read -r -a device; do
|
||||
mkfifo /tmp/mqtt-hanlder-${device[0]//\//}
|
||||
# Send status message process
|
||||
tail -F /tmp/mqtt-hanlder-${device[0]//\//} | mosquitto_pub -l -t "${topic}/${device[0]//\//}/state" &
|
||||
done <<<"${device_scan}"
|
||||
|
||||
while read -r -a device; do
|
||||
cdevice=${device[0]//\//}
|
||||
shaOldMessage="-"
|
||||
sleepTime=$AVG_TIMEFRAME
|
||||
while true; do
|
||||
#bashio::log.info "[$cdevice] ${shaOldMessage} $sleepTime"
|
||||
status=$(smartctl -A -l error -l selftest "${device[0]}" -j | jq -c '. * (.ata_smart_attributes.table // [] | INDEX(.name)) | del(.ata_smart_attributes.table) | del(.local_time) | del(.smartctl) | del (.json_format_version)')
|
||||
# Debug
|
||||
shaMessage=$(sha1sum <<<"$status")
|
||||
# bashio::log.green "[$cdevice] SleepTimes: $sleepTime $shaOldMessage] ${status}"
|
||||
if [ "${shaOldMessage}" = "$shaMessage" ]; then
|
||||
sleepTime=$((sleepTime * 2))
|
||||
[ ${sleepTime} -gt $MAX_TIMEFRAME ] && sleepTime=$MAX_TIMEFRAME
|
||||
else
|
||||
# Send status message
|
||||
if [ $sleepTime -gt $AVG_TIMEFRAME ]; then
|
||||
sleepTime=$((sleepTime / 2))
|
||||
else
|
||||
sleepTime=$((sleepTime - MIN_TIMEFRAME))
|
||||
fi
|
||||
[ $sleepTime -le $MIN_TIMEFRAME ] && sleepTime=$MIN_TIMEFRAME
|
||||
fi
|
||||
jq -c --arg st "$sleepTime" --arg sh "$shaMessage" '. +
|
||||
{
|
||||
ref: {
|
||||
mws: $st,
|
||||
sha: $sh
|
||||
}
|
||||
}' <<<"${status}" >/tmp/mqtt-hanlder-$cdevice
|
||||
shaOldMessage=$shaMessage
|
||||
# Sleep
|
||||
sleep ${sleepTime}
|
||||
done &
|
||||
sleep $((10 / $(wc -l <<<"$device_scan")))
|
||||
done <<<"${device_scan}"
|
||||
wait
|
||||
else
|
||||
exec sleep infinity
|
||||
fi
|
||||
Reference in New Issue
Block a user