Add way to customize WIndowViewController

Author: wtmoose

CHANGELOG.md

@@ -3,6 +3,9 @@ All notable changes to this project will be documented in this file.
 
 ## 4.1.2
 
+### Features
+* #164 Added an optional `windowViewController` property to `SwiftMessages.Config` for supplying a custom subclass of `WindowViewController`.
+
 ### Bug Fixes
 * Custom presentation styles using `TopBottomAnimation` now display properly under top and bottom bars.
 

Demo/Podfile.lock

@@ -1,5 +1,5 @@
 PODS:
-  - SwiftMessages (4.0.1)
+  - SwiftMessages (4.1.2)
 
 DEPENDENCIES:
   - SwiftMessages (from `../`)
@@ -9,8 +9,8 @@ EXTERNAL SOURCES:
     :path: ../
 
 SPEC CHECKSUMS:
-  SwiftMessages: 794cbec72392c020e15e4c45dfcfce15781e07cb
+  SwiftMessages: 7f883c0b717808e9557e9b385f8956418dddd1cb
 
 PODFILE CHECKSUM: 6431c980c9207084d738b6ba87b2101dd9eb5097
 
-COCOAPODS: 1.3.1
+COCOAPODS: 1.4.0

Demo/Pods/Local Podspecs/SwiftMessages.podspec.json

@@ -246,6 +246,7 @@
 			children = (
 				86589D461D64B6E40041676C /* BaseView.swift */,
 				86AAF82C1D580F410031EE32 /* Theme.swift */,
+				8644955C1D4FAF7C0056EB2A /* WindowViewController.swift */,
 				867BED201D622793005212E3 /* BackgroundViewable.swift */,
 				864495551D4F7C390056EB2A /* Identifiable.swift */,
 				86AAF81D1D5549680031EE32 /* MarginAdjustable.swift */,
@@ -283,7 +284,6 @@
 				86AAF8171D54F0650031EE32 /* PassthroughView.swift */,
 				22E01F631E74EC8B00ACE19A /* MaskingView.swift */,
 				86AAF8191D54F0850031EE32 /* PassthroughWindow.swift */,
-				8644955C1D4FAF7C0056EB2A /* WindowViewController.swift */,
 				86AAF81B1D551FE60031EE32 /* UIViewController+Utils.swift */,
 				862C0CB01D5911C100D06168 /* NSBundle+Utils.swift */,
 			);

Demo/Pods/Manifest.lock

@@ -268,7 +268,7 @@ class Presenter: NSObject {
     private func getPresentationContext() throws -> PresentationContext {
 
         func newWindowViewController(_ windowLevel: UIWindowLevel) -> UIViewController {
-            let viewController = WindowViewController(windowLevel: windowLevel, config: config)
+            let viewController = WindowViewController.newInstance(windowLevel: windowLevel, config: config)
             return viewController
         }
 

Demo/Pods/Pods.xcodeproj/project.pbxproj

@@ -309,6 +309,13 @@ open class SwiftMessages {
          label, e.g. "dismiss" when the `interactive` option is used.
         */
         public var dimModeAccessibilityLabel: String = "dismiss"
+
+        /**
+         If specified, SwiftMessages calls this closure when an instance of
+         `WindowViewController` is needed. Use this if you need to supply a custom subclass
+         of `WindowViewController`.
+         */
+        public var windowViewController: ((_ windowLevel: UIWindowLevel?, _ config: SwiftMessages.Config) -> WindowViewController)?
     }
 
     /**

Demo/Pods/Target Support Files/Pods-Demo/Pods-Demo-frameworks.sh

@@ -21,7 +21,7 @@ extension UIViewController {
         } else if case .bottom? = topBottomStyle, let tabBarController = sm_selectTabBarControllerTopDown() {
             return tabBarController
         }
-        return WindowViewController(windowLevel: self.view.window?.windowLevel ?? UIWindowLevelNormal, config: config)
+        return WindowViewController.newInstance(windowLevel: self.view.window?.windowLevel, config: config)
     }
 
     fileprivate func sm_selectNavigationControllerTopDown() -> UINavigationController? {
@@ -80,7 +80,7 @@ extension UIViewController {
             if let parent = self.parent {
                 return parent.sm_selectPresentationContextBottomUp(config)
             } else {
-                return WindowViewController(windowLevel: self.view.window?.windowLevel ?? UIWindowLevelNormal, config: config)
+                return WindowViewController.newInstance(windowLevel: self.view.window?.windowLevel, config: config)
             }
         }
         return self

Demo/Pods/Target Support Files/Pods-Demo/Pods-Demo.debug.xcconfig

@@ -8,27 +8,27 @@
 
 import UIKit
 
-class WindowViewController: UIViewController
+open class WindowViewController: UIViewController
 {
     fileprivate var window: UIWindow?
 
     let windowLevel: UIWindowLevel
     let config: SwiftMessages.Config
 
-    override var shouldAutorotate: Bool {
+    override open var shouldAutorotate: Bool {
         return config.shouldAutorotate
     }
 
-    init(windowLevel: UIWindowLevel = UIWindowLevelNormal, config: SwiftMessages.Config)
+    public init(windowLevel: UIWindowLevel?, config: SwiftMessages.Config)
     {
-        self.windowLevel = windowLevel
+        self.windowLevel = windowLevel ?? UIWindowLevelNormal
         self.config = config
         let window = PassthroughWindow(frame: UIScreen.main.bounds)
         self.window = window
         super.init(nibName: nil, bundle: nil)
         self.view = PassthroughView()
         window.rootViewController = self
-        window.windowLevel = windowLevel
+        window.windowLevel = windowLevel ?? UIWindowLevelNormal
     }
 
     func install(becomeKey: Bool) {
@@ -45,11 +45,17 @@ class WindowViewController: UIViewController
         window = nil
     }
 
-    required init?(coder aDecoder: NSCoder) {
+    required public init?(coder aDecoder: NSCoder) {
         fatalError("init(coder:) has not been implemented")
     }
 
-    override public var preferredStatusBarStyle: UIStatusBarStyle {
+    override open var preferredStatusBarStyle: UIStatusBarStyle {
         return config.preferredStatusBarStyle ?? super.preferredStatusBarStyle
     }
 }
+
+extension WindowViewController {
+    static func newInstance(windowLevel: UIWindowLevel?, config: SwiftMessages.Config) -> WindowViewController {
+        return config.windowViewController?(windowLevel, config) ?? WindowViewController(windowLevel: windowLevel, config: config)
+    }
+}