SideStore/AltStore/Operations/SendAppOperation.swift

//
//  SendAppOperation.swift
//  AltStore
//
//  Created by Riley Testut on 6/7/19.
//  Copyright © 2019 Riley Testut. All rights reserved.
//

import Foundation
import Network

import AltStoreCore

@objc(SendAppOperation)
class SendAppOperation: ResultOperation<ServerConnection>
{
    let context: InstallAppOperationContext
    
    private let dispatchQueue = DispatchQueue(label: "com.altstore.SendAppOperation")
    
    private var serverConnection: ServerConnection?
    
    init(context: InstallAppOperationContext)
    {
        self.context = context
        
        super.init()
        
        self.progress.totalUnitCount = 1
    }
    
    override func main()
    {
        super.main()
        
        if let error = self.context.error
        {
            self.finish(.failure(error))
            return
        }
        
        guard let resignedApp = self.context.resignedApp, let server = self.context.server else { return self.finish(.failure(OperationError.invalidParameters)) }
        
        // self.context.resignedApp.fileURL points to the app bundle, but we want the .ipa.
        let app = AnyApp(name: resignedApp.name, bundleIdentifier: self.context.bundleIdentifier, url: resignedApp.url)
        let fileURL = InstalledApp.refreshedIPAURL(for: app)
        
        // Connect to server.
        ServerManager.shared.connect(to: server) { (result) in
            switch result
            {
            case .failure(let error): self.finish(.failure(error))
            case .success(let serverConnection):
                self.serverConnection = serverConnection
                
                // Send app to server.
                self.sendApp(at: fileURL, via: serverConnection) { (result) in
                    switch result
                    {
                    case .failure(let error): self.finish(.failure(error))
                    case .success:
                        self.progress.completedUnitCount += 1
                        self.finish(.success(serverConnection))
                    }
                }
            }
        }
    }
}

private extension SendAppOperation
{
    func sendApp(at fileURL: URL, via connection: ServerConnection, completionHandler: @escaping (Result<Void, Error>) -> Void)
    {
        do
        {
            guard let appData = try? Data(contentsOf: fileURL) else { throw OperationError.invalidApp }
            guard let udid = Bundle.main.object(forInfoDictionaryKey: Bundle.Info.deviceID) as? String else { throw OperationError.unknownUDID }
            
            var request = PrepareAppRequest(udid: udid, contentSize: appData.count)
            
            if connection.server.connectionType == .local
            {
                // Background daemons have low memory limit (~6MB as of 13.5),
                // so send just the file URL rather than the app data itself.
                request.fileURL = fileURL
            }
            
            connection.send(request) { (result) in
                switch result
                {
                case .failure(let error): completionHandler(.failure(error))
                case .success:
                    
                    if connection.server.connectionType == .local
                    {
                        // Sent file URL, so don't need to send any more.
                        completionHandler(.success(()))
                    }
                    else
                    {
                        print("Sending app data (\(appData.count) bytes)...")
                        
                        connection.send(appData, prependSize: false) { (result) in
                            switch result
                            {
                            case .failure(let error):
                                print("Failed to send app data (\(appData.count) bytes)")
                                completionHandler(.failure(error))
                                
                            case .success:
                                print("Successfully sent app data (\(appData.count) bytes)")
                                completionHandler(.success(()))
                            }
                        }
                    }
                }
            }
        }
        catch
        {
            completionHandler(.failure(error))
        }
    }
}