function EnableRDP {
    Write-Host "Enabling RDP..."
    Set-ItemProperty -Path "HKLM:\System\CurrentControlSet\Control\Terminal Server" -Name "fDenyTSConnections" -Value 0
    Enable-NetFirewallRule -Group "@FirewallAPI.dll,-28752"
    Write-Host "RDP enabled."
}

function DisableRDP {
    Write-Host "Disabling RDP..."
    Set-ItemProperty -Path "HKLM:\System\CurrentControlSet\Control\Terminal Server" -Name "fDenyTSConnections" -Value 1
    Disable-NetFirewallRule -Group "@FirewallAPI.dll,-28752"
    Write-Host "RDP disabled."
}

function SetupRDP {
    param (
        [bool] $enable
    )
    if ($enable) {
        EnableRDP
    } else {
        DisableRDP
    }
}