Public/Set-CCMSite.ps1
|
function Set-CCMSite { <# .SYNOPSIS Sets the current MEMCM Site for the MEMCM Client .DESCRIPTION This function will set the current MEMCM Site for the MEMCM Client. This is done using the Microsoft.SMS.Client COM Object. .PARAMETER SiteCode The desired MEMCM Site that will be set for the specified computers/cimsessions .PARAMETER CimSession Provides CimSessions to set the current MEMCM Site for .PARAMETER ComputerName Provides computer names to set the current MEMCM Site for .PARAMETER PSSession Provides PSSession to set the current MEMCM Site for .PARAMETER ConnectionPreference Determines if the 'Get-CCMConnection' function should check for a PSSession, or a CIMSession first when a ComputerName is passed to the function. This is ultimately going to result in the function running faster. The typical use case is when you are using the pipeline. In the pipeline scenario, the 'ComputerName' parameter is what is passed along the pipeline. The 'Get-CCMConnection' function is used to find the available connections, falling back from the preference specified in this parameter, to the the alternative (eg. you specify, PSSession, it falls back to CIMSession), and then falling back to ComputerName. Keep in mind that the 'ConnectionPreference' also determines what type of connection / command the ComputerName parameter is passed to. .EXAMPLE C:\PS> Set-CCMSite -SiteCode 'TST' Sets the local computer's MEMCM Site to TST .EXAMPLE C:\PS> Set-CCMSite -ComputerName 'Workstation1234','Workstation4321' -SiteCode 'TST' Sets the MEMCM Site for Workstation1234, and Workstation4321 to TST .NOTES FileName: Set-CCMSite.ps1 Author: Cody Mathis Contact: @CodyMathis123 Created: 2020-01-18 Updated: 2020-03-01 #> [CmdletBinding(SupportsShouldProcess = $true, DefaultParameterSetName = 'ComputerName')] param( [parameter(Mandatory = $true)] [string]$SiteCode, [Parameter(Mandatory = $false, ValueFromPipelineByPropertyName = $true, ParameterSetName = 'CimSession')] [Microsoft.Management.Infrastructure.CimSession[]]$CimSession, [Parameter(Mandatory = $false, ValueFromPipelineByPropertyName = $true, ParameterSetName = 'ComputerName')] [Alias('Connection', 'PSComputerName', 'PSConnectionName', 'IPAddress', 'ServerName', 'HostName', 'DNSHostName')] [string[]]$ComputerName = $env:ComputerName, [Parameter(Mandatory = $false, ParameterSetName = 'PSSession')] [Alias('Session')] [System.Management.Automation.Runspaces.PSSession[]]$PSSession, [Parameter(Mandatory = $false, ParameterSetName = 'ComputerName')] [ValidateSet('CimSession', 'PSSession')] [string]$ConnectionPreference ) begin { $SetAssignedSiteCodeScriptBlockString = [string]::Format('(New-Object -ComObject Microsoft.SMS.Client).SetAssignedSite("{0}", 0)', $SiteCode) $SetAssignedSiteCodeScriptBlock = [scriptblock]::Create($SetAssignedSiteCodeScriptBlockString) $invokeCommandSplat = @{ ScriptBlock = $SetAssignedSiteCodeScriptBlock } } process { foreach ($Connection in (Get-Variable -Name $PSCmdlet.ParameterSetName -ValueOnly)) { $getConnectionInfoSplat = @{ $PSCmdlet.ParameterSetName = $Connection } switch ($PSBoundParameters.ContainsKey('ConnectionPreference')) { $true { $getConnectionInfoSplat['Prefer'] = $ConnectionPreference } } $ConnectionInfo = Get-CCMConnection @getConnectionInfoSplat $Computer = $ConnectionInfo.ComputerName $connectionSplat = $ConnectionInfo.connectionSplat $Result = [ordered]@{ } $Result['ComputerName'] = $Computer if ($PSCmdlet.ShouldProcess("[ComputerName = '$Computer'] [Site = '$SiteCode']", "Set-CCMSite")) { try { switch ($Computer -eq $env:ComputerName) { $true { $SetAssignedSiteCodeScriptBlock.Invoke() $Result['SiteSet'] = $true [pscustomobject]$Result } $false { Invoke-CCMCommand @invokeCommandSplat @connectionSplat } } } catch { $Result['SiteSet'] = $false Write-Error "Failure to set MEMCM Site to $SiteCode for $Computer - $($_.Exception.Message)" } } } } } |