BUGFIX: MessageDisplayer was broken

Author: Chris-Perkins

Lifting Buddy/AppDelegate.swift

@@ -16,6 +16,8 @@ import GBVersionTracking
 class AppDelegate: UIResponder, UIApplicationDelegate {
 
     var window: UIWindow?
+    
+    var mainViewController: MainViewController?
 
     func application(_ application: UIApplication, didFinishLaunchingWithOptions launchOptions: [UIApplicationLaunchOptionsKey: Any]?) -> Bool {
         // We need this for views to not be hidden under the keyboard

Lifting Buddy/MainViewController.swift

@@ -47,6 +47,7 @@ class MainViewController: UIViewController {
     override func viewDidLoad() {
         super.viewDidLoad()
 
+        (UIApplication.shared.delegate as? AppDelegate)?.mainViewController = self
         view.backgroundColor = .lightestBlackWhiteColor
     }
 

Lifting Buddy/MessageQueue.swift

@@ -26,18 +26,18 @@ public class MessageQueue {
 
     private var messageQueue = [Message]()
     private var queueTimer: Timer?
-    private var md: MessageDisplayer? {
-        return (UIApplication.shared.delegate as? AppDelegate)?.window?.rootViewController as? MessageDisplayer
+    private var messageDisplayer: MessageDisplayer? {
+        return (UIApplication.shared.delegate as? AppDelegate)?.mainViewController
     }
 
     public func append(_ message: Message) {
         messageQueue.append(message)
 
         if queueTimer == nil {
-            md?.messageQueueStarted?()
+            messageDisplayer?.messageQueueStarted?()
 
             queueTimer = Timer.scheduledTimer(withTimeInterval: messageDisplayTime, repeats: true) { (timer) in
-                if let md = self.md {
+                if let md = self.messageDisplayer {
 
                     if self.messageQueue.count == 0 {
                         self.queueTimer?.invalidate()