// // AttachmentView.swift // Tusker // // Created by Shadowfacts on 8/31/18. // Copyright © 2018 Shadowfacts. All rights reserved. // import UIKit import Pachyderm import Gifu import AVFoundation protocol AttachmentViewDelegate: class { func showAttachmentsGallery(startingAt index: Int) } class AttachmentView: UIImageView, GIFAnimatable { weak var delegate: AttachmentViewDelegate? var playImageView: UIImageView? var attachment: Attachment! var index: Int! private var attachmentRequest: ImageCache.Request? var gifData: Data? public lazy var animator: Animator? = Animator(withDelegate: self) init(attachment: Attachment, index: Int) { super.init(image: nil) commonInit() self.attachment = attachment self.index = index loadAttachment() } deinit { attachmentRequest?.cancel() } required init?(coder aDecoder: NSCoder) { super.init(coder: aDecoder) commonInit() } func commonInit() { contentMode = .scaleAspectFill layer.masksToBounds = true isUserInteractionEnabled = true addGestureRecognizer(UITapGestureRecognizer(target: self, action: #selector(imagePressed))) } func loadAttachment() { guard AttachmentsContainerView.supportedAttachmentTypes.contains(attachment.kind) else { preconditionFailure("invalid attachment type") } switch attachment.kind { case .image: loadImage() case .video: loadVideo() case .audio: loadAudio() default: preconditionFailure("invalid attachment type") } } func loadImage() { attachmentRequest = ImageCache.attachments.get(attachment.url) { [weak self] (data) in guard let self = self, let data = data else { return } self.attachmentRequest = nil DispatchQueue.main.async { if self.attachment.url.pathExtension == "gif" { self.animate(withGIFData: data) self.gifData = data } else { self.image = UIImage(data: data) } } } } func loadVideo() { let attachmentURL = self.attachment.url DispatchQueue.global(qos: .userInitiated).async { let asset = AVURLAsset(url: attachmentURL) let generator = AVAssetImageGenerator(asset: asset) generator.appliesPreferredTrackTransform = true guard let image = try? generator.copyCGImage(at: CMTime(seconds: 0, preferredTimescale: 1), actualTime: nil) else { return } DispatchQueue.main.async { guard self.attachment.url == attachmentURL else { return } self.image = UIImage(cgImage: image) } } let playImageView = UIImageView(image: UIImage(systemName: "play.circle.fill")) playImageView.translatesAutoresizingMaskIntoConstraints = false addSubview(playImageView) NSLayoutConstraint.activate([ playImageView.widthAnchor.constraint(equalToConstant: 50), playImageView.heightAnchor.constraint(equalToConstant: 50), playImageView.centerXAnchor.constraint(equalTo: centerXAnchor), playImageView.centerYAnchor.constraint(equalTo: centerYAnchor), ]) } func loadAudio() { let label = UILabel() label.text = "Audio Only" let playImageView = UIImageView(image: UIImage(systemName: "play.circle.fill")) let stack = UIStackView(arrangedSubviews: [ label, playImageView ]) stack.translatesAutoresizingMaskIntoConstraints = false stack.axis = .vertical stack.spacing = 8 stack.alignment = .center addSubview(stack) NSLayoutConstraint.activate([ stack.centerXAnchor.constraint(equalTo: centerXAnchor), stack.centerYAnchor.constraint(equalTo: centerYAnchor), playImageView.widthAnchor.constraint(equalToConstant: 50), playImageView.heightAnchor.constraint(equalToConstant: 50), ]) } override func display(_ layer: CALayer) { updateImageIfNeeded() } @objc func imagePressed() { delegate?.showAttachmentsGallery(startingAt: index) } }