DSCResources/CTX_PVSJoinFarm/CTX_PVSJoinFarm.psm1


function Get-TargetResource {
    [CmdletBinding()]
    [Diagnostics.CodeAnalysis.SuppressMessageAttribute('PSUseShouldProcessForStateChangingFunctions', '')]
    [Diagnostics.CodeAnalysis.SuppressMessageAttribute('PSUseDeclaredVarsMoreThanAssignments', 'DSCMachineStatus')]
    [Diagnostics.CodeAnalysis.SuppressMessageAttribute('PSAvoidGlobalVars', 'global:DSCMachineStatus')]
    param (

        [Parameter(Mandatory)]
        [ValidateSet('Yes')]
        [System.String] $IsSingleInstance = 'Yes',

        [Parameter(Mandatory)]
        [ValidateNotNullOrEmpty()]
        [System.String] $DatabaseServer,

        [Parameter()]
        [ValidateNotNullOrEmpty()]
        [System.String] $DatabaseInstance,

        [Parameter()]
        [ValidateNotNullOrEmpty()]
        [System.String] $DatabaseName = 'ProvisioningServices',

        [Parameter(Mandatory)]
        [ValidateNotNullOrEmpty()]
        [System.String] $FarmName,

        [Parameter()]
        [ValidateNotNullOrEmpty()]
        [System.String] $SiteName = 'Site',

        [Parameter()]
        [ValidateNotNullOrEmpty()]
        [System.String] $StoreName = 'Store',

        [Parameter(Mandatory)]
        [ValidateNotNullOrEmpty()]
        [System.String] $LicenseServer,

        [Parameter()]
        [ValidateNotNullOrEmpty()]
        [System.UInt64] $LicenseServerPort = 27000,

        [Parameter()]
        [System.String] $StreamingIP = '',

        [Parameter()]
        [System.String] $ManagementIP = '',

        [Parameter()]
        [System.String[]] $StreamingIPs = '',

        [Parameter()]
        [ValidateNotNullOrEmpty()]
        [System.UInt64] $FirstStreamingPort = 6890,

        [Parameter()]
        [ValidateNotNullOrEmpty()]
        [System.UInt64] $LastStreamingPort = 6910,

        [Parameter()]
        [ValidateNotNullOrEmpty()]
        [System.UInt64] $SoapPort = 54321,

        [Parameter(Mandatory)]
        [ValidateNotNullOrEmpty()]
        [System.String] $Username,

        [Parameter(Mandatory)]
        [ValidateNotNullOrEmpty()]
        [System.Management.Automation.PSCredential]
        [System.Management.Automation.CredentialAttribute()]
        $Password,

        [Parameter()]
        [ValidateNotNullOrEmpty()]
        [System.Boolean] $PXEService = $true,

        [Parameter()]
        [ValidateNotNullOrEmpty()]
        [System.String] $BootstrapFile = 'C:\ProgramData\Citrix\Provisioning Services\Tftpboot\ARDBP32.BIN',

        [Parameter()]
        [ValidateNotNullOrEmpty()]
        [System.UInt64] $PasswordManagementInverval = 7,

        [Parameter()]
        [ValidateSet('Present', 'Absent')]
        [System.String] $Ensure = 'Present'
    )

    try {
        Write-Verbose "Loading module 'C:\Program Files\Citrix\Provisioning Services Console\Citrix.PVS.SnapIn.dll'"
        Import-Module 'C:\Program Files\Citrix\Provisioning Services Console\Citrix.PVS.SnapIn.dll' -Verbose:$false
    }
    catch {
       throw "Error loading PVS Powershell module..."
    }

    try {
        $pvsfarm = Get-PvsFarm
        $pvsStore = Get-PvsStore
        $pvsAuthGroup = Get-PvsAuthGroup | Select-Object -ExpandProperty AuthGroupName
        $pvsCollection = Get-PvsCollection | Select-Object -ExpandProperty CollectionName
        $pvsServerInfo = Get-PvsServerInfo
        $pvsConnection = Get-PvsConnection
        $pvsBootServer = Get-PvsServerBootstrap -ServerName $env:computername -Name ARDBP32.BIN | Select-Object Bootserver*_Ip
    }
    catch {
        Write-Verbose "Couldn't get PVS configuration..."

        $targetResource = @{
            DatabaseServer              = ""
            DatabaseInstance            = ""
            DatabaseName                = ""
            FarmName                    = ""
            SiteName                    = ""
            CollectionName              = ""
            FarmAdminGroupName          = ""
            StoreName                   = ""
            StorePath                   = ""
            LicenseServer               = ""
            LicenseServerPort           = ""
            StreamingIP                 = ""
            ManagementIP                = ""
            FirstStreamingPort          = 0
            LastStreamingPort           = 0
            SoapPort                    = 0
            BootstrapFile               = ""
            PasswordManagementInverval  = 0
            StreamingIPs           = ""            
            Ensure                      = "Absent"            
        }

    }

    $targetResource = @{
        DatabaseServer              = $pvsfarm.DatabaseServerName
        DatabaseInstance            = $pvsfarm.DatabaseInstanceName
        DatabaseName                = $pvsfarm.DatabaseName
        FarmName                    = $pvsfarm.FarmName
        SiteName                    = $pvsfarm.DefaultSiteName
        CollectionName              = $pvsCollection
        FarmAdminGroupName          = $pvsAuthGroup
        StoreName                   = $pvsStore.StoreName
        StorePath                   = $pvsStore.Path
        LicenseServer               = $pvsfarm.LicenseServer
        LicenseServerPort           = $pvsfarm.LicenseServerPort
        StreamingIP                 = $pvsServerInfo.ContactIp
        ManagementIP                = $pvsServerInfo.ManagementIp
        FirstStreamingPort          = $pvsServerInfo.FirstPort
        LastStreamingPort           = $pvsServerInfo.LastPort
        SoapPort                    = $pvsConnection.Port
        BootstrapFile               = Get-ItemPropertyValue HKLM:\Software\Citrix\ProvisioningServices\Admin\ Bootstrap -ErrorAction SilentlyContinue
        PasswordManagementInverval  = $pvsServerInfo.AdMaxPasswordAge
        StreamingIPs           = $pvsBootServer.Bootserver1_Ip.IPAddressToString, $pvsBootServer.Bootserver2_Ip.IPAddressToString, $pvsBootServer.Bootserver3_Ip.IPAddressToString,$pvsBootServer.Bootserver4_Ip.IPAddressToString           
        Ensure                      = "Absent"
    }

    if ($pvsfarm.FarmName -like $FarmName)
    {
        Write-Verbose "$FarmName has been found... "
        $targetResource["Ensure"] = "Present"
    }

    return $targetResource

} #end function Get-TargetResource

function Test-TargetResource {
    [CmdletBinding()]
    [OutputType([System.Boolean])]
    param (

        [Parameter(Mandatory)]
        [ValidateSet('Yes')]
        [System.String] $IsSingleInstance = 'Yes',

        [Parameter(Mandatory)]
        [ValidateNotNullOrEmpty()]
        [System.String] $DatabaseServer,

        [Parameter()]
        [ValidateNotNullOrEmpty()]
        [System.String] $DatabaseInstance,

        [Parameter()]
        [ValidateNotNullOrEmpty()]
        [System.String] $DatabaseName = 'ProvisioningServices',

        [Parameter(Mandatory)]
        [ValidateNotNullOrEmpty()]
        [System.String] $FarmName,

        [Parameter()]
        [ValidateNotNullOrEmpty()]
        [System.String] $SiteName = 'Site',

        [Parameter()]
        [ValidateNotNullOrEmpty()]
        [System.String] $StoreName = 'Store',

        [Parameter(Mandatory)]
        [ValidateNotNullOrEmpty()]
        [System.String] $LicenseServer,

        [Parameter()]
        [ValidateNotNullOrEmpty()]
        [System.UInt64] $LicenseServerPort = 27000,

        [Parameter()]
        [System.String] $StreamingIP = '',

        [Parameter()]
        [System.String] $ManagementIP = '',

        [Parameter()]
        [System.String[]] $StreamingIPs = '',

        [Parameter()]
        [ValidateNotNullOrEmpty()]
        [System.UInt64] $FirstStreamingPort = 6890,

        [Parameter()]
        [ValidateNotNullOrEmpty()]
        [System.UInt64] $LastStreamingPort = 6910,

        [Parameter()]
        [ValidateNotNullOrEmpty()]
        [System.UInt64] $SoapPort = 54321,

        [Parameter(Mandatory)]
        [ValidateNotNullOrEmpty()]
        [System.String] $Username,

        [Parameter(Mandatory)]
        [ValidateNotNullOrEmpty()]
        [System.Management.Automation.PSCredential]
        [System.Management.Automation.CredentialAttribute()]
        $Password,

        [Parameter()]
        [ValidateNotNullOrEmpty()]
        [System.Boolean] $PXEService = $true,

        [Parameter()]
        [ValidateNotNullOrEmpty()]
        [System.String] $BootstrapFile = 'C:\ProgramData\Citrix\Provisioning Services\Tftpboot\ARDBP32.BIN',

        [Parameter()]
        [ValidateNotNullOrEmpty()]
        [System.UInt64] $PasswordManagementInverval = 7,

        [Parameter()]
        [ValidateSet('Present', 'Absent')]
        [System.String] $Ensure = 'Present'
    )

    $targetResource = Get-TargetResource @PSBoundParameters;
    $inDesiredState = $true;

    Write-Verbose "Ensure: $Ensure"
    Write-Verbose "TargetResource Ensure: $targetResource.Ensure"

    if ($Ensure -ne $targetResource.Ensure) {
        Write-Verbose "Not in desired state..."
        $inDesiredState = $false;
    }

    return $inDesiredState

} #end function Test-TargetResource

function Set-TargetResource {
    [CmdletBinding()]
    [Diagnostics.CodeAnalysis.SuppressMessageAttribute('PSUseShouldProcessForStateChangingFunctions', '')]
    [Diagnostics.CodeAnalysis.SuppressMessageAttribute('PSUseDeclaredVarsMoreThanAssignments', 'DSCMachineStatus')]
    [Diagnostics.CodeAnalysis.SuppressMessageAttribute('PSAvoidGlobalVars', 'global:DSCMachineStatus')]
    param (

        [Parameter(Mandatory)]
        [ValidateSet('Yes')]
        [System.String] $IsSingleInstance = 'Yes',

        [Parameter(Mandatory)]
        [ValidateNotNullOrEmpty()]
        [System.String] $DatabaseServer,

        [Parameter()]
        [ValidateNotNullOrEmpty()]
        [System.String] $DatabaseInstance,

        [Parameter()]
        [ValidateNotNullOrEmpty()]
        [System.String] $DatabaseName = 'ProvisioningServices',

        [Parameter(Mandatory)]
        [ValidateNotNullOrEmpty()]
        [System.String] $FarmName,

        [Parameter()]
        [ValidateNotNullOrEmpty()]
        [System.String] $SiteName = 'Site',

        [Parameter()]
        [ValidateNotNullOrEmpty()]
        [System.String] $StoreName = 'Store',

        [Parameter(Mandatory)]
        [ValidateNotNullOrEmpty()]
        [System.String] $LicenseServer,

        [Parameter()]
        [ValidateNotNullOrEmpty()]
        [System.UInt64] $LicenseServerPort = 27000,

        [Parameter()]
        [System.String] $StreamingIP = '',

        [Parameter()]
        [System.String] $ManagementIP = '',

        [Parameter()]
        [System.String[]] $StreamingIPs = '',

        [Parameter()]
        [ValidateNotNullOrEmpty()]
        [System.UInt64] $FirstStreamingPort = 6890,

        [Parameter()]
        [ValidateNotNullOrEmpty()]
        [System.UInt64] $LastStreamingPort = 6910,

        [Parameter()]
        [ValidateNotNullOrEmpty()]
        [System.UInt64] $SoapPort = 54321,

        [Parameter(Mandatory)]
        [ValidateNotNullOrEmpty()]
        [System.String] $Username,

        [Parameter(Mandatory)]
        [ValidateNotNullOrEmpty()]
        [System.Management.Automation.PSCredential]
        [System.Management.Automation.CredentialAttribute()]
        $Password,

        [Parameter()]
        [ValidateNotNullOrEmpty()]
        [System.Boolean] $PXEService = $true,

        [Parameter()]
        [ValidateNotNullOrEmpty()]
        [System.String] $BootstrapFile = 'C:\ProgramData\Citrix\Provisioning Services\Tftpboot\ARDBP32.BIN',

        [Parameter()]
        [ValidateNotNullOrEmpty()]
        [System.UInt64] $PasswordManagementInverval = 7,

        [Parameter()]
        [ValidateSet('Present', 'Absent')]
        [System.String] $Ensure = 'Present'
    )

    # Get current IP address
    try {
        $AvailableNICs = gwmi Win32_NetworkAdapter -Filter "NetEnabled='True'"
        switch ($AvailableNICs.Count)
        { 
            {$_ -eq 0 } {
                Write-Verbose "No enabled network card could be found!"
                Write-Verbose "Please make sure at least one network card exists and is connected to the network."
                Exit 1
            }
            {$_ -gt 1 } {
                Write-Verbose "More than one enabled network card was found! This script cannot determine which network card to use."
                Write-Verbose "Please enter the preferred IP address in the DSC configuration."
            }
            default {
                Write-Verbose "One enabled network card was found."
            }
        }
            ForEach ($Adapter in $AvailableNICs) {
                $IPv4Address = $(gwmi Win32_NetworkAdapterConfiguration -Filter "Index = '$($Adapter.Index)'").IPAddress[0]
            }
            Write-Verbose "The IPv4 address of the main network card is $IPv4Address"

    
    } catch {
        Write-Verbose "An error occurred trying to retrieve the IPv4 address (error: $($Error[0]))"
        Exit 1
    }

<# # Make sure the Store path exists
    if (!(Test-Path $StorePath))
    {
        [ref] $null = New-Item -ItemType Directory -Force -Path $StorePath
    } #>


    # At the moment it's only possible to join an existing Site and existing Store
    $PVSConfig += "FarmConfiguration=2" + "`r`n"   # 0 = farm already configures, 1= create farm, 2 = join farm

    if ($PXEService) {
        $PVSConfig += "PXEServiceType=1" + "`r`n"
    }

    $PVSConfig += "DatabaseServer=" + $DatabaseServer + "`r`n"   # 0 = farm already configures, 1= create farm, 2 = join farm
    if ( !([string]::IsNullOrEmpty($DatabaseInstance)) ) {
        $PVSConfig += "DatabaseInstance=" + $DatabaseInstance + "`r`n"
    }
    $PVSConfig += "FarmExisting=" + $DatabaseName + "`r`n"
    $PVSConfig += "ExistingSite=" + $SiteName + "`r`n"        
    $PVSConfig += "ExistingStore=" + $StoreName + "`r`n"
    $PVSConfig += "LicenseServer=" + $LicenseServer + "`r`n"
    $PVSConfig += "LicenseServerPort=" + $LicenseServerPort + "`r`n"
    $PVSConfig += "UserName=" + $Username + "`r`n"
    $PVSConfig += "UserPass=" + $Password.GetNetworkCredential().Password + "`r`n"
    $PVSConfig += "PasswordManagementInterval=" + $PasswordManagementInverval + "`r`n"

    if ( [string]::IsNullOrEmpty($StreamingIP) ) {
        $PVSConfig += "StreamNetworkAdapterIP=" + $IPv4Address + "`r`n"
    }
    else {
        $PVSConfig += "StreamNetworkAdapterIP=" + $StreamingIP + "`r`n"
    }

    if ( [string]::IsNullOrEmpty($ManagementIP) ) {
        $PVSConfig += "ManagementNetwworkAdapterIP=" + $IPv4Address + "`r`n"
    }
    else {
        $PVSConfig += "ManagementNetwworkAdapterIP=" + $ManagementIP + "`r`n"
    }

    $PVSConfig += "IpcPortBase=" + $FirstStreamingPort + "`r`n"
    $PVSConfig += "IpcPortCount=" + ($LastStreamingPort - $FirstStreamingPort) + "`r`n"
    $PVSConfig += "SoapPort=" + $SoapPort + "`r`n"
    $PVSConfig += "BootstrapFile=" + $BootstrapFile + "`r`n"

    if ( [string]::IsNullOrEmpty($StreamingIPs) ) {
        if ( [string]::IsNullOrEmpty($StreamingIP) ) {
            $PVSConfig += "LS1=" + $IPv4Address + ",0.0.0.0,0.0.0.0,6910" + "`r`n"
        }
        else {
            $PVSConfig += "LS1=" + $StreamingIP + ",0.0.0.0,0.0.0.0,6910" + "`r`n"
        }
    }
    else {
        # ALL Streaming IPs hinzufügen
        $i = 1
        foreach ($ip in $StreamingIPs) {
            $PVSConfig += "LS" + $i + "=" + $StreamingIPs[$i - 1] + ",0.0.0.0,0.0.0.0,6910" + "`r`n"   
            ++$i
        }
    }
   

    # Create answer file
    $ConfWizardANSFile = "$env:Temp\ConfigWizardPVS.ans"

    Set-Content $ConfWizardANSFile -value ($PVSConfig) -Encoding Unicode -Force

    # Execute the ConfigWizard and either join or create the Provisioning Server farm
    $ConfigWizardEXE = "$env:ProgramFiles\Citrix\Provisioning Services\ConfigWizard.exe"
    $ConfigWizardLogFile = "$env:Temp\PVSConfigWizard.log"
    if ( Test-Path $ConfigWizardEXE )
    {
        $params = "/a:$ConfWizardANSFile /o:$ConfigWizardLogFile"
        Write-Verbose "Command: $ConfigWizardEXE $params -WindowStyle Hidden -Wait"   
        start-process -FilePath $ConfigWizardEXE -ArgumentList $params -WindowStyle Hidden -Wait  # There is no need for a try / catch statement since the ConfigWizard always exists with code 0

        # Create Store folders of all existing Stores
        Import-Module "C:\Program Files\Citrix\Provisioning Services Console\Citrix.PVS.SnapIn.dll"
        $stores = (Get-PvsStore).Path
        foreach ($StorePath in $stores)
        {
            # Make sure the Store path exists
            if (!(Test-Path $StorePath)) {
                [ref] $null = New-Item -ItemType Directory -Force -Path $StorePath
            }
        }

    }
    else
    {
        Write-Error "ConfigWizard.exe not found..."
    }

} #end function Set-TargetResource


$moduleRoot = Split-Path -Path $MyInvocation.MyCommand.Path -Parent;

## Import the PVSCommon library functions
$moduleParent = Split-Path -Path $moduleRoot -Parent;
Import-Module (Join-Path -Path $moduleParent -ChildPath 'CTX_PVSCommon');

Export-ModuleMember -Function *-TargetResource;