Animate UIWindow rootViewController switch

UIWindow+Extension.swift

extension UIWindow {

    func setRootViewController(_ rootViewController: UIViewController?, animated: Bool) {

        guard let viewContoller = rootViewController else {
            self.rootViewController = rootViewController
            return
        }

        var snapShotView: UIView?
        if (animated) {
            if let snapShot = snapshotView(afterScreenUpdates: true) {
                snapShotView = snapShot
                viewContoller.view.addSubview(snapShot)
            }
        }
        self.rootViewController = viewContoller

        if (animated) {
            UIView.animate(withDuration: 0.25, animations: {
                snapShotView?.layer.opacity = 0
                snapShotView?.layer.transform = CATransform3DMakeScale(1.5, 1.5, 1.5)
            }) { (_) in
                snapShotView?.removeFromSuperview()
            }
        }
    }
}