CodexPowerMode.ps1

param(
    [ValidateSet("on", "off", "status")]
    [string]$Mode = "status"
)

$ErrorActionPreference = "Stop"

function Test-IsAdmin {
    $identity = [Security.Principal.WindowsIdentity]::GetCurrent()
    $principal = [Security.Principal.WindowsPrincipal]::new($identity)
    return $principal.IsInRole([Security.Principal.WindowsBuiltInRole]::Administrator)
}

function Ensure-Admin {
    if (Test-IsAdmin) {
        return
    }

    $args = "-NoProfile -ExecutionPolicy Bypass -File `"$PSCommandPath`" -Mode $Mode"
    Write-Host "Requesting administrator permission..."
    Start-Process -FilePath "powershell.exe" -ArgumentList $args -Verb RunAs -Wait -WindowStyle Normal
    exit
}

function Set-PowerValue {
    param(
        [string]$Subgroup,
        [string]$Setting,
        [int]$Ac,
        [int]$Dc
    )

    & powercfg /setacvalueindex SCHEME_CURRENT $Subgroup $Setting $Ac | Out-Null
    & powercfg /setdcvalueindex SCHEME_CURRENT $Subgroup $Setting $Dc | Out-Null
}

function Set-Dword {
    param(
        [string]$Path,
        [string]$Name,
        [int]$Value
    )

    if (-not (Test-Path $Path)) {
        New-Item -Path $Path -Force | Out-Null
    }
    New-ItemProperty -Path $Path -Name $Name -PropertyType DWord -Value $Value -Force | Out-Null
}

function Set-StringValue {
    param(
        [string]$Path,
        [string]$Name,
        [string]$Value
    )

    if (-not (Test-Path $Path)) {
        New-Item -Path $Path -Force | Out-Null
    }
    New-ItemProperty -Path $Path -Name $Name -PropertyType String -Value $Value -Force | Out-Null
}

function Remove-ValueIfPresent {
    param(
        [string]$Path,
        [string]$Name
    )

    Remove-ItemProperty -Path $Path -Name $Name -ErrorAction SilentlyContinue
}

function Set-ServiceMode {
    param(
        [string]$Name,
        [ValidateSet("Disabled", "Automatic")]
        [string]$StartupType,
        [bool]$ShouldRun
    )

    $service = Get-Service -Name $Name -ErrorAction SilentlyContinue
    if (-not $service) {
        return
    }

    if ($ShouldRun) {
        Set-Service -Name $Name -StartupType $StartupType
        Start-Service -Name $Name -ErrorAction SilentlyContinue
    } else {
        Stop-Service -Name $Name -Force -ErrorAction SilentlyContinue
        Set-Service -Name $Name -StartupType $StartupType
    }
}

$SUB_SLEEP = "238c9fa8-0aad-41ed-83f4-97be242c8f20"
$STANDBYIDLE = "29f6c1db-86da-48c5-9fdb-f2b67b1f44da"
$HYBRIDSLEEP = "94ac6d29-73ce-41a6-809f-6363ba21b47e"
$HIBERNATEIDLE = "9d7815a6-7ee4-497e-8888-515a05f02364"
$RTCWAKE = "bd3b718a-0680-4d9d-8ab2-e1d2b4ac806d"
$UNATTENDSLEEP = "7bc4a2f9-d8fc-4469-b07b-33eb785aaca0"
$REMOTEFILESLEEP = "d4c1d4c8-d5cc-43d3-b83e-fc51215cb04d"

$SUB_VIDEO = "7516b95f-f776-4464-8c53-06167f40cc99"
$VIDEOIDLE = "3c0bc021-c8a8-4e07-a973-6b14cbcb2b7e"
$VIDEONORMALLEVEL = "aded5e82-b909-4619-9949-f5d71dac0bcb"

$SUB_DISK = "0012ee47-9041-4b5d-9b77-535fba8b1442"
$DISKIDLE = "6738e2c4-e8a5-4a42-b16a-e040e769756e"

$SUB_BUTTONS = "4f971e89-eebd-4455-a8de-9e59040e7347"
$LIDACTION = "5ca83367-6e45-459f-a27b-476b1d01c936"
$PBUTTONACTION = "7648efa3-dd9c-4e3e-b566-50f929386280"
$SBUTTONACTION = "96996bc0-ad50-47ec-923b-6f41874dd9eb"

$SUB_NONE = "fea3413e-7e05-4911-9a71-700331f1c294"
$CONSOLELOCK = "0e796bdb-100d-47d6-a2d5-f7d2daa51f51"
$DEVICEIDLE = "4faab71a-92e5-4726-b531-224559672d19"
$CONNECTIVITYINSTANDBY = "f15576e8-98b7-4186-b944-eafa664402d9"

$SUB_PCIEXPRESS = "501a4d13-42af-4429-9fd1-a8218c268e20"
$ASPM = "ee12f906-d277-404b-b6da-e5fa1a576df5"

$SUB_USB = "2a737441-1930-4402-8d77-b2bebba308a3"
$USBSELECTIVE = "48e6b7a6-50f5-4782-a5d4-53bb8f07e226"

$SUB_WIFI = "19cbb8fa-5279-450e-9fac-8a3d5fedd0c1"
$WIFI_POWER_SAVE = "12bbebe6-58d6-4636-95bb-3217ef867c1a"

$SUB_PRESENCE = "8619b916-e004-4dd8-9b66-dae86f806698"
$HUPRVIDEOIDLE = "0a7d6ab6-ac83-4ad1-8282-eca5b58308f3"
$USERPRESENCEPREDICTION = "82011705-fb95-4d46-8d35-4042b1d20def"
$HUPRVIDEODIMAWAY = "a79c8e0e-f271-482d-8f8a-5db9a18312de"
$HUPRVIDEODIM = "cf8c6097-12b8-4279-bbdd-44601ee5209d"
$HUPRVIDEOIDLEINATTENTIVE = "ee16691e-6ab3-4619-bb48-1c77c9357e5a"

$DesktopKey = "HKCU:\Control Panel\Desktop"
$DynamicLockKey = "HKCU:\Software\Microsoft\Windows NT\CurrentVersion\Winlogon"
$SystemPolicyKey = "HKLM:\SOFTWARE\Microsoft\Windows\CurrentVersion\Policies\System"
$WindowsUpdateAuKey = "HKLM:\SOFTWARE\Policies\Microsoft\Windows\WindowsUpdate\AU"
$LenovoPresenceServices = @(
    "SmartSense",
    "Lenovo Smart Communication Intelligent Sensing Service"
)

function Enable-CodexMode {
    Ensure-Admin

    # No sleep, hibernate, display-off, lid sleep, or device idle on AC or battery.
    Set-PowerValue $SUB_SLEEP $STANDBYIDLE 0 0
    Set-PowerValue $SUB_SLEEP $HIBERNATEIDLE 0 0
    Set-PowerValue $SUB_SLEEP $HYBRIDSLEEP 0 0
    Set-PowerValue $SUB_SLEEP $UNATTENDSLEEP 0 0
    Set-PowerValue $SUB_SLEEP $REMOTEFILESLEEP 0 0
    Set-PowerValue $SUB_SLEEP $RTCWAKE 0 0

    Set-PowerValue $SUB_VIDEO $VIDEOIDLE 0 0
    Set-PowerValue $SUB_VIDEO $VIDEONORMALLEVEL 15 15
    Set-PowerValue $SUB_DISK $DISKIDLE 0 0

    Set-PowerValue $SUB_BUTTONS $LIDACTION 0 0
    Set-PowerValue $SUB_BUTTONS $PBUTTONACTION 4 4
    Set-PowerValue $SUB_BUTTONS $SBUTTONACTION 0 0

    Set-PowerValue $SUB_NONE $CONSOLELOCK 0 0
    Set-PowerValue $SUB_NONE $DEVICEIDLE 0 0
    Set-PowerValue $SUB_NONE $CONNECTIVITYINSTANDBY 1 1

    Set-PowerValue $SUB_PCIEXPRESS $ASPM 0 0
    Set-PowerValue $SUB_USB $USBSELECTIVE 0 0
    Set-PowerValue $SUB_WIFI $WIFI_POWER_SAVE 0 0

    Set-PowerValue $SUB_PRESENCE $HUPRVIDEOIDLE 0 0
    Set-PowerValue $SUB_PRESENCE $USERPRESENCEPREDICTION 0 0
    Set-PowerValue $SUB_PRESENCE $HUPRVIDEODIMAWAY 0 0
    Set-PowerValue $SUB_PRESENCE $HUPRVIDEODIM 0 0
    Set-PowerValue $SUB_PRESENCE $HUPRVIDEOIDLEINATTENTIVE 0 0

    Set-Dword $DynamicLockKey "EnableGoodbye" 0
    Set-StringValue $DesktopKey "ScreenSaveActive" "0"
    Set-StringValue $DesktopKey "ScreenSaverIsSecure" "0"
    Set-StringValue $DesktopKey "ScreenSaveTimeOut" "0"
    Set-Dword $SystemPolicyKey "InactivityTimeoutSecs" 0
    Set-Dword $WindowsUpdateAuKey "NoAutoRebootWithLoggedOnUsers" 1

    foreach ($serviceName in $LenovoPresenceServices) {
        Set-ServiceMode -Name $serviceName -StartupType Disabled -ShouldRun $false
    }

    Stop-Process -Name "UserSSCtrl" -Force -ErrorAction SilentlyContinue
    & powercfg /requestsoverride PROCESS UserSSCtrl.exe DISPLAY | Out-Null
    & powercfg /setactive SCHEME_CURRENT | Out-Null

    Set-Dword $DynamicLockKey "EnableGoodbye" 0
    Set-StringValue $DesktopKey "ScreenSaveActive" "0"
    Set-StringValue $DesktopKey "ScreenSaverIsSecure" "0"
    Set-StringValue $DesktopKey "ScreenSaveTimeOut" "0"
    Set-Dword $SystemPolicyKey "InactivityTimeoutSecs" 0
    Set-Dword $WindowsUpdateAuKey "NoAutoRebootWithLoggedOnUsers" 1

    Write-Host "Codex overnight mode is ON."
    Write-Host "Warning: battery sleep is also disabled in this mode."
}

function Disable-CodexMode {
    Ensure-Admin

    # Restore the normal-ish values observed before the overnight tuning.
    Set-PowerValue $SUB_SLEEP $STANDBYIDLE 3600 1800
    Set-PowerValue $SUB_SLEEP $HIBERNATEIDLE 10800 3600
    Set-PowerValue $SUB_SLEEP $HYBRIDSLEEP 1 1
    Set-PowerValue $SUB_SLEEP $UNATTENDSLEEP 120 120
    Set-PowerValue $SUB_SLEEP $REMOTEFILESLEEP 1 1
    Set-PowerValue $SUB_SLEEP $RTCWAKE 0 0

    Set-PowerValue $SUB_VIDEO $VIDEOIDLE 900 1800
    Set-PowerValue $SUB_VIDEO $VIDEONORMALLEVEL 50 50
    Set-PowerValue $SUB_DISK $DISKIDLE 1200 600

    Set-PowerValue $SUB_BUTTONS $LIDACTION 1 1
    Set-PowerValue $SUB_BUTTONS $PBUTTONACTION 1 1
    Set-PowerValue $SUB_BUTTONS $SBUTTONACTION 1 1

    Set-PowerValue $SUB_NONE $CONSOLELOCK 1 1
    Set-PowerValue $SUB_NONE $DEVICEIDLE 0 1
    Set-PowerValue $SUB_NONE $CONNECTIVITYINSTANDBY 1 2

    Set-PowerValue $SUB_PCIEXPRESS $ASPM 2 2
    Set-PowerValue $SUB_USB $USBSELECTIVE 1 1
    Set-PowerValue $SUB_WIFI $WIFI_POWER_SAVE 0 2

    Set-PowerValue $SUB_PRESENCE $HUPRVIDEOIDLE 30 30
    Set-PowerValue $SUB_PRESENCE $USERPRESENCEPREDICTION 0 1
    Set-PowerValue $SUB_PRESENCE $HUPRVIDEODIMAWAY 0 0
    Set-PowerValue $SUB_PRESENCE $HUPRVIDEODIM 5 5
    Set-PowerValue $SUB_PRESENCE $HUPRVIDEOIDLEINATTENTIVE 0 0

    Set-Dword $DynamicLockKey "EnableGoodbye" 1
    Set-StringValue $DesktopKey "ScreenSaveActive" "1"
    Remove-ValueIfPresent $DesktopKey "ScreenSaverIsSecure"
    Remove-ValueIfPresent $DesktopKey "ScreenSaveTimeOut"
    Remove-ValueIfPresent $SystemPolicyKey "InactivityTimeoutSecs"
    Remove-ValueIfPresent $WindowsUpdateAuKey "NoAutoRebootWithLoggedOnUsers"

    foreach ($serviceName in $LenovoPresenceServices) {
        Set-ServiceMode -Name $serviceName -StartupType Automatic -ShouldRun $true
    }

    & powercfg /requestsoverride PROCESS UserSSCtrl.exe | Out-Null
    & powercfg /setactive SCHEME_CURRENT | Out-Null

    Set-Dword $DynamicLockKey "EnableGoodbye" 1
    Set-StringValue $DesktopKey "ScreenSaveActive" "1"
    Remove-ValueIfPresent $DesktopKey "ScreenSaverIsSecure"
    Remove-ValueIfPresent $DesktopKey "ScreenSaveTimeOut"
    Remove-ValueIfPresent $SystemPolicyKey "InactivityTimeoutSecs"
    Remove-ValueIfPresent $WindowsUpdateAuKey "NoAutoRebootWithLoggedOnUsers"

    Write-Host "Normal mode is restored."
    Write-Host "Hibernate is restored: AC after 3 hours, battery after 1 hour."
    Write-Host "Note: normal mode does not close third-party apps; check power requests if sleep is still blocked."
}

function Show-Status {
    Write-Host "Administrator: $(Test-IsAdmin)"
    Write-Host ""
    Write-Host "Power requests:"
    & powercfg /requests
    Write-Host ""
    Write-Host "Request overrides:"
    & powercfg /requestsoverride
    Write-Host ""
    Write-Host "Lenovo presence services:"
    Get-CimInstance Win32_Service |
        Where-Object { $_.Name -in $LenovoPresenceServices } |
        Select-Object Name, State, StartMode |
        Format-Table -AutoSize
    Write-Host ""
    Write-Host "Lock-related registry values:"
    Get-ItemProperty $DynamicLockKey -ErrorAction SilentlyContinue |
        Select-Object EnableGoodbye |
        Format-List
    Get-ItemProperty $DesktopKey -ErrorAction SilentlyContinue |
        Select-Object ScreenSaveActive, ScreenSaverIsSecure, ScreenSaveTimeOut |
        Format-List
    Get-ItemProperty $SystemPolicyKey -ErrorAction SilentlyContinue |
        Select-Object InactivityTimeoutSecs |
        Format-List
}

switch ($Mode) {
    "on" { Enable-CodexMode }
    "off" { Disable-CodexMode }
    "status" { Show-Status }
}