73 lines
2.3 KiB
Swift
73 lines
2.3 KiB
Swift
import CoreData
|
|
import Foundation
|
|
import Models
|
|
import SwiftGraphQL
|
|
|
|
public extension DataService {
|
|
func setLabelsForHighlight(highlightID: String, labelIDs: [String]) {
|
|
backgroundContext.perform { [weak self] in
|
|
guard let self = self else { return }
|
|
guard let highlight = Highlight.lookup(byID: highlightID, inContext: self.backgroundContext) else { return }
|
|
|
|
if let existingLabels = highlight.labels {
|
|
highlight.removeFromLabels(existingLabels)
|
|
}
|
|
|
|
for labelID in labelIDs {
|
|
if let labelObject = LinkedItemLabel.lookup(byID: labelID, inContext: self.backgroundContext) {
|
|
highlight.addToLabels(labelObject)
|
|
}
|
|
}
|
|
|
|
// Send update to server
|
|
self.syncHighlightLabelUpdates(highlightID: highlightID, labelIDs: labelIDs)
|
|
}
|
|
}
|
|
|
|
func syncHighlightLabelUpdates(highlightID: String, labelIDs: [String]) {
|
|
enum MutationResult {
|
|
case saved(labels: [InternalLinkedItemLabel])
|
|
case error(errorCode: Enums.SetLabelsErrorCode)
|
|
}
|
|
|
|
let selection = Selection<MutationResult, Unions.SetLabelsResult> {
|
|
try $0.on(
|
|
setLabelsError: .init { .error(errorCode: try $0.errorCodes().first ?? .badRequest) },
|
|
setLabelsSuccess: .init { .saved(labels: try $0.labels(selection: highlightLabelSelection.list)) }
|
|
)
|
|
}
|
|
|
|
let mutation = Selection.Mutation {
|
|
try $0.setLabelsForHighlight(
|
|
input: InputObjects.SetLabelsForHighlightInput(
|
|
highlightId: highlightID,
|
|
labelIds: labelIDs
|
|
),
|
|
selection: selection
|
|
)
|
|
}
|
|
|
|
let path = appEnvironment.graphqlPath
|
|
let headers = networker.defaultHeaders
|
|
let context = backgroundContext
|
|
|
|
send(mutation, to: path, headers: headers) { result in
|
|
let data = try? result.get()
|
|
let syncStatus: ServerSyncStatus = data == nil ? .needsUpdate : .isNSync
|
|
|
|
context.perform {
|
|
guard let highlight = Highlight.lookup(byID: highlightID, inContext: context) else { return }
|
|
highlight.serverSyncStatus = Int64(syncStatus.rawValue)
|
|
|
|
do {
|
|
try context.save()
|
|
logger.debug("Highlight labels updated succesfully")
|
|
} catch {
|
|
context.rollback()
|
|
logger.debug("Failed to update highlight labels: \(error.localizedDescription)")
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|