snapshot

Signed-off-by: Joseph Mattello <mail@joemattiello.com>

SideStoreApp/Sources/SideStoreUIKit/TabBarController.swift

@@ -0,0 +1,121 @@
+//
+//  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
+    }
+}