disable hibernate on open torrents

server/cmd/preconfig_win.go

@@ -6,6 +6,9 @@ package main
 import (
 	"syscall"
 	"time"
+
+	"server/torr"
+	"server/torr/state"
 )
 
 const (
@@ -16,14 +19,27 @@ const (
 var pulseTime = 1 * time.Minute
 
 func Preconfig(kill bool) {
-	// don't sleep/hibernate windows
+	go func() {
-	kernel32 := syscall.NewLazyDLL("kernel32.dll")
+		// don't sleep/hibernate windows
-	setThreadExecStateProc := kernel32.NewProc("SetThreadExecutionState")
+		kernel32 := syscall.NewLazyDLL("kernel32.dll")
-	pulse := time.NewTicker(pulseTime)
+		setThreadExecStateProc := kernel32.NewProc("SetThreadExecutionState")
-	for {
+		pulse := time.NewTicker(pulseTime)
-		select {
+		for {
-		case <-pulse.C:
+			select {
-			setThreadExecStateProc.Call(uintptr(EsSystemRequired))
+			case <-pulse.C:
+				{
+					send := false
+					for i, torrent := range torr.ListTorrent() {
+						if torrent.Stat != state.TorrentInDB {
+							send = true
+							break
+						}
+					}
+					if send {
+						setThreadExecStateProc.Call(uintptr(EsSystemRequired))
+					}
+				}
+			}
 		}
-	}
+	}()
 }