DSCResources/MSFT_xRDSessionDeployment/MSFT_xRDSessionDeployment.psm1
|
Import-Module -Name "$PSScriptRoot\..\..\Modules\xRemoteDesktopSessionHostCommon.psm1" if (-not (Test-xRemoteDesktopSessionHostOsRequirement)) { throw 'The minimum OS requirement was not met.' } Import-Module RemoteDesktop ####################################################################### # The Get-TargetResource cmdlet. ####################################################################### function Get-TargetResource { [CmdletBinding()] [OutputType([System.Collections.Hashtable])] param ( [Parameter(Mandatory = $true)] [string[]] $SessionHost, [Parameter(Mandatory = $true)] [string] $ConnectionBroker, [Parameter(Mandatory = $true)] [string[]] $WebAccessServer ) Write-Verbose 'Getting list of RD Server roles.' # Start service RDMS is needed because otherwise a reboot loop could happen due to # the RDMS Service being on Delay-Start by default, and DSC kicks in too quickly after a reboot. if ((Get-Service -Name RDMS -ErrorAction SilentlyContinue | Select-Object -ExpandProperty Status) -ne 'Running') { try { Start-Service -Name RDMS -ErrorAction Stop } catch { Write-Warning "Failed to start RDMS service. Error: '$_'." } } $deployed = Get-RDServer -ConnectionBroker $ConnectionBroker -ErrorAction SilentlyContinue @{ SessionHost = [System.String[]] ($deployed | Where-Object Roles -Contains 'RDS-RD-SERVER' | ForEach-Object Server) ConnectionBroker = $deployed | Where-Object Roles -Contains 'RDS-CONNECTION-BROKER' | ForEach-Object Server WebAccessServer = $deployed | Where-Object Roles -Contains 'RDS-WEB-ACCESS' | ForEach-Object Server } } ######################################################################## # The Set-TargetResource cmdlet. ######################################################################## function Set-TargetResource { [Diagnostics.CodeAnalysis.SuppressMessageAttribute('PSAvoidGlobalVars', 'global:DSCMachineStatus')] [CmdletBinding()] param ( [Parameter(Mandatory = $true)] [string[]] $SessionHost, [Parameter(Mandatory = $true)] [string] $ConnectionBroker, [Parameter(Mandatory = $true)] [string[]] $WebAccessServer ) $currentStatus = Get-TargetResource @PSBoundParameters if ($null -eq $currentStatus) { Write-Verbose 'Initiating new RDSH deployment.' $parameters = @{ ConnectionBroker = $ConnectionBroker SessionHost = $SessionHost WebAccessServer = $WebAccessServer | Select-Object -First 1 } New-RDSessionDeployment @parameters $global:DSCMachineStatus = 1 return } foreach ($server in ($SessionHost | Where-Object { $_ -notin $currentStatus.SessionHost })) { Write-Verbose "Adding server '$server' to deployment." Add-RDServer -Server $server -Role 'RDS-RD-SERVER' -ConnectionBroker $ConnectionBroker } foreach ($server in ($WebAccessServer | Select-Object -Skip 1 | Where-Object { $_ -notin $currentStatus.WebAccessServer })) { Write-Verbose "Adding server '$server' to deployment." Add-RDServer -Server $server -Role 'RDS-WEB-ACCESS' -ConnectionBroker $ConnectionBroker } } ####################################################################### # The Test-TargetResource cmdlet. ####################################################################### function Test-TargetResource { [CmdletBinding()] [OutputType([System.Boolean])] param ( [Parameter(Mandatory = $true)] [string[]] $SessionHost, [Parameter(Mandatory = $true)] [string] $ConnectionBroker, [Parameter(Mandatory = $true)] [string[]] $WebAccessServer ) Write-Verbose 'Checking RDSH role is deployed on this node.' $currentStatus = Get-TargetResource @PSBoundParameters if ($currentStatus.ConnectionBroker -ne $ConnectionBroker) { Write-Verbose -Message "Found connection broker '$($currentStatus.ConnectionBroker)', expected '$ConnectionBroker'" return $false } if ($WebAccessServer.Count -gt 0 -and $null -eq $currentStatus.WebAccessServer) { Write-Verbose -Message "Desired list of Web Access Servers is empty, while $($WebAccessServer.Count) Web Access Servers should have been configured." return $false } $compare = Compare-Object -ReferenceObject $WebAccessServer -DifferenceObject $currentStatus.WebAccessServer if ($null -ne $compare) { Write-Verbose -Message "Desired list of Web Access Servers not equal`r`n$($compare | Out-String)" return $false } if ($SessionHost.Count -gt 0 -and $null -eq $currentStatus.SessionHost) { Write-Verbose -Message "Desired list of session hosts is empty, while $($SessionHost.Count) session hosts should have been configured." return $false } $compare = Compare-Object -ReferenceObject $SessionHost -DifferenceObject $currentStatus.SessionHost if ($null -ne $compare) { Write-Verbose -Message "Desired list of session hosts not equal`r`n$($compare | Out-String)" return $false } $true } Export-ModuleMember -Function *-TargetResource |