94 lines
2.3 KiB
Swift
94 lines
2.3 KiB
Swift
//
|
|
// AuthorAvatarDownloader.swift
|
|
// NetNewsWire
|
|
//
|
|
// Created by Brent Simmons on 11/26/17.
|
|
// Copyright © 2017 Ranchero Software. All rights reserved.
|
|
//
|
|
|
|
import Foundation
|
|
import Articles
|
|
import RSCore
|
|
|
|
extension Notification.Name {
|
|
|
|
static let AvatarDidBecomeAvailable = Notification.Name("AvatarDidBecomeAvailableNotification") // UserInfoKey.imageURL (which is an avatarURL)
|
|
}
|
|
|
|
final class AuthorAvatarDownloader {
|
|
|
|
private let imageDownloader: ImageDownloader
|
|
private var cache = [String: RSImage]() // avatarURL: RSImage
|
|
private var waitingForAvatarURLs = Set<String>()
|
|
|
|
init(imageDownloader: ImageDownloader) {
|
|
|
|
self.imageDownloader = imageDownloader
|
|
NotificationCenter.default.addObserver(self, selector: #selector(imageDidBecomeAvailable(_:)), name: .ImageDidBecomeAvailable, object: imageDownloader)
|
|
}
|
|
|
|
func resetCache() {
|
|
cache = [String: RSImage]()
|
|
}
|
|
|
|
func image(for author: Author) -> RSImage? {
|
|
|
|
guard let avatarURL = author.avatarURL else {
|
|
return nil
|
|
}
|
|
|
|
if let cachedImage = cache[avatarURL] {
|
|
return cachedImage
|
|
}
|
|
|
|
if let imageData = imageDownloader.image(for: avatarURL) {
|
|
scaleAndCacheImageData(imageData, avatarURL)
|
|
}
|
|
else {
|
|
waitingForAvatarURLs.insert(avatarURL)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
@objc func imageDidBecomeAvailable(_ note: Notification) {
|
|
guard let avatarURL = note.userInfo?[UserInfoKey.url] as? String else {
|
|
return
|
|
}
|
|
guard waitingForAvatarURLs.contains(avatarURL) else {
|
|
return
|
|
}
|
|
guard let imageData = imageDownloader.image(for: avatarURL) else {
|
|
return
|
|
}
|
|
scaleAndCacheImageData(imageData, avatarURL)
|
|
}
|
|
}
|
|
|
|
private extension AuthorAvatarDownloader {
|
|
|
|
func scaleAndCacheImageData(_ imageData: Data, _ avatarURL: String) {
|
|
RSImage.scaledForAvatar(imageData) { (image) in
|
|
if let image = image {
|
|
self.handleImageDidBecomeAvailable(avatarURL, image)
|
|
}
|
|
}
|
|
}
|
|
|
|
func handleImageDidBecomeAvailable(_ avatarURL: String, _ image: RSImage) {
|
|
if cache[avatarURL] == nil {
|
|
cache[avatarURL] = image
|
|
}
|
|
if waitingForAvatarURLs.contains(avatarURL) {
|
|
waitingForAvatarURLs.remove(avatarURL)
|
|
postAvatarDidBecomeAvailableNotification(avatarURL)
|
|
}
|
|
}
|
|
|
|
func postAvatarDidBecomeAvailableNotification(_ avatarURL: String) {
|
|
DispatchQueue.main.async {
|
|
NotificationCenter.default.post(name: .AvatarDidBecomeAvailable, object: self, userInfo: [UserInfoKey.url: avatarURL])
|
|
}
|
|
}
|
|
}
|