107 lines
2.6 KiB
Swift
107 lines
2.6 KiB
Swift
import Combine
|
|
import CoreData
|
|
import Foundation
|
|
import Models
|
|
import Services
|
|
|
|
public final class PDFViewerViewModel: ObservableObject {
|
|
@Published public var errorMessage: String?
|
|
@Published public var readerView: Bool = false
|
|
|
|
public let pdfItem: PDFItem
|
|
private var storedURL: URL?
|
|
|
|
var subscriptions = Set<AnyCancellable>()
|
|
|
|
public init(pdfItem: PDFItem) {
|
|
self.pdfItem = pdfItem
|
|
}
|
|
|
|
public func dataURL(remoteURL: URL) -> URL {
|
|
if let storedURL = storedURL {
|
|
return storedURL
|
|
}
|
|
|
|
guard let data = pdfItem.documentData else { return remoteURL }
|
|
|
|
let subPath = pdfItem.title.isEmpty ? UUID().uuidString : pdfItem.title
|
|
|
|
let path = FileManager.default
|
|
.urls(for: .cachesDirectory, in: .userDomainMask)[0]
|
|
.appendingPathComponent(subPath)
|
|
|
|
do {
|
|
try data.write(to: path)
|
|
storedURL = path
|
|
return path
|
|
} catch {
|
|
return remoteURL
|
|
}
|
|
}
|
|
|
|
public func loadHighlightPatches(completion onComplete: @escaping ([String]) -> Void) {
|
|
onComplete(pdfItem.highlights.map { $0.patch ?? "" })
|
|
}
|
|
|
|
public func createHighlight(
|
|
dataService: DataService,
|
|
shortId: String,
|
|
highlightID: String,
|
|
quote: String,
|
|
patch: String
|
|
) {
|
|
_ = dataService.createHighlight(
|
|
shortId: shortId,
|
|
highlightID: highlightID,
|
|
quote: quote,
|
|
patch: patch,
|
|
articleId: pdfItem.itemID
|
|
)
|
|
}
|
|
|
|
public func mergeHighlight(
|
|
dataService: DataService,
|
|
shortId: String,
|
|
highlightID: String,
|
|
quote: String,
|
|
patch: String,
|
|
overlapHighlightIdList: [String]
|
|
) {
|
|
_ = dataService.mergeHighlights(
|
|
shortId: shortId,
|
|
highlightID: highlightID,
|
|
quote: quote,
|
|
patch: patch,
|
|
articleId: pdfItem.itemID,
|
|
overlapHighlightIdList: overlapHighlightIdList
|
|
)
|
|
}
|
|
|
|
public func removeHighlights(dataService: DataService, highlightIds: [String]) {
|
|
highlightIds.forEach { highlightID in
|
|
dataService.deleteHighlight(highlightID: highlightID)
|
|
}
|
|
}
|
|
|
|
public func updateItemReadProgress(dataService: DataService, percent: Double, anchorIndex: Int) {
|
|
dataService.updateLinkReadingProgress(
|
|
itemID: pdfItem.itemID,
|
|
readingProgress: percent,
|
|
anchorIndex: anchorIndex
|
|
)
|
|
}
|
|
|
|
public func highlightShareURL(dataService: DataService, shortId: String) -> URL? {
|
|
let baseURL = dataService.appEnvironment.serverBaseURL
|
|
var components = URLComponents(url: baseURL, resolvingAgainstBaseURL: false)
|
|
|
|
if let username = dataService.currentViewer?.username {
|
|
components?.path = "/\(username)/\(pdfItem.slug)/highlights/\(shortId)"
|
|
} else {
|
|
return nil
|
|
}
|
|
|
|
return components?.url
|
|
}
|
|
}
|