Feature/prevent sleep (#6)

Prevent sleep only when there's something to download

Author: Chris Jeong

NicoDownloader.xcodeproj/project.pbxproj

@@ -10,6 +10,7 @@
 		476658D2A46286BC3170742D /* Pods_NicoDownloaderTests.framework in Frameworks */ = {isa = PBXBuildFile; fileRef = B2992436EA5924EDB8CFAADF /* Pods_NicoDownloaderTests.framework */; };
 		572B875C1E332A79008BC5AE /* ProgressViewController+Tableview.swift in Sources */ = {isa = PBXBuildFile; fileRef = 572B875B1E332A79008BC5AE /* ProgressViewController+Tableview.swift */; };
 		572B875E1E332AB3008BC5AE /* ProgressViewController+Download.swift in Sources */ = {isa = PBXBuildFile; fileRef = 572B875D1E332AB3008BC5AE /* ProgressViewController+Download.swift */; };
+		573B9A0D1E3DC755001C4F5D /* PowerManager.swift in Sources */ = {isa = PBXBuildFile; fileRef = 573B9A0C1E3DC755001C4F5D /* PowerManager.swift */; };
 		575D45CB1E3093B000F9E623 /* ProgressTableCellView.swift in Sources */ = {isa = PBXBuildFile; fileRef = 575D45CA1E3093B000F9E623 /* ProgressTableCellView.swift */; };
 		575D45D11E30A80F00F9E623 /* Item.swift in Sources */ = {isa = PBXBuildFile; fileRef = 575D45D01E30A80F00F9E623 /* Item.swift */; };
 		5784FDBB1E3075AC004BD160 /* AppDelegate.swift in Sources */ = {isa = PBXBuildFile; fileRef = 5784FDBA1E3075AC004BD160 /* AppDelegate.swift */; };
@@ -37,6 +38,7 @@
 /* Begin PBXFileReference section */
 		572B875B1E332A79008BC5AE /* ProgressViewController+Tableview.swift */ = {isa = PBXFileReference; fileEncoding = 4; lastKnownFileType = sourcecode.swift; name = "ProgressViewController+Tableview.swift"; path = "NicoDownloader/Extensions/ProgressViewController+Tableview.swift"; sourceTree = SOURCE_ROOT; };
 		572B875D1E332AB3008BC5AE /* ProgressViewController+Download.swift */ = {isa = PBXFileReference; fileEncoding = 4; lastKnownFileType = sourcecode.swift; name = "ProgressViewController+Download.swift"; path = "NicoDownloader/Extensions/ProgressViewController+Download.swift"; sourceTree = SOURCE_ROOT; };
+		573B9A0C1E3DC755001C4F5D /* PowerManager.swift */ = {isa = PBXFileReference; fileEncoding = 4; lastKnownFileType = sourcecode.swift; path = PowerManager.swift; sourceTree = "<group>"; };
 		575D45CA1E3093B000F9E623 /* ProgressTableCellView.swift */ = {isa = PBXFileReference; fileEncoding = 4; lastKnownFileType = sourcecode.swift; path = ProgressTableCellView.swift; sourceTree = "<group>"; };
 		575D45D01E30A80F00F9E623 /* Item.swift */ = {isa = PBXFileReference; fileEncoding = 4; lastKnownFileType = sourcecode.swift; name = Item.swift; path = Models/Item.swift; sourceTree = "<group>"; };
 		5784FDB71E3075AB004BD160 /* NicoDownloader.app */ = {isa = PBXFileReference; explicitFileType = wrapper.application; includeInIndex = 0; path = NicoDownloader.app; sourceTree = BUILT_PRODUCTS_DIR; };
@@ -89,6 +91,14 @@
 			name = Frameworks;
 			sourceTree = "<group>";
 		};
+		573B9A0A1E3DC731001C4F5D /* Utils */ = {
+			isa = PBXGroup;
+			children = (
+				573B9A0C1E3DC755001C4F5D /* PowerManager.swift */,
+			);
+			path = Utils;
+			sourceTree = "<group>";
+		};
 		575D45CC1E30A6A500F9E623 /* ViewControllers */ = {
 			isa = PBXGroup;
 			children = (
@@ -138,6 +148,7 @@
 		5784FDB91E3075AB004BD160 /* NicoDownloader */ = {
 			isa = PBXGroup;
 			children = (
+				573B9A0A1E3DC731001C4F5D /* Utils */,
 				57F30F721E30E6DC00BB2445 /* Extensions */,
 				575D45CF1E30A76A00F9E623 /* Models */,
 				575D45CD1E30A6F900F9E623 /* UI */,
@@ -388,6 +399,7 @@
 				572B875C1E332A79008BC5AE /* ProgressViewController+Tableview.swift in Sources */,
 				575D45D11E30A80F00F9E623 /* Item.swift in Sources */,
 				5784FDD81E307B49004BD160 /* ProgressViewController.swift in Sources */,
+				573B9A0D1E3DC755001C4F5D /* PowerManager.swift in Sources */,
 				57F30F741E30E72D00BB2445 /* Date+XMLPubDate.swift in Sources */,
 				5784FDBD1E3075AC004BD160 /* ViewController.swift in Sources */,
 				575D45CB1E3093B000F9E623 /* ProgressTableCellView.swift in Sources */,

NicoDownloader/Extensions/ProgressViewController+Download.swift

@@ -125,6 +125,7 @@ extension ProgressViewController {
                     })
                 }.then { _ -> Void in
                     self.items[idx].status = .done
+                    self.togglePreventSleep()
                     semaphore.signal()
                     DispatchQueue.main.async {
                         self.downloadProgressTableView.reloadData()
@@ -135,6 +136,8 @@ extension ProgressViewController {
                     }
                 }.catch { error in
                     self.items[idx].status = .error
+                    semaphore.signal()
+                    self.togglePreventSleep()
                     DispatchQueue.main.async {
                         self.downloadProgressTableView.reloadData()
                     }

NicoDownloader/Utils/PowerManager.swift

@@ -0,0 +1,47 @@
+//
+//  PowerManager.swift
+//  NicoDownloader
+//
+//  Created by Jeong on 2017. 1. 29..
+//  Copyright © 2017년 Jeong. All rights reserved.
+//
+
+import CoreFoundation
+import Foundation
+import IOKit.pwr_mgt
+
+let kIOPMAssertionTypeNoDisplaySleep = "PreventUserIdleDisplaySleep" as CFString
+
+class PowerManager {
+    
+    var powerAssertion: IOReturn = -100
+    var powerId: IOPMAssertionID = IOPMAssertionID(0)
+    
+    func preventSleep(time: NSInteger = 0) {
+        if powerAssertion == kIOReturnSuccess {
+            print("Sleep already prevented; releasing existing assertion first.")
+            releaseSleepAssertion()
+        }
+        
+        powerAssertion = IOPMAssertionCreateWithName(
+            kIOPMAssertionTypeNoDisplaySleep,
+            IOPMAssertionLevel(kIOPMAssertionLevelOn),
+            "Keep screen on for set time" as CFString!,
+            &powerId
+        )
+        
+        if powerAssertion == kIOReturnSuccess {
+            if time != 0 {
+                print("Disable screen sleep for %i minute(s)", time)
+            } else {
+                print("Disable screen sleep indefinitely")
+            }
+        }
+    }
+    
+    func releaseSleepAssertion() {
+        print("Enable display sleep")
+        IOPMAssertionRelease(powerId)
+    }
+    
+}

NicoDownloader/ViewControllers/ProgressViewController.swift

@@ -33,11 +33,12 @@ class ProgressViewController: NSViewController {
     var semaphore: DispatchSemaphore?
     var allDone: Bool {
         get {
-            return self.items.reduce(true) { $0.0 && ($0.1.status == .done) }
+            return self.items.reduce(true) { $0.0 && ($0.1.status == .done || $0.1.status == .error) }
         }
     }
 
     let cookies = HTTPCookieStorage.shared
+    let powerManager = PowerManager()
 
     override func viewDidLoad() {
         super.viewDidLoad()
@@ -127,3 +128,19 @@ extension ProgressViewController: NSWindowDelegate {
         dismissViewController(self)
     }
 }
+
+extension ProgressViewController {
+    override func viewWillAppear() {
+        super.viewWillAppear()
+        powerManager.preventSleep()
+    }
+    
+    override func viewWillDisappear() {
+        super.viewWillDisappear()
+        powerManager.releaseSleepAssertion()
+    }
+    
+    func togglePreventSleep() {
+        !allDone ? powerManager.preventSleep() : powerManager.releaseSleepAssertion()
+    }
+}