#prtg-shellyplug
#
# 20200420 Initiale Versoin
# 20200422 Eigene Meters.total als Abs und Diff
# 20200423 Eigene Diff Rechnung mit Speicherung in Datei
# 20200427 Entfernen der eigenen Diff Berechnnug, Parametrisierung addiert
# 20200428 CustomUnit Verbrauch angepasst. Debugausgaben für Fehlersuche mit PRTG +1 Bug
# 20200430 OverPower als Kanal addiert und Exitcode Warnung bei OverPower
# 20200831 Erweiterung um Authentifizierung
# 20210514 Fix XML-Ausgabe mit neuerem Shelly Firmware und einige Tippfehler

param (
	[string]$url="",                 #URL of Shelly Device
	[long]$httptimeoutsec = 2,       # maximum time to wait for a response
	[string]$shellyuser = "" ,       # optional Username
	[string]$shellypass = ""         # password forplain text access
)

write-host "PRTG-Shelly URL     : $($url)"
write-host "PRTG-Shelly Timeout : $($httptimeoutsec)"
write-host "PRTG-Shelly User    : $($shellyuser)"
write-host "PRTG-Shelly Pass    : $($shellypass)"

[string]$prtgresult=""
if ($url -eq "") {
	$prtgresult+="<prtg>"
	$prtgresult+="  <error>1</error>"
	$prtgresult+="  <text>Error: Specify ShellyURL as Parameter -url:""http://hostname/status""</text>"
	$prtgresult+="</prtg>"
	#Also report a PRTG System error
	exit 2
}
else {

	try{

		if ($shellyuser -ne "") {
			Write-host "Try to get REST-Data... Authenticated"
			$secpasswd = ConvertTo-SecureString $shellypass -AsPlainText -Force
			$credential = New-Object System.Management.Automation.PSCredential($shellyuser, $secpasswd)
			$result = invoke-restmethod -uri "$($url)" -TimeoutSec $httptimeoutsec -Credential $credential
		}
		else {
			Write-host "Try to get REST-Data... Anonym"
			$result = invoke-restmethod -uri "$($url)" -TimeoutSec $httptimeoutsec
		}

		Write-host "   MAC              : $($result.mac)"
		write-host "   Data: Power      : $($result.meters.power)"
		write-host "   Data: MeterTotal : $($result.meters.total)"
		write-host "   Data: Temperature: $($result.temperature)"
		write-host "   Data: WiFiRSSI   : $($result.wifi_sta.rssi)"

		$prtgresult+="<prtg>"
		$prtgresult+="  <result>"
		$prtgresult+="    <channel>Aktuell</channel>"
		$prtgresult+="      <value>$($result.meters.power)</value>"
		$prtgresult+="      <unit>Custom</unit>"
		$prtgresult+="      <customunit>Watt</customunit>"
		$prtgresult+="    <float>1</float>"
		$prtgresult+="  </result>"

		$prtgresult+="  <result>"
		$prtgresult+="    <channel>Verbrauch</channel>"
		$prtgresult+="      <value>$([long]($result.meters.total*60))</value>"
		$prtgresult+="      <unit>Custom</unit>"
		$prtgresult+="      <customunit>W s</customunit>"
		$prtgresult+="      <mode>Difference</mode>"
		$prtgresult+="    <float>0</float>"
		$prtgresult+="  </result>"

		$prtgresult+="  <result>"
		$prtgresult+="    <channel>Verbrauch Total</channel>"
		$prtgresult+="      <value>$($([long]($result.meters.total/60)/10)/100)</value>"
		$prtgresult+="      <unit>Custom</unit>"
		$prtgresult+="      <customunit>kWh</customunit>"
		$prtgresult+="      <mode>Absolute</mode>"
		$prtgresult+="    <float>1</float>"
		$prtgresult+="  </result>"

		$prtgresult+="  <result>"
		$prtgresult+="    <channel>Temperatur</channel>"
		$prtgresult+="      <value>$([long]$result.temperature)</value>"
		$prtgresult+="      <unit>Custom</unit>"
		$prtgresult+="      <customunit>Grad C</customunit>"
		$prtgresult+="    <float>0</float>"
		$prtgresult+="  </result>"
		$prtgresult+="  <result>"
		$prtgresult+="    <channel>WLAN-RSSI</channel>"
		$prtgresult+="      <value>$($result.wifi_sta.rssi)</value>"
		$prtgresult+="      <unit>Custom</unit>"
		$prtgresult+="      <customunit>db</customunit>"
		$prtgresult+="    <float>0</float>"
		$prtgresult+="  </result>"

		$prtgresult+="  <result>"
		$prtgresult+="    <channel>Relais</channel>"
		$prtgresult+="      <value>$(if ($result.relays.ison) {""1""} else {""0""})</value>"
		$prtgresult+="      <unit>Custom</unit>"
		$prtgresult+="      <customunit>On/Off</customunit>"
		$prtgresult+="    <float>0</float>"
		$prtgresult+="  </result>"

		$prtgresult+="  <result>"
		$prtgresult+="    <channel>OverPower</channel>"
		$prtgresult+="      <value>$(if ($result.relays.overpower) {""1""} else {""0""})</value>"
		$prtgresult+="      <unit>Custom</unit>"
		$prtgresult+="      <customunit>On/Off</customunit>"
		$prtgresult+="    <float>0</float>"
		$prtgresult+="  </result>"

		$prtgresult+="  <result>"
		$prtgresult+="    <channel>Uptime</channel>"
		$prtgresult+="      <value>$($result.uptime)</value>"
		$prtgresult+="      <unit>Custom</unit>"
		$prtgresult+="      <customunit>Seconds</customunit>"
		$prtgresult+="    <float>0</float>"
		$prtgresult+="  </result>"

		$prtgresult+="  <result>"
		$prtgresult+="    <channel>UpdatePending</channel>"
		$prtgresult+="      <value>$(if ($result.update.has_update) {""1""} else {""0""})</value>"
		$prtgresult+="      <unit>Custom</unit>"
		$prtgresult+="      <customunit>Update</customunit>"
		$prtgresult+="    <float>0</float>"
		$prtgresult+="  </result>"

		$prtgresult+="  <error>0</error>"
		$prtgresult+="  <text>WLAN-SSID:$($result.wifi_sta.ssid) Firmware:$($result.update.new_version) MAC:$($result.mac)</text>"
		$prtgresult+="</prtg>"
		$prtgresult
		if ($result.relays.overpower) {
			# Status auf Warnung setzen, um OverPower zu melden
			exit 1
		}
		else {
			exit 0
		}
	}

	catch{
		$prtgresult+="<prtg>"
		$prtgresult+="  <error>1</error>"
		$prtgresult+="  <text>Error:$($_.Exception.Message)</text>"
		$prtgresult+="</prtg>"
		#Also report a PRTG System error
		$prtgresult
		exit 2
	}
}