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 Display

public func makeTelegramAccountAuxiliaryMethods(uploadInBackground: ((Postbox, MediaResource) -> Signal<String?, NoError>)?) -> AccountAuxiliaryMethods {
    return AccountAuxiliaryMethods(fetchResource: { postbox, resource, ranges, _ in
        if let resource = resource as? VideoLibraryMediaResource {
            return fetchVideoLibraryMediaResource(postbox: postbox, resource: resource)
        } else if let resource = resource as? LocalFileVideoMediaResource {
            return fetchLocalFileVideoMediaResource(postbox: postbox, resource: resource)
        } else if let resource = resource as? LocalFileGifMediaResource {
            return fetchLocalFileGifMediaResource(resource: resource)
        } else if let photoLibraryResource = resource as? PhotoLibraryMediaResource {
            return fetchPhotoLibraryResource(localIdentifier: photoLibraryResource.localIdentifier)
        } 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
}