import Foundation
import Display
import SafariServices
import TelegramCore
import SwiftSignalKit

public func openExternalUrl(account: Account, url: String, presentationData: PresentationData, applicationContext: TelegramApplicationContext, navigationController: NavigationController?) {
    if url.lowercased().hasPrefix("tel:") {
        applicationContext.applicationBindings.openUrl(url)
        return
    }
    
    var parsedUrlValue: URL?
    if let parsed = URL(string: url) {
        parsedUrlValue = parsed
    } else if let encoded = (url as NSString).addingPercentEscapes(using: String.Encoding.utf8.rawValue), let parsed = URL(string: encoded) {
        parsedUrlValue = parsed
    }
    if let parsed = parsedUrlValue, parsed.scheme == nil {
        parsedUrlValue = URL(string: "https://" + parsed.absoluteString)
    }
    
    guard let parsedUrl = parsedUrlValue else {
        return
    }
    
    if parsedUrl.scheme == "mailto" {
        applicationContext.applicationBindings.openUrl(url)
        return
    }
    
    if let host = parsedUrl.host?.lowercased() {
        if host == "itunes.apple.com" {
            if applicationContext.applicationBindings.canOpenUrl(parsedUrl.absoluteString) {
                applicationContext.applicationBindings.openUrl(url)
                return
            }
        }
        if host == "twitter.com" || host == "mobile.twitter.com" {
            if applicationContext.applicationBindings.canOpenUrl("twitter://status") {
                applicationContext.applicationBindings.openUrl(url)
                return
            }
        } else if host == "instagram.com" {
            if applicationContext.applicationBindings.canOpenUrl("instagram://photo") {
                applicationContext.applicationBindings.openUrl(url)
                return
            }
        }
    }
    
    if parsedUrl.scheme == "tg", let query = parsedUrl.query {
        var convertedUrl: String?
        if parsedUrl.host == "resolve" {
            if let components = URLComponents(string: "/?" + query) {
                var domain: String?
                var start: String?
                var startGroup: String?
                var game: String?
                var post: String?
                if let queryItems = components.queryItems {
                    for queryItem in queryItems {
                        if let value = queryItem.value {
                            if queryItem.name == "domain" {
                                domain = value
                            } else if queryItem.name == "start" {
                                start = value
                            } else if queryItem.name == "startgroup" {
                                startGroup = value
                            } else if queryItem.name == "game" {
                                game = value
                            } else if queryItem.name == "post" {
                                post = value
                            }
                        }
                    }
                }
                
                if let domain = domain {
                    var result = "https://t.me/\(domain)"
                    if let post = post, let postValue = Int(post) {
                        result += "/\(postValue)"
                    }
                    if let start = start {
                        result += "?start=\(start)"
                    } else if let startGroup = startGroup {
                        result += "?startgroup=\(startGroup)"
                    } else if let game = game {
                        result += "?game=\(game)"
                    }
                    convertedUrl = result
                }
            }
        } else if parsedUrl.host == "join" {
            if let components = URLComponents(string: "/?" + query) {
                var invite: String?
                if let queryItems = components.queryItems {
                    for queryItem in queryItems {
                        if let value = queryItem.value {
                            if queryItem.name == "invite" {
                                invite = value
                            }
                        }
                    }
                }
                if let invite = invite {
                    convertedUrl = "https://t.me/joinchat/\(invite)"
                }
            }
        } else if parsedUrl.host == "addstickers" {
            if let components = URLComponents(string: "/?" + query) {
                var set: String?
                if let queryItems = components.queryItems {
                    for queryItem in queryItems {
                        if let value = queryItem.value {
                            if queryItem.name == "set" {
                                set = value
                            }
                        }
                    }
                }
                if let set = set {
                    convertedUrl = "https://t.me/addstickers/\(set)"
                }
            }
        } else if parsedUrl.host == "msg_url" {
            if let components = URLComponents(string: "/?" + query) {
                var shareUrl: String?
                var shareText: String?
                if let queryItems = components.queryItems {
                    for queryItem in queryItems {
                        if let value = queryItem.value {
                            if queryItem.name == "url" {
                                shareUrl = value
                            } else if queryItem.name == "text" {
                                shareText = value
                            }
                        }
                    }
                }
                if let shareUrl = shareUrl {
                    let controller = PeerSelectionController(account: account)
                    controller.peerSelected = { [weak controller] peerId in
                        if let strongController = controller {
                            strongController.dismiss()
                            
                            let textInputState: ChatTextInputState
                            if let shareText = shareText, !shareText.isEmpty {
                                let urlString = NSMutableAttributedString(string: "\(shareUrl)\n")
                                let textString = NSAttributedString(string: "\(shareText)")
                                let selectionRange: Range<Int> = urlString.length ..< (urlString.length + textString.length)
                                urlString.append(textString)
                                textInputState = ChatTextInputState(inputText: urlString, selectionRange: selectionRange)
                            } else {
                                textInputState = ChatTextInputState(inputText: NSAttributedString(string: "\(shareUrl)"))
                            }
                            
                            let _ = (account.postbox.modify({ modifier -> Void in
                                modifier.updatePeerChatInterfaceState(peerId, update: { currentState in
                                    if let currentState = currentState as? ChatInterfaceState {
                                        return currentState.withUpdatedComposeInputState(textInputState)
                                    } else {
                                        return ChatInterfaceState().withUpdatedComposeInputState(textInputState)
                                    }
                                })
                            })
                            |> deliverOnMainQueue).start(completed: {
                                navigationController?.pushViewController(ChatController(account: account, chatLocation: .peer(peerId), messageId: nil))
                            })
                        }
                    }
                    if let navigationController = navigationController {
                        (navigationController.viewControllers.last as? ViewController)?.present(controller, in: .window(.root), with: ViewControllerPresentationArguments(presentationAnimation: ViewControllerPresentationAnimation.modalSheet))
                    }
                }
            }
        } else if parsedUrl.host == "socks" {
            if let components = URLComponents(string: "/?" + query) {
                var server: String?
                var port: String?
                var user: String?
                var pass: String?
                if let queryItems = components.queryItems {
                    for queryItem in queryItems {
                        if let value = queryItem.value {
                            if queryItem.name == "server" || queryItem.name == "proxy" {
                                server = value
                            } else if queryItem.name == "port" {
                                port = value
                            } else if queryItem.name == "user" {
                                user = value
                            } else if queryItem.name == "pass" {
                                pass = value
                            }
                        }
                    }
                }
                
                if let server = server, !server.isEmpty, let port = port, let _ = Int32(port) {
                    var result = "https://t.me/proxy?proxy=\(server)&port=\(port)"
                    if let user = user {
                        result += "&user=\(user)"
                        if let pass = pass {
                            result += "&pass=\(pass)"
                        }
                    }
                    convertedUrl = result
                }
            }
        }
        
        if let convertedUrl = convertedUrl {
            let _ = (resolveUrl(account: account, url: convertedUrl)
            |> deliverOnMainQueue).start(next: { resolved in
                if case let .externalUrl(value) = resolved {
                    applicationContext.applicationBindings.openUrl(value)
                } else {
                    openResolvedUrl(resolved, account: account, navigationController: navigationController, openPeer: { peerId, navigation in
                        switch navigation {
                            case .info:
                                let _ = (account.postbox.loadedPeerWithId(peerId)
                                |> deliverOnMainQueue).start(next: { peer in
                                    if let infoController = peerInfoController(account: account, peer: peer) {
                                        navigationController?.pushViewController(infoController)
                                    }
                                })
                            case .chat:
                                if let navigationController = navigationController {
                                    navigateToChatController(navigationController: navigationController, account: account, chatLocation: .peer(peerId))
                                }
                            case .withBotStartPayload:
                                if let navigationController = navigationController {
                                    navigateToChatController(navigationController: navigationController, account: account, chatLocation: .peer(peerId))
                                }
                        }
                    }, present: { c, a in
                        if let navigationController = navigationController {
                            (navigationController.viewControllers.last as? ViewController)?.present(c, in: .window(.root), with: a)
                        }
                    })
                }
            })
        }
        return
    }
    
    if parsedUrl.scheme == "http" || parsedUrl.scheme == "https" {
        if #available(iOSApplicationExtension 9.0, *) {
            if let window = navigationController?.view.window {
                let controller = SFSafariViewController(url: parsedUrl)
                if #available(iOSApplicationExtension 10.0, *) {
                    controller.preferredBarTintColor = presentationData.theme.rootController.navigationBar.backgroundColor
                    controller.preferredControlTintColor = presentationData.theme.rootController.navigationBar.accentTextColor
                }
                window.rootViewController?.present(controller, animated: true)
            } else {
                applicationContext.applicationBindings.openUrl(parsedUrl.absoluteString)
            }
        } else {
            applicationContext.applicationBindings.openUrl(url)
        }
    } else {
        applicationContext.applicationBindings.openUrl(url)
    }
}