GBA002/Delta/Importing/ImportController.swift

//
//  ImportController.swift
//  Delta
//
//  Created by Riley Testut on 10/10/15.
//  Copyright © 2015 Riley Testut. All rights reserved.
//

import UIKit
import MobileCoreServices
import ObjectiveC

import DeltaCore

import Roxas

protocol ImportControllerDelegate
{
    func importController(_ importController: ImportController, didImportItemsAt urls: Set<URL>, errors: [Error])
    
    /** Optional **/
    func importControllerDidCancel(_ importController: ImportController)
}

extension ImportControllerDelegate
{
    func importControllerDidCancel(_ importController: ImportController)
    {
        // Empty Implementation
    }
}

class ImportController: NSObject
{
    let documentTypes: Set<String>
    
    var delegate: ImportControllerDelegate?
    var importOptions: [ImportOption]?
    
    fileprivate weak var presentingViewController: UIViewController?
    
    // Store presentedViewController separately, since when we dismiss we don't know if it has already been dismissed.
    // Calling dismiss on presentingViewController in that case would dismiss presentingViewController, which is bad.
    fileprivate weak var presentedViewController: UIViewController?
    
    fileprivate let importQueue: OperationQueue
    fileprivate let fileCoordinator: NSFileCoordinator
    
    init(documentTypes: Set<String>)
    {
        self.documentTypes = documentTypes
        
        let dispatchQueue = DispatchQueue(label: "com.rileytestut.Delta.ImportController.dispatchQueue", qos: .userInitiated, attributes: .concurrent)
        
        self.importQueue = OperationQueue()
        self.importQueue.name = "com.rileytestut.Delta.ImportController.importQueue"
        self.importQueue.underlyingQueue = dispatchQueue
        
        self.fileCoordinator = NSFileCoordinator(filePresenter: nil)
        
        super.init()
    }
    
    fileprivate func presentImportController(from presentingViewController: UIViewController, animated: Bool, completionHandler: (() -> Void)?)
    {
        self.presentingViewController = presentingViewController
        
        #if IMPACTOR
            
            let alertController = UIAlertController(title: nil, message: nil, preferredStyle: .actionSheet)
            alertController.addAction(UIAlertAction.cancel)
            
            if let importOptions = self.importOptions
            {
                for importOption in importOptions
                {
                    alertController.add(importOption) { [unowned self] (urls) in
                        self.finish(with: urls, errors: [])
                    }
                }
            }
            
            self.presentedViewController = alertController
            self.presentingViewController?.present(alertController, animated: true, completion: nil)
            
        #else
            
            let documentMenuController = UIDocumentMenuViewController(documentTypes: Array(self.documentTypes), in: .import)
            documentMenuController.delegate = self
            
            if let reversedImportOptions = self.importOptions?.reversed()
            {
                for importOption in reversedImportOptions
                {
                    documentMenuController.add(importOption, order: .first) { [unowned self] (urls) in
                        self.finish(with: urls, errors: [])
                    }
                }
            }
            
            self.presentedViewController = documentMenuController
            self.presentingViewController?.present(documentMenuController, animated: true, completion: nil)
            
        #endif
    }
    
    @objc fileprivate func cancel()
    {
        self.finish(with: nil, errors: [])
    }
    
    fileprivate func finish(with urls: Set<URL>?, errors: [Error])
    {
        if let urls = urls
        {
            self.delegate?.importController(self, didImportItemsAt: urls, errors: errors)
        }
        else
        {
            self.delegate?.importControllerDidCancel(self)
        }
        
        self.presentedViewController?.dismiss(animated: true)
        
        self.presentingViewController?.importController = nil
    }
}


extension ImportController: UIDocumentMenuDelegate
{
    func documentMenu(_ documentMenu: UIDocumentMenuViewController, didPickDocumentPicker documentPicker: UIDocumentPickerViewController)
    {
        if #available(iOS 11.0, *)
        {
            let cancelButton = UIBarButtonItem(barButtonSystemItem: .cancel, target: self, action: #selector(ImportController.cancel))
            
            let documentBrowserViewController = UIDocumentBrowserViewController(forOpeningFilesWithContentTypes: Array(self.documentTypes))
            documentBrowserViewController.delegate = self
            documentBrowserViewController.browserUserInterfaceStyle = .dark
            documentBrowserViewController.allowsPickingMultipleItems = true
            documentBrowserViewController.allowsDocumentCreation = false
            documentBrowserViewController.additionalTrailingNavigationBarButtonItems = [cancelButton]
            
            self.presentedViewController = documentBrowserViewController
            self.presentingViewController?.present(documentBrowserViewController, animated: true, completion: nil)
        }
        else
        {
            documentPicker.delegate = self
            
            self.presentedViewController = documentPicker
            self.presentingViewController?.present(documentPicker, animated: true, completion: nil)
        }
    }
    
    func documentMenuWasCancelled(_ documentMenu: UIDocumentMenuViewController)
    {
        self.finish(with: nil, errors: [])
    }
}

extension ImportController: UIDocumentPickerDelegate
{
    func documentPicker(_ controller: UIDocumentPickerViewController, didPickDocumentAt url: URL)
    {
        self.finish(with: [url], errors: [])
    }
    
    func documentPicker(_ controller: UIDocumentPickerViewController, didPickDocumentsAt urls: [URL])
    {
        self.finish(with: Set(urls), errors: [])
    }
    
    func documentPickerWasCancelled(_ controller: UIDocumentPickerViewController)
    {
        self.finish(with: nil, errors: [])
    }
}

@available(iOS 11.0, *)
extension ImportController: UIDocumentBrowserViewControllerDelegate
{
    func documentBrowser(_ controller: UIDocumentBrowserViewController, didPickDocumentURLs documentURLs: [URL])
    {
        var coordinatedURLs = Set<URL>()
        var errors = [Error]()
        
        let dispatchGroup = DispatchGroup()
        
        for url in documentURLs
        {
            dispatchGroup.enter()
            
            let intent = NSFileAccessIntent.readingIntent(with: url)
            self.fileCoordinator.coordinate(with: [intent], queue: self.importQueue) { (error) in
                if let error = error
                {
                    errors.append(error)
                }
                else
                {
                    let temporaryURL = FileManager.default.temporaryDirectory.appendingPathComponent(url.lastPathComponent)
                    
                    do
                    {
                        // Always access intent.url, as the system may have updated it when requesting access.
                        try FileManager.default.copyItem(at: intent.url, to: temporaryURL)
                        
                        coordinatedURLs.insert(temporaryURL)
                    }
                    catch
                    {
                        errors.append(error)
                    }
                }
                
                dispatchGroup.leave()
            }
        }
        
        dispatchGroup.notify(queue: self.importQueue.underlyingQueue!) {
            self.finish(with: coordinatedURLs, errors: errors)
        }
    }
}

private var ImportControllerKey: UInt8 = 0

extension UIViewController
{
    fileprivate(set) var importController: ImportController?
    {
        set
        {
            objc_setAssociatedObject(self, &ImportControllerKey, newValue, .OBJC_ASSOCIATION_RETAIN_NONATOMIC)
        }
        get
        {
            return objc_getAssociatedObject(self, &ImportControllerKey) as? ImportController
        }
    }
    
    func present(_ importController: ImportController, animated: Bool, completion: (() -> Void)?)
    {
        self.importController = importController
        
        importController.presentImportController(from: self, animated: animated, completionHandler: completion)
    }
}