Telegram-iOS/submodules/WebSearchUI/Sources/LegacyWebSearchEditor.swift

import Foundation
import UIKit
import LegacyComponents
import SwiftSignalKit
import TelegramCore
import SSignalKit
import Display
import TelegramPresentationData
import AccountContext
import LegacyUI
import LegacyMediaPickerUI

func presentLegacyWebSearchEditor(context: AccountContext, theme: PresentationTheme, result: ChatContextResult, initialLayout: ContainerViewLayout?, updateHiddenMedia: @escaping (String?) -> Void, transitionHostView: @escaping () -> UIView?, transitionView: @escaping (ChatContextResult) -> UIView?, completed: @escaping (UIImage) -> Void, present: @escaping (ViewController, Any?) -> Void) {
    guard let item = legacyWebSearchItem(engine: context.engine, result: result) else {
        return
    }
    
    var screenImage: Signal<UIImage?, NoError> = .single(nil)
    if let resource = item.thumbnailResource {
        screenImage = context.engine.resources.data(resource: EngineMediaResource(resource), attemptSynchronously: true)
        |> map { maybeData -> UIImage? in
            if maybeData.isComplete {
                if let loadedData = try? Data(contentsOf: URL(fileURLWithPath: maybeData.path), options: []), let image = UIImage(data: loadedData) {
                    return image
                }
            }
            return nil
        }
    }
    
    let _ = (screenImage
    |> take(1)
    |> deliverOnMainQueue).start(next: { screenImage in
        let legacyController = LegacyController(presentation: .custom, theme: theme, initialLayout: initialLayout)
        legacyController.statusBar.statusBarStyle = theme.rootController.statusBarStyle.style
        
        let paintStickersContext = LegacyPaintStickersContext(context: context)
        
        let controller = TGPhotoEditorController(context: legacyController.context, item: item, intent: TGPhotoEditorControllerAvatarIntent, adjustments: nil, caption: nil, screenImage: screenImage ?? UIImage(), availableTabs: TGPhotoEditorController.defaultTabs(forAvatarIntent: true), selectedTab: .cropTab)!
        controller.stickersContext = paintStickersContext
        legacyController.bind(controller: controller)
        
        controller.editingContext = TGMediaEditingContext()
        controller.didFinishEditing = { [weak controller] _, result, _, hasChanges, commit in
            if !hasChanges {
                return
            }
            if let result = result {
                completed(result)
            }
            commit?()
            controller?.dismiss(animated: true)
        }
        controller.requestThumbnailImage = { _ -> SSignal in
            return item.thumbnailImageSignal()
        }
        controller.requestOriginalScreenSizeImage = { _, position -> SSignal in
            return item.screenImageSignal(position)
        }
        controller.requestOriginalFullSizeImage = { _, position -> SSignal in
            return item.originalImageSignal(position)
        }
        
        let fromView = transitionView(result)!
        let transition = TGMediaAvatarEditorTransition(controller: controller, from: fromView)!
        transition.transitionHostView = transitionHostView()
        transition.referenceFrame = {
            return fromView.frame
        }
        transition.referenceImageSize = {
            return item.dimensions
        }
        transition.referenceScreenImageSignal = {
            return item.screenImageSignal(0.0)
        }
        transition.imageReady = {
            updateHiddenMedia(result.id)
        }
        
        controller.beginCustomTransitionOut = { [weak legacyController] outFrame, outView, completion in
            transition.outReferenceFrame = outFrame
            transition.repView = outView
            transition.dismiss(animated: true, completion: {
                updateHiddenMedia(nil)
                if let completion = completion {
                    DispatchQueue.main.async {
                        completion()
                    }
                }
                legacyController?.dismiss()
            })
        }
        
        present(legacyController, nil)
        transition.present(animated: true)
    })
}