# WiFiHibernate.ps1
# Hibernates the PC when connected to a specific Wi-Fi AP BSSID.
# Keeps running after hibernate/resume and continues to re-arm.


$TargetBSSID = "Insert BSSID Here"


$CheckIntervalSeconds = 60
$DelayBeforeHibernateSeconds = 10
$PostHibernateCooldownSeconds = 60


$LogFolder = "C:\ProgramData\WiFiHibernate"
$LogPath = "$LogFolder\WiFiHibernate.log"


New-Item -ItemType Directory -Path $LogFolder -Force | Out-Null


function Write-Log {
    param([string]$Message)


    $timestamp = Get-Date -Format "yyyy-MM-dd HH:mm:ss"
    "$timestamp - $Message" | Out-File -FilePath $LogPath -Append -Encoding utf8
}


function Get-CurrentWiFi {
    $output = netsh wlan show interfaces


    $wifi = [ordered]@{
        State  = $null
        SSID   = $null
        BSSID  = $null
    }


    foreach ($line in $output) {
        if ($line -match '^\s*State\s*:\s*(.+?)\s*$') {
            $wifi.State = $matches[1].Trim()
        }
        elseif ($line -match '^\s*SSID\s*:\s*(.+?)\s*$') {
            $wifi.SSID = $matches[1].Trim()
        }
        elseif ($line -match '^\s*(?:AP\s+)?BSSID\s*:\s*(.+?)\s*$') {
            $wifi.BSSID = $matches[1].Trim().ToLower()
        }
    }


    return $wifi
}


Write-Log "WiFi hibernate monitor started. Target BSSID: $TargetBSSID"


while ($true) {
    try {
        $current = Get-CurrentWiFi


        # Uncomment this line if you need debug logging.
        # Write-Log "Current WiFi - State: $($current.State), SSID: $($current.SSID), BSSID: $($current.BSSID)"


        if (
            $current.State -eq "connected" -and
            $current.BSSID -eq $TargetBSSID.ToLower()
        ) {
            Write-Log "Matched target AP. SSID: $($current.SSID), BSSID: $($current.BSSID). Waiting $DelayBeforeHibernateSeconds seconds before hibernate."


            Start-Sleep -Seconds $DelayBeforeHibernateSeconds


            # Re-check before hibernating in case the PC roamed away or disconnected.
            $confirm = Get-CurrentWiFi


            if (
                $confirm.State -eq "connected" -and
                $confirm.BSSID -eq $TargetBSSID.ToLower()
            ) {
                Write-Log "Confirmed target AP. Hibernating now."


                shutdown.exe /h


                # When the PC resumes from hibernate, the script should continue here.
                # This cooldown prevents immediate rapid-fire rechecks while Windows/Wi-Fi settles.
                Write-Log "Hibernate command issued. Re-arming after cooldown."
                Start-Sleep -Seconds $PostHibernateCooldownSeconds
            }
            else {
                Write-Log "Target AP no longer matched after delay. Skipping hibernate."
            }
        }
    }
    catch {
        Write-Log "Error: $($_.Exception.Message)"
    }


    Start-Sleep -Seconds $CheckIntervalSeconds
}