Update DebugHelper.swift

Flipkart · Nov 22, 2023 · 0c367fc · 0c367fc
1 parent 4e7a49d
commit 0c367fc
Showing 1 changed file with 21 additions and 15 deletions.
diff --git a/Source/Classes/Debug/DebugHelper.swift b/Source/Classes/Debug/DebugHelper.swift
@@ -88,27 +88,33 @@ internal class DebugHelper: NSObject {
         isShown = true
 
         // create chat head image
-        let bundle: Bundle = Bundle(for: Self.self)
-        let imageView: UIImageView = UIImageView(image: UIImage(named: "Logo", in: bundle, compatibleWith: nil))
-        imageView.layer.cornerRadius = 20
-        imageView.clipsToBounds = true
-
-        // create and display chat head with debug options
-        let chatHead: FCChatHeadsController = FCChatHeadsController()
-        chatHead.datasource = self
-        chatHead.delegate = self
-        chatHead.presentChatHead(with: imageView, chatID: "NowYouSeeMe.debugView")
-        self.chatHead = chatHead
+        let bundle: Bundle = Bundle(for: DebugView.self)
+        if let NYSMBundleUrl: URL = bundle.url(forResource: "NowYouSeeMe", withExtension: "bundle"),
+           let NYSMBundle: Bundle = Bundle(url: NYSMBundleUrl) {
+            let imageView: UIImageView = UIImageView(image: UIImage(named: "Logo", in: NYSMBundle, compatibleWith: nil))
+            imageView.layer.cornerRadius = 20
+            imageView.clipsToBounds = true
+
+            // create and display chat head with debug options
+            let chatHead: FCChatHeadsController = FCChatHeadsController()
+            chatHead.datasource = self
+            chatHead.delegate = self
+            chatHead.presentChatHead(with: imageView, chatID: "NowYouSeeMe.debugView")
+            self.chatHead = chatHead
+        }
     }
 }
 
 extension DebugHelper: FCChatHeadsControllerDatasource {
     internal func chatHeadsController(_ chatHeadsController: FCChatHeadsController!, viewForPopoverForChatHeadWithChatID chatID: String!) -> UIView! {
         // provide view for debug options
-        let bundle: Bundle = Bundle(for: Self.self)
-        let myView: DebugView? = bundle.loadNibNamed("DebugView", owner: nil, options: nil)?.first as? DebugView
-
-        return myView ?? UIView()
+        let bundle: Bundle = Bundle(for: DebugView.self)
+        if let NYSMBundleUrl: URL = bundle.url(forResource: "NowYouSeeMe", withExtension: "bundle"),
+           let NYSMBundle: Bundle = Bundle(url: NYSMBundleUrl),
+           let myView: DebugView = NYSMBundle.loadNibNamed("DebugView", owner: nil, options: nil)?.first as? DebugView {
+            return myView
+        }
+        return UIView()
     }
 }