fixup! Added sleep prevention during file transfer

NearbyShare/NearbyConnection.swift

@@ -69,7 +69,6 @@ class NearbyConnection{
 
 	internal func handleConnectionClosure(){
 		print("Connection closed")
-        SleepManager.shared.enableSleep()
 	}
 
 	internal func protocolError(){
@@ -120,7 +119,6 @@ class NearbyConnection{
 				self.protocolError()
 				return
 			}
-            SleepManager.shared.disableSleep()
 			self.receiveFrameAsync(length: frameLength)
 		}
 	}

NearbyShare/NearbyConnectionManager.swift

@@ -183,6 +183,7 @@ 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()
 		}
@@ -226,6 +227,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()
 	}
 
 	public func submitUserConsent(transferID:String, accept:Bool){
@@ -385,5 +387,9 @@ public class NearbyConnectionManager : NSObject, NetServiceDelegate, InboundNear
 		}
 		outgoingTransfers.removeValue(forKey: connection.id)
 	}
+
+    public func getActiveConnections() -> Int {
+        return activeConnections.count
+    }
 }
 

NearbyShare/SleepManager.swift

@@ -27,7 +27,7 @@ class SleepManager{
     }
 
     public func enableSleep(){
-        if(assertionID == 0){
+        if(assertionID == 0 || NearbyConnectionManager.shared.getActiveConnections() != 0){
             return
         }