firebase-ios-sdk/Firebase/InAppMessagingDisplay/FIRIAMDefaultDisplayImpl.m

/*
 * Copyright 2018 Google
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#import <Foundation/Foundation.h>

#import <FirebaseCore/FIRAppInternal.h>
#import <FirebaseInAppMessaging/FIRInAppMessaging.h>
#import <FirebaseInAppMessaging/FIRInAppMessagingRendering.h>

#import "FIDBannerViewController.h"
#import "FIDImageOnlyViewController.h"
#import "FIDModalViewController.h"
#import "FIDRenderingWindowHelper.h"
#import "FIDTimeFetcher.h"
#import "FIRCore+InAppMessagingDisplay.h"
#import "FIRIAMDefaultDisplayImpl.h"

@implementation FIRIAMDefaultDisplayImpl

+ (void)load {
  [[NSNotificationCenter defaultCenter] addObserver:self
                                           selector:@selector(didReceiveConfigureSDKNotification:)
                                               name:kFIRAppReadyToConfigureSDKNotification
                                             object:nil];
}

+ (void)didReceiveConfigureSDKNotification:(NSNotification *)notification {
  FIRLogDebug(kFIRLoggerInAppMessagingDisplay, @"I-FID100010",
              @"Got notification for kFIRAppReadyToConfigureSDKNotification. Setting display "
               "component on headless SDK.");

  FIRIAMDefaultDisplayImpl *display = [[FIRIAMDefaultDisplayImpl alloc] init];
  [FIRInAppMessaging inAppMessaging].messageDisplayComponent = display;
}

+ (NSBundle *)getViewResourceBundle {
  static NSBundle *resourceBundle;
  static dispatch_once_t onceToken;

  dispatch_once(&onceToken, ^{
    // TODO. This logic of finding the resource bundle may need to change once it's open
    // sourced
    NSBundle *containingBundle = [NSBundle mainBundle];
    // This is assuming the display resource bundle is contained in the main bundle
    NSURL *bundleURL = [containingBundle URLForResource:@"InAppMessagingDisplayResources"
                                          withExtension:@"bundle"];
    resourceBundle = [NSBundle bundleWithURL:bundleURL];

    if (resourceBundle == nil) {
      FIRLogWarning(kFIRLoggerInAppMessagingDisplay, @"I-FID100007",
                    @"FIAM Display Resource bundle "
                     "is missing: not contained within bundle %@",
                    containingBundle);
    }
  });
  return resourceBundle;
}

+ (void)displayModalViewWithMessageDefinition:(FIRInAppMessagingModalDisplay *)modalMessage
                              displayDelegate:
                                  (id<FIRInAppMessagingDisplayDelegate>)displayDelegate {
  NSBundle *resourceBundle = [self getViewResourceBundle];

  if (resourceBundle == nil) {
    NSError *error = [NSError errorWithDomain:kFirebaseInAppMessagingDisplayErrorDomain
                                         code:FIAMDisplayRenderErrorTypeUnspecifiedError
                                     userInfo:@{@"message" : @"resource bundle is missing"}];
    [displayDelegate displayErrorEncountered:error];
    return;
  }

  dispatch_async(dispatch_get_main_queue(), ^{
    FIDTimerWithNSDate *timeFetcher = [[FIDTimerWithNSDate alloc] init];
    FIDModalViewController *modalVC =
        [FIDModalViewController instantiateViewControllerWithResourceBundle:resourceBundle
                                                             displayMessage:modalMessage
                                                            displayDelegate:displayDelegate
                                                                timeFetcher:timeFetcher];

    if (modalVC == nil) {
      FIRLogWarning(kFIRLoggerInAppMessagingDisplay, @"I-FID100004",
                    @"View controller can not be created.");
      NSError *error = [NSError errorWithDomain:kFirebaseInAppMessagingDisplayErrorDomain
                                           code:FIAMDisplayRenderErrorTypeUnspecifiedError
                                       userInfo:@{}];
      [displayDelegate displayErrorEncountered:error];
      return;
    }

    UIWindow *displayUIWindow = [FIDRenderingWindowHelper UIWindowForModalView];
    displayUIWindow.rootViewController = modalVC;
    [displayUIWindow setHidden:NO];
  });
}

+ (void)displayBannerViewWithMessageDefinition:(FIRInAppMessagingBannerDisplay *)bannerMessage
                               displayDelegate:
                                   (id<FIRInAppMessagingDisplayDelegate>)displayDelegate {
  NSBundle *resourceBundle = [self getViewResourceBundle];

  if (resourceBundle == nil) {
    NSError *error = [NSError errorWithDomain:kFirebaseInAppMessagingDisplayErrorDomain
                                         code:FIAMDisplayRenderErrorTypeUnspecifiedError
                                     userInfo:@{}];
    [displayDelegate displayErrorEncountered:error];
    return;
  }

  dispatch_async(dispatch_get_main_queue(), ^{
    FIDTimerWithNSDate *timeFetcher = [[FIDTimerWithNSDate alloc] init];
    FIDBannerViewController *bannerVC =
        [FIDBannerViewController instantiateViewControllerWithResourceBundle:resourceBundle
                                                              displayMessage:bannerMessage
                                                             displayDelegate:displayDelegate
                                                                 timeFetcher:timeFetcher];

    if (bannerVC == nil) {
      FIRLogWarning(kFIRLoggerInAppMessagingDisplay, @"I-FID100008",
                    @"Banner view controller can not be created.");
      NSError *error = [NSError errorWithDomain:kFirebaseInAppMessagingDisplayErrorDomain
                                           code:FIAMDisplayRenderErrorTypeUnspecifiedError
                                       userInfo:@{}];
      [displayDelegate displayErrorEncountered:error];
      return;
    }

    UIWindow *displayUIWindow = [FIDRenderingWindowHelper UIWindowForBannerView];
    displayUIWindow.rootViewController = bannerVC;
    [displayUIWindow setHidden:NO];
  });
}

+ (void)displayImageOnlyViewWithMessageDefinition:
            (FIRInAppMessagingImageOnlyDisplay *)imageOnlyMessage
                                  displayDelegate:
                                      (id<FIRInAppMessagingDisplayDelegate>)displayDelegate {
  NSBundle *resourceBundle = [self getViewResourceBundle];

  if (resourceBundle == nil) {
    NSError *error = [NSError errorWithDomain:kFirebaseInAppMessagingDisplayErrorDomain
                                         code:FIAMDisplayRenderErrorTypeUnspecifiedError
                                     userInfo:@{}];
    [displayDelegate displayErrorEncountered:error];
    return;
  }

  dispatch_async(dispatch_get_main_queue(), ^{
    FIDTimerWithNSDate *timeFetcher = [[FIDTimerWithNSDate alloc] init];
    FIDImageOnlyViewController *imageOnlyVC =
        [FIDImageOnlyViewController instantiateViewControllerWithResourceBundle:resourceBundle
                                                                 displayMessage:imageOnlyMessage
                                                                displayDelegate:displayDelegate
                                                                    timeFetcher:timeFetcher];

    if (imageOnlyVC == nil) {
      FIRLogWarning(kFIRLoggerInAppMessagingDisplay, @"I-FID100006",
                    @"Image only view controller can not be created.");
      NSError *error = [NSError errorWithDomain:kFirebaseInAppMessagingDisplayErrorDomain
                                           code:FIAMDisplayRenderErrorTypeUnspecifiedError
                                       userInfo:@{}];
      [displayDelegate displayErrorEncountered:error];
      return;
    }

    UIWindow *displayUIWindow = [FIDRenderingWindowHelper UIWindowForImageOnlyView];
    displayUIWindow.rootViewController = imageOnlyVC;
    [displayUIWindow setHidden:NO];
  });
}

#pragma mark - protocol FIRInAppMessagingDisplay
- (void)displayMessage:(FIRInAppMessagingDisplayMessageBase *)messageForDisplay
       displayDelegate:(id<FIRInAppMessagingDisplayDelegate>)displayDelegate {
  if ([messageForDisplay isKindOfClass:[FIRInAppMessagingModalDisplay class]]) {
    FIRLogDebug(kFIRLoggerInAppMessagingDisplay, @"I-FID100000", @"Display a modal message");
    [self.class displayModalViewWithMessageDefinition:(FIRInAppMessagingModalDisplay *)
                                                          messageForDisplay
                                      displayDelegate:displayDelegate];

  } else if ([messageForDisplay isKindOfClass:[FIRInAppMessagingBannerDisplay class]]) {
    FIRLogDebug(kFIRLoggerInAppMessagingDisplay, @"I-FID100001", @"Display a banner message");
    [self.class displayBannerViewWithMessageDefinition:(FIRInAppMessagingBannerDisplay *)
                                                           messageForDisplay
                                       displayDelegate:displayDelegate];
  } else if ([messageForDisplay isKindOfClass:[FIRInAppMessagingImageOnlyDisplay class]]) {
    FIRLogDebug(kFIRLoggerInAppMessagingDisplay, @"I-FID100002", @"Display an image only message");
    [self.class displayImageOnlyViewWithMessageDefinition:(FIRInAppMessagingImageOnlyDisplay *)
                                                              messageForDisplay
                                          displayDelegate:displayDelegate];
  } else {
    FIRLogWarning(kFIRLoggerInAppMessagingDisplay, @"I-FID100003",
                  @"Unknown message type %@ "
                   "Don't know how to handle it.",
                  messageForDisplay.class);
    NSError *error = [NSError errorWithDomain:kFirebaseInAppMessagingDisplayErrorDomain
                                         code:FIAMDisplayRenderErrorTypeUnspecifiedError
                                     userInfo:@{}];
    [displayDelegate displayErrorEncountered:error];
  }
}
@end