fixup! fixup! Added sleep prevention during file transfer

NearbyShare/InboundNearbyConnection.swift

@@ -329,6 +329,7 @@ class InboundNearbyConnection: NearbyConnection{
 	private func acceptTransfer(){
 		do{
 			for (id, file) in transferredFiles{
+				SleepManager.shared.disableSleep()
 				FileManager.default.createFile(atPath: file.destinationURL.path, contents: nil)
 				let handle=try FileHandle(forWritingTo: file.destinationURL)
 				transferredFiles[id]!.fileHandle=handle
@@ -350,7 +351,6 @@ class InboundNearbyConnection: NearbyConnection{
 			try sendTransferSetupFrame(frame)
 		}catch{
 			lastError=error
-            SleepManager.shared.enableSleep()
 			protocolError()
 		}
 	}

NearbyShare/NearbyConnectionManager.swift

@@ -183,7 +183,6 @@ public class NearbyConnectionManager : NSObject, NetServiceDelegate, InboundNear
 			let id=UUID().uuidString
 			let conn=InboundNearbyConnection(connection: connection, id: id)
 			self.activeConnections[id]=conn
-            SleepManager.shared.disableSleep()
 			conn.delegate=self
 			conn.start()
 		}
@@ -227,7 +226,7 @@ public class NearbyConnectionManager : NSObject, NetServiceDelegate, InboundNear
 		guard let delegate=mainAppDelegate else {return}
 		delegate.incomingTransfer(id: connection.id, didFinishWith: error)
 		activeConnections.removeValue(forKey: connection.id)
-        SleepManager.shared.enableSleep()
+		SleepManager.shared.enableSleep()
 	}
 
 	public func submitUserConsent(transferID:String, accept:Bool){
@@ -387,9 +386,9 @@ public class NearbyConnectionManager : NSObject, NetServiceDelegate, InboundNear
 		}
 		outgoingTransfers.removeValue(forKey: connection.id)
 	}
-    
-    public func getActiveConnections() -> Int {
-        return activeConnections.count
-    }
+	
+	public func getActiveConnections() -> Int {
+		return activeConnections.count
+	}
 }
 

NearbyShare/SleepManager.swift

@@ -10,30 +10,30 @@ import IOKit
 import IOKit.pwr_mgt
 
 class SleepManager{
-    public static let shared=SleepManager()
-    private var assertionID: IOPMAssertionID=0
-
-    public func disableSleep(){
-        if(assertionID != 0){
-            return
-        }
+	public static let shared=SleepManager()
+	private var assertionID: IOPMAssertionID=0
 
-        IOPMAssertionCreateWithName(
-            kIOPMAssertionTypePreventUserIdleSystemSleep as CFString,
-            IOPMAssertionLevel(kIOPMAssertionLevelOn),
-            "Data transfer over NearDrop" as CFString,
-            &assertionID
-        )
-    }
-
-    public func enableSleep(){
-        if(assertionID == 0 || NearbyConnectionManager.shared.getActiveConnections() != 0){
-            return
-        }
-
-        IOPMAssertionRelease(assertionID)
-        assertionID = 0
-    }
-
-    private init(){}
+	public func disableSleep(){
+		if(assertionID != 0){
+			return
+		}
+
+		IOPMAssertionCreateWithName(
+			kIOPMAssertionTypePreventUserIdleSystemSleep as CFString,
+			IOPMAssertionLevel(kIOPMAssertionLevelOn),
+			"Data transfer over NearDrop" as CFString,
+			&assertionID
+		)
+	}
+
+	public func enableSleep(){
+		if(assertionID == 0 || NearbyConnectionManager.shared.getActiveConnections() != 0){
+			return
+		}
+
+		IOPMAssertionRelease(assertionID)
+		assertionID = 0
+	}
+
+	private init(){}
 }