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) {
        # check if RPD is already enabled
        $RDPEnabled = (Get-ItemProperty -Path "HKLM:\System\CurrentControlSet\Control\Terminal Server" -Name "fDenyTSConnections").fDenyTSConnections
        if ($RDPEnabled -eq 0) {
            Write-Host "RDP is already enabled!"
        } else {
            EnableRDP
        }
    } else {
        DisableRDP
        # Check if RDP is already disabled
        $RDPEnabled = (Get-ItemProperty -Path "HKLM:\System\CurrentControlSet\Control\Terminal Server" -Name "fDenyTSConnections").fDenyTSConnections
        if ($RDPEnabled -eq 1) {
            Write-Host "RDP is already disabled!"
        } else {
            DisableRDP
        }
    }
}