GLEGram-iOS/submodules/TelegramUI/Components/TelegramAccountAuxiliaryMethods/Sources/TelegramAccountAuxiliaryMethods.swift

import Foundation
import TelegramCore
import Postbox
import MediaResources
import PassportUI
import OpenInExternalAppUI
import MusicAlbumArtResources
import LocalMediaResources
import LocationResources
import ChatInterfaceState
import WallpaperResources
import AppBundle
import SwiftSignalKit
import ICloudResources
import FetchVideoMediaResource
import FetchAudioMediaResource
import Display
import UIKit

public func makeTelegramAccountAuxiliaryMethods(uploadInBackground: ((Postbox, MediaResource) -> Signal<String?, NoError>)?) -> AccountAuxiliaryMethods {
    return AccountAuxiliaryMethods(fetchResource: { postbox, resource, ranges, _ in
        if let resource = resource as? VideoLibraryMediaResource {
            return postbox.transaction { transaction -> Bool in
                var useModernPipeline = true
                let appConfig = currentAppConfiguration(transaction: transaction)
                if let data = appConfig.data, let _ = data["ios_killswitch_disable_modern_video_pipeline"] {
                    useModernPipeline = false
                }
                return useModernPipeline
            }
            |> castError(MediaResourceDataFetchError.self)
            |> mapToSignal { useModernPipeline -> Signal<MediaResourceDataFetchResult, MediaResourceDataFetchError> in
                return fetchVideoLibraryMediaResource(postbox: postbox, resource: resource, alwaysUseModernPipeline: useModernPipeline)
            }
        } else if let resource = resource as? LocalFileVideoMediaResource {
            return postbox.transaction { transaction -> Bool in
                var useModernPipeline = true
                let appConfig = currentAppConfiguration(transaction: transaction)
                if let data = appConfig.data, let _ = data["ios_killswitch_disable_modern_video_pipeline"] {
                    useModernPipeline = false
                }
                return useModernPipeline
            }
            |> castError(MediaResourceDataFetchError.self)
            |> mapToSignal { useModernPipeline -> Signal<MediaResourceDataFetchResult, MediaResourceDataFetchError> in
                return fetchLocalFileVideoMediaResource(postbox: postbox, resource: resource, alwaysUseModernPipeline: useModernPipeline)
            }
        } else if let resource = resource as? LocalFileAudioMediaResource {
            return fetchLocalFileAudioMediaResource(postbox: postbox, resource: resource)
        } else if let resource = resource as? LocalFileGifMediaResource {
            return fetchLocalFileGifMediaResource(resource: resource)
        } else if let photoLibraryResource = resource as? PhotoLibraryMediaResource {
            return postbox.transaction { transaction -> Bool in
                var useExif = true
                let appConfig = currentAppConfiguration(transaction: transaction)
                if let data = appConfig.data, let _ = data["ios_killswitch_disable_use_photo_exif"] {
                    useExif = false
                }
                return useExif
            }
            |> castError(MediaResourceDataFetchError.self)
            |> mapToSignal { useExif -> Signal<MediaResourceDataFetchResult, MediaResourceDataFetchError> in
                return fetchPhotoLibraryResource(localIdentifier: photoLibraryResource.localIdentifier, width: photoLibraryResource.width, height: photoLibraryResource.height, format: photoLibraryResource.format, quality: photoLibraryResource.quality, hd: photoLibraryResource.forceHd, useExif: useExif)
            }
        } else if let resource = resource as? ICloudFileResource {
            return fetchICloudFileResource(resource: resource)
        } else if let resource = resource as? SecureIdLocalImageResource {
            return fetchSecureIdLocalImageResource(postbox: postbox, resource: resource)
        } else if let resource = resource as? BundleResource {
            return Signal { subscriber in
                subscriber.putNext(.reset)
                if let data = try? Data(contentsOf: URL(fileURLWithPath: resource.path), options: .mappedRead) {
                    subscriber.putNext(.dataPart(resourceOffset: 0, data: data, range: 0 ..< Int64(data.count), complete: true))
                }
                return EmptyDisposable
            }
        } else if let wallpaperResource = resource as? WallpaperDataResource {
            let builtinWallpapers: [String] = [
                "fqv01SQemVIBAAAApND8LDRUhRU",
                "Ye7DfT2kCVIKAAAAhzXfrkdOjxs"
            ]
            if builtinWallpapers.contains(wallpaperResource.slug) {
                if let url = getAppBundle().url(forResource: wallpaperResource.slug, withExtension: "tgv") {
                    return Signal { subscriber in
                        subscriber.putNext(.reset)
                        if let data = try? Data(contentsOf: url, options: .mappedRead) {
                            subscriber.putNext(.dataPart(resourceOffset: 0, data: data, range: 0 ..< Int64(data.count), complete: true))
                        }
                        
                        return EmptyDisposable
                    }
                } else {
                    return nil
                }
            }
            return nil
        } else if let cloudDocumentMediaResource = resource as? CloudDocumentMediaResource {
            if cloudDocumentMediaResource.fileId == 5789658100176783156 {
                if let url = getAppBundle().url(forResource: "fqv01SQemVIBAAAApND8LDRUhRU", withExtension: "tgv") {
                    return Signal { subscriber in
                        subscriber.putNext(.reset)
                        if let data = try? Data(contentsOf: url, options: .mappedRead) {
                            subscriber.putNext(.dataPart(resourceOffset: 0, data: data, range: 0 ..< Int64(data.count), complete: true))
                        }
                        
                        return EmptyDisposable
                    }
                } else {
                    return nil
                }
            } else if cloudDocumentMediaResource.fileId == 5911315028815907420 {
                if let url = getAppBundle().url(forResource: "Ye7DfT2kCVIKAAAAhzXfrkdOjxs", withExtension: "tgv") {
                    return Signal { subscriber in
                        subscriber.putNext(.reset)
                        if let data = try? Data(contentsOf: url, options: .mappedRead) {
                            subscriber.putNext(.dataPart(resourceOffset: 0, data: data, range: 0 ..< Int64(data.count), complete: true))
                        }
                        
                        return EmptyDisposable
                    }
                } else {
                    return nil
                }
            }
        } else if let cloudDocumentSizeMediaResource = resource as? CloudDocumentSizeMediaResource {
            if cloudDocumentSizeMediaResource.documentId == 5789658100176783156 && cloudDocumentSizeMediaResource.sizeSpec == "m" {
                if let url = getAppBundle().url(forResource: "5789658100176783156-m", withExtension: "resource") {
                    return Signal { subscriber in
                        subscriber.putNext(.reset)
                        if let data = try? Data(contentsOf: url, options: .mappedRead) {
                            subscriber.putNext(.dataPart(resourceOffset: 0, data: data, range: 0 ..< Int64(data.count), complete: true))
                        }
                        
                        return EmptyDisposable
                    }
                } else {
                    return nil
                }
            }
            return nil
        }
        return nil
    }, fetchResourceMediaReferenceHash: { resource in
        if let resource = resource as? VideoLibraryMediaResource {
            return fetchVideoLibraryMediaResourceHash(resource: resource)
        }
        return .single(nil)
    }, prepareSecretThumbnailData: { data in
        return prepareSecretThumbnailData(EngineMediaResource.ResourceData(data)).flatMap { size, data in
            return (PixelDimensions(size), data)
        }
    }, backgroundUpload: { postbox, _, resource in
        if let uploadInBackground {
            return uploadInBackground(postbox, resource)
        }
        return .single(nil)
    })
}

private func prepareSecretThumbnailData(_ data: EngineMediaResource.ResourceData) -> (CGSize, Data)? {
    if data.isComplete, let image = UIImage(contentsOfFile: data.path) {
        if image.size.width < 100 && image.size.height < 100 {
            if let resultData = try? Data(contentsOf: URL(fileURLWithPath: data.path)) {
                return (image.size, resultData)
            }
        }
        let scaledSize = image.size.fitted(CGSize(width: 90.0, height: 90.0))
        if let scaledImage = generateScaledImage(image: image, size: scaledSize, scale: 1.0), let scaledData = scaledImage.jpegData(compressionQuality: 0.4) {
            return (scaledSize, scaledData)
        }
    }
    return nil
}