mirror of
https://github.com/Swiftgram/Telegram-iOS.git
synced 2025-06-16 05:55:20 +00:00
716 lines
36 KiB
Swift
716 lines
36 KiB
Swift
import Foundation
|
|
import Postbox
|
|
import SwiftSignalKit
|
|
|
|
|
|
extension MediaResourceReference {
|
|
var apiFileReference: Data? {
|
|
if let resource = self.resource as? TelegramCloudMediaResourceWithFileReference {
|
|
return resource.fileReference
|
|
} else {
|
|
return nil
|
|
}
|
|
}
|
|
}
|
|
|
|
final class TelegramCloudMediaResourceFetchInfo: MediaResourceFetchInfo {
|
|
let reference: MediaResourceReference
|
|
let preferBackgroundReferenceRevalidation: Bool
|
|
let continueInBackground: Bool
|
|
|
|
init(reference: MediaResourceReference, preferBackgroundReferenceRevalidation: Bool, continueInBackground: Bool) {
|
|
self.reference = reference
|
|
self.preferBackgroundReferenceRevalidation = preferBackgroundReferenceRevalidation
|
|
self.continueInBackground = continueInBackground
|
|
}
|
|
}
|
|
|
|
public func fetchedMediaResource(mediaBox: MediaBox, reference: MediaResourceReference, range: (Range<Int>, MediaBoxFetchPriority)? = nil, statsCategory: MediaResourceStatsCategory = .generic, reportResultStatus: Bool = false, preferBackgroundReferenceRevalidation: Bool = false, continueInBackground: Bool = false) -> Signal<FetchResourceSourceType, FetchResourceError> {
|
|
return fetchedMediaResource(mediaBox: mediaBox, reference: reference, ranges: range.flatMap({ [$0] }), statsCategory: statsCategory, reportResultStatus: reportResultStatus, preferBackgroundReferenceRevalidation: preferBackgroundReferenceRevalidation, continueInBackground: continueInBackground)
|
|
}
|
|
|
|
public func fetchedMediaResource(mediaBox: MediaBox, reference: MediaResourceReference, ranges: [(Range<Int>, MediaBoxFetchPriority)]?, statsCategory: MediaResourceStatsCategory = .generic, reportResultStatus: Bool = false, preferBackgroundReferenceRevalidation: Bool = false, continueInBackground: Bool = false) -> Signal<FetchResourceSourceType, FetchResourceError> {
|
|
var isRandomAccessAllowed = true
|
|
switch reference {
|
|
case let .media(media, _):
|
|
if let file = media.media as? TelegramMediaFile {
|
|
if file.fileId.namespace == Namespaces.Media.CloudSecretFile {
|
|
isRandomAccessAllowed = false
|
|
}
|
|
}
|
|
default:
|
|
break
|
|
}
|
|
|
|
if let ranges = ranges {
|
|
let signals = ranges.map { (range, priority) -> Signal<Void, FetchResourceError> in
|
|
return mediaBox.fetchedResourceData(reference.resource, in: range, priority: priority, parameters: MediaResourceFetchParameters(tag: TelegramMediaResourceFetchTag(statsCategory: statsCategory), info: TelegramCloudMediaResourceFetchInfo(reference: reference, preferBackgroundReferenceRevalidation: preferBackgroundReferenceRevalidation, continueInBackground: continueInBackground), isRandomAccessAllowed: isRandomAccessAllowed))
|
|
}
|
|
return combineLatest(signals)
|
|
|> ignoreValues
|
|
|> map { _ -> FetchResourceSourceType in .local }
|
|
|> then(.single(.local))
|
|
} else {
|
|
return mediaBox.fetchedResource(reference.resource, parameters: MediaResourceFetchParameters(tag: TelegramMediaResourceFetchTag(statsCategory: statsCategory), info: TelegramCloudMediaResourceFetchInfo(reference: reference, preferBackgroundReferenceRevalidation: preferBackgroundReferenceRevalidation, continueInBackground: continueInBackground), isRandomAccessAllowed: isRandomAccessAllowed), implNext: reportResultStatus)
|
|
}
|
|
}
|
|
|
|
enum RevalidateMediaReferenceError {
|
|
case generic
|
|
}
|
|
|
|
public func stickerPackFileReference(_ file: TelegramMediaFile) -> FileMediaReference {
|
|
for attribute in file.attributes {
|
|
if case let .Sticker(sticker) = attribute, let stickerPack = sticker.packReference {
|
|
return .stickerPack(stickerPack: stickerPack, media: file)
|
|
}
|
|
}
|
|
return .standalone(media: file)
|
|
}
|
|
|
|
private func areResourcesEqual(_ lhs: MediaResource, _ rhs: MediaResource) -> Bool {
|
|
if let lhsResource = lhs as? CloudDocumentMediaResource, let rhsResource = rhs as? CloudDocumentMediaResource {
|
|
if lhsResource.fileId == rhsResource.fileId {
|
|
return true
|
|
}
|
|
} else if let lhsResource = lhs as? CloudDocumentSizeMediaResource, let rhsResource = rhs as? CloudDocumentSizeMediaResource {
|
|
if lhsResource.documentId == rhsResource.documentId && lhsResource.sizeSpec == rhsResource.sizeSpec {
|
|
return true
|
|
}
|
|
}
|
|
return lhs.id.isEqual(to: rhs.id)
|
|
}
|
|
|
|
private func findMediaResource(media: Media, previousMedia: Media?, resource: MediaResource) -> TelegramMediaResource? {
|
|
if let image = media as? TelegramMediaImage {
|
|
for representation in image.representations {
|
|
if let updatedResource = representation.resource as? CloudPhotoSizeMediaResource, let previousResource = resource as? CloudPhotoSizeMediaResource {
|
|
if updatedResource.photoId == previousResource.photoId && updatedResource.sizeSpec == previousResource.sizeSpec {
|
|
return representation.resource
|
|
}
|
|
}
|
|
if representation.resource.id.isEqual(to: resource.id) {
|
|
return representation.resource
|
|
}
|
|
}
|
|
for representation in image.videoRepresentations {
|
|
if representation.resource.id.isEqual(to: resource.id) {
|
|
return representation.resource
|
|
}
|
|
}
|
|
} else if let file = media as? TelegramMediaFile {
|
|
if areResourcesEqual(file.resource, resource) {
|
|
return file.resource
|
|
} else {
|
|
for representation in file.previewRepresentations {
|
|
if areResourcesEqual(representation.resource, resource) {
|
|
return representation.resource
|
|
}
|
|
}
|
|
}
|
|
} else if let webPage = media as? TelegramMediaWebpage, case let .Loaded(content) = webPage.content {
|
|
if let image = content.image, let result = findMediaResource(media: image, previousMedia: previousMedia, resource: resource) {
|
|
return result
|
|
}
|
|
if let file = content.file, let result = findMediaResource(media: file, previousMedia: previousMedia, resource: resource) {
|
|
return result
|
|
}
|
|
if let instantPage = content.instantPage {
|
|
for pageMedia in instantPage.media.values {
|
|
if let result = findMediaResource(media: pageMedia, previousMedia: previousMedia, resource: resource) {
|
|
return result
|
|
}
|
|
}
|
|
}
|
|
} else if let game = media as? TelegramMediaGame {
|
|
if let image = game.image, let result = findMediaResource(media: image, previousMedia: previousMedia, resource: resource) {
|
|
return result
|
|
}
|
|
if let file = game.file, let result = findMediaResource(media: file, previousMedia: previousMedia, resource: resource) {
|
|
return result
|
|
}
|
|
} else if let action = media as? TelegramMediaAction {
|
|
switch action.action {
|
|
case let .photoUpdated(image):
|
|
if let image = image, let result = findMediaResource(media: image, previousMedia: previousMedia, resource: resource) {
|
|
return result
|
|
}
|
|
default:
|
|
break
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
private func findUpdatedMediaResource(media: Media, previousMedia: Media?, resource: MediaResource) -> TelegramMediaResource? {
|
|
if let foundResource = findMediaResource(media: media, previousMedia: previousMedia, resource: resource) {
|
|
return foundResource
|
|
} else {
|
|
return nil
|
|
}
|
|
}
|
|
|
|
private enum MediaReferenceRevalidationKey: Hashable {
|
|
case message(message: MessageReference)
|
|
case webPage(webPage: WebpageReference)
|
|
case stickerPack(stickerPack: StickerPackReference)
|
|
case savedGifs
|
|
case peer(peer: PeerReference)
|
|
case wallpaper(wallpaper: WallpaperReference)
|
|
case wallpapers
|
|
case themes
|
|
case peerAvatars(peer: PeerReference)
|
|
}
|
|
|
|
private final class MediaReferenceRevalidationItemContext {
|
|
let subscribers = Bag<(Any) -> Void>()
|
|
let disposable: Disposable
|
|
|
|
init(disposable: Disposable) {
|
|
self.disposable = disposable
|
|
}
|
|
|
|
deinit {
|
|
self.disposable.dispose()
|
|
}
|
|
|
|
var isEmpty: Bool {
|
|
return self.subscribers.isEmpty
|
|
}
|
|
|
|
func addSubscriber(_ f: @escaping (Any) -> Void) -> Int {
|
|
return self.subscribers.add(f)
|
|
}
|
|
|
|
func removeSubscriber(_ index: Int) {
|
|
self.subscribers.remove(index)
|
|
}
|
|
}
|
|
|
|
private struct MediaReferenceRevalidationKeyAndPlacement: Hashable {
|
|
let key: MediaReferenceRevalidationKey
|
|
let background: Bool
|
|
}
|
|
|
|
private final class MediaReferenceRevalidationContextImpl {
|
|
let queue: Queue
|
|
|
|
var itemContexts: [MediaReferenceRevalidationKeyAndPlacement: MediaReferenceRevalidationItemContext] = [:]
|
|
|
|
init(queue: Queue) {
|
|
self.queue = queue
|
|
}
|
|
|
|
func genericItem(key: MediaReferenceRevalidationKey, background: Bool, request: @escaping (@escaping (Any) -> Void, @escaping (RevalidateMediaReferenceError) -> Void) -> Disposable, _ f: @escaping (Any) -> Void) -> Disposable {
|
|
let queue = self.queue
|
|
|
|
let itemKey = MediaReferenceRevalidationKeyAndPlacement(key: key, background: background)
|
|
|
|
let context: MediaReferenceRevalidationItemContext
|
|
if let current = self.itemContexts[itemKey] {
|
|
context = current
|
|
} else {
|
|
let disposable = MetaDisposable()
|
|
context = MediaReferenceRevalidationItemContext(disposable: disposable)
|
|
self.itemContexts[itemKey] = context
|
|
disposable.set(request({ [weak self] result in
|
|
queue.async {
|
|
guard let strongSelf = self else {
|
|
return
|
|
}
|
|
if let current = strongSelf.itemContexts[itemKey], current === context {
|
|
strongSelf.itemContexts.removeValue(forKey: itemKey)
|
|
for subscriber in current.subscribers.copyItems() {
|
|
subscriber(result)
|
|
}
|
|
}
|
|
}
|
|
}, { _ in
|
|
}))
|
|
}
|
|
|
|
let index = context.addSubscriber(f)
|
|
|
|
return ActionDisposable { [weak self, weak context] in
|
|
queue.async {
|
|
guard let strongSelf = self else {
|
|
return
|
|
}
|
|
if let current = strongSelf.itemContexts[itemKey], current === context {
|
|
current.removeSubscriber(index)
|
|
if current.isEmpty {
|
|
current.disposable.dispose()
|
|
strongSelf.itemContexts.removeValue(forKey: itemKey)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
final class MediaReferenceRevalidationContext {
|
|
private let queue: Queue
|
|
private let impl: QueueLocalObject<MediaReferenceRevalidationContextImpl>
|
|
|
|
init() {
|
|
self.queue = Queue()
|
|
let queue = self.queue
|
|
self.impl = QueueLocalObject(queue: self.queue, generate: {
|
|
return MediaReferenceRevalidationContextImpl(queue: queue)
|
|
})
|
|
}
|
|
|
|
private func genericItem(key: MediaReferenceRevalidationKey, background: Bool, request: @escaping (@escaping (Any) -> Void, @escaping (RevalidateMediaReferenceError) -> Void) -> Disposable) -> Signal<Any, RevalidateMediaReferenceError> {
|
|
return Signal { subscriber in
|
|
let disposable = MetaDisposable()
|
|
self.impl.with { impl in
|
|
disposable.set(impl.genericItem(key: key, background: background, request: request, { result in
|
|
subscriber.putNext(result)
|
|
subscriber.putCompletion()
|
|
}))
|
|
}
|
|
return disposable
|
|
}
|
|
}
|
|
|
|
func message(postbox: Postbox, network: Network, background: Bool, message: MessageReference) -> Signal<Message, RevalidateMediaReferenceError> {
|
|
return self.genericItem(key: .message(message: message), background: background, request: { next, error in
|
|
let source: Signal<FetchMessageHistoryHoleSource, NoError>
|
|
if background {
|
|
source = network.background()
|
|
|> map(FetchMessageHistoryHoleSource.download)
|
|
} else {
|
|
source = .single(.network(network))
|
|
}
|
|
let signal = source
|
|
|> mapToSignal { source -> Signal<Message?, NoError> in
|
|
return fetchRemoteMessage(postbox: postbox, source: source, message: message)
|
|
}
|
|
return signal.start(next: { value in
|
|
if let value = value {
|
|
next(value)
|
|
} else {
|
|
error(.generic)
|
|
}
|
|
}, error: { _ in
|
|
error(.generic)
|
|
})
|
|
}) |> mapToSignal { next -> Signal<Message, RevalidateMediaReferenceError> in
|
|
if let next = next as? Message {
|
|
return .single(next)
|
|
} else {
|
|
return .fail(.generic)
|
|
}
|
|
}
|
|
}
|
|
|
|
func stickerPack(postbox: Postbox, network: Network, background: Bool, stickerPack: StickerPackReference) -> Signal<(StickerPackCollectionInfo, [ItemCollectionItem]), RevalidateMediaReferenceError> {
|
|
return self.genericItem(key: .stickerPack(stickerPack: stickerPack), background: background, request: { next, error in
|
|
return (updatedRemoteStickerPack(postbox: postbox, network: network, reference: stickerPack)
|
|
|> mapError { _ -> RevalidateMediaReferenceError in
|
|
return .generic
|
|
}).start(next: { value in
|
|
if let value = value {
|
|
next(value)
|
|
} else {
|
|
error(.generic)
|
|
}
|
|
}, error: { _ in
|
|
error(.generic)
|
|
})
|
|
}) |> mapToSignal { next -> Signal<(StickerPackCollectionInfo, [ItemCollectionItem]), RevalidateMediaReferenceError> in
|
|
if let next = next as? (StickerPackCollectionInfo, [ItemCollectionItem]) {
|
|
return .single(next)
|
|
} else {
|
|
return .fail(.generic)
|
|
}
|
|
}
|
|
}
|
|
|
|
func webPage(postbox: Postbox, network: Network, background: Bool, webPage: WebpageReference) -> Signal<TelegramMediaWebpage, RevalidateMediaReferenceError> {
|
|
return self.genericItem(key: .webPage(webPage: webPage), background: background, request: { next, error in
|
|
return (updatedRemoteWebpage(postbox: postbox, network: network, webPage: webPage)
|
|
|> mapError { _ -> RevalidateMediaReferenceError in
|
|
return .generic
|
|
}).start(next: { value in
|
|
if let value = value {
|
|
next(value)
|
|
} else {
|
|
error(.generic)
|
|
}
|
|
}, error: { _ in
|
|
error(.generic)
|
|
})
|
|
}) |> mapToSignal { next -> Signal<TelegramMediaWebpage, RevalidateMediaReferenceError> in
|
|
if let next = next as? TelegramMediaWebpage {
|
|
return .single(next)
|
|
} else {
|
|
return .fail(.generic)
|
|
}
|
|
}
|
|
}
|
|
|
|
func savedGifs(postbox: Postbox, network: Network, background: Bool) -> Signal<[TelegramMediaFile], RevalidateMediaReferenceError> {
|
|
return self.genericItem(key: .savedGifs, background: background, request: { next, error in
|
|
let loadRecentGifs: Signal<[TelegramMediaFile], NoError> = postbox.transaction { transaction -> [TelegramMediaFile] in
|
|
return transaction.getOrderedListItems(collectionId: Namespaces.OrderedItemList.CloudRecentGifs).compactMap({ item -> TelegramMediaFile? in
|
|
if let contents = item.contents as? RecentMediaItem, let file = contents.media as? TelegramMediaFile {
|
|
return file
|
|
}
|
|
return nil
|
|
})
|
|
}
|
|
return (managedRecentGifs(postbox: postbox, network: network, forceFetch: true)
|
|
|> mapToSignal { _ -> Signal<[TelegramMediaFile], NoError> in
|
|
return .complete()
|
|
}
|
|
|> then(loadRecentGifs)
|
|
|> castError(RevalidateMediaReferenceError.self)).start(next: { value in
|
|
next(value)
|
|
}, error: { _ in
|
|
error(.generic)
|
|
})
|
|
}) |> mapToSignal { next -> Signal<[TelegramMediaFile], RevalidateMediaReferenceError> in
|
|
if let next = next as? [TelegramMediaFile] {
|
|
return .single(next)
|
|
} else {
|
|
return .fail(.generic)
|
|
}
|
|
}
|
|
}
|
|
|
|
func peer(postbox: Postbox, network: Network, background: Bool, peer: PeerReference) -> Signal<Peer, RevalidateMediaReferenceError> {
|
|
return self.genericItem(key: .peer(peer: peer), background: background, request: { next, error in
|
|
return (_internal_updatedRemotePeer(postbox: postbox, network: network, peer: peer)
|
|
|> mapError { _ -> RevalidateMediaReferenceError in
|
|
return .generic
|
|
}).start(next: { value in
|
|
next(value)
|
|
}, error: { _ in
|
|
error(.generic)
|
|
})
|
|
}) |> mapToSignal { next -> Signal<Peer, RevalidateMediaReferenceError> in
|
|
if let next = next as? Peer {
|
|
return .single(next)
|
|
} else {
|
|
return .fail(.generic)
|
|
}
|
|
}
|
|
}
|
|
|
|
func wallpapers(postbox: Postbox, network: Network, background: Bool, wallpaper: WallpaperReference?) -> Signal<[TelegramWallpaper], RevalidateMediaReferenceError> {
|
|
let key: MediaReferenceRevalidationKey
|
|
if let wallpaper = wallpaper {
|
|
key = .wallpaper(wallpaper: wallpaper)
|
|
} else {
|
|
key = .wallpapers
|
|
}
|
|
return self.genericItem(key: key, background: background, request: { next, error in
|
|
let signal: Signal<[TelegramWallpaper]?, RevalidateMediaReferenceError>
|
|
if let wallpaper = wallpaper, case let .slug(slug) = wallpaper {
|
|
signal = getWallpaper(network: network, slug: slug)
|
|
|> mapError { _ -> RevalidateMediaReferenceError in
|
|
return .generic
|
|
}
|
|
|> map { [$0] }
|
|
} else {
|
|
signal = telegramWallpapers(postbox: postbox, network: network, forceUpdate: true)
|
|
|> last
|
|
|> mapError { _ -> RevalidateMediaReferenceError in
|
|
}
|
|
}
|
|
return (signal
|
|
).start(next: { value in
|
|
if let value = value {
|
|
next(value)
|
|
} else {
|
|
error(.generic)
|
|
}
|
|
}, error: { _ in
|
|
error(.generic)
|
|
})
|
|
}) |> mapToSignal { next -> Signal<[TelegramWallpaper], RevalidateMediaReferenceError> in
|
|
if let next = next as? [TelegramWallpaper] {
|
|
return .single(next)
|
|
} else {
|
|
return .fail(.generic)
|
|
}
|
|
}
|
|
}
|
|
|
|
func themes(postbox: Postbox, network: Network, background: Bool) -> Signal<[TelegramTheme], RevalidateMediaReferenceError> {
|
|
return self.genericItem(key: .themes, background: background, request: { next, error in
|
|
return (telegramThemes(postbox: postbox, network: network, accountManager: nil, forceUpdate: true)
|
|
|> take(1)
|
|
|> mapError { _ -> RevalidateMediaReferenceError in
|
|
return .generic
|
|
}).start(next: { value in
|
|
next(value)
|
|
}, error: { _ in
|
|
error(.generic)
|
|
})
|
|
}) |> mapToSignal { next -> Signal<[TelegramTheme], RevalidateMediaReferenceError> in
|
|
if let next = next as? [TelegramTheme] {
|
|
return .single(next)
|
|
} else {
|
|
return .fail(.generic)
|
|
}
|
|
}
|
|
}
|
|
|
|
func peerAvatars(postbox: Postbox, network: Network, background: Bool, peer: PeerReference) -> Signal<[TelegramPeerPhoto], RevalidateMediaReferenceError> {
|
|
return self.genericItem(key: .peerAvatars(peer: peer), background: background, request: { next, error in
|
|
return (_internal_requestPeerPhotos(postbox: postbox, network: network, peerId: peer.id)
|
|
|> mapError { _ -> RevalidateMediaReferenceError in
|
|
return .generic
|
|
}).start(next: { value in
|
|
next(value)
|
|
}, error: { _ in
|
|
error(.generic)
|
|
})
|
|
}) |> mapToSignal { next -> Signal<[TelegramPeerPhoto], RevalidateMediaReferenceError> in
|
|
if let next = next as? [TelegramPeerPhoto] {
|
|
return .single(next)
|
|
} else {
|
|
return .fail(.generic)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
struct RevalidatedMediaResource {
|
|
let updatedResource: TelegramMediaResource
|
|
let updatedReference: MediaResourceReference?
|
|
}
|
|
|
|
func revalidateMediaResourceReference(postbox: Postbox, network: Network, revalidationContext: MediaReferenceRevalidationContext, info: TelegramCloudMediaResourceFetchInfo, resource: MediaResource) -> Signal<RevalidatedMediaResource, RevalidateMediaReferenceError> {
|
|
var updatedReference = info.reference
|
|
if case let .media(media, resource) = updatedReference {
|
|
if case let .message(messageReference, mediaValue) = media {
|
|
if let file = mediaValue as? TelegramMediaFile {
|
|
if let partialReference = file.partialReference {
|
|
updatedReference = partialReference.mediaReference(media.media).resourceReference(resource)
|
|
}
|
|
|
|
var revalidateWithStickerpack = false
|
|
if file.isSticker {
|
|
if messageReference.isSecret == true {
|
|
revalidateWithStickerpack = true
|
|
} else if case .none = messageReference.content {
|
|
revalidateWithStickerpack = true
|
|
}
|
|
}
|
|
|
|
if revalidateWithStickerpack {
|
|
var stickerPackReference: StickerPackReference?
|
|
for attribute in file.attributes {
|
|
if case let .Sticker(sticker) = attribute {
|
|
if let packReference = sticker.packReference {
|
|
stickerPackReference = packReference
|
|
}
|
|
}
|
|
}
|
|
if let stickerPackReference = stickerPackReference {
|
|
updatedReference = .media(media: .stickerPack(stickerPack: stickerPackReference, media: mediaValue), resource: resource)
|
|
}
|
|
}
|
|
} else if let image = mediaValue as? TelegramMediaImage {
|
|
if let partialReference = image.partialReference {
|
|
updatedReference = partialReference.mediaReference(media.media).resourceReference(resource)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
switch updatedReference {
|
|
case let .media(media, _):
|
|
switch media {
|
|
case let .message(message, previousMedia):
|
|
return revalidationContext.message(postbox: postbox, network: network, background: info.preferBackgroundReferenceRevalidation, message: message)
|
|
|> mapToSignal { message -> Signal<RevalidatedMediaResource, RevalidateMediaReferenceError> in
|
|
for media in message.media {
|
|
if let updatedResource = findUpdatedMediaResource(media: media, previousMedia: previousMedia, resource: resource) {
|
|
return .single(RevalidatedMediaResource(updatedResource: updatedResource, updatedReference: nil))
|
|
}
|
|
}
|
|
return .fail(.generic)
|
|
}
|
|
case let .stickerPack(stickerPack, media):
|
|
return revalidationContext.stickerPack(postbox: postbox, network: network, background: info.preferBackgroundReferenceRevalidation, stickerPack: stickerPack)
|
|
|> mapToSignal { result -> Signal<RevalidatedMediaResource, RevalidateMediaReferenceError> in
|
|
for item in result.1 {
|
|
if let item = item as? StickerPackItem {
|
|
if media.id != nil && item.file.id == media.id {
|
|
if let updatedResource = findUpdatedMediaResource(media: item.file, previousMedia: media, resource: resource) {
|
|
return postbox.transaction { transaction -> RevalidatedMediaResource in
|
|
if let id = media.id {
|
|
var attributes = item.file.attributes
|
|
if !attributes.contains(where: { attribute in
|
|
if case .hintIsValidated = attribute {
|
|
return true
|
|
} else {
|
|
return false
|
|
}
|
|
}) {
|
|
attributes.append(.hintIsValidated)
|
|
}
|
|
let file = item.file.withUpdatedAttributes(attributes)
|
|
updateMessageMedia(transaction: transaction, id: id, media: file)
|
|
}
|
|
return RevalidatedMediaResource(updatedResource: updatedResource, updatedReference: nil)
|
|
}
|
|
|> castError(RevalidateMediaReferenceError.self)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
return .fail(.generic)
|
|
}
|
|
case let .webPage(webPage, previousMedia):
|
|
return revalidationContext.webPage(postbox: postbox, network: network, background: info.preferBackgroundReferenceRevalidation, webPage: webPage)
|
|
|> mapToSignal { result -> Signal<RevalidatedMediaResource, RevalidateMediaReferenceError> in
|
|
if let updatedResource = findUpdatedMediaResource(media: result, previousMedia: previousMedia, resource: resource) {
|
|
return .single(RevalidatedMediaResource(updatedResource: updatedResource, updatedReference: nil))
|
|
}
|
|
return .fail(.generic)
|
|
}
|
|
case let .savedGif(media):
|
|
return revalidationContext.savedGifs(postbox: postbox, network: network, background: info.preferBackgroundReferenceRevalidation)
|
|
|> mapToSignal { result -> Signal<RevalidatedMediaResource, RevalidateMediaReferenceError> in
|
|
for file in result {
|
|
if media.id != nil && file.id == media.id {
|
|
if let updatedResource = findUpdatedMediaResource(media: file, previousMedia: media, resource: resource) {
|
|
return .single(RevalidatedMediaResource(updatedResource: updatedResource, updatedReference: nil))
|
|
}
|
|
}
|
|
}
|
|
return .fail(.generic)
|
|
}
|
|
case let .avatarList(peer, media):
|
|
return revalidationContext.peerAvatars(postbox: postbox, network: network, background: info.preferBackgroundReferenceRevalidation, peer: peer)
|
|
|> mapToSignal { result -> Signal<RevalidatedMediaResource, RevalidateMediaReferenceError> in
|
|
for photo in result {
|
|
if let updatedResource = findUpdatedMediaResource(media: photo.image, previousMedia: media, resource: resource) {
|
|
return .single(RevalidatedMediaResource(updatedResource: updatedResource, updatedReference: nil))
|
|
}
|
|
}
|
|
return .fail(.generic)
|
|
}
|
|
case let .standalone(media):
|
|
if let file = media as? TelegramMediaFile {
|
|
for attribute in file.attributes {
|
|
if case let .Sticker(sticker) = attribute, let stickerPack = sticker.packReference {
|
|
return revalidationContext.stickerPack(postbox: postbox, network: network, background: info.preferBackgroundReferenceRevalidation, stickerPack: stickerPack)
|
|
|> mapToSignal { result -> Signal<RevalidatedMediaResource, RevalidateMediaReferenceError> in
|
|
for item in result.1 {
|
|
if let item = item as? StickerPackItem {
|
|
if media.id != nil && item.file.id == media.id {
|
|
if let updatedResource = findUpdatedMediaResource(media: item.file, previousMedia: media, resource: resource) {
|
|
return .single(RevalidatedMediaResource(updatedResource: updatedResource, updatedReference: nil))
|
|
}
|
|
}
|
|
}
|
|
}
|
|
return .fail(.generic)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
return .fail(.generic)
|
|
}
|
|
case let .avatar(peer, _):
|
|
return revalidationContext.peer(postbox: postbox, network: network, background: info.preferBackgroundReferenceRevalidation, peer: peer)
|
|
|> mapToSignal { updatedPeer -> Signal<RevalidatedMediaResource, RevalidateMediaReferenceError> in
|
|
for representation in updatedPeer.profileImageRepresentations {
|
|
if let updatedResource = representation.resource as? CloudPeerPhotoSizeMediaResource, let previousResource = resource as? CloudPeerPhotoSizeMediaResource {
|
|
if updatedResource.sizeSpec == previousResource.sizeSpec {
|
|
return .single(RevalidatedMediaResource(updatedResource: representation.resource, updatedReference: nil))
|
|
}
|
|
}
|
|
if representation.resource.id.isEqual(to: resource.id) {
|
|
return .single(RevalidatedMediaResource(updatedResource: representation.resource, updatedReference: nil))
|
|
}
|
|
}
|
|
return .fail(.generic)
|
|
}
|
|
case let .avatarList(peer, _):
|
|
return revalidationContext.peerAvatars(postbox: postbox, network: network, background: info.preferBackgroundReferenceRevalidation, peer: peer)
|
|
|> mapToSignal { updatedPeerAvatars -> Signal<RevalidatedMediaResource, RevalidateMediaReferenceError> in
|
|
for photo in updatedPeerAvatars {
|
|
if let updatedResource = findUpdatedMediaResource(media: photo.image, previousMedia: nil, resource: resource) {
|
|
return .single(RevalidatedMediaResource(updatedResource: updatedResource, updatedReference: nil))
|
|
}
|
|
}
|
|
return .fail(.generic)
|
|
}
|
|
case let .messageAuthorAvatar(message, _):
|
|
return revalidationContext.message(postbox: postbox, network: network, background: info.preferBackgroundReferenceRevalidation, message: message)
|
|
|> mapToSignal { updatedMessage -> Signal<RevalidatedMediaResource, RevalidateMediaReferenceError> in
|
|
guard let author = updatedMessage.author, let authorReference = PeerReference(author) else {
|
|
return .fail(.generic)
|
|
}
|
|
for representation in author.profileImageRepresentations {
|
|
if representation.resource.id.isEqual(to: resource.id) {
|
|
return .single(RevalidatedMediaResource(updatedResource: representation.resource, updatedReference: .avatar(peer: authorReference, resource: representation.resource)))
|
|
}
|
|
}
|
|
if let legacyResource = resource as? CloudFileMediaResource {
|
|
for representation in author.profileImageRepresentations {
|
|
if let updatedResource = representation.resource as? CloudPeerPhotoSizeMediaResource {
|
|
if updatedResource.localId == legacyResource.localId && updatedResource.volumeId == legacyResource.volumeId {
|
|
return .single(RevalidatedMediaResource(updatedResource: updatedResource, updatedReference: .avatar(peer: authorReference, resource: updatedResource)))
|
|
}
|
|
}
|
|
}
|
|
}
|
|
return .fail(.generic)
|
|
}
|
|
case let .wallpaper(wallpaper, _):
|
|
return revalidationContext.wallpapers(postbox: postbox, network: network, background: info.preferBackgroundReferenceRevalidation, wallpaper: wallpaper)
|
|
|> mapToSignal { wallpapers -> Signal<RevalidatedMediaResource, RevalidateMediaReferenceError> in
|
|
for wallpaper in wallpapers {
|
|
switch wallpaper {
|
|
case let .image(representations, _):
|
|
for representation in representations {
|
|
if representation.resource.id.isEqual(to: resource.id) {
|
|
return .single(RevalidatedMediaResource(updatedResource: representation.resource, updatedReference: nil))
|
|
}
|
|
}
|
|
case let .file(_, _, _, _, _, _, _, file, _):
|
|
if let updatedResource = findUpdatedMediaResource(media: file, previousMedia: nil, resource: resource) {
|
|
return .single(RevalidatedMediaResource(updatedResource: updatedResource, updatedReference: nil))
|
|
}
|
|
default:
|
|
break
|
|
}
|
|
}
|
|
return .fail(.generic)
|
|
}
|
|
case let .stickerPackThumbnail(packReference, resource):
|
|
return revalidationContext.stickerPack(postbox: postbox, network: network, background: info.preferBackgroundReferenceRevalidation, stickerPack: packReference)
|
|
|> mapToSignal { result -> Signal<RevalidatedMediaResource, RevalidateMediaReferenceError> in
|
|
if let thumbnail = result.0.thumbnail {
|
|
if thumbnail.resource.id.isEqual(to: resource.id) {
|
|
return .single(RevalidatedMediaResource(updatedResource: thumbnail.resource, updatedReference: nil))
|
|
}
|
|
if let _ = thumbnail.resource as? CloudStickerPackThumbnailMediaResource, let _ = resource as? CloudStickerPackThumbnailMediaResource {
|
|
return .single(RevalidatedMediaResource(updatedResource: thumbnail.resource, updatedReference: nil))
|
|
}
|
|
}
|
|
return .fail(.generic)
|
|
}
|
|
case let .theme(_, resource):
|
|
return revalidationContext.themes(postbox: postbox, network: network, background: info.preferBackgroundReferenceRevalidation)
|
|
|> mapToSignal { themes -> Signal<RevalidatedMediaResource, RevalidateMediaReferenceError> in
|
|
for theme in themes {
|
|
if let file = theme.file, file.resource.id.isEqual(to: resource.id) {
|
|
return .single(RevalidatedMediaResource(updatedResource: file.resource, updatedReference: nil))
|
|
}
|
|
}
|
|
return .fail(.generic)
|
|
}
|
|
case .standalone:
|
|
return .fail(.generic)
|
|
}
|
|
}
|