mirror of
https://github.com/Swiftgram/Telegram-iOS.git
synced 2025-06-16 05:55:20 +00:00
286 lines
16 KiB
Swift
286 lines
16 KiB
Swift
import Foundation
|
|
import AsyncDisplayKit
|
|
import Postbox
|
|
import SwiftSignalKit
|
|
import Display
|
|
import TelegramCore
|
|
|
|
private struct FetchControls {
|
|
let fetch: () -> Void
|
|
let cancel: () -> Void
|
|
}
|
|
|
|
final class ChatMessageInteractiveMediaNode: ASTransformNode {
|
|
private let imageNode: TransformImageNode
|
|
private var progressNode: RadialProgressNode?
|
|
private var tapRecognizer: UITapGestureRecognizer?
|
|
|
|
private var account: Account?
|
|
private var messageIdAndFlags: (MessageId, MessageFlags)?
|
|
private var media: Media?
|
|
|
|
private let statusDisposable = MetaDisposable()
|
|
private let fetchControls = Atomic<FetchControls?>(value: nil)
|
|
private var fetchStatus: MediaResourceStatus?
|
|
private let fetchDisposable = MetaDisposable()
|
|
|
|
var activateLocalContent: () -> Void = { }
|
|
|
|
init() {
|
|
self.imageNode = TransformImageNode()
|
|
|
|
super.init(layerBacked: false)
|
|
|
|
self.imageNode.displaysAsynchronously = false
|
|
self.addSubnode(self.imageNode)
|
|
}
|
|
|
|
deinit {
|
|
self.statusDisposable.dispose()
|
|
self.fetchDisposable.dispose()
|
|
}
|
|
|
|
override func didLoad() {
|
|
let tapRecognizer = UITapGestureRecognizer(target: self, action: #selector(self.imageTap(_:)))
|
|
self.imageNode.view.addGestureRecognizer(tapRecognizer)
|
|
self.tapRecognizer = tapRecognizer
|
|
}
|
|
|
|
@objc func progressPressed() {
|
|
if let fetchStatus = self.fetchStatus {
|
|
switch fetchStatus {
|
|
case .Fetching:
|
|
if let account = self.account, let (messageId, flags) = self.messageIdAndFlags, flags.contains(.Unsent) && !flags.contains(.Failed) {
|
|
account.postbox.modify({ modifier -> Void in
|
|
modifier.deleteMessages([messageId])
|
|
}).start()
|
|
}
|
|
if let cancel = self.fetchControls.with({ return $0?.cancel }) {
|
|
cancel()
|
|
}
|
|
case .Remote:
|
|
if let fetch = self.fetchControls.with({ return $0?.fetch }) {
|
|
fetch()
|
|
}
|
|
case .Local:
|
|
break
|
|
}
|
|
}
|
|
}
|
|
|
|
@objc func imageTap(_ recognizer: UITapGestureRecognizer) {
|
|
if case .ended = recognizer.state {
|
|
if let file = media as? TelegramMediaFile, (file.isVideo || file.isAnimated || file.mimeType.hasPrefix("video/")) {
|
|
self.activateLocalContent()
|
|
} else {
|
|
if let fetchStatus = self.fetchStatus, case .Local = fetchStatus {
|
|
self.activateLocalContent()
|
|
} else {
|
|
self.progressPressed()
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
func asyncLayout() -> (_ account: Account, _ message: Message, _ media: Media, _ corners: ImageCorners, _ automaticDownload: Bool, _ constrainedSize: CGSize, _ layoutConstants: ChatMessageItemLayoutConstants) -> (CGFloat, (CGSize) -> (CGFloat, (CGFloat) -> (CGSize, () -> Void))) {
|
|
let currentMessageIdAndFlags = self.messageIdAndFlags
|
|
let currentMedia = self.media
|
|
let imageLayout = self.imageNode.asyncLayout()
|
|
|
|
return { account, message, media, corners, automaticDownload, constrainedSize, layoutConstants in
|
|
var nativeSize: CGSize
|
|
|
|
if let image = media as? TelegramMediaImage, let dimensions = largestImageRepresentation(image.representations)?.dimensions {
|
|
nativeSize = CGSize(width: floor(dimensions.width * 0.5), height: floor(dimensions.height * 0.5)).fitted(constrainedSize)
|
|
} else if let file = media as? TelegramMediaFile, let dimensions = file.dimensions {
|
|
nativeSize = CGSize(width: floor(dimensions.width * 0.5), height: floor(dimensions.height * 0.5)).fitted(constrainedSize)
|
|
} else {
|
|
nativeSize = CGSize(width: 54.0, height: 54.0)
|
|
}
|
|
|
|
return (layoutConstants.image.maxDimensions.width, { constrainedSize in
|
|
return (min(layoutConstants.image.maxDimensions.width, nativeSize.width), { boundingWidth in
|
|
let drawingSize = nativeSize.fittedToWidthOrSmaller(boundingWidth)
|
|
let boundingSize = CGSize(width: max(boundingWidth, drawingSize.width), height: drawingSize.height).cropped(layoutConstants.image.maxDimensions)
|
|
|
|
var updateImageSignal: Signal<(TransformImageArguments) -> DrawingContext?, NoError>?
|
|
var updatedStatusSignal: Signal<MediaResourceStatus, NoError>?
|
|
var updatedFetchControls: FetchControls?
|
|
|
|
var mediaUpdated = false
|
|
if let currentMedia = currentMedia {
|
|
mediaUpdated = !media.isEqual(currentMedia)
|
|
} else {
|
|
mediaUpdated = true
|
|
}
|
|
|
|
var statusUpdated = mediaUpdated
|
|
if currentMessageIdAndFlags?.0 != message.id || currentMessageIdAndFlags?.1 != message.flags {
|
|
statusUpdated = true
|
|
}
|
|
|
|
if mediaUpdated {
|
|
if let image = media as? TelegramMediaImage {
|
|
updateImageSignal = chatMessagePhoto(account: account, photo: image)
|
|
|
|
updatedFetchControls = FetchControls(fetch: { [weak self] in
|
|
if let strongSelf = self {
|
|
strongSelf.fetchDisposable.set(chatMessagePhotoInteractiveFetched(account: account, photo: image).start())
|
|
}
|
|
}, cancel: {
|
|
chatMessagePhotoCancelInteractiveFetch(account: account, photo: image)
|
|
})
|
|
} else if let file = media as? TelegramMediaFile {
|
|
updateImageSignal = chatMessageVideo(account: account, video: file)
|
|
updatedFetchControls = FetchControls(fetch: { [weak self] in
|
|
if let strongSelf = self {
|
|
strongSelf.fetchDisposable.set(chatMessageFileInteractiveFetched(account: account, file: file).start())
|
|
}
|
|
}, cancel: {
|
|
chatMessageFileCancelInteractiveFetch(account: account, file: file)
|
|
})
|
|
}
|
|
}
|
|
|
|
if statusUpdated {
|
|
if let image = media as? TelegramMediaImage {
|
|
if message.flags.contains(.Unsent) && !message.flags.contains(.Failed) {
|
|
updatedStatusSignal = combineLatest(chatMessagePhotoStatus(account: account, photo: image), account.pendingMessageManager.pendingMessageStatus(message.id))
|
|
|> map { resourceStatus, pendingStatus -> MediaResourceStatus in
|
|
if let pendingStatus = pendingStatus {
|
|
return .Fetching(progress: pendingStatus.progress)
|
|
} else {
|
|
return resourceStatus
|
|
}
|
|
}
|
|
} else {
|
|
updatedStatusSignal = chatMessagePhotoStatus(account: account, photo: image)
|
|
}
|
|
} else if let file = media as? TelegramMediaFile {
|
|
updatedStatusSignal = combineLatest(chatMessageFileStatus(account: account, file: file), account.pendingMessageManager.pendingMessageStatus(message.id))
|
|
|> map { resourceStatus, pendingStatus -> MediaResourceStatus in
|
|
if let pendingStatus = pendingStatus {
|
|
return .Fetching(progress: pendingStatus.progress)
|
|
} else {
|
|
return resourceStatus
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
let arguments = TransformImageArguments(corners: corners, imageSize: drawingSize, boundingSize: boundingSize, intrinsicInsets: UIEdgeInsets())
|
|
|
|
let imageFrame = CGRect(origin: CGPoint(x: -arguments.insets.left, y: -arguments.insets.top), size: arguments.drawingSize)
|
|
|
|
let imageApply = imageLayout(arguments)
|
|
|
|
return (boundingSize, { [weak self] in
|
|
if let strongSelf = self {
|
|
strongSelf.account = account
|
|
strongSelf.messageIdAndFlags = (message.id, message.flags)
|
|
strongSelf.media = media
|
|
strongSelf.imageNode.frame = imageFrame
|
|
strongSelf.progressNode?.position = CGPoint(x: imageFrame.midX, y: imageFrame.midY)
|
|
|
|
if let updateImageSignal = updateImageSignal {
|
|
strongSelf.imageNode.setSignal(account: account, signal: updateImageSignal)
|
|
}
|
|
|
|
if let updatedStatusSignal = updatedStatusSignal {
|
|
strongSelf.statusDisposable.set((updatedStatusSignal |> deliverOnMainQueue).start(next: { [weak strongSelf] status in
|
|
displayLinkDispatcher.dispatch {
|
|
if let strongSelf = strongSelf {
|
|
strongSelf.fetchStatus = status
|
|
|
|
if let file = media as? TelegramMediaFile, (file.isVideo || file.mimeType.hasPrefix("video/")) {
|
|
if let progressNode = strongSelf.progressNode {
|
|
progressNode.removeFromSupernode()
|
|
strongSelf.progressNode = nil
|
|
}
|
|
} else {
|
|
if case .Local = status {
|
|
if let progressNode = strongSelf.progressNode {
|
|
progressNode.removeFromSupernode()
|
|
strongSelf.progressNode = nil
|
|
}
|
|
} else {
|
|
if strongSelf.progressNode == nil {
|
|
let progressNode = RadialProgressNode()
|
|
progressNode.frame = CGRect(origin: CGPoint(), size: CGSize(width: 50.0, height: 50.0))
|
|
progressNode.position = strongSelf.imageNode.position
|
|
strongSelf.progressNode = progressNode
|
|
strongSelf.addSubnode(progressNode)
|
|
}
|
|
}
|
|
|
|
switch status {
|
|
case let .Fetching(progress):
|
|
strongSelf.progressNode?.state = .Fetching(progress: progress)
|
|
case .Local:
|
|
var state: RadialProgressState = .None
|
|
if let file = media as? TelegramMediaFile {
|
|
if file.isVideo {
|
|
state = .Play
|
|
}
|
|
}
|
|
strongSelf.progressNode?.state = state
|
|
case .Remote:
|
|
strongSelf.progressNode?.state = .Remote
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}))
|
|
}
|
|
|
|
if let updatedFetchControls = updatedFetchControls {
|
|
let _ = strongSelf.fetchControls.swap(updatedFetchControls)
|
|
if automaticDownload {
|
|
if let image = media as? TelegramMediaImage {
|
|
strongSelf.fetchDisposable.set(chatMessagePhotoInteractiveFetched(account: account, photo: image).start())
|
|
}
|
|
}
|
|
}
|
|
|
|
imageApply()
|
|
}
|
|
})
|
|
})
|
|
})
|
|
}
|
|
}
|
|
|
|
static func asyncLayout(_ node: ChatMessageInteractiveMediaNode?) -> (_ account: Account, _ message: Message, _ media: Media, _ corners: ImageCorners, _ automaticDownload: Bool, _ constrainedSize: CGSize, _ layoutConstants: ChatMessageItemLayoutConstants) -> (CGFloat, (CGSize) -> (CGFloat, (CGFloat) -> (CGSize, () -> ChatMessageInteractiveMediaNode))) {
|
|
let currentAsyncLayout = node?.asyncLayout()
|
|
|
|
return { account, message, media, corners, automaticDownload, constrainedSize, layoutConstants in
|
|
var imageNode: ChatMessageInteractiveMediaNode
|
|
var imageLayout: (_ account: Account, _ message: Message, _ media: Media, _ corners: ImageCorners, _ automaticDownload: Bool, _ constrainedSize: CGSize, _ layoutConstants: ChatMessageItemLayoutConstants) -> (CGFloat, (CGSize) -> (CGFloat, (CGFloat) -> (CGSize, () -> Void)))
|
|
|
|
if let node = node, let currentAsyncLayout = currentAsyncLayout {
|
|
imageNode = node
|
|
imageLayout = currentAsyncLayout
|
|
} else {
|
|
imageNode = ChatMessageInteractiveMediaNode()
|
|
imageLayout = imageNode.asyncLayout()
|
|
}
|
|
|
|
let (initialWidth, continueLayout) = imageLayout(account, message, media, corners, automaticDownload, constrainedSize, layoutConstants)
|
|
|
|
return (initialWidth, { constrainedSize in
|
|
let (finalWidth, finalLayout) = continueLayout(constrainedSize)
|
|
|
|
return (finalWidth, { boundingWidth in
|
|
let (finalSize, apply) = finalLayout(boundingWidth)
|
|
|
|
return (finalSize, {
|
|
apply()
|
|
return imageNode
|
|
})
|
|
})
|
|
})
|
|
}
|
|
}
|
|
}
|
|
|