forked from Zero-zhou/iOS-Extensions
-
Notifications
You must be signed in to change notification settings - Fork 0
/
TopMostViewController.swift
86 lines (74 loc) · 2.79 KB
/
TopMostViewController.swift
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
//
// TopMostViewController.swift
//
// Created by Augus on 9/26/15.
// Copyright © 2015 iAugus. All rights reserved.
//
import UIKit
/**
Description: the toppest view controller of presenting view controller
How to use: UIApplication.topMostViewController
Where to use: controllers are not complex
*/
extension UIApplication {
var topMostViewController: UIViewController? {
var topController = UIApplication.shared.keyWindow?.rootViewController
while topController?.presentedViewController != nil {
topController = topController?.presentedViewController
}
return topController
}
/// App has more than one window and just want to get topMostViewController of the AppDelegate window.
var appDelegateWindowTopMostViewController: UIViewController? {
let delegate = UIApplication.shared.delegate as? AppDelegate
var topController = delegate?.window?.rootViewController
while topController?.presentedViewController != nil {
topController = topController?.presentedViewController
}
return topController
}
}
/**
Description: the toppest view controller of presenting view controller
How to use: UIApplication.sharedApplication().keyWindow?.rootViewController?.topMostViewController
Where to use: There are lots of kinds of controllers (UINavigationControllers, UITabbarControllers, UIViewController)
*/
extension UIViewController {
var topMostViewController: UIViewController? {
// Handling Modal views
if let presentedViewController = self.presentedViewController {
return presentedViewController.topMostViewController
}
// Handling UIViewController's added as subviews to some other views.
else {
for view in self.view.subviews
{
// Key property which most of us are unaware of / rarely use.
if let subViewController = view.next {
if subViewController is UIViewController {
let viewController = subViewController as! UIViewController
return viewController.topMostViewController
}
}
}
return self
}
}
}
extension UITabBarController {
override var topMostViewController: UIViewController? {
return self.selectedViewController?.topMostViewController
}
var topVisibleViewController: UIViewController? {
var top = selectedViewController
while top?.presentedViewController != nil {
top = top?.presentedViewController
}
return top
}
}
extension UINavigationController {
override var topMostViewController: UIViewController? {
return self.visibleViewController?.topMostViewController
}
}