param(
[Parameter(mandatory = $false)]
[object]$WebHookData
)
# If runbook was called from Webhook, WebhookData will not be null.
if ($WebHookData) {

# Collect properties of WebhookData
$WebhookName = $WebHookData.WebhookName
$WebhookHeaders = $WebHookData.RequestHeader
$WebhookBody = $WebHookData.RequestBody
# debug print
Write-Output (“WebhookName Variable: ” + $WebhookName)
Write-Output (“WebhookHeaders Variable: ” + $WebhookHeaders)
Write-Output (“WebhookBody Variable: ” + $WebhookBody)

# Collect individual headers. Input converted from JSON.
$From = $WebhookHeaders.From
$Input = (ConvertFrom-Json -InputObject $WebhookBody)
}
else
{
Write-Error -Message ‘Runbook was not started from Webhook’ -ErrorAction stop
}
# Initialize
$UserPrincipalName=$Input.UserPrincipalName
$Action=$Input.Action
$AADTenantId=Get-AutomationVariable -Name ‘AADTenantId’
$SubscriptionID=Get-AutomationVariable -Name ‘SubscriptionID’
$RDBrokerURL=Get-AutomationVariable -Name ‘RDBrokerURL’
$TenantName=Get-AutomationVariable -Name ‘TenantName’
$HostpoolName=Get-AutomationVariable -Name ‘HostpoolName’

Set-ExecutionPolicy -ExecutionPolicy Undefined -Scope Process -Force -Confirm:$false
Set-ExecutionPolicy -ExecutionPolicy Unrestricted -Scope LocalMachine -Force -Confirm:$false
# Setting ErrorActionPreference to stop script execution when error occurs
$ErrorActionPreference = “Stop”
[Net.ServicePointManager]::SecurityProtocol = [Net.SecurityProtocolType]::Tls12

### MAIN ###
# Collect the credentials from Azure Automation Account Assets
$Connection = Get-AutomationConnection -Name ‘AzureRunAsConnection’

# Authenticating to Azure
Clear-AzContext -Force
$AZAuthentication = Connect-AzAccount -ApplicationId $Connection.ApplicationId -TenantId $AADTenantId -CertificateThumbprint $Connection.CertificateThumbprint -ServicePrincipal
if ($AZAuthentication -eq $null) {
Write-Output “Failed to authenticate Azure: $($_.exception.message)”
exit
} else {
$AzObj = $AZAuthentication | Out-String
Write-Output “Authenticating as service principal for Azure. Result: `n$AzObj”
}
# Set the Azure context with Subscription
$AzContext = Set-AzContext -SubscriptionId $SubscriptionID
if ($AzContext -eq $null) {
Write-Error “Please provide a valid subscription”
exit
} else {
$AzSubObj = $AzContext | Out-String
Write-Output “Sets the Azure subscription. Result: `n$AzSubObj”
}

# Authenticating to WVD
try {
$WVDAuthentication = Add-RdsAccount -DeploymentUrl $RDBrokerURL -ApplicationId $Connection.ApplicationId -CertificateThumbprint $Connection.CertificateThumbprint -AADTenantId $AadTenantId
}
catch {
Write-Output “Failed to authenticate WVD: $($_.exception.message)”
exit
}
$WVDObj = $WVDAuthentication | Out-String
Write-Output “Authenticating as service principal for WVD. Result: `n$WVDObj”

# Function to Start the assigned VM
function Start-VM
{
param(
[string]$SessionHostName
)
$vm = Get-AzVM -Name $SessionHostName
$StartRtn = $vm | Start-AzVM -ErrorAction Continue
# debug Pring
Write-Output (“vm Variable” + $vm)
Write-Output (“StartRtn Variable” + $StartRtn)

if ($StartRtn.Status -ne ‘Succeeded’)
{
# The VM failed to start, so send notice
Write-Output ($vm.Name + ” failed to start”)
Write-Error ($vm.Name + ” failed to start. Error was:”) -ErrorAction Continue
Write-Error (ConvertTo-Json $StartRtn.Error) -ErrorAction Continue
}
else
{
# The VM stopped, so send notice
Write-Output ($vm.Name + ” has been started”)
}
}

# Function to Stop the assigned VM
function Stop-VM
{
param(
[string]$SessionHostName
)
$vm = Get-AzVM -Name $SessionHostName
$StopRtn = $vm | Stop-AzVM -Force -ErrorAction Continue
# debug Pring
Write-Output (“vm Variable” + $vm)
Write-Output (“StartRtn Variable” + $StartRtn)

if ($StopRtn.Status -ne ‘Succeeded’)
{
# The VM failed to stop, so send notice
Write-Output ($VM.Name + ” failed to stop”)
Write-Error ($VM.Name + ” failed to stop. Error was:”) -ErrorAction Continue
Write-Error (ConvertTo-Json $StopRtn.Error) -ErrorAction Continue
}
else
{
# The VM stopped, so send notice
Write-Output ($VM.Name + ” has been stopped”)
}
}

# Get Assigned VM Info
$VMInfo = Get-RdsSessionHost -TenantName $TenantName -HostPoolName $HostPoolName | Where-Object{ $_.AssignedUser -eq $UserPrincipalName}
$SessionHostName = $VMInfo.SessionHostName -replace ‘^([^\.]+).*’,’$1′
# debug print
Write-Output (“VMInfo Variable: ” + $VMInfo)
Write-Output (“SessionHostName Variable: ” + $SessionHostName)
Write-Output (“Action Variable: ” + $Action)
#　Start or Stop Assigned VM
switch($Action)
{
“Start” {Start-VM -SessionHostName $SessionHostName}
“Stop” {Stop-VM -SessionHostName $SessionHostName}
}