source-code

source-code/ALTs/AltStore/Operations/Operation.swift

@@ -0,0 +1,93 @@
+//
+//  Operation.swift
+//  AltStore
+//
+//  Created by Riley Testut on 6/7/19.
+//  Copyright © 2019 Riley Testut. All rights reserved.
+//
+
+import Foundation
+import Roxas
+
+class ResultOperation<ResultType>: Operation
+{
+    var resultHandler: ((Result<ResultType, Error>) -> Void)?
+    
+    @available(*, unavailable)
+    override func finish()
+    {
+        super.finish()
+    }
+
+    func finish(_ result: Result<ResultType, Error>)
+    {
+        guard !self.isFinished else { return }
+        
+        if self.isCancelled
+        {
+            self.resultHandler?(.failure(OperationError.cancelled))
+        }
+        else
+        {
+            self.resultHandler?(result)
+        }
+        
+        super.finish()
+    }
+}
+
+class Operation: RSTOperation, ProgressReporting
+{
+    let progress = Progress.discreteProgress(totalUnitCount: 1)
+    
+    private var backgroundTaskID: UIBackgroundTaskIdentifier?
+    
+    override var isAsynchronous: Bool {
+        return true
+    }
+    
+    override init()
+    {
+        super.init()
+        
+        self.progress.cancellationHandler = { [weak self] in self?.cancel() }
+    }
+    
+    override func cancel()
+    {
+        super.cancel()
+        
+        if !self.progress.isCancelled
+        {
+            self.progress.cancel()
+        }
+    }
+    
+    override func main()
+    {
+        super.main()
+        
+        let name = "com.altstore." + NSStringFromClass(type(of: self))
+        self.backgroundTaskID = UIApplication.shared.beginBackgroundTask(withName: name) { [weak self] in
+            guard let backgroundTask = self?.backgroundTaskID else { return }
+            
+            self?.cancel()
+            
+            UIApplication.shared.endBackgroundTask(backgroundTask)
+            self?.backgroundTaskID = .invalid
+        }        
+    }
+    
+    override func finish()
+    {
+        guard !self.isFinished else { return }
+        
+        super.finish()
+        
+        if let backgroundTaskID = self.backgroundTaskID
+        {
+            UIApplication.shared.endBackgroundTask(backgroundTaskID)
+            self.backgroundTaskID = .invalid
+        }
+    }
+}