Prevent PC from sleeping while actively compressing. Closes #110

Author: Iridium-IO

CompactGUI.Core/SharedMethods.vb

@@ -145,6 +145,14 @@ Public Module SharedMethods
         Return oneDrivePaths.Any(Function(odPath) normalizedFolderPath.StartsWith(Path.GetFullPath(odPath).TrimEnd(Path.DirectorySeparatorChar).ToLowerInvariant()))
     End Function
 
+    Public Sub PreventSleep()
+        SetThreadExecutionState(EXECUTION_STATE.ES_CONTINUOUS Or EXECUTION_STATE.ES_SYSTEM_REQUIRED Or EXECUTION_STATE.ES_DISPLAY_REQUIRED)
+    End Sub
+    Public Sub RestoreSleep()
+        SetThreadExecutionState(EXECUTION_STATE.ES_CONTINUOUS)
+    End Sub
+
+
 #Region "DLL Imports"
 
     <DllImport("kernel32.dll", SetLastError:=True)>
@@ -173,6 +181,18 @@ Public Module SharedMethods
     End Function
 
 
+    <DllImport("kernel32.dll", SetLastError:=True)>
+    Private Function SetThreadExecutionState(esFlags As EXECUTION_STATE) As EXECUTION_STATE
+    End Function
+
+    <Flags()>
+    Private Enum EXECUTION_STATE As UInteger
+        ES_AWAYMODE_REQUIRED = &H40
+        ES_CONTINUOUS = &H80000000UI
+        ES_DISPLAY_REQUIRED = &H2
+        ES_SYSTEM_REQUIRED = &H1
+    End Enum
+
 #End Region
 
 

CompactGUI/ViewModels/HomeViewModel.vb

@@ -178,6 +178,7 @@ Partial Public Class HomeViewModel
         Await Application.GetService(Of Watcher.Watcher).DisableBackgrounding()
 
         Compressing = True
+        Core.PreventSleep()
         Dim tasks As New List(Of Task)()
         Dim foldersToCompress = Folders.Where(Function(f) f.FolderActionState = ActionState.Idle).ToList
         For Each folder In foldersToCompress
@@ -209,6 +210,7 @@ Partial Public Class HomeViewModel
             AddOrUpdateFolderWatcher(folder)
         Next
 
+        Core.RestoreSleep()
 
         Await Application.GetService(Of Watcher.Watcher).EnableBackgrounding()
     End Function