SideStore/SideStoreApp/Sources/SideStoreUIKit/TabBarController.swift

//
//  TabBarController.swift
//  AltStore
//
//  Created by Riley Testut on 9/19/19.
//  Copyright © 2019 Riley Testut. All rights reserved.
//

import SideStoreCore
import UIKit

extension TabBarController {
    private enum Tab: Int, CaseIterable {
        case news
        case browse
        case myApps
        case settings
    }
}

public final class TabBarController: UITabBarController {
    private var initialSegue: (identifier: String, sender: Any?)?

    private var _viewDidAppear = false

    required init?(coder aDecoder: NSCoder) {
        super.init(coder: aDecoder)

        NotificationCenter.default.addObserver(self, selector: #selector(TabBarController.openPatreonSettings(_:)), name: SideStoreAppDelegate.openPatreonSettingsDeepLinkNotification, object: nil)
        NotificationCenter.default.addObserver(self, selector: #selector(TabBarController.importApp(_:)), name: SideStoreAppDelegate.importAppDeepLinkNotification, object: nil)
        NotificationCenter.default.addObserver(self, selector: #selector(TabBarController.presentSources(_:)), name: SideStoreAppDelegate.addSourceDeepLinkNotification, object: nil)
    }

	public override func viewDidAppear(_ animated: Bool) {
        super.viewDidAppear(animated)

        _viewDidAppear = true

        if let (identifier, sender) = initialSegue {
            initialSegue = nil
            performSegue(withIdentifier: identifier, sender: sender)
        } else if let patchedApps = UserDefaults.standard.patchedApps, !patchedApps.isEmpty {
            // Check if we need to finish installing untethered jailbreak.
            let activeApps = InstalledApp.fetchActiveApps(in: DatabaseManager.shared.viewContext)
            guard let patchedApp = activeApps.first(where: { patchedApps.contains($0.bundleIdentifier) }) else { return }

            performSegue(withIdentifier: "finishJailbreak", sender: patchedApp)
        }
    }

	public override func prepare(for segue: UIStoryboardSegue, sender: Any?) {
        guard let identifier = segue.identifier else { return }

        switch identifier {
        case "presentSources":
            guard let notification = sender as? Notification,
                  let sourceURL = notification.userInfo?[SideStoreAppDelegate.addSourceDeepLinkURLKey] as? URL
            else { return }

            let navigationController = segue.destination as! UINavigationController
            let sourcesViewController = navigationController.viewControllers.first as! SourcesViewController
            sourcesViewController.deepLinkSourceURL = sourceURL

        case "finishJailbreak":
            guard let installedApp = sender as? InstalledApp, #available(iOS 14, *) else { return }

            let navigationController = segue.destination as! UINavigationController

            let patchViewController = navigationController.viewControllers.first as! PatchViewController
            patchViewController.installedApp = installedApp
            patchViewController.completionHandler = { [weak self] _ in
                self?.dismiss(animated: true, completion: nil)
            }

        default: break
        }
    }

	public override func performSegue(withIdentifier identifier: String, sender: Any?) {
        guard _viewDidAppear else {
            initialSegue = (identifier, sender)
            return
        }

        super.performSegue(withIdentifier: identifier, sender: sender)
    }
}

extension TabBarController {
    @objc func presentSources(_ sender: Any) {
        if let presentedViewController = presentedViewController {
            if let navigationController = presentedViewController as? UINavigationController,
               let sourcesViewController = navigationController.viewControllers.first as? SourcesViewController {
                if let notification = (sender as? Notification),
                   let sourceURL = notification.userInfo?[SideStoreAppDelegate.addSourceDeepLinkURLKey] as? URL {
                    sourcesViewController.deepLinkSourceURL = sourceURL
                } else {
                    // Don't dismiss SourcesViewController if it's already presented.
                }
            } else {
                presentedViewController.dismiss(animated: true) {
                    self.presentSources(sender)
                }
            }

            return
        }

        performSegue(withIdentifier: "presentSources", sender: sender)
    }
}

private extension TabBarController {
    @objc func openPatreonSettings(_: Notification) {
        selectedIndex = Tab.settings.rawValue
    }

    @objc func importApp(_: Notification) {
        selectedIndex = Tab.myApps.rawValue
    }
}