Update RefdsApplication.swift

rafaelesantos · Aug 22, 2024 · 2c13d74 · 2c13d74
1 parent 7840961
commit 2c13d74
Showing 1 changed file with 10 additions and 17 deletions.
diff --git a/Sources/RefdsShared/Application/RefdsApplication.swift b/Sources/RefdsShared/Application/RefdsApplication.swift
@@ -7,22 +7,15 @@ import UIKit
 
 public final class RefdsApplication {
     public static let shared = RefdsApplication()
-#if os(macOS)
-    public var rootViewController: NSViewController? {
-        if let window = NSApplication.shared.windows.first,
-           let rootViewController = window.contentViewController {
-            return rootViewController
-        }
-        return nil
-    }
-#else
-    public var rootViewController: UIViewController? {
-        if let windowScene = UIApplication.shared.connectedScenes.first as? UIWindowScene,
-           let window = windowScene.windows.first,
-           let rootViewController = window.rootViewController {
-            return rootViewController
-        }
-        return nil
+
+    public var rootViewController: RefdsViewController? {
+        #if os(macOS)
+        let window = NSApplication.shared.windows.first
+        return window?.contentViewController?.refdsViewController
+        #else
+        let windowScene = UIApplication.shared.connectedScenes.first as? UIWindowScene
+        let window = windowScene?.windows.first
+        return window?.rootViewController?.refdsViewController
+        #endif
     }
-#endif
 }