Show GIFs in AttachmentView

This commit is contained in:
Shadowfacts 2018-11-08 19:20:02 -05:00
parent 6393829174
commit 9d19c76e4c
8 changed files with 66 additions and 38 deletions

View File

@ -15,31 +15,31 @@ class ImageCache {
static let headers = ImageCache(name: "Headers", memoryExpiry: .seconds(60 * 60), diskExpiry: .seconds(60 * 60 * 24))
static let attachments = ImageCache(name: "Attachments", memoryExpiry: .seconds(60 * 2))
let cache: Cache<UIImage>
let cache: Cache<Data>
var requests = [URL: Request]()
init(name: String, memoryExpiry expiry: Expiry) {
let storage = MemoryStorage<UIImage>(config: MemoryConfig(expiry: expiry))
let storage = MemoryStorage<Data>(config: MemoryConfig(expiry: expiry))
self.cache = .memory(storage)
}
init(name: String, diskExpiry expiry: Expiry) {
let storage = try! DiskStorage<UIImage>(config: DiskConfig(name: name, expiry: expiry), transformer: TransformerFactory.forImage())
let storage = try! DiskStorage<Data>(config: DiskConfig(name: name, expiry: expiry), transformer: TransformerFactory.forData())
self.cache = .disk(storage)
}
init(name: String, memoryExpiry: Expiry, diskExpiry: Expiry) {
let memory = MemoryStorage<UIImage>(config: MemoryConfig(expiry: memoryExpiry))
let disk = try! DiskStorage<UIImage>(config: DiskConfig(name: name, expiry: diskExpiry), transformer: TransformerFactory.forImage())
let memory = MemoryStorage<Data>(config: MemoryConfig(expiry: memoryExpiry))
let disk = try! DiskStorage<Data>(config: DiskConfig(name: name, expiry: diskExpiry), transformer: TransformerFactory.forData())
self.cache = .hybrid(HybridStorage(memoryStorage: memory, diskStorage: disk))
}
func get(_ url: URL, completion: ((UIImage?) -> Void)?) {
func get(_ url: URL, completion: ((Data?) -> Void)?) {
let key = url.absoluteString
if (try? cache.existsObject(forKey: key)) ?? false,
let image = try? cache.object(forKey: key) {
completion?(image)
let data = try? cache.object(forKey: key) {
completion?(data)
} else {
if let completion = completion, let request = requests[url] {
request.callbacks.append(completion)
@ -53,7 +53,7 @@ class ImageCache {
}
}
func get(_ url: URL) -> UIImage? {
func get(_ url: URL) -> Data? {
return try? cache.object(forKey: url.absoluteString)
}
@ -64,9 +64,9 @@ class ImageCache {
class Request {
let url: URL
var task: URLSessionDataTask?
var callbacks: [(UIImage?) -> Void]
var callbacks: [(Data?) -> Void]
init(url: URL, completion: ((UIImage?) -> Void)?) {
init(url: URL, completion: ((Data?) -> Void)?) {
if let completion = completion {
self.callbacks = [completion]
} else {
@ -75,14 +75,14 @@ class ImageCache {
self.url = url
}
func run(cache: @escaping (UIImage) -> Void) {
func run(cache: @escaping (Data) -> Void) {
task = URLSession.shared.dataTask(with: url, completionHandler: { (data, response, error) in
guard error == nil, let data = data, let image = UIImage(data: data) else {
guard error == nil, let data = data else {
self.complete(with: nil)
return
}
cache(image)
self.complete(with: image)
cache(data)
self.complete(with: data)
})
task!.resume()
}
@ -92,8 +92,8 @@ class ImageCache {
complete(with: nil)
}
func complete(with image: UIImage?) {
callbacks.forEach { $0(image) }
func complete(with data: Data?) {
callbacks.forEach { $0(data) }
}
}

View File

@ -98,9 +98,10 @@ class ComposeViewController: UIViewController {
inReplyToAvatarImageView.layer.cornerRadius = Preferences.shared.avatarStyle.cornerRadius(for: inReplyToAvatarImageView)
inReplyToAvatarImageView.layer.masksToBounds = true
inReplyToAvatarImageView.image = nil
ImageCache.avatars.get(inReplyTo.account.avatar) { (image) in
ImageCache.avatars.get(inReplyTo.account.avatar) { (data) in
guard let data = data else { return }
DispatchQueue.main.async {
self.inReplyToAvatarImageView.image = image
self.inReplyToAvatarImageView.image = UIImage(data: data)
}
}
inReplyToLabel.text = "In reply to \(inReplyTo.account.realDisplayName)"

View File

@ -8,6 +8,7 @@
import UIKit
import Pachyderm
import WebKit
protocol AttachmentViewDelegate {
func showLargeAttachment(for attachmentView: AttachmentView)
@ -43,9 +44,27 @@ class AttachmentView: UIImageView {
}
func loadImage() {
ImageCache.attachments.get(attachment.url) { (image) in
DispatchQueue.main.async {
self.image = image
ImageCache.attachments.get(attachment.url) { (data) in
guard let data = data else {
// TODO: error icon
return
}
// TODO: better way of detecting gifs
if self.attachment.url.pathExtension == "gif" {
DispatchQueue.main.async {
// I hate this; this is awful
let webView = WKWebView(frame: self.bounds)
webView.isUserInteractionEnabled = false
webView.backgroundColor = .clear
webView.isOpaque = false
webView.load(data, mimeType: "image/gif", characterEncodingName: "utf-8", baseURL: self.attachment.url)
self.addSubview(webView)
}
} else {
DispatchQueue.main.async {
self.image = UIImage(data: data)
}
}
}
}

View File

@ -79,17 +79,19 @@ class ActionNotificationTableViewCell: UITableViewCell, PreferencesAdaptive {
usernameLabel.text = "@\(status.account.acct)"
opAvatarImageView.image = nil
opAvatarURL = status.account.avatar
ImageCache.avatars.get(status.account.avatar) { (image) in
ImageCache.avatars.get(status.account.avatar) { (data) in
guard let data = data else { return }
DispatchQueue.main.async {
self.opAvatarImageView.image = image
self.opAvatarImageView.image = UIImage(data: data)
self.opAvatarURL = nil
}
}
actionAvatarImageView.image = nil
actionAvatarURL = notification.account.avatar
ImageCache.avatars.get(notification.account.avatar) { (image) in
ImageCache.avatars.get(notification.account.avatar) { (data) in
guard let data = data else { return }
DispatchQueue.main.async {
self.actionAvatarImageView.image = image
self.actionAvatarImageView.image = UIImage(data: data)
self.actionAvatarURL = nil
}
}

View File

@ -51,9 +51,10 @@ class FollowNotificationTableViewCell: UITableViewCell, PreferencesAdaptive {
usernameLabel.text = "@\(account.acct)"
avatarImageView.image = nil
avatarURL = account.avatar
ImageCache.avatars.get(account.avatar) { (image) in
ImageCache.avatars.get(account.avatar) { (data) in
guard let data = data else { return }
DispatchQueue.main.async {
self.avatarImageView.image = image
self.avatarImageView.image = UIImage(data: data)
self.avatarURL = nil
}
}

View File

@ -61,15 +61,17 @@ class ProfileHeaderTableViewCell: UITableViewCell, PreferencesAdaptive {
avatarImageView.image = nil
avatarURL = account.avatar
ImageCache.avatars.get(account.avatar) { (image) in
ImageCache.avatars.get(account.avatar) { (data) in
guard let data = data else { return }
DispatchQueue.main.async {
self.avatarImageView.image = image
self.avatarImageView.image = UIImage(data: data)
self.avatarURL = nil
}
}
ImageCache.headers.get(account.header) { (image) in
ImageCache.headers.get(account.header) { (data) in
guard let data = data else { return }
DispatchQueue.main.async {
self.headerImageView.image = image
self.headerImageView.image = UIImage(data: data)
self.headerURL = nil
}
}

View File

@ -80,9 +80,10 @@ class ConversationMainStatusTableViewCell: UITableViewCell, PreferencesAdaptive
usernameLabel.text = "@\(account.acct)"
avatarImageView.image = nil
avatarURL = account.avatar
ImageCache.avatars.get(account.avatar) { (image) in
ImageCache.avatars.get(account.avatar) { (data) in
guard let data = data else { return }
DispatchQueue.main.async {
self.avatarImageView.image = image
self.avatarImageView.image = UIImage(data: data)
self.avatarURL = nil
}
}

View File

@ -97,10 +97,12 @@ class StatusTableViewCell: UITableViewCell, PreferencesAdaptive {
usernameLabel.text = "@\(account.acct)"
avatarImageView.image = nil
avatarURL = account.avatar
ImageCache.avatars.get(account.avatar) { (image) in
ImageCache.avatars.get(account.avatar) { (data) in
self.avatarURL = nil
guard let data = data else { return }
DispatchQueue.main.async {
self.avatarImageView.image = image
self.avatarURL = nil
self.avatarImageView.image = UIImage(data: data)
}
}
updateTimestamp()