Copy-SqlPolicyManagement.ps1

Function Copy-SqlPolicyManagement
{
<#
.SYNOPSIS
Migrates SQL Policy Based Management Objects, including both policies and conditions.

.DESCRIPTION
By default, all policies and conditions are copied. If an object already exist on the destination, it will be skipped unless -Force is used. 
	
The -Policies and -Conditions parameters are autopopulated for command-line completion and can be used to copy only specific objects.

THIS CODE IS PROVIDED "AS IS", WITH NO WARRANTIES.

.PARAMETER Source
Source SQL Server.You must have sysadmin access and server version must be SQL Server version 2008 or higher.

.PARAMETER Destination
Destination Sql Server. You must have sysadmin access and server version must be SQL Server version 2008 or higher.

.PARAMETER SourceSqlCredential
Allows you to login to servers using SQL Logins as opposed to Windows Auth/Integrated/Trusted. To use:

$scred = Get-Credential, then pass $scred object to the -SourceSqlCredential parameter. 

Windows Authentication will be used if DestinationSqlCredential is not specified. SQL Server does not accept Windows credentials being passed as credentials. 	
To connect as a different Windows user, run PowerShell as that user.

.PARAMETER DestinationSqlCredential
Allows you to login to servers using SQL Logins as opposed to Windows Auth/Integrated/Trusted. To use:

$dcred = Get-Credential, then pass this $dcred to the -DestinationSqlCredential parameter. 

Windows Authentication will be used if DestinationSqlCredential is not specified. SQL Server does not accept Windows credentials being passed as credentials. 	
To connect as a different Windows user, run PowerShell as that user.

.PARAMETER Force
If policies exists on destination server, it will be dropped and recreated.

.NOTES 
Author: Chrissy LeMaire (@cl), netnerds.net
Requires: sysadmin access on SQL Servers

dbatools PowerShell module (https://dbatools.io, clemaire@gmail.com)
Copyright (C) 2016 Chrissy LeMaire

This program is free software: you can redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free Software Foundation, either version 3 of the License, or (at your option) any later version.

This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more details.

You should have received a copy of the GNU General Public License along with this program.  If not, see <http://www.gnu.org/licenses/>.

.LINK
https://dbatools.io/Copy-SqlPolicyManagement 

.EXAMPLE   
Copy-SqlPolicyManagement -Source sqlserver2014a -Destination sqlcluster

Copies all policies and conditions from sqlserver2014a to sqlcluster, using Windows credentials. 

.EXAMPLE   
Copy-SqlPolicyManagement -Source sqlserver2014a -Destination sqlcluster -SourceSqlCredential $cred

Copies all policies and conditions from sqlserver2014a to sqlcluster, using SQL credentials for sqlserver2014a and Windows credentials for sqlcluster.

.EXAMPLE   
Copy-SqlPolicyManagement -Source sqlserver2014a -Destination sqlcluster -WhatIf

Shows what would happen if the command were executed.
	
.EXAMPLE   
Copy-SqlPolicyManagement -Source sqlserver2014a -Destination sqlcluster -Policy 'xp_cmdshell must be disabled'

Copies only one policy, 'xp_cmdshell must be disabled' from sqlserver2014a to sqlcluster. No conditions are migrated.
	
#>
	[CmdletBinding(DefaultParameterSetName = "Default", SupportsShouldProcess = $true)]
	param (
		[parameter(Mandatory = $true)]
		[object]$Source,
		[parameter(Mandatory = $true)]
		[object]$Destination,
		[System.Management.Automation.PSCredential]$SourceSqlCredential,
		[System.Management.Automation.PSCredential]$DestinationSqlCredential,
		[switch]$Force
	)
	
	DynamicParam { if ($source) { return (Get-ParamSqlPolicyManagement -SqlServer $Source -SqlCredential $SourceSqlCredential) } }
	
	BEGIN
	{
		if ([System.Reflection.Assembly]::LoadWithPartialName("Microsoft.SqlServer.Dmf") -eq $null)
		{
			throw "SMO version is too old. To migrate Policies, you must have SQL Server Management Studio 2008 R2 or higher installed."
		}
		
		$sourceserver = Connect-SqlServer -SqlServer $Source -SqlCredential $SourceSqlCredential
		$destserver = Connect-SqlServer -SqlServer $Destination -SqlCredential $DestinationSqlCredential
		
		$source = $sourceserver.DomainInstanceName
		$destination = $destserver.DomainInstanceName
		$policies = $psboundparameters.policies
		$conditions = $psboundparameters.conditions
		
		if ($sourceserver.versionMajor -lt 10 -or $destserver.versionMajor -lt 10)
		{
			throw "Policy Management is only supported in SQL Server 2008 and above. Quitting."
		}
		
	}
	PROCESS
	{
		
		
		$sourceSqlConn = $sourceserver.ConnectionContext.SqlConnectionObject
		$sourceSqlStoreConnection = New-Object Microsoft.SqlServer.Management.Sdk.Sfc.SqlStoreConnection $sourceSqlConn
		$sourceStore = New-Object  Microsoft.SqlServer.Management.DMF.PolicyStore $sourceSqlStoreConnection
		
		$destSqlConn = $destserver.ConnectionContext.SqlConnectionObject
		$destSqlStoreConnection = New-Object Microsoft.SqlServer.Management.Sdk.Sfc.SqlStoreConnection $destSqlConn
		$destStore = New-Object  Microsoft.SqlServer.Management.DMF.PolicyStore $destSqlStoreConnection
		
		$storepolicies = $sourceStore.policies | Where-Object { $_.IsSystemObject -eq $false }
		$storeconditions = $sourceStore.conditions | Where-Object { $_.IsSystemObject -eq $false }
		
		if ($policies.length -gt 0) { $storepolicies = $storepolicies | Where-Object { $policies -contains $_.Name } }
		if ($conditions.length -gt 0) { $storeconditions = $storeconditions | Where-Object { $conditions -contains $_.Name } }
		
		if ($policies.length -gt 0 -and $conditions.length -eq 0) { $storeconditions = $null }
		if ($conditions.length -gt 0 -and $policies.length -eq 0) { $storepolicies = $null }
		
		<# 
						Conditions
		#>
		
		Write-Output "Migrating conditions"
		foreach ($condition in $storeconditions)
		{
			$conditionName = $condition.name
			if ($deststore.conditions[$conditionName] -ne $null)
			{
				if ($force -eq $false)
				{
					Write-Warning "condition '$conditionName' was skipped because it already exists on $destination"
					Write-Warning "Use -Force to drop and recreate"
					continue
				}
				else
				{
					if ($Pscmdlet.ShouldProcess($destination, "Attempting to drop $conditionName"))
					{
						Write-Verbose "Condition '$conditionName' exists on $destination"
						Write-Verbose "Force specified. Dropping $conditionName."
						
						try
						{
							$dependentpolicies = $deststore.conditions[$conditionName].EnumDependentPolicies()
							foreach ($dependent in $dependentpolicies)
							{
								$dependent.Drop()
								$deststore.conditions.Refresh()
							}
							$deststore.conditions[$conditionName].Drop()
						}
						catch
						{
							Write-Exception $_
							continue
						}
					}
				}
			}
			
			if ($Pscmdlet.ShouldProcess($destination, "Migrating condition $conditionName"))
			{
				try
				{
					$sql = $condition.ScriptCreate().GetScript() | Out-String
					$sql = $sql -replace [Regex]::Escape("'$source'"), [Regex]::Escape("'$destination'")
					Write-Verbose $sql
					Write-Output "Copying condition $conditionName"
					$null = $destserver.ConnectionContext.ExecuteNonQuery($sql)
					$deststore.Conditions.Refresh()
				}
				catch
				{
					Write-Exception $_
				}
			}
		}
		
		<# 
						Policies
		#>
		
		Write-Output "Migrating policies"
		foreach ($policy in $storepolicies)
		{
			$policyName = $policy.name
			if ($deststore.policies[$policyName] -ne $null)
			{
				if ($force -eq $false)
				{
					Write-Warning "Policy '$policyName' was skipped because it already exists on $destination"
					Write-Warning "Use -Force to drop and recreate"
					continue
				}
				else
				{
					if ($Pscmdlet.ShouldProcess($destination, "Attempting to drop $policyName"))
					{
						Write-Verbose "Policy '$policyName' exists on $destination"
						Write-Verbose "Force specified. Dropping $policyName."
						
						try
						{
							$deststore.policies[$policyName].Drop()
							$deststore.policies.refresh()
						}
						catch
						{
							Write-Exception $_
							continue
						}
					}
				}
			}
			
			if ($Pscmdlet.ShouldProcess($destination, "Migrating policy $policyName"))
			{
				try
				{
					$deststore.conditions.Refresh()
					$deststore.policies.Refresh()
					$sql = $policy.ScriptCreateWithDependencies().GetScript() | Out-String
					$sql = $sql -replace [Regex]::Escape("'$source'"), [Regex]::Escape("'$destination'")
					Write-Verbose $sql
					Write-Output "Copying policy $policyName"
					$null = $destserver.ConnectionContext.ExecuteNonQuery($sql)
				}
				catch
				{
					# This is usually because of a duplicate dependent from above. Just skip for now.
					# Write-Exception $_
				}
			}
		}
	}
	
	end
	{
		$sourceserver.ConnectionContext.Disconnect()
		$destserver.ConnectionContext.Disconnect()
		If ($Pscmdlet.ShouldProcess("console", "Showing finished message")) { Write-Output "Policy Management migration finished" }
	}
}