closes #705 Blank view from #TARGET_IS_EXTENSION

HeroTransitions · Mar 7, 2021 · 44e9a75 · 44e9a75
1 parent 5715db1
commit 44e9a75
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 6 deletions.
diff --git a/Sources/Extensions/UIViewController+Hero.swift b/Sources/Extensions/UIViewController+Hero.swift
@@ -336,9 +336,7 @@ public extension HeroExtension where Base: UIViewController {
             parentVC.present(next, animated: false, completion: completion)
           }
         } else {
-          #if TARGET_IS_EXTENSION
-          UIApplication.shared.keyWindow?.rootViewController = next
-          #endif
+          parentVC?.view.window?.rootViewController = next
         }
       }
     }

diff --git a/Sources/Transition/HeroTransition+Complete.swift b/Sources/Transition/HeroTransition+Complete.swift
@@ -89,9 +89,7 @@ extension HeroTransition {
       if isPresenting != finished, !inContainerController, transitionContext != nil {
         // only happens when present a .overFullScreen VC
         // bug: http://openradar.appspot.com/radar?id=5320103646199808
-        #if TARGET_IS_EXTENSION
-        UIApplication.shared.keyWindow?.addSubview(isPresenting ? fromView : toView)
-        #endif
+        container.window?.addSubview(isPresenting ? fromView : toView)
       }
     }