Быстрое обнаружение супервизора данного класса с дженериками

Question

Думаю, я борюсь с дженериками. Я хочу создать простое расширение UIView, чтобы найти рекурсивно супервизор класса, переданный в параметре функции. Я хочу, чтобы функция возвращала необязательный элемент, содержащий, очевидно, либо нуль, либо объект, видимый как экземпляр предоставленного класса.

extension UIView { 
    func superviewOfClass<T>(ofClass: T.Type) -> T? { 
     var currentView: UIView? = self 

     while currentView != nil { 
      if currentView is T { 
       break 
      } else { 
       currentView = currentView?.superview 
      } 
     } 

     return currentView as? T 
    } 
} 

Любая помощь очень ценится.

Answer 1

Swift 3

Это более краткий способ:

extension UIView { 

    func superview<T>(of type: T.Type) -> T? { 
     return superview as? T ?? superview.flatMap { $0.superview(of: type) } 
    } 

    func subview<T>(of type: T.Type) -> T? { 
     return subviews.flatMap { $0 as? T ?? $0.subview(of: type) }.first 
    } 

}