mirror of
https://github.com/yattee/yattee.git
synced 2024-12-14 14:20:32 +05:30
157 lines
5.1 KiB
Swift
157 lines
5.1 KiB
Swift
import Foundation
|
|
import Logging
|
|
|
|
struct URLBookmarkModel {
|
|
static let bookmarkPrefix = "urlbookmark-"
|
|
static var shared = URLBookmarkModel()
|
|
|
|
var logger = Logger(label: "stream.yattee.url-bookmark")
|
|
|
|
var allBookmarksKeys: [String] {
|
|
guard let defaults = BookmarksCacheModel.shared.defaults else { return [] }
|
|
|
|
return defaults.dictionaryRepresentation().keys.filter { $0.starts(with: Self.bookmarkPrefix) }
|
|
}
|
|
|
|
var allURLs: [URL] {
|
|
allBookmarksKeys.compactMap { urlFromBookmark($0) }
|
|
}
|
|
|
|
func saveBookmark(_ url: URL) {
|
|
var urlForBookmark = url
|
|
if let yatteeSanitizedUrl = url.byReplacingYatteeProtocol() {
|
|
urlForBookmark = yatteeSanitizedUrl
|
|
}
|
|
|
|
guard urlForBookmark.isFileURL else {
|
|
logger.error("trying to save bookmark for something that is not a file")
|
|
logger.error("not a file: \(urlForBookmark.absoluteString)")
|
|
return
|
|
}
|
|
|
|
guard let defaults = BookmarksCacheModel.shared.defaults else {
|
|
logger.error("could not open bookmarks defaults")
|
|
return
|
|
}
|
|
|
|
if let bookmarkData = try? urlForBookmark.bookmarkData(options: bookmarkCreationOptions, includingResourceValuesForKeys: nil, relativeTo: nil) {
|
|
defaults.set(bookmarkData, forKey: bookmarkKey(urlForBookmark))
|
|
logger.info("saved bookmark for \(bookmarkKey(urlForBookmark))")
|
|
} else {
|
|
logger.error("no bookmark data for \(urlForBookmark)")
|
|
}
|
|
}
|
|
|
|
func saveBookmark(_ url: NSURL) {
|
|
guard url.isFileURL else {
|
|
logger.error("trying to save bookmark for something that is not a file")
|
|
logger.error("not a file: \(url.absoluteString)")
|
|
return
|
|
}
|
|
|
|
guard let defaults = BookmarksCacheModel.shared.defaults else {
|
|
logger.error("could not open bookmarks defaults")
|
|
return
|
|
}
|
|
|
|
if let bookmarkData = try? url.bookmarkData(options: bookmarkCreationOptions, includingResourceValuesForKeys: nil, relativeTo: nil) {
|
|
defaults.set(bookmarkData, forKey: bookmarkKey(url))
|
|
logger.info("saved bookmark for \(bookmarkKey(url))")
|
|
} else {
|
|
logger.error("no bookmark data for \(bookmarkKey(url))")
|
|
}
|
|
}
|
|
|
|
func loadBookmark(_ url: URL) -> URL? {
|
|
var urlForBookmark = url
|
|
if let yatteeSanitizedUrl = url.byReplacingYatteeProtocol() {
|
|
urlForBookmark = yatteeSanitizedUrl
|
|
}
|
|
|
|
logger.info("loading bookmark for \(bookmarkKey(urlForBookmark))")
|
|
|
|
guard let defaults = BookmarksCacheModel.shared.defaults else {
|
|
logger.error("could not open bookmarks defaults")
|
|
return nil
|
|
}
|
|
|
|
if let data = defaults.data(forKey: bookmarkKey(urlForBookmark)) {
|
|
do {
|
|
var isStale = false
|
|
let url = try URL(
|
|
resolvingBookmarkData: data,
|
|
options: bookmarkResolutionOptions,
|
|
relativeTo: nil,
|
|
bookmarkDataIsStale: &isStale
|
|
)
|
|
if isStale {
|
|
saveBookmark(urlForBookmark)
|
|
}
|
|
logger.info("loaded bookmark for \(bookmarkKey(urlForBookmark))")
|
|
|
|
return url
|
|
} catch {
|
|
print("Error resolving bookmark:", error)
|
|
return nil
|
|
}
|
|
} else {
|
|
logger.warning("could not find bookmark for \(bookmarkKey(urlForBookmark))")
|
|
return nil
|
|
}
|
|
}
|
|
|
|
func removeBookmark(_ url: URL) {
|
|
logger.info("removing bookmark for \(bookmarkKey(url))")
|
|
|
|
guard let defaults = BookmarksCacheModel.shared.defaults else {
|
|
logger.error("could not open bookmarks defaults")
|
|
return
|
|
}
|
|
|
|
defaults.removeObject(forKey: bookmarkKey(url))
|
|
}
|
|
|
|
func refreshAll() {
|
|
logger.info("refreshing all bookmarks")
|
|
|
|
allURLs.forEach { url in
|
|
if loadBookmark(url) != nil {
|
|
logger.info("bookmark for \(url) exists")
|
|
} else {
|
|
logger.info("bookmark does not exist")
|
|
}
|
|
}
|
|
}
|
|
|
|
private func bookmarkKey(_ url: URL) -> String {
|
|
"\(Self.bookmarkPrefix)\(NSString(string: url.absoluteString).standardizingPath)"
|
|
}
|
|
|
|
private func bookmarkKey(_ url: NSURL) -> String {
|
|
"\(Self.bookmarkPrefix)\(url.standardizingPath?.absoluteString ?? "unknown")"
|
|
}
|
|
|
|
private func urlFromBookmark(_ key: String) -> URL? {
|
|
if let urlString = key.components(separatedBy: Self.bookmarkPrefix).last {
|
|
return URL(string: urlString)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
private var bookmarkCreationOptions: URL.BookmarkCreationOptions {
|
|
#if os(macOS)
|
|
return [.withSecurityScope, .securityScopeAllowOnlyReadAccess]
|
|
#else
|
|
return []
|
|
#endif
|
|
}
|
|
|
|
private var bookmarkResolutionOptions: URL.BookmarkResolutionOptions {
|
|
#if os(macOS)
|
|
return [.withSecurityScope]
|
|
#else
|
|
return []
|
|
#endif
|
|
}
|
|
}
|