diff options
Diffstat (limited to 'Firebase/Messaging/FIRMessagingRemoteNotificationsProxy.m')
| -rw-r--r-- | Firebase/Messaging/FIRMessagingRemoteNotificationsProxy.m | 7 |
1 files changed, 6 insertions, 1 deletions
diff --git a/Firebase/Messaging/FIRMessagingRemoteNotificationsProxy.m b/Firebase/Messaging/FIRMessagingRemoteNotificationsProxy.m index c5ad337..7cea178 100644 --- a/Firebase/Messaging/FIRMessagingRemoteNotificationsProxy.m +++ b/Firebase/Messaging/FIRMessagingRemoteNotificationsProxy.m @@ -21,6 +21,7 @@ #import "FIRMessagingConstants.h" #import "FIRMessagingLogger.h" +#import "FIRMessagingUtilities.h" #import "FIRMessaging_Private.h" static const BOOL kDefaultAutoRegisterEnabledValue = YES; @@ -98,7 +99,11 @@ static NSString *kReceiveDataMessageSelectorString = @"messaging:didReceiveMessa return; } - NSObject<UIApplicationDelegate> *appDelegate = [[UIApplication sharedApplication] delegate]; + UIApplication *application = FIRMessagingUIApplication(); + if (!application) { + return; + } + NSObject<UIApplicationDelegate> *appDelegate = [application delegate]; [self swizzleAppDelegateMethods:appDelegate]; // Add KVO listener on [UNUserNotificationCenter currentNotificationCenter]'s delegate property |