Как сделать этот UINavigationController более удобным для кода?

Question

-(UINavigationController *) navigationControllerOfParentOrSelf //These 2 functions are short so I just go ahead 
{ 
    UIViewController * current=self; 
    while (current) { 
     UINavigationController * nav = current.navigationController; 
     if (nav) { 
      return nav; 
     } 
     current=current.parentViewController; 
    } 
    return nil; 
} 

-(UITabBarController *) tabBarControllerOfParentOrSelf 
{ 
    UIViewController * current=self; 
    while (current) { 
     UITabBarController * tc = current.tabBarController; 
     if (tc) { 
      return tc; 
     } 
     current=current.parentViewController; 
    } 
    return nil; 
} 

Похож на много повторяющегося кода.

В основном я просто хочу знать, находится ли UIViewController внутри UINavigationController или нет. Дело в том, navigationController свойство часто NIL, когда UIViewController является childViewController

Answer 1

Я хотел бы предложить что-то вроде этого:

-(UINavigationController *) navigationControllerOfParentOrSelf 
{ 
    return [self parrentControllerOfParrentOrSelfWithGetter: @selector(navigationController)]; 
} 

-(UITabBarController *) tabBarControllerOfParentOrSelf 
{ 
    return [self parrentControllerOfParrentOrSelfWithGetter: @selector(tabBarController)]; 

} 

- (id) parrentControllerOfParrentOrSelfWithGetter: (SEL) getter 
{ 
    UIViewController * current=self; 
    while (current) { 
     id res = [current performSelector: getter]; 
     if (res) { 
      return tc; 
     } 
     current=current.parentViewController; 
    } 
    return nil; 
} 

Answer 2

Вы можете сделать это нравится:

-(id) getViewController:(BOOL)isNavController 
{ 
    id controller = nil; 
    if(isNavController) 
    { 
     controller = self.navigationController; 
    } 
    else 
    { 
     controller = self.tabBarController; 
    } 

    return controller; 
}