# PRTG-O365Status
#
# CustomXML-Sensor to read all Office 365 Status messages and generate a PRTG-XML-File
# Simply Counting "open" issues per Product without 
#
# 20160515 Version 1.0  FC initial Version based on some other scripts

param (
	$Username = "",
	$password = ""
)

write-host "PRTG-O365Status:Start" 
$error.clear()
[bool]$errorfound = $false

[string]$prtgresult=""
$prtgresult+="<?xml version=""1.0"" encoding=""Windows-1252"" ?>`r`n"
$prtgresult+="<prtg>`r`n"

if ($Username -eq "") {
	$cred = get-credential -UserName $Username -Message "Office 365 Account"
}
else {
	$Securepassword = ConvertTo-SecureString -String $password -AsPlainText -Force
	$cred = new-object -typename System.Management.Automation.PSCredential -argumentlist $Username, $Securepassword
}

write-host "PRTG-O365Status: Generate JSON Request"
# JSON Anfrage formulieren
$jsonPayload = (`
   @{ UserName=$cred.Username; `
      password=$cred.GetNetworkCredential().password;} `
   | convertto-json).tostring()
# Anmeldecookie anfordern

if ($error) {
	write-host "Error: Invalid Parameter"
	$prtgresult+="   <error>2</error>`r`n"
	$prtgresult+="   <text>Error: Invalid Parameter`r`n</text>"
	$errorfound = $true
}

if (!$errorfound) {

	try {
	write-host "PRTG-O365Status: Requesting Session Cookie"
		$cookie = (invoke-restmethod `
			  -contenttype "application/json" `
			  -method Post `
			  -uri "https://api.admin.microsoftonline.com/shdtenantcommunications.svc/Register" `
			  -body $jsonPayload).RegistrationCookie

		$jsonPayload = (@{lastCookie=$cookie;locale="en-US";preferredEventTypes=@(0,1)} `
		   | convertto-json).tostring()
		}
	catch {
		write-host "Error: Unable to get cookie"
		$prtgresult+="   <error>2</error>`r`n"
		$prtgresult+="   <text>Error: Unable to get cookie`r`n</text>"
		$errorfound = $true
	}
}

if (!$errorfound) {
	try {   
		write-host "PRTG-O365Status: Retrieving Service Endpoints"
		$Serviceenpoints = (invoke-restmethod `
			  -contenttype "application/json"  `
			  -method Post  `
			  -uri "https://api.admin.microsoftonline.com/shdtenantcommunications.svc/GetServiceInformation"  `
			  -body $jsonPayload)
		}
	catch {
		write-host "Error: Unable to get ServiceEndpoints"
		$prtgresult+="   <error>2</error>`r`n"
		$prtgresult+="   <text>Error: Unable to get events`r`n</text>"
		$errorfound = $true
	}
}

if (!$errorfound) {
	try {   
		write-host "PRTG-O365Status: Retrieving Status"
		$events = (invoke-restmethod `
			  -contenttype "application/json"  `
			  -method Post  `
			  -uri "https://api.admin.microsoftonline.com/shdtenantcommunications.svc/GetEvents"  `
			  -body $jsonPayload)
		}
	catch {
		write-host "Error: Unable to get events"
		$prtgresult+="   <error>2</error>`r`n"
		$prtgresult+="   <text>Error: Unable to get events`r`n</text>"
		$errorfound = $true
	}
}

if (!$errorfound) {
	write-host "PRTG-O365Status: Parsing Status Result"
	[hashtable]$statuslist = @{}
	$totalopen = 0
	foreach ($serviceenpoint in $Serviceenpoints) {
		$productname = $serviceenpoint.ServiceName
		if (!$statuslist.containskey($productname)) {
			$statuslist[$productname] = [int]0
		}
		foreach ($event in $events.Events) {
			if (($event.AffectedServiceHealthStatus.ServiceName -eq $productname) `
			-and ($event.endtime -eq $null)){
				$statuslist[$productname]++
				$totalopen++
			}
		}
	}

	write-host "PRTG-O365Status: Generating Result"
	$prtgresult+="    <result>`r`n"
	$prtgresult+="        <channel>Total</channel>`r`n"
	$prtgresult+="        <value>"+$totalopen+"</value>`r`n"
	$prtgresult+="        <float>0</float>`r`n"
	$prtgresult+="        <mode>Absolute</mode>"
	$prtgresult+="        <CustomUnit>Events</CustomUnit>"
	$prtgresult+="        <LimitMaxWarning>1</LimitMaxWarning>"
	$prtgresult+="        <LimitMode>1</LimitMode>"
	$prtgresult+="        <Warning>1</Warning>"
	$prtgresult+="    </result>`r`n"

	foreach ($key in $statuslist.keys) {
		$prtgresult+="    <result>`r`n"
		$prtgresult+="        <channel>"+$key+"</channel>`r`n"
		$prtgresult+="        <value>"+$statuslist[$key]+"</value>`r`n"
		$prtgresult+="        <float>0</float>`r`n"
		$prtgresult+="        <mode>Absolute</mode>"
		$prtgresult+="        <CustomUnit>Events</CustomUnit>"
		$prtgresult+="        <LimitMaxWarning>1</LimitMaxWarning>"
		$prtgresult+="        <LimitMode>1</LimitMode>"
		$prtgresult+="        <Warning>1</Warning>"
		$prtgresult+="    </result>`r`n"
	}
}

$prtgresult+="</prtg>"
if ($errorfound) {
	write-host "Error Found. Ending with EXIT Code" ([xml]$prtgresult).prtg.error
}
write-host "PRTG-O365Status: Sending PRTGRESULT to STDOUT"
$prtgresult

if ($errorfound) {
	exit ([xml]$prtgresult).prtg.error
}