Fix bug where app launch messages weren't appearing on iOS (#3237)

* Fix bug where messages with `app_launch` as a trigger weren't being parsed correctly

* Add a call to display executor check for app launch messages upon completion of initial app launch message fetch.

* Verify that the check for app launch messages gets called on initial app launch fetch

* Whitespace fix

* Fix duplicated error codes in checkAndDisplayNextAppLaunchMessage

* Refactor out logic for looking into app launch/foreground trigger messages to share more code

* Whitespace fix

Author: christibbs

Firebase/InAppMessaging/Data/FIRIAMFetchResponseParser.m

@@ -101,9 +101,12 @@ - (instancetype)initWithTimeFetcher:(id<FIRIAMTimeFetcher>)timeFetcher {
   NSMutableArray<FIRIAMDisplayTriggerDefinition *> *triggers = [[NSMutableArray alloc] init];
 
   for (NSDictionary *nextTriggerCondition in triggerConditions) {
+    // Handle app_launch and on_foreground cases.
     if (nextTriggerCondition[@"fiamTrigger"]) {
       if ([nextTriggerCondition[@"fiamTrigger"] isEqualToString:@"ON_FOREGROUND"]) {
         [triggers addObject:[[FIRIAMDisplayTriggerDefinition alloc] initForAppForegroundTrigger]];
+      } else if ([nextTriggerCondition[@"fiamTrigger"] isEqualToString:@"APP_LAUNCH"]) {
+        [triggers addObject:[[FIRIAMDisplayTriggerDefinition alloc] initForAppLaunchTrigger]];
       }
     } else if ([nextTriggerCondition[@"event"] isKindOfClass:[NSDictionary class]]) {
       NSDictionary *triggeringEvent = (NSDictionary *)nextTriggerCondition[@"event"];

Firebase/InAppMessaging/Data/FIRIAMMessageDefinition.h

@@ -15,6 +15,7 @@
  */
 #import <Foundation/Foundation.h>
 
+#import "FIRIAMDisplayTriggerDefinition.h"
 #import "FIRIAMMessageRenderData.h"
 
 @class FIRIAMDisplayTriggerDefinition;
@@ -52,8 +53,9 @@ NS_ASSUME_NONNULL_BEGIN
 - (BOOL)messageHasExpired;
 - (BOOL)messageHasStarted;
 
-// should this message be rendered when the app gets foregrounded?
-- (BOOL)messageRenderedOnAppForegroundEvent;
+// should this message be rendered given the FIAM trigger type? only use this method for app launch
+// and foreground trigger, use messageRenderedOnAnalyticsEvent: for analytics triggers
+- (BOOL)messageRenderedOnTrigger:(FIRIAMRenderTrigger)trigger;
 // should this message be rendered when a given analytics event is fired?
 - (BOOL)messageRenderedOnAnalyticsEvent:(NSString *)eventName;
 @end

Firebase/InAppMessaging/Data/FIRIAMMessageDefinition.m

@@ -15,7 +15,6 @@
  */
 
 #import "FIRIAMMessageDefinition.h"
-#import "FIRIAMDisplayTriggerDefinition.h"
 
 @implementation FIRIAMMessageRenderData
 
@@ -60,9 +59,9 @@ - (BOOL)messageHasExpired {
   return self.endTime < [[NSDate date] timeIntervalSince1970];
 }
 
-- (BOOL)messageRenderedOnAppForegroundEvent {
+- (BOOL)messageRenderedOnTrigger:(FIRIAMRenderTrigger)trigger {
   for (FIRIAMDisplayTriggerDefinition *nextTrigger in self.renderTriggers) {
-    if (nextTrigger.triggerType == FIRIAMRenderTriggerOnAppForeground) {
+    if (nextTrigger.triggerType == trigger) {
       return YES;
     }
   }

Firebase/InAppMessaging/DisplayTrigger/FIRIAMDisplayTriggerDefinition.h

@@ -17,6 +17,7 @@
 #import <Foundation/Foundation.h>
 
 typedef NS_ENUM(NSInteger, FIRIAMRenderTrigger) {
+  FIRIAMRenderTriggerOnAppLaunch,
   FIRIAMRenderTriggerOnAppForeground,
   FIRIAMRenderTriggerOnFirebaseAnalyticsEvent
 };
@@ -28,6 +29,7 @@ NS_ASSUME_NONNULL_BEGIN
 // applicable only when triggerType == FIRIAMRenderTriggerOnFirebaseAnalyticsEvent
 @property(nonatomic, copy, nullable, readonly) NSString *firebaseEventName;
 
+- (instancetype)initForAppLaunchTrigger;
 - (instancetype)initForAppForegroundTrigger;
 - (instancetype)initWithFirebaseAnalyticEvent:(NSString *)title;
 @end

Firebase/InAppMessaging/DisplayTrigger/FIRIAMDisplayTriggerDefinition.m

@@ -17,6 +17,14 @@
 #import "FIRIAMDisplayTriggerDefinition.h"
 
 @implementation FIRIAMDisplayTriggerDefinition
+
+- (instancetype)initForAppLaunchTrigger {
+  if (self = [super init]) {
+    _triggerType = FIRIAMRenderTriggerOnAppLaunch;
+  }
+  return self;
+}
+
 - (instancetype)initForAppForegroundTrigger {
   if (self = [super init]) {
     _triggerType = FIRIAMRenderTriggerOnAppForeground;

Firebase/InAppMessaging/Flows/FIRIAMDisplayExecutor.h

@@ -49,6 +49,8 @@ NS_ASSUME_NONNULL_BEGIN
                         activityLogger:(FIRIAMActivityLogger *)activityLogger
                   analyticsEventLogger:(id<FIRIAMAnalyticsEventLogger>)analyticsEventLogger;
 
+// Check and display next in-app message eligible for app launch trigger
+- (void)checkAndDisplayNextAppLaunchMessage;
 // Check and display next in-app message eligible for app open trigger
 - (void)checkAndDisplayNextAppForegroundMessage;
 // Check and display next in-app message eligible for analytics event trigger with given event name.

Firebase/InAppMessaging/Flows/FIRIAMDisplayExecutor.m

@@ -604,6 +604,49 @@ - (BOOL)enoughIntervalFromLastDisplay {
   return intervalFromLastDisplayInSeconds >= self.setting.displayMinIntervalInMinutes * 60.0;
 }
 
+- (void)checkAndDisplayNextAppLaunchMessage {
+  // synchronizing on self so that we won't potentially enter the render flow from two
+  // threads.
+  @synchronized(self) {
+    if (!self.messageDisplayComponent) {
+      FIRLogDebug(kFIRLoggerInAppMessaging, @"I-IAM400028",
+                  @"Message display component is not present yet. No display should happen.");
+      return;
+    }
+
+    if (self.suppressMessageDisplay) {
+      FIRLogDebug(kFIRLoggerInAppMessaging, @"I-IAM400029",
+                  @"Message display is being suppressed. No regular message rendering.");
+      return;
+    }
+
+    if (self.isMsgBeingDisplayed) {
+      FIRLogDebug(kFIRLoggerInAppMessaging, @"I-IAM400030",
+                  @"An in-app message display is in progress, do not over-display on top of it.");
+      return;
+    }
+
+    if ([self.messageCache hasTestMessage] || [self enoughIntervalFromLastDisplay]) {
+      // We can display test messages anytime or display regular messages when
+      // the display time interval has been reached
+      FIRIAMMessageDefinition *nextAppLaunchMessage = [self.messageCache nextOnAppLaunchDisplayMsg];
+
+      if (nextAppLaunchMessage) {
+        [self displayForMessage:nextAppLaunchMessage
+                    triggerType:FIRInAppMessagingDisplayTriggerTypeOnAnalyticsEvent];
+        self.lastDisplayTime = [self.timeFetcher currentTimestampInSeconds];
+      } else {
+        FIRLogDebug(kFIRLoggerInAppMessaging, @"I-IAM400040",
+                    @"No appropriate in-app message detected for display.");
+      }
+    } else {
+      FIRLogDebug(kFIRLoggerInAppMessaging, @"I-IAM400041",
+                  @"Minimal display interval of %lf seconds has not been reached yet.",
+                  self.setting.displayMinIntervalInMinutes * 60.0);
+    }
+  }
+}
+
 - (void)checkAndDisplayNextAppForegroundMessage {
   // synchronizing on self so that we won't potentially enter the render flow from two
   // threads: example like showing analytics triggered message and a regular app open

Firebase/InAppMessaging/Flows/FIRIAMFetchFlow.h

@@ -54,6 +54,7 @@ typedef void (^FIRIAMFetchMessageCompletionHandler)(
 
 // Triggers a potential fetch of in-app messaging from the source. It would check and respect the
 // the fetchMinIntervalInMinutes defined in setting
-- (void)checkAndFetch;
+- (void)checkAndFetchForInitialAppLaunch:(BOOL)forInitialAppLaunch;
+
 @end
 NS_ASSUME_NONNULL_END

Firebase/InAppMessaging/Flows/FIRIAMFetchFlow.m

@@ -19,6 +19,7 @@
 #import "FIRCore+InAppMessaging.h"
 #import "FIRIAMClearcutLogger.h"
 #import "FIRIAMFetchFlow.h"
+#import "FIRIAMRuntimeManager.h"
 
 @implementation FIRIAMFetchSetting
 @end
@@ -124,7 +125,7 @@ - (void)handleSuccessullyFetchedMessages:(NSArray<FIRIAMMessageDefinition *> *)m
                                    nextFetchWaitTime:fetchWaitTime];
 }
 
-- (void)checkAndFetch {
+- (void)checkAndFetchForInitialAppLaunch:(BOOL)forInitialAppLaunch {
   NSTimeInterval intervalFromLastFetchInSeconds =
       [self.timeFetcher currentTimestampInSeconds] - self.fetchBookKeeper.lastFetchTime;
 
@@ -229,6 +230,10 @@ - (void)checkAndFetch {
                              [self handleSuccessullyFetchedMessages:messages
                                                   withFetchWaitTime:nextFetchWaitTime
                                                  requestImpressions:impressions];
+
+                             if (forInitialAppLaunch) {
+                               [self checkForAppLaunchMessage];
+                             }
                            }
                            // Send this regardless whether fetch is successful or not.
                            [self sendFetchIsDoneNotification];
@@ -250,4 +255,9 @@ - (void)checkAndFetch {
     [self.activityLogger addLogRecord:record];
   }
 }
+
+- (void)checkForAppLaunchMessage {
+  [[FIRIAMRuntimeManager getSDKRuntimeInstance]
+          .displayExecutor checkAndDisplayNextAppLaunchMessage];
+}
 @end

Firebase/InAppMessaging/Flows/FIRIAMFetchOnAppForegroundFlow.m

@@ -35,7 +35,7 @@ - (void)appWillEnterForeground:(UIApplication *)application {
   // to a concurrent global queue instead of serial queue since app open event won't happen at
   // fast speed to cause race conditions
   dispatch_async(dispatch_get_global_queue(DISPATCH_QUEUE_PRIORITY_DEFAULT, 0ul), ^{
-    [self checkAndFetch];
+    [self checkAndFetchForInitialAppLaunch:NO];
   });
 }