Compare commits

..

2 Commits

4 changed files with 111 additions and 67 deletions

View File

@ -8,6 +8,8 @@
<dict> <dict>
<key>NSExtensionActivationRule</key> <key>NSExtensionActivationRule</key>
<dict> <dict>
<key>NSExtensionActivationSupportsText</key>
<true/>
<key>NSExtensionActivationSupportsImageWithMaxCount</key> <key>NSExtensionActivationSupportsImageWithMaxCount</key>
<integer>4</integer> <integer>4</integer>
<key>NSExtensionActivationSupportsMovieWithMaxCount</key> <key>NSExtensionActivationSupportsMovieWithMaxCount</key>

View File

@ -71,36 +71,43 @@ class ShareViewController: UIViewController {
private func getDraftConfigurationFromExtensionContext() async -> (String, [DraftAttachment]) { private func getDraftConfigurationFromExtensionContext() async -> (String, [DraftAttachment]) {
guard let extensionContext, guard let extensionContext,
let inputItem = (extensionContext.inputItems as? [NSExtensionItem])?.first, let inputItem = (extensionContext.inputItems as? [NSExtensionItem])?.first else {
let itemProvider = inputItem.attachments?.first else {
return ("", []) return ("", [])
} }
if let url: NSURL = await getObject(from: itemProvider) { var text: String = ""
if let title = inputItem.attributedTitle ?? inputItem.attributedContentText { var url: URL?
return ("\n\n\(title.string)\n\(url.absoluteString ?? "")", []) var attachments: [DraftAttachment] = []
} else {
return ("\n\n\(url.absoluteString ?? "")", []) for itemProvider in inputItem.attachments ?? [] {
if let attached: NSURL = await getObject(from: itemProvider) {
if url == nil {
url = attached as URL
} }
} else if let text: NSString = await getObject(from: itemProvider) { } else if let s: NSString = await getObject(from: itemProvider) {
return ("\n\n\(text)", []) if text.isEmpty {
} else if let attributedContent = inputItem.attributedContentText { text = s as String
return ("\n\n\(attributedContent.string)", []) }
} else { } else if let attachment: DraftAttachment = await getObject(from: itemProvider) {
let attachments = await withTaskGroup(of: DraftAttachment?.self, returning: [DraftAttachment].self) { group in attachments.append(attachment)
for provider in inputItem.attachments! {
group.addTask { @MainActor in
await self.getObject(from: provider)
} }
} }
return await group.reduce(into: [], { partialResult, result in if text.isEmpty,
if let result { let s = inputItem.attributedTitle ?? inputItem.attributedContentText {
partialResult.append(result) text = s.string
} }
})
if let url {
if !text.isEmpty {
text += "\n"
} }
return ("", attachments) text += url.absoluteString
} }
if !text.isEmpty {
text = "\n\n\(text)"
}
return (text, attachments)
} }
private func getObject<T: NSItemProviderReading>(from itemProvider: NSItemProvider) async -> T? { private func getObject<T: NSItemProviderReading>(from itemProvider: NSItemProvider) async -> T? {

View File

@ -9,7 +9,7 @@
import UIKit import UIKit
struct ImageGrayscalifier { struct ImageGrayscalifier {
static let queue = DispatchQueue(label: "ImageGrayscalifier", qos: .default) static let queue = DispatchQueue(label: "ImageGrayscalifier", qos: .userInitiated)
private static let context = CIContext() private static let context = CIContext()
private static let cache = NSCache<NSURL, UIImage>() private static let cache = NSCache<NSURL, UIImage>()
@ -24,6 +24,17 @@ struct ImageGrayscalifier {
} }
} }
static func convert(url: URL?, image: UIImage) -> UIImage? {
if let url,
let cached = cache.object(forKey: url as NSURL) {
return cached
}
guard let cgImage = image.cgImage else {
return nil
}
return doConvert(CIImage(cgImage: cgImage), url: url)
}
static func convert(url: URL?, data: Data) -> UIImage? { static func convert(url: URL?, data: Data) -> UIImage? {
if let url = url, if let url = url,
let cached = cache.object(forKey: url as NSURL) { let cached = cache.object(forKey: url as NSURL) {

View File

@ -75,10 +75,8 @@ class AttachmentView: GIFImageView {
gifPlaybackModeChanged() gifPlaybackModeChanged()
if isGrayscale != Preferences.shared.grayscaleImages { if isGrayscale != Preferences.shared.grayscaleImages {
ImageGrayscalifier.queue.async {
self.displayImage() self.displayImage()
} }
}
if getBadges().isEmpty != Preferences.shared.showAttachmentBadges { if getBadges().isEmpty != Preferences.shared.showAttachmentBadges {
createBadgesView(getBadges()) createBadgesView(getBadges())
@ -189,53 +187,59 @@ class AttachmentView: GIFImageView {
func loadImage() { func loadImage() {
let attachmentURL = attachment.url let attachmentURL = attachment.url
attachmentRequest = ImageCache.attachments.get(attachmentURL) { [weak self] (data, _) in attachmentRequest = ImageCache.attachments.get(attachmentURL) { [weak self] (data, image) in
guard let self = self, let data = data else { return } guard let self = self,
self.attachment.url == attachmentURL else {
return
}
DispatchQueue.main.async { DispatchQueue.main.async {
self.attachmentRequest = nil self.attachmentRequest = nil
}
if self.attachment.url.pathExtension == "gif" { if attachmentURL.pathExtension == "gif",
self.source = .gifData(attachmentURL, data) let data {
let controller = GIFController(gifData: data) self.source = .gifData(attachmentURL, data, image)
DispatchQueue.main.async {
controller.attach(to: self)
if self.autoplayGifs { if self.autoplayGifs {
let controller = GIFController(gifData: data)
controller.attach(to: self)
controller.startAnimating() controller.startAnimating()
}
}
if !self.autoplayGifs {
self.displayImage()
}
} else { } else {
self.source = .imageData(attachmentURL, data)
self.displayImage() self.displayImage()
} }
} else if let image {
self.source = .image(attachmentURL, image)
self.displayImage()
}
}
} }
} }
func loadVideo() { func loadVideo() {
if let previewURL = self.attachment.previewURL { if let previewURL = self.attachment.previewURL {
attachmentRequest = ImageCache.attachments.get(previewURL, completion: { [weak self] (data, _ )in attachmentRequest = ImageCache.attachments.get(previewURL, completion: { [weak self] (_, image) in
guard let self = self, let data = data else { return } guard let self, let image else { return }
DispatchQueue.main.async { DispatchQueue.main.async {
self.attachmentRequest = nil self.attachmentRequest = nil
self.source = .imageData(previewURL, data) self.source = .image(previewURL, image)
self.displayImage() self.displayImage()
} }
}) })
} else { } else {
let attachmentURL = self.attachment.url let attachmentURL = self.attachment.url
AttachmentView.queue.async { AttachmentView.queue.async { [weak self] in
let asset = AVURLAsset(url: attachmentURL) let asset = AVURLAsset(url: attachmentURL)
let generator = AVAssetImageGenerator(asset: asset) let generator = AVAssetImageGenerator(asset: asset)
generator.appliesPreferredTrackTransform = true generator.appliesPreferredTrackTransform = true
guard let image = try? generator.copyCGImage(at: .zero, actualTime: nil) else { return } guard let image = try? generator.copyCGImage(at: .zero, actualTime: nil) else { return }
self.source = .cgImage(attachmentURL, image) UIImage(cgImage: image).prepareForDisplay { [weak self] image in
DispatchQueue.main.async { [weak self] in
guard let self, let image else { return }
self.source = .image(attachmentURL, image)
self.displayImage() self.displayImage()
} }
} }
}
}
let playImageView = UIImageView(image: UIImage(systemName: "play.circle.fill")) let playImageView = UIImageView(image: UIImage(systemName: "play.circle.fill"))
playImageView.translatesAutoresizingMaskIntoConstraints = false playImageView.translatesAutoresizingMaskIntoConstraints = false
@ -276,9 +280,11 @@ class AttachmentView: GIFImageView {
let generator = AVAssetImageGenerator(asset: asset) let generator = AVAssetImageGenerator(asset: asset)
generator.appliesPreferredTrackTransform = true generator.appliesPreferredTrackTransform = true
guard let image = try? generator.copyCGImage(at: .zero, actualTime: nil) else { return } guard let image = try? generator.copyCGImage(at: .zero, actualTime: nil) else { return }
DispatchQueue.main.async {
self.source = .cgImage(attachmentURL, image) self.source = .cgImage(attachmentURL, image)
self.displayImage() self.displayImage()
} }
}
let gifvView = GifvAttachmentView(asset: asset, gravity: .resizeAspectFill) let gifvView = GifvAttachmentView(asset: asset, gravity: .resizeAspectFill)
self.gifvView = gifvView self.gifvView = gifvView
@ -295,33 +301,51 @@ class AttachmentView: GIFImageView {
]) ])
} }
@MainActor
private func displayImage() { private func displayImage() {
isGrayscale = Preferences.shared.grayscaleImages isGrayscale = Preferences.shared.grayscaleImages
let image: UIImage?
switch source { switch source {
case nil: case nil:
image = nil self.image = nil
case let .imageData(url, data), let .gifData(url, data): case let .image(url, sourceImage):
if isGrayscale { if isGrayscale {
image = ImageGrayscalifier.convert(url: url, data: data) ImageGrayscalifier.queue.async { [weak self] in
let grayscale = ImageGrayscalifier.convert(url: url, image: sourceImage)
DispatchQueue.main.async { [weak self] in
self?.image = grayscale
}
}
} else { } else {
image = UIImage(data: data) self.image = sourceImage
}
case let .gifData(url, _, sourceImage):
if isGrayscale,
let sourceImage {
ImageGrayscalifier.queue.async { [weak self] in
let grayscale = ImageGrayscalifier.convert(url: url, image: sourceImage)
DispatchQueue.main.async { [weak self] in
self?.image = grayscale
}
}
} else {
self.image = sourceImage
} }
case let .cgImage(url, cgImage): case let .cgImage(url, cgImage):
if isGrayscale { if isGrayscale {
image = ImageGrayscalifier.convert(url: url, cgImage: cgImage) ImageGrayscalifier.queue.async { [weak self] in
let grayscale = ImageGrayscalifier.convert(url: url, cgImage: cgImage)
DispatchQueue.main.async { [weak self] in
self?.image = grayscale
}
}
} else { } else {
image = UIImage(cgImage: cgImage) image = UIImage(cgImage: cgImage)
} }
} }
DispatchQueue.main.async {
self.image = image
}
} }
private func createBadgesView(_ badges: Badges) { private func createBadgesView(_ badges: Badges) {
@ -409,8 +433,8 @@ class AttachmentView: GIFImageView {
fileprivate extension AttachmentView { fileprivate extension AttachmentView {
enum Source { enum Source {
case imageData(URL, Data) case image(URL, UIImage)
case gifData(URL, Data) case gifData(URL, Data, UIImage?)
case cgImage(URL, CGImage) case cgImage(URL, CGImage)
} }