Revision control
Copy as Markdown
// This Source Code Form is subject to the terms of the Mozilla Public
// License, v. 2.0. If a copy of the MPL was not distributed with this
import Foundation
protocol FaviconURLHandler {
func getFaviconURL(site: SiteImageModel) async throws -> SiteImageModel
func cacheFaviconURL(cacheKey: String, faviconURL: URL)
func clearCache()
}
struct DefaultFaviconURLHandler: FaviconURLHandler {
private let urlFetcher: FaviconURLFetcher
private let urlCache: FaviconURLCache
init(urlFetcher: FaviconURLFetcher = DefaultFaviconURLFetcher(),
urlCache: FaviconURLCache = DefaultFaviconURLCache.shared) {
self.urlFetcher = urlFetcher
self.urlCache = urlCache
}
func getFaviconURL(site: SiteImageModel) async throws -> SiteImageModel {
// Don't fetch favicon URL if we don't have a URL or domain for it
guard let siteURL = site.siteURL else {
throw SiteImageError.noFaviconURLFound
}
var imageModel = site
do {
let url = try await urlCache.getURLFromCache(cacheKey: imageModel.cacheKey)
imageModel.faviconURL = url
return imageModel
} catch {
do {
let url = try await urlFetcher.fetchFaviconURL(siteURL: siteURL)
await urlCache.cacheURL(cacheKey: imageModel.cacheKey, faviconURL: url)
imageModel.faviconURL = url
return imageModel
} catch {
throw SiteImageError.noFaviconURLFound
}
}
}
func cacheFaviconURL(cacheKey: String, faviconURL: URL) {
Task {
await urlCache.cacheURL(cacheKey: cacheKey, faviconURL: faviconURL)
}
}
func clearCache() {
Task {
await urlCache.clearCache()
}
}
}