handle cascade childViewController

kyoji2 · kyoji2 · commit b85562c42516 · 2013-11-01T10:18:03.000+08:00
diff --git a/SIAlertView/SIAlertView.m b/SIAlertView/SIAlertView.m
@@ -69,22 +69,20 @@ - (UIViewController *)viewControllerForStatusBarStyle
 {
     UIViewController *currentViewController = [self currentViewController];
     
-    if ([currentViewController childViewControllerForStatusBarStyle]) {
-        return [currentViewController childViewControllerForStatusBarStyle];
-    } else {
-        return currentViewController;
+    while ([currentViewController childViewControllerForStatusBarStyle]) {
+        currentViewController = [currentViewController childViewControllerForStatusBarStyle];
     }
+    return currentViewController;
 }
 
 - (UIViewController *)viewControllerForStatusBarHidden
 {
     UIViewController *currentViewController = [self currentViewController];
     
-    if ([currentViewController childViewControllerForStatusBarHidden]) {
-        return [currentViewController childViewControllerForStatusBarHidden];
-    } else {
-        return currentViewController;
+    while ([currentViewController childViewControllerForStatusBarHidden]) {
+        currentViewController = [currentViewController childViewControllerForStatusBarHidden];
     }
+    return currentViewController;
 }
 
 @end

Original file line number	Diff line number	Diff line change
`@@ -69,22 +69,20 @@ - (UIViewController *)viewControllerForStatusBarStyle`
`69`	`69`	`{`
`70`	`70`	`UIViewController *currentViewController = [self currentViewController];`
`71`	`71`
`72`		`- if ([currentViewController childViewControllerForStatusBarStyle]) {`
`73`		`- return [currentViewController childViewControllerForStatusBarStyle];`
`74`		`- } else {`
`75`		`- return currentViewController;`
	`72`	`+ while ([currentViewController childViewControllerForStatusBarStyle]) {`
	`73`	`+ currentViewController = [currentViewController childViewControllerForStatusBarStyle];`
`76`	`74`	`}`
	`75`	`+ return currentViewController;`
`77`	`76`	`}`
`78`	`77`
`79`	`78`	`- (UIViewController *)viewControllerForStatusBarHidden`
`80`	`79`	`{`
`81`	`80`	`UIViewController *currentViewController = [self currentViewController];`
`82`	`81`
`83`		`- if ([currentViewController childViewControllerForStatusBarHidden]) {`
`84`		`- return [currentViewController childViewControllerForStatusBarHidden];`
`85`		`- } else {`
`86`		`- return currentViewController;`
	`82`	`+ while ([currentViewController childViewControllerForStatusBarHidden]) {`
	`83`	`+ currentViewController = [currentViewController childViewControllerForStatusBarHidden];`
`87`	`84`	`}`
	`85`	`+ return currentViewController;`
`88`	`86`	`}`
`89`	`87`
`90`	`88`	`@end`