<#
Optimize-WiFi-BT-WithRollback.ps1
Run in PowerShell ISE as Administrator.
- Backs up adapter advanced properties, DNS, Bluetooth service state, and power plan to JSON.
- Applies recommended settings for gaming over Wi-Fi while enabling Bluetooth coexistence.
- Writes a revert script that restores the saved state.
#>
# Require admin
If (-not ([Security.Principal.WindowsPrincipal] [Security.Principal.WindowsIdentity]::GetCurrent()).IsInRole([Security.Principal.WindowsBuiltInRole] "Administrator")) {
    Write-Error "Run PowerShell ISE as Administrator."
    Break
}
Write-Host "Starting Wi‑Fi and Bluetooth optimization with rollback generation..." -ForegroundColor Cyan
# 1) Identify active Wi-Fi adapter
$wifi = Get-NetAdapter -Physical | Where-Object { $_.Status -eq 'Up' -and ($_.InterfaceDescription -match 'Wireless|Wi-Fi|WLAN' -or $_.Name -match 'Wi-Fi|Wireless|WLAN') } | Select-Object -First 1
if (-not $wifi) {
    Write-Error "No active Wi‑Fi adapter found. Ensure Wi‑Fi is enabled and connected."
    exit 1
}
$adapterName = $wifi.Name
Write-Host "Active adapter:" $adapterName -ForegroundColor Green
# 2) Prepare backup folder and file
$backupDir = Join-Path $env:USERPROFILE "WiFiGamingBackup"
if (-not (Test-Path $backupDir)) { New-Item -Path $backupDir -ItemType Directory | Out-Null }
$timestamp = (Get-Date).ToString("yyyyMMdd_HHmmss")
$backupFile = Join-Path $backupDir "WiFiGamingBackup_$timestamp.json"
# 3) Collect current state
$advProps = Get-NetAdapterAdvancedProperty -Name $adapterName | Select-Object DisplayName, DisplayValue
$ifIndex = (Get-NetAdapter -Name $adapterName).ifIndex
$currentDns = (Get-DnsClientServerAddress -InterfaceIndex $ifIndex -ErrorAction SilentlyContinue).ServerAddresses
$btService = Get-Service -Name bthserv -ErrorAction SilentlyContinue
$btState = if ($btService) { @{ Exists = $true; Status = $btService.Status; StartType = (Get-WmiObject -Class Win32_Service -Filter "Name='bthserv'").StartMode } } else { @{ Exists = $false } }
$powerOut = powercfg -getactivescheme 2>&1
$powerGuid = if ($powerOut -match '\(([^)]+)\)') { ($powerOut -split '\s+')[3] } else { $null }
$backup = @{
    Timestamp = $timestamp
    AdapterName = $adapterName
    AdvancedProperties = $advProps
    DnsServers = $currentDns
    BluetoothService = $btState
    PowerPlanGuid = $powerGuid
}
$backup | ConvertTo-Json -Depth 6 | Out-File -FilePath $backupFile -Encoding UTF8
Write-Host "Backup written to $backupFile" -ForegroundColor Green
# 4) Helper to set adapter advanced property if present
function Set-IfAvailable {
    param([string]$Adapter, [string]$DisplayName, [string]$DisplayValue)
    $prop = Get-NetAdapterAdvancedProperty -Name $Adapter | Where-Object { $_.DisplayName -eq $DisplayName }
    if ($prop) {
        try {
            Set-NetAdapterAdvancedProperty -Name $Adapter -DisplayName $DisplayName -DisplayValue $DisplayValue -NoRestart -ErrorAction Stop
            Write-Host "Set '$DisplayName' -> '$DisplayValue'"
        } catch {
            Write-Warning "Failed to set '$DisplayName' -> '$DisplayValue': $_"
        }
    } else {
        Write-Host "Skipping missing property: $DisplayName" -ForegroundColor DarkYellow
    }
}
# 5) Apply recommended settings (tries multiple common DisplayName variants)
$settings = @(
    @{Name='Preferred Band'; Value='Prefer 5 GHz band'},
    @{Name='Preferred Band'; Value='3. Prefer 5GHz band'},
    @{Name='Band'; Value='5 GHz'},
    @{Name='Wireless Mode'; Value='802.11ax/802.11ac'},
    @{Name='Wireless Mode'; Value='802.11ax'},
    @{Name='Roaming Aggressiveness'; Value='Lowest'},
    @{Name='Roaming Aggressiveness'; Value='1 - Lowest'},
    @{Name='Power Save Mode'; Value='Disabled'},
    @{Name='Power Save Mode'; Value='Off'},
    @{Name='U-APSD Support'; Value='Disabled'},
    @{Name='Transmit Power'; Value='Highest'},
    @{Name='Transmit Power'; Value='100%'},
    @{Name='Bluetooth Collaboration'; Value='Enabled'},
    @{Name='Bluetooth Coexistence'; Value='Enabled'},
    @{Name='Bluetooth Collaboration Mode'; Value='Enabled'}
)
Write-Host "`nApplying adapter settings..." -ForegroundColor Cyan
foreach ($s in $settings) { Set-IfAvailable -Adapter $adapterName -DisplayName $s.Name -DisplayValue $s.Value }
# Channel width attempts
Set-IfAvailable -Adapter $adapterName -DisplayName 'Channel Width for 5GHz' -DisplayValue '80 MHz'
Set-IfAvailable -Adapter $adapterName -DisplayName 'Channel Width for 5GHz' -DisplayValue 'Auto'
Set-IfAvailable -Adapter $adapterName -DisplayName 'Channel Width' -DisplayValue '80 MHz'
# 6) Set DNS servers to Cloudflare and Google
try {
    Write-Host "`nSetting DNS servers to 1.1.1.1 and 8.8.8.8"
    Set-DnsClientServerAddress -InterfaceIndex $ifIndex -ServerAddresses @('1.1.1.1','8.8.8.8') -ErrorAction Stop
    Write-Host "DNS updated" -ForegroundColor Green
} catch {
    Write-Warning "Failed to set DNS: $_"
}
# 7) Flush DNS and reset network stacks
Write-Host "`nFlushing DNS and resetting network stacks..."
try {
    ipconfig /flushdns | Out-Null
    netsh winsock reset | Out-Null
    netsh int ip reset | Out-Null
    Write-Host "Network stacks reset. Reboot recommended." -ForegroundColor Green
} catch {
    Write-Warning "Network reset commands encountered an error: $_"
}
# 8) Bluetooth service handling option (default: do not stop)
$stopBluetooth = $false
if ($stopBluetooth) {
    try {
        if ($btService -and $btService.Status -ne 'Stopped') {
            Stop-Service -Name bthserv -Force -ErrorAction Stop
            Set-Service -Name bthserv -StartupType Disabled
            Write-Host "Bluetooth service stopped and disabled."
        }
    } catch {
        Write-Warning "Could not stop Bluetooth service: $_"
    }
} else {
    Write-Host "`nBluetooth service left unchanged. To auto-stop Bluetooth set `$stopBluetooth = $true` in this script and re-run." -ForegroundColor Yellow
}
# 9) Set power plan to High Performance if available
try {
    $high = powercfg -l | Select-String -Pattern "High performance|High Performance"
    if ($high) {
        $guid = ($high -split '\s+')[3]
        powercfg -setactive $guid
        Write-Host "Power plan set to High Performance."
    } else {
        Write-Host "High Performance plan not found; skipping."
    }
} catch {
    Write-Warning "Power plan change failed: $_"
}
# 10) Generate revert script that restores saved values
$revertPath = Join-Path $backupDir "Revert-WiFiGamingChanges_$timestamp.ps1"
$revertScript = @"
<#
Revert-WiFiGamingChanges_$timestamp.ps1
Restores adapter advanced properties, DNS, Bluetooth service state, and power plan from:
$backupFile
Run as Administrator.
#>
If (-not ([Security.Principal.WindowsPrincipal] [Security.Principal.WindowsIdentity]::GetCurrent()).IsInRole([Security.Principal.WindowsBuiltInRole] "Administrator")) {
    Write-Error "Run PowerShell as Administrator."
    Break
}
\$backup = \$null
try {
    \$backup = Get-Content -Path '$backupFile' -Raw | ConvertFrom-Json
} catch {
    Write-Error "Could not read backup file: $backupFile"
    exit 1
}
\$adapter = \$backup.AdapterName
# Restore advanced properties
foreach (\$p in \$backup.AdvancedProperties) {
    try {
        Set-NetAdapterAdvancedProperty -Name \$adapter -DisplayName \$p.DisplayName -DisplayValue \$p.DisplayValue -NoRestart -ErrorAction Stop
        Write-Host "Restored \$($p.DisplayName) -> \$($p.DisplayValue)"
    } catch {
        Write-Warning "Failed to restore \$($p.DisplayName): \$_"
    }
}
# Restore DNS
try {
    if (\$backup.DnsServers) {
        \$ifIndex = (Get-NetAdapter -Name \$adapter).ifIndex
        Set-DnsClientServerAddress -InterfaceIndex \$ifIndex -ServerAddresses \$backup.DnsServers -ErrorAction Stop
        Write-Host "Restored DNS servers: \$($backup.DnsServers -join ', ')"
    }
} catch {
    Write-Warning "Failed to restore DNS: \$_"
}
# Restore Bluetooth service state
if (\$backup.BluetoothService.Exists -eq \$true) {
    try {
        \$svc = Get-Service -Name bthserv -ErrorAction SilentlyContinue
        if (\$svc) {
            if (\$backup.BluetoothService.Status -ne \$svc.Status) {
                if (\$backup.BluetoothService.Status -eq 'Running') { Start-Service -Name bthserv -ErrorAction SilentlyContinue }
                if (\$backup.BluetoothService.Status -eq 'Stopped') { Stop-Service -Name bthserv -Force -ErrorAction SilentlyContinue }
            }
            \$wmi = Get-WmiObject -Class Win32_Service -Filter "Name='bthserv'"
            if (\$wmi -and \$backup.BluetoothService.StartType) {
                \$wmi.ChangeStartMode(\$backup.BluetoothService.StartType) | Out-Null
                Write-Host "Restored Bluetooth service start mode to \$($backup.BluetoothService.StartType)"
            }
        }
    } catch {
        Write-Warning "Failed to restore Bluetooth service state: \$_"
    }
}
# Restore power plan
try {
    if (\$backup.PowerPlanGuid) {
        powercfg -setactive \$backup.PowerPlanGuid
        Write-Host "Restored power plan GUID \$($backup.PowerPlanGuid)"
    }
} catch {
    Write-Warning "Failed to restore power plan: \$_"
}
Write-Host "Revert complete. Reboot recommended."
"@
$revertScript | Out-File -FilePath $revertPath -Encoding UTF8
Write-Host "`nRevert script created at: $revertPath" -ForegroundColor Green
# 11) Short ping test
Write-Host "`nRunning 20-second ping test to 8.8.8.8. Observe for spikes." -ForegroundColor Cyan
ping 8.8.8.8 -n 20
Write-Host "`nDone. Backup: $backupFile" -ForegroundColor Cyan
Write-Host "Revert script: $revertPath" -ForegroundColor Cyan
Write-Host "Reboot recommended to ensure all changes take effect." -ForegroundColor Yellow

Set-Location -Path "$env:USERPROFILE\WiFiGamingBackup"
.\Revert-WiFiGamingChanges_YYYYMMDD_HHMMSS.ps1

ping 8.8.8.8 -t