Public/Set-StatusCakeHelperTest.ps1
|
<# .Synopsis Set the configuration of a StatusCake test .EXAMPLE Set-StatusCakeHelperTest -Username "Username" -ApiKey "APIKEY" -TestID "123456" -TestName "http://www.example.com" .INPUTS baseTestURL - Base URL endpoint of the statuscake test API Username - Username associated with the API key ApiKey - APIKey to access the StatusCake API TestID - The Test ID to modify the details for <optional parameters> TestName - Name of the Test to be displayed in StatusCake TestURL - Test location, either an IP (for TCP and Ping) or a fully qualified URL for other TestTypes CheckRate - The interval in seconds between checks TestType - The type of test to create Port - The port to use on a TCP test NodeLocations - Test locations to use separated by commas. Test location servercodes are required Paused - The state of the test should be after it is created. 0 for unpaused, 1 for paused Timeout - Time in seconds before a test times out PingURL - A URL to ping if a site goes down CustomHeader - Custom HTTP header for the test, must be supplied as JSON Confirmation - Number of confirmation servers to use must be between 0 and 10 DNSServer - DNS Tests only. Hostname or IP of DNS server to use. DNSIP - DNS Tests only. IP to compare against WebsiteURL value. BasicUser - A Basic Auth User account to use to login for a HTTP check BasicPass - If BasicUser is set then this should be the password for the BasicUser for a HTTP check Public - Set 1 to enable public reporting, 0 to disable LogoImage - A URL to a image to use for public reporting UseJar - Set to 1 to enable the Cookie Jar. Required for some redirects. Branding - Set to 0 to use branding (default) or 1 to disable public reporting branding) WebsiteHost - Used internally by StatusCake Virus - Enable virus checking or not. 1 to enable FindString - A string that should either be found or not found. DoNotFind - If the above string should be found to trigger a alert. 1 = will trigger if FindString found ContactGroup - A contact group ID assoicated with account to use. RealBrowser - Use 1 to TURN OFF real browser testing TriggerRate - How many minutes to wait before sending an alert TestTags - Tags should be seperated by a comma - no spacing between tags (this,is,a set,of,tags) StatusCodes - Comma Seperated List of StatusCodes to Trigger Error on (on Update will replace, so send full list each time) EnableSSLWarning - HTTP Tests only. If enabled, tests will send warnings if the SSL certificate is about to expire. Paid users only FollowRedirect - HTTP Tests only. If enabled, our tests will follow redirects and logo the status of the final page. .FUNCTIONALITY #> function Set-StatusCakeHelperTest { [CmdletBinding(PositionalBinding=$false,SupportsShouldProcess=$true)] Param( [Parameter(ParameterSetName='SetByTestName')] [Parameter(ParameterSetName='SetByTestID')] [Parameter(ParameterSetName='SetNewTest')] $baseTestURL = "https://app.statuscake.com/API/Tests/Update", [Parameter(ParameterSetName='SetByTestName',Mandatory=$true)] [Parameter(ParameterSetName='SetByTestID',Mandatory=$true)] [Parameter(ParameterSetName='SetNewTest',Mandatory=$true)] [Parameter(Mandatory=$true)] $Username, [Parameter(ParameterSetName='SetByTestName',Mandatory=$true)] [Parameter(ParameterSetName='SetByTestID',Mandatory=$true)] [Parameter(ParameterSetName='SetNewTest',Mandatory=$true)] [Parameter(Mandatory=$true)] $ApiKey, [Parameter(ParameterSetName='SetByTestID',Mandatory=$true)] [ValidatePattern('^\d{1,}$')] $TestID, [Parameter(ParameterSetName='SetByTestName',Mandatory=$true)] [switch]$SetByTestName, [Parameter(ParameterSetName='SetByTestName',Mandatory=$true)] [Parameter(ParameterSetName='SetByTestID')] [Parameter(ParameterSetName='SetNewTest',Mandatory=$true)] [ValidateNotNullOrEmpty()] $TestName, [Parameter(ParameterSetName='SetByTestName')] [Parameter(ParameterSetName='SetByTestID')] [Parameter(ParameterSetName='SetNewTest',Mandatory=$true)] [ValidatePattern('^((http|https):\/\/)?([a-zA-Z0-9\-]+(\.[a-zA-Z]+)+.*)$|^(?!^.*,$)((([0-9]|[1-9][0-9]|1[0-9]{2}|2[0-4][0-9]|25[0-5])\.){3}([0-9]|[1-9][0-9]|1[0-9]{2}|2[0-4][0-9]|25[0-5]))*$')] $TestURL, [Parameter(ParameterSetName='SetByTestName')] [Parameter(ParameterSetName='SetByTestID')] [Parameter(ParameterSetName='SetNewTest',Mandatory=$true)] [ValidateRange(0,24000)] $CheckRate, [Parameter(ParameterSetName='SetByTestName')] [Parameter(ParameterSetName='SetByTestID')] [Parameter(ParameterSetName='SetNewTest',Mandatory=$true)] [ValidateSet("HTTP","TCP","PING","DNS")] [String]$TestType, [Parameter(ParameterSetName='SetByTestName')] [Parameter(ParameterSetName='SetByTestID')] [Parameter(ParameterSetName='SetNewTest')] [ValidatePattern('^\d{1,}$')] $ContactGroup, [Parameter(ParameterSetName='SetByTestName')] [Parameter(ParameterSetName='SetByTestID')] [Parameter(ParameterSetName='SetNewTest')] [object]$TestTags, [Parameter(ParameterSetName='SetByTestName')] [Parameter(ParameterSetName='SetByTestID')] [Parameter(ParameterSetName='SetNewTest')] [ValidatePattern('^\d{2,}$')] $Port, [Parameter(ParameterSetName='SetByTestName')] [Parameter(ParameterSetName='SetByTestID')] [Parameter(ParameterSetName='SetNewTest')] [object]$NodeLocations, [Parameter(ParameterSetName='SetByTestName')] [Parameter(ParameterSetName='SetByTestID')] [Parameter(ParameterSetName='SetNewTest')] [ValidateRange(0,1)] $Paused, [Parameter(ParameterSetName='SetByTestName')] [Parameter(ParameterSetName='SetByTestID')] [Parameter(ParameterSetName='SetNewTest')] [ValidateRange(5,100)] $Timeout, [Parameter(ParameterSetName='SetByTestName')] [Parameter(ParameterSetName='SetByTestID')] [Parameter(ParameterSetName='SetNewTest')] [ValidatePattern('^((http|https):\/\/)([a-zA-Z0-9\-]+(\.[a-zA-Z]+)+.*)$|^(?!^.*,$)')] $PingURL, [Parameter(ParameterSetName='SetByTestName')] [Parameter(ParameterSetName='SetByTestID')] [Parameter(ParameterSetName='SetNewTest')] [hashtable]$CustomHeader, [Parameter(ParameterSetName='SetByTestName')] [Parameter(ParameterSetName='SetByTestID')] [Parameter(ParameterSetName='SetNewTest')] [ValidateRange(0,10)] $Confirmation, [Parameter(ParameterSetName='SetByTestName')] [Parameter(ParameterSetName='SetByTestID')] [Parameter(ParameterSetName='SetNewTest')] [ValidatePattern('^([a-zA-Z0-9]{2,}\.[a-zA-Z]{2,})(\.[a-zA-Z]{2,})?|^(?!^.*,$)((([0-9]|[1-9][0-9]|1[0-9]{2}|2[0-4][0-9]|25[0-5])\.){3}([0-9]|[1-9][0-9]|1[0-9]{2}|2[0-4][0-9]|25[0-5]))*$')] [string]$DNSServer, [Parameter(ParameterSetName='SetByTestName')] [Parameter(ParameterSetName='SetByTestID')] [Parameter(ParameterSetName='SetNewTest')] [ValidatePattern('^(?!^.*,$)((([0-9]|[1-9][0-9]|1[0-9]{2}|2[0-4][0-9]|25[0-5])\.){3}([0-9]|[1-9][0-9]|1[0-9]{2}|2[0-4][0-9]|25[0-5]))*$')] [string]$DNSIP, [Parameter(ParameterSetName='SetByTestName')] [Parameter(ParameterSetName='SetByTestID')] [Parameter(ParameterSetName='SetNewTest')] [string]$BasicUser, [Parameter(ParameterSetName='SetByTestName')] [Parameter(ParameterSetName='SetByTestID')] [Parameter(ParameterSetName='SetNewTest')] [securestring]$BasicPass, [Parameter(ParameterSetName='SetByTestName')] [Parameter(ParameterSetName='SetByTestID')] [Parameter(ParameterSetName='SetNewTest')] [ValidateRange(0,1)] $Public, [Parameter(ParameterSetName='SetByTestName')] [Parameter(ParameterSetName='SetByTestID')] [Parameter(ParameterSetName='SetNewTest')] [ValidatePattern('^((http|https):\/\/)([a-zA-Z0-9\-]+(\.[a-zA-Z]+)+.*)$|^(?!^.*,$)')] $LogoImage, [Parameter(ParameterSetName='SetByTestName')] [Parameter(ParameterSetName='SetByTestID')] [Parameter(ParameterSetName='SetNewTest')] [ValidateRange(0,1)] $UseJar, [Parameter(ParameterSetName='SetByTestName')] [Parameter(ParameterSetName='SetByTestID')] [Parameter(ParameterSetName='SetNewTest')] [ValidateRange(0,1)] $Branding, [Parameter(ParameterSetName='SetByTestName')] [Parameter(ParameterSetName='SetByTestID')] [Parameter(ParameterSetName='SetNewTest')] [string]$WebsiteHost, [Parameter(ParameterSetName='SetByTestName')] [Parameter(ParameterSetName='SetByTestID')] [Parameter(ParameterSetName='SetNewTest')] [ValidateRange(0,1)] $Virus, [Parameter(ParameterSetName='SetByTestName')] [Parameter(ParameterSetName='SetByTestID')] [Parameter(ParameterSetName='SetNewTest')] [string]$FindString, [Parameter(ParameterSetName='SetByTestName')] [Parameter(ParameterSetName='SetByTestID')] [Parameter(ParameterSetName='SetNewTest')] [ValidateRange(0,1)] $DoNotFind, [Parameter(ParameterSetName='SetByTestName')] [Parameter(ParameterSetName='SetByTestID')] [Parameter(ParameterSetName='SetNewTest')] [ValidateRange(0,1)] $RealBrowser, [Parameter(ParameterSetName='SetByTestName')] [Parameter(ParameterSetName='SetByTestID')] [Parameter(ParameterSetName='SetNewTest')] [ValidateRange(0,60)] $TriggerRate, [Parameter(ParameterSetName='SetByTestName')] [Parameter(ParameterSetName='SetByTestID')] [Parameter(ParameterSetName='SetNewTest')] [object]$StatusCodes, [Parameter(ParameterSetName='SetByTestName')] [Parameter(ParameterSetName='SetByTestID')] [Parameter(ParameterSetName='SetNewTest')] [ValidateRange(0,1)] $EnableSSLWarning, [Parameter(ParameterSetName='SetByTestName')] [Parameter(ParameterSetName='SetByTestID')] [Parameter(ParameterSetName='SetNewTest')] [ValidateRange(0,1)] $FollowRedirect ) $authenticationHeader = @{"Username"="$username";"API"="$ApiKey"} if($SetByTestName -and $TestName) { #If setting test by name check if a test or tests with that name exists if( $pscmdlet.ShouldProcess("StatusCake API", "Retrieve StatusCake Tests")) { $testCheck = Get-StatusCakeHelperTest -Username $username -apikey $ApiKey -TestName $TestName if(!$testCheck) { $result = [PSCustomObject]@{"Success" = "False";"Message" = "No Test with Specified Name Exists";"Data" = $testCheck;"InsertID" = -1} Write-Error "$($result.Message) [$($result.Data)]" Return $null } elseif($testCheck.GetType().Name -eq 'Object[]') { $result = [PSCustomObject]@{"Success" = "False";"Message" = "Multiple Tests with the same name";"Data" = $testCheck;"InsertID" = -1} Write-Error "$($result.Message) [$($result.Data)]" Return $null } $TestID = $testCheck.TestID } } elseif($TestID) { #If setting by TestID verify that TestID already exists if( $pscmdlet.ShouldProcess("StatusCake API", "Retrieve StatusCake Tests")) { $testCheck = Get-StatusCakeHelperTest -Username $username -apikey $ApiKey -TestID $TestID if(!$testCheck) { $result = [PSCustomObject]@{"Success" = "False";"Message" = "No Test with Specified ID Exists";"Data" = $testCheck;"InsertID" = -1} Write-Error "$($result.Message) [$($result.Data)]" Return $null } $TestID = $testCheck.TestID } } else { #Setup a test with the supplied detiails if( $pscmdlet.ShouldProcess("StatusCake API", "Retrieve StatusCake Tests") ) { $testCheck = Get-StatusCakeHelperTest -Username $username -apikey $ApiKey -TestName $TestName if($testCheck) { $result = [PSCustomObject]@{"Success" = "False";"Message" = "Test with specified name already exists";"Data" = $testCheck;"InsertID" = -1} Write-Error "$($result.Message) [$($result.Data)]" Return $null } } } $convertTestURL = $false switch($TestType) { "DNS"{ If(!$DNSIP) { $result = [PSCustomObject]@{"Success" = "False";"Message" = "No DNSIP supplied for DNS check";"Data" = $testName;"InsertID" = -1} Write-Error "$($result.Message) [$($result.Data)]" Return $null } $convertTestURL = $true } "PING"{$convertTestURL = $true} "TCP"{ If(!$Port) { $result = [PSCustomObject]@{"Success" = "False";"Message" = "No Port supplied for TCP check";"Data" = $testName;"InsertID" = -1} Write-Error "$($result.Message) [$($result.Data)]" Return $null } $convertTestURL = $true } Default{} } #Certain test types require only the domain name so remove protocol if it is part of the TestURL if($convertTestURL -and $TestURL) { $TestURL = $TestURL | ConvertTo-StatusCakeHelperDomainName } if($NodeLocations) { foreach($node in $NodeLocations) { Write-Verbose "Validating node location [$node]" if(!$($node | Test-StatusCakeHelperNodeLocation)) { Write-Error "Node Location Server code invalid [$node]" Return $null } } } $psParams = @{} $ParameterList = (Get-Command -Name $MyInvocation.InvocationName).Parameters $ParamsToIgnore = @("baseTestURL","Username","ApiKey","SetByTestName") foreach ($key in $ParameterList.keys) { $var = Get-Variable -Name $key -ErrorAction SilentlyContinue; if($ParamsToIgnore -contains $var.Name) { continue } elseif($var.value -or $var.value -eq 0) { $psParams.Add($var.name,$var.value) } } $statusCakeAPIParams = $psParams | ConvertTo-StatusCakeHelperAPIParams if($statusCakeAPIParams.BasicPass) { $Credentials = New-Object System.Management.Automation.PSCredential -ArgumentList $statusCakeAPIParams.BasicUser, $statusCakeAPIParams.BasicPass $statusCakeAPIParams.BasicPass = $Credentials.GetNetworkCredential().Password } $putRequestParams = @{ uri = $baseTestURL Headers = $authenticationHeader UseBasicParsing = $true method = "Put" ContentType = "application/x-www-form-urlencoded" body = $statusCakeAPIParams } if( $pscmdlet.ShouldProcess("StatusCake API", "Set StatusCake Test") ) { $jsonResponse = Invoke-WebRequest @putRequestParams $response = $jsonResponse | ConvertFrom-Json if($response.Success -ne "True") { Write-Error "$($response.Message) [$($response.Issues)]" Return $null } Return $response } } |