From 4e1935ef194f79b600d58d98f6f6497818bafacc Mon Sep 17 00:00:00 2001
From: yourok <8yourok8@mail.ru>
Date: Wed, 26 Apr 2023 19:01:55 +0300
Subject: [PATCH] fix sleep win

---
 server/cmd/preconfig_win.go | 25 +++++++++++++++++++------
 1 file changed, 19 insertions(+), 6 deletions(-)

diff --git a/server/cmd/preconfig_win.go b/server/cmd/preconfig_win.go
index d36c65a..80a3c04 100644
--- a/server/cmd/preconfig_win.go
+++ b/server/cmd/preconfig_win.go
@@ -12,8 +12,9 @@ import (
 )
 
 const (
-	EsSystemRequired = 0x00000001
-	EsContinuous     = 0x80000000
+	EsSystemRequired   = 0x00000001
+	EsAwaymodeRequired = 0x00000040 // Added for future improvements
+	EsContinuous       = 0x80000000
 )
 
 var pulseTime = 1 * time.Minute
@@ -23,20 +24,32 @@ func Preconfig(kill bool) {
 		// don't sleep/hibernate windows
 		kernel32 := syscall.NewLazyDLL("kernel32.dll")
 		setThreadExecStateProc := kernel32.NewProc("SetThreadExecutionState")
+		currentExecState := uintptr(EsContinuous)
+		normalExecutionState := uintptr(EsContinuous)
+		systemRequireState := uintptr(EsSystemRequired | EsContinuous)
 		pulse := time.NewTicker(pulseTime)
 		for {
 			select {
 			case <-pulse.C:
 				{
-					send := false
+					systemRequired := false
 					for _, torrent := range torr.ListTorrent() {
 						if torrent.Stat != state.TorrentInDB {
-							send = true
+							systemRequired = true
 							break
 						}
 					}
-					if send {
-						setThreadExecStateProc.Call(uintptr(EsSystemRequired))
+					if systemRequired && currentExecState != systemRequireState {
+						// Looks like sending just EsSystemRequired to clear timer is broken in Win11.
+						// Enable system required to avoid the system to idle to sleep.
+						currentExecState = systemRequireState
+						setThreadExecStateProc.Call(systemRequireState)
+					}
+
+					if !systemRequired && currentExecState != normalExecutionState {
+						// Clear EXECUTION_STATE flags to disable away mode and allow the system to idle to sleep normally.
+						currentExecState = normalExecutionState
+						setThreadExecStateProc.Call(normalExecutionState)
 					}
 				}
 			}