Add support for whatif

Author: lazywinadmin

TOOL-Set-RDPDisable/Set-RDPDisable.ps1

@@ -23,20 +23,22 @@ function Set-RDPDisable {
         lazywinadmin.com
         @lazywinadmin
 #>
-    [CmdletBinding()]
+    [CmdletBinding(SupportsShouldProcess = $true)]
     PARAM (
         [String[]]$ComputerName = $env:COMPUTERNAME
     )
     PROCESS {
         FOREACH ($Computer in $ComputerName) {
             TRY {
-                IF (Test-Connection -ComputerName $Computer -Count 1 -Quiet) {
-                    $regKey = [Microsoft.Win32.RegistryKey]::OpenRemoteBaseKey([Microsoft.Win32.RegistryHive]::LocalMachine, $Computer)
-                    $regKey = $regKey.OpenSubKey("SYSTEM\\CurrentControlSet\\Control\\Terminal Server", $True)
-                    $regkey.SetValue("fDenyTSConnections", 1)
-                    $regKey.flush()
-                    $regKey.Close()
-                } #IF Test-Connection
+                IF ($PSCmdlet.ShouldProcess($Computer, "Disable Remote Desktop")) {
+                    IF (Test-Connection -ComputerName $Computer -Count 1 -Quiet) {
+                        $regKey = [Microsoft.Win32.RegistryKey]::OpenRemoteBaseKey([Microsoft.Win32.RegistryHive]::LocalMachine, $Computer)
+                        $regKey = $regKey.OpenSubKey("SYSTEM\\CurrentControlSet\\Control\\Terminal Server", $True)
+                        $regkey.SetValue("fDenyTSConnections", 1)
+                        $regKey.flush()
+                        $regKey.Close()
+                    } #IF Test-Connection
+                }
             } #Try
             CATCH {
                 $PSCmdlet.ThrowTerminatingError($_)