space-ios/File Provider Extension/FileProviderDomain.swift

//
//  FileProviderDomain.swift
//  Nextcloud
//
//  Created by Marino Faggiana on 04/06/2019.
//  Copyright © 2018 Marino Faggiana. All rights reserved.
//
//  Author Marino Faggiana <marino.faggiana@nextcloud.com>
//
//  This program is free software: you can redistribute it and/or modify
//  it under the terms of the GNU General Public License as published by
//  the Free Software Foundation, either version 3 of the License, or
//  (at your option) any later version.
//
//  This program is distributed in the hope that it will be useful,
//  but WITHOUT ANY WARRANTY; without even the implied warranty of
//  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
//  GNU General Public License for more details.
//
//  You should have received a copy of the GNU General Public License
//  along with this program.  If not, see <http://www.gnu.org/licenses/>.
//

import Foundation

class FileProviderDomain: NSObject {
    @objc static let sharedInstance: FileProviderDomain = {
        let instance = FileProviderDomain()
        return instance
    }()

    @objc func registerDomain() {
        
        NSFileProviderManager.getDomainsWithCompletionHandler { (fileProviderDomain, error) in
            
            var domains = [String]()
            let pathRelativeToDocumentStorage = NSFileProviderManager.default.documentStorageURL.absoluteString
            let tableAccounts = NCManageDatabase.sharedInstance.getAllAccount()
            
            for domain in fileProviderDomain {
                domains.append(domain.identifier.rawValue)
            }
            
            // Delete all domains
            for domain in domains {
                let domainRawValue = NSFileProviderDomain(identifier: NSFileProviderDomainIdentifier(rawValue: domain), displayName: domain, pathRelativeToDocumentStorage: pathRelativeToDocumentStorage)
                NSFileProviderManager.remove(domainRawValue, completionHandler: { (error) in
                    if error != nil {
                        print("Error  domain: \(domainRawValue) error: \(String(describing: error))")
                    }
                })
            }
            
            // Check account->domain & (add)
            for tableAccount in tableAccounts {
                guard let url = NSURL(string: tableAccount.url) else {
                    continue
                }
                guard let host = url.host else {
                    continue
                }
                let accountDomain =  tableAccount.userID + " (" + host + ")"
                let domainRawValue = NSFileProviderDomain(identifier: NSFileProviderDomainIdentifier(rawValue: accountDomain), displayName: accountDomain, pathRelativeToDocumentStorage: pathRelativeToDocumentStorage)
                NSFileProviderManager.add(domainRawValue, completionHandler: { (error) in
                    if error != nil {
                        print("Error  domain: \(domainRawValue) error: \(String(describing: error))")
                    }
                })
            }
        }
    }
    
    @objc func removeAllDomain() {
        
        NSFileProviderManager.getDomainsWithCompletionHandler { (fileProviderDomain, error) in
            
            var domains = [String]()
            let pathRelativeToDocumentStorage = NSFileProviderManager.default.documentStorageURL.absoluteString

            for domain in fileProviderDomain {
                domains.append(domain.identifier.rawValue)
            }
            for domain in domains {
                let domainRawValue = NSFileProviderDomain(identifier: NSFileProviderDomainIdentifier(rawValue: domain), displayName: domain, pathRelativeToDocumentStorage: pathRelativeToDocumentStorage)
                NSFileProviderManager.remove(domainRawValue, completionHandler: { (error) in
                    if error != nil {
                        print("Error  domain: \(domainRawValue) error: \(String(describing: error))")
                    }
                })
            }
        }
    }
}