mirror of
https://github.com/nextcloud/desktop.git
synced 2024-11-22 13:05:51 +03:00
417 lines
15 KiB
Swift
417 lines
15 KiB
Swift
/*
|
|
* Copyright (C) 2022 by Claudio Cambra <claudio.cambra@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 2 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.
|
|
*/
|
|
|
|
import FileProvider
|
|
import NCDesktopClientSocketKit
|
|
import NextcloudKit
|
|
import NextcloudFileProviderKit
|
|
import OSLog
|
|
|
|
@objc class FileProviderExtension: NSObject, NSFileProviderReplicatedExtension {
|
|
let domain: NSFileProviderDomain
|
|
let ncKit = NextcloudKit()
|
|
let appGroupIdentifier = Bundle.main.object(forInfoDictionaryKey: "SocketApiPrefix") as? String
|
|
var ncAccount: Account?
|
|
var changeObserver: RemoteChangeObserver?
|
|
lazy var ncKitBackground = NKBackground(nkCommonInstance: ncKit.nkCommonInstance)
|
|
lazy var socketClient: LocalSocketClient? = {
|
|
guard let containerUrl = pathForAppGroupContainer() else {
|
|
Logger.fileProviderExtension.critical("Won't start socket client, no container url")
|
|
return nil;
|
|
}
|
|
|
|
let socketPath = containerUrl.appendingPathComponent(
|
|
".fileprovidersocket", conformingTo: .archive)
|
|
let lineProcessor = FileProviderSocketLineProcessor(delegate: self)
|
|
return LocalSocketClient(socketPath: socketPath.path, lineProcessor: lineProcessor)
|
|
}()
|
|
|
|
var syncActions = Set<UUID>()
|
|
var errorActions = Set<UUID>()
|
|
var actionsLock = NSLock()
|
|
|
|
// Whether or not we are going to recursively scan new folders when they are discovered.
|
|
// Apple's recommendation is that we should always scan the file hierarchy fully.
|
|
// This does lead to long load times when a file provider domain is initially configured.
|
|
// We can instead do a fast enumeration where we only scan folders as the user navigates through
|
|
// them, thereby avoiding this issue; the trade-off is that we will be unable to detect
|
|
// materialised file moves to unexplored folders, therefore deleting the item when we could have
|
|
// just moved it instead.
|
|
//
|
|
// Since it's not desirable to cancel a long recursive enumeration half-way through, we do the
|
|
// fast enumeration by default. We prompt the user on the client side to run a proper, full
|
|
// enumeration if they want for safety.
|
|
lazy var config = FileProviderConfig(domainIdentifier: domain.identifier)
|
|
|
|
required init(domain: NSFileProviderDomain) {
|
|
// The containing application must create a domain using
|
|
// `NSFileProviderManager.add(_:, completionHandler:)`. The system will then launch the
|
|
// application extension process, call `FileProviderExtension.init(domain:)` to instantiate
|
|
// the extension for that domain, and call methods on the instance.
|
|
self.domain = domain
|
|
super.init()
|
|
socketClient?.start()
|
|
}
|
|
|
|
func invalidate() {
|
|
// TODO: cleanup any resources
|
|
Logger.fileProviderExtension.debug(
|
|
"Extension for domain \(self.domain.displayName, privacy: .public) is being torn down"
|
|
)
|
|
}
|
|
|
|
func insertSyncAction(_ actionId: UUID) {
|
|
actionsLock.lock()
|
|
let oldActions = syncActions
|
|
syncActions.insert(actionId)
|
|
actionsLock.unlock()
|
|
updatedSyncStateReporting(oldActions: oldActions)
|
|
}
|
|
|
|
func insertErrorAction(_ actionId: UUID) {
|
|
actionsLock.lock()
|
|
let oldActions = syncActions
|
|
syncActions.remove(actionId)
|
|
errorActions.insert(actionId)
|
|
actionsLock.unlock()
|
|
updatedSyncStateReporting(oldActions: oldActions)
|
|
}
|
|
|
|
func removeSyncAction(_ actionId: UUID) {
|
|
actionsLock.lock()
|
|
let oldActions = syncActions
|
|
syncActions.remove(actionId)
|
|
errorActions.remove(actionId)
|
|
actionsLock.unlock()
|
|
updatedSyncStateReporting(oldActions: oldActions)
|
|
}
|
|
|
|
// MARK: - NSFileProviderReplicatedExtension protocol methods
|
|
|
|
func item(
|
|
for identifier: NSFileProviderItemIdentifier,
|
|
request _: NSFileProviderRequest,
|
|
completionHandler: @escaping (NSFileProviderItem?, Error?) -> Void
|
|
) -> Progress {
|
|
if let item = Item.storedItem(identifier: identifier, remoteInterface: ncKit) {
|
|
completionHandler(item, nil)
|
|
} else {
|
|
completionHandler(nil, NSFileProviderError(.noSuchItem))
|
|
}
|
|
return Progress()
|
|
}
|
|
|
|
func fetchContents(
|
|
for itemIdentifier: NSFileProviderItemIdentifier,
|
|
version requestedVersion: NSFileProviderItemVersion?,
|
|
request: NSFileProviderRequest,
|
|
completionHandler: @escaping (URL?, NSFileProviderItem?, Error?) -> Void
|
|
) -> Progress {
|
|
let actionId = UUID()
|
|
insertSyncAction(actionId)
|
|
|
|
Logger.fileProviderExtension.debug(
|
|
"Received request to fetch contents of item with identifier: \(itemIdentifier.rawValue, privacy: .public)"
|
|
)
|
|
|
|
guard requestedVersion == nil else {
|
|
// TODO: Add proper support for file versioning
|
|
Logger.fileProviderExtension.error(
|
|
"Can't return contents for a specific version as this is not supported."
|
|
)
|
|
insertErrorAction(actionId)
|
|
completionHandler(
|
|
nil,
|
|
nil,
|
|
NSError(domain: NSCocoaErrorDomain, code: NSFeatureUnsupportedError)
|
|
)
|
|
return Progress()
|
|
}
|
|
|
|
guard ncAccount != nil else {
|
|
Logger.fileProviderExtension.error(
|
|
"""
|
|
Not fetching contents for item: \(itemIdentifier.rawValue, privacy: .public)
|
|
as account not set up yet.
|
|
"""
|
|
)
|
|
insertErrorAction(actionId)
|
|
completionHandler(nil, nil, NSFileProviderError(.notAuthenticated))
|
|
return Progress()
|
|
}
|
|
|
|
guard let item = Item.storedItem(identifier: itemIdentifier, remoteInterface: ncKit) else {
|
|
Logger.fileProviderExtension.error(
|
|
"""
|
|
Not fetching contents for item: \(itemIdentifier.rawValue, privacy: .public)
|
|
as item not found.
|
|
"""
|
|
)
|
|
completionHandler(nil, nil, NSFileProviderError(.noSuchItem))
|
|
insertErrorAction(actionId)
|
|
return Progress()
|
|
}
|
|
|
|
let progress = Progress()
|
|
Task {
|
|
let (localUrl, updatedItem, error) = await item.fetchContents(
|
|
domain: self.domain, progress: progress
|
|
)
|
|
removeSyncAction(actionId)
|
|
completionHandler(localUrl, updatedItem, error)
|
|
}
|
|
return progress
|
|
}
|
|
|
|
func createItem(
|
|
basedOn itemTemplate: NSFileProviderItem,
|
|
fields: NSFileProviderItemFields,
|
|
contents url: URL?,
|
|
options: NSFileProviderCreateItemOptions = [],
|
|
request: NSFileProviderRequest,
|
|
completionHandler: @escaping (
|
|
NSFileProviderItem?, NSFileProviderItemFields, Bool, Error?
|
|
) -> Void
|
|
) -> Progress {
|
|
let actionId = UUID()
|
|
insertSyncAction(actionId)
|
|
|
|
let tempId = itemTemplate.itemIdentifier.rawValue
|
|
Logger.fileProviderExtension.debug(
|
|
"""
|
|
Received create item request for item with identifier: \(tempId, privacy: .public)
|
|
and filename: \(itemTemplate.filename, privacy: .public)
|
|
"""
|
|
)
|
|
|
|
guard let ncAccount else {
|
|
Logger.fileProviderExtension.error(
|
|
"""
|
|
Not creating item: \(itemTemplate.itemIdentifier.rawValue, privacy: .public)
|
|
as account not set up yet
|
|
"""
|
|
)
|
|
insertErrorAction(actionId)
|
|
completionHandler(
|
|
itemTemplate,
|
|
NSFileProviderItemFields(),
|
|
false,
|
|
NSFileProviderError(.notAuthenticated)
|
|
)
|
|
return Progress()
|
|
}
|
|
|
|
let progress = Progress()
|
|
Task {
|
|
let (item, error) = await Item.create(
|
|
basedOn: itemTemplate,
|
|
fields: fields,
|
|
contents: url,
|
|
request: request,
|
|
domain: self.domain,
|
|
remoteInterface: ncKit,
|
|
progress: progress
|
|
)
|
|
|
|
if error != nil {
|
|
insertErrorAction(actionId)
|
|
signalEnumerator(completionHandler: { _ in })
|
|
} else {
|
|
removeSyncAction(actionId)
|
|
}
|
|
|
|
completionHandler(
|
|
item ?? itemTemplate,
|
|
NSFileProviderItemFields(),
|
|
false,
|
|
error
|
|
)
|
|
}
|
|
return progress
|
|
}
|
|
|
|
func modifyItem(
|
|
_ item: NSFileProviderItem,
|
|
baseVersion: NSFileProviderItemVersion,
|
|
changedFields: NSFileProviderItemFields,
|
|
contents newContents: URL?,
|
|
options: NSFileProviderModifyItemOptions = [],
|
|
request: NSFileProviderRequest,
|
|
completionHandler: @escaping (
|
|
NSFileProviderItem?, NSFileProviderItemFields, Bool, Error?
|
|
) -> Void
|
|
) -> Progress {
|
|
// An item was modified on disk, process the item's modification
|
|
// TODO: Handle finder things like tags, other possible item changed fields
|
|
let actionId = UUID()
|
|
insertSyncAction(actionId)
|
|
|
|
let identifier = item.itemIdentifier
|
|
let ocId = identifier.rawValue
|
|
Logger.fileProviderExtension.debug(
|
|
"""
|
|
Received modify item request for item with identifier: \(ocId, privacy: .public)
|
|
and filename: \(item.filename, privacy: .public)
|
|
"""
|
|
)
|
|
|
|
guard let ncAccount else {
|
|
Logger.fileProviderExtension.error(
|
|
"Not modifying item: \(ocId, privacy: .public) as account not set up yet."
|
|
)
|
|
insertErrorAction(actionId)
|
|
completionHandler(item, [], false, NSFileProviderError(.notAuthenticated))
|
|
return Progress()
|
|
}
|
|
|
|
guard let existingItem = Item.storedItem(identifier: identifier, remoteInterface: ncKit) else {
|
|
Logger.fileProviderExtension.error(
|
|
"Not modifying item: \(ocId, privacy: .public) as item not found."
|
|
)
|
|
insertErrorAction(actionId)
|
|
completionHandler(item, [], false, NSFileProviderError(.noSuchItem))
|
|
return Progress()
|
|
}
|
|
|
|
let progress = Progress()
|
|
Task {
|
|
let (modifiedItem, error) = await existingItem.modify(
|
|
itemTarget: item,
|
|
baseVersion: baseVersion,
|
|
changedFields: changedFields,
|
|
contents: newContents,
|
|
options: options,
|
|
request: request,
|
|
domain: domain,
|
|
progress: progress
|
|
)
|
|
|
|
if error != nil {
|
|
insertErrorAction(actionId)
|
|
signalEnumerator(completionHandler: { _ in })
|
|
} else {
|
|
removeSyncAction(actionId)
|
|
}
|
|
|
|
completionHandler(modifiedItem ?? item, [], false, error)
|
|
}
|
|
return progress
|
|
}
|
|
|
|
func deleteItem(
|
|
identifier: NSFileProviderItemIdentifier,
|
|
baseVersion _: NSFileProviderItemVersion,
|
|
options _: NSFileProviderDeleteItemOptions = [],
|
|
request _: NSFileProviderRequest,
|
|
completionHandler: @escaping (Error?) -> Void
|
|
) -> Progress {
|
|
let actionId = UUID()
|
|
insertSyncAction(actionId)
|
|
|
|
Logger.fileProviderExtension.debug(
|
|
"Received delete request for item: \(identifier.rawValue, privacy: .public)"
|
|
)
|
|
|
|
guard ncAccount != nil else {
|
|
Logger.fileProviderExtension.error(
|
|
"Not deleting item \(identifier.rawValue, privacy: .public), account not set up yet"
|
|
)
|
|
insertErrorAction(actionId)
|
|
completionHandler(NSFileProviderError(.notAuthenticated))
|
|
return Progress()
|
|
}
|
|
|
|
guard let item = Item.storedItem(identifier: identifier, remoteInterface: ncKit) else {
|
|
Logger.fileProviderExtension.error(
|
|
"Not deleting item \(identifier.rawValue, privacy: .public), item not found"
|
|
)
|
|
insertErrorAction(actionId)
|
|
completionHandler(NSFileProviderError(.noSuchItem))
|
|
return Progress()
|
|
}
|
|
|
|
let progress = Progress(totalUnitCount: 1)
|
|
Task {
|
|
let error = await item.delete()
|
|
if error != nil {
|
|
insertErrorAction(actionId)
|
|
signalEnumerator(completionHandler: { _ in })
|
|
} else {
|
|
removeSyncAction(actionId)
|
|
}
|
|
progress.completedUnitCount = 1
|
|
completionHandler(await item.delete())
|
|
}
|
|
return progress
|
|
}
|
|
|
|
func enumerator(
|
|
for containerItemIdentifier: NSFileProviderItemIdentifier, request _: NSFileProviderRequest
|
|
) throws -> NSFileProviderEnumerator {
|
|
guard let ncAccount else {
|
|
Logger.fileProviderExtension.error(
|
|
"Not providing enumerator for container with identifier \(containerItemIdentifier.rawValue, privacy: .public) yet as account not set up"
|
|
)
|
|
throw NSFileProviderError(.notAuthenticated)
|
|
}
|
|
|
|
return Enumerator(
|
|
enumeratedItemIdentifier: containerItemIdentifier,
|
|
remoteInterface: ncKit,
|
|
domain: domain,
|
|
fastEnumeration: config.fastEnumerationEnabled,
|
|
listener: self
|
|
)
|
|
}
|
|
|
|
func materializedItemsDidChange(completionHandler: @escaping () -> Void) {
|
|
guard let ncAccount else {
|
|
Logger.fileProviderExtension.error(
|
|
"Not purging stale local file metadatas, account not set up")
|
|
completionHandler()
|
|
return
|
|
}
|
|
|
|
guard let fpManager = NSFileProviderManager(for: domain) else {
|
|
Logger.fileProviderExtension.error(
|
|
"Could not get file provider manager for domain: \(self.domain.displayName, privacy: .public)"
|
|
)
|
|
completionHandler()
|
|
return
|
|
}
|
|
|
|
let materialisedEnumerator = fpManager.enumeratorForMaterializedItems()
|
|
let materialisedObserver = MaterialisedEnumerationObserver(
|
|
ncKitAccount: ncAccount.ncKitAccount
|
|
) { _ in
|
|
completionHandler()
|
|
}
|
|
let startingPage = NSFileProviderPage(NSFileProviderPage.initialPageSortedByName as Data)
|
|
|
|
materialisedEnumerator.enumerateItems(for: materialisedObserver, startingAt: startingPage)
|
|
}
|
|
|
|
// MARK: - Helper functions
|
|
|
|
func signalEnumerator(completionHandler: @escaping (_ error: Error?) -> Void) {
|
|
guard let fpManager = NSFileProviderManager(for: domain) else {
|
|
Logger.fileProviderExtension.error(
|
|
"Could not get file provider manager for domain, could not signal enumerator. This might lead to future conflicts."
|
|
)
|
|
return
|
|
}
|
|
|
|
fpManager.signalEnumerator(for: .workingSet, completionHandler: completionHandler)
|
|
}
|
|
}
|