firebase-ios-sdk/FirebaseInAppMessaging/Tests/Integration/DefaultUITestApp/FiamDisplaySwiftExample/BannerMessageViewController.swift

/*
 * 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 UIKit

class BannerMessageViewController: CommonMessageTestVC {
  class TestableBannerMessage: InAppMessagingBannerDisplay {
    var writableCampaignInfo: InAppMessagingCampaignInfo
    var writableTitle: String
    var writableBody: String?
    var writableTextColor: UIColor
    var writableImageData: InAppMessagingImageData?
    var writableBackgroundColor: UIColor
    var writableActionURL: URL?
    var writableMessageType: FIRInAppMessagingDisplayMessageType
    var writableTriggerType: FIRInAppMessagingDisplayTriggerType

    override var campaignInfo: InAppMessagingCampaignInfo {
      return writableCampaignInfo
    }

    override var title: String {
      return writableTitle
    }

    override var bodyText: String? {
      return writableBody
    }

    override var textColor: UIColor {
      return writableTextColor
    }

    override var imageData: InAppMessagingImageData? {
      return writableImageData
    }

    override var displayBackgroundColor: UIColor {
      return writableBackgroundColor
    }

    override var actionURL: URL? {
      return writableActionURL
    }

    override var type: FIRInAppMessagingDisplayMessageType {
      return writableMessageType
    }

    override var triggerType: FIRInAppMessagingDisplayTriggerType {
      return writableTriggerType
    }

    init(titleText: String,
         bodyText: String?,
         textColor: UIColor,
         backgroundColor: UIColor,
         imageData: InAppMessagingImageData?,
         actionURL: URL?) {
      writableTitle = titleText
      writableBody = bodyText
      writableTextColor = textColor
      writableImageData = imageData
      writableBackgroundColor = backgroundColor
      writableActionURL = actionURL
      writableCampaignInfo = TestableCampaignInfo(messageID: "testID",
                                                  campaignName: "testCampaign",
                                                  isTestMessage: false)
      writableMessageType = FIRInAppMessagingDisplayMessageType.banner
      writableTriggerType = FIRInAppMessagingDisplayTriggerType.onAnalyticsEvent
      super.init(messageID: "testID",
                 campaignName: "testCampaign",
                 renderAsTestMessage: false,
                 messageType: .imageOnly,
                 triggerType: .onAnalyticsEvent)
    }
  }

  let displayImpl = InAppMessagingDefaultDisplayImpl()

  @IBOutlet var verifyLabel: UILabel!

  override func messageClicked(_ inAppMessage: InAppMessagingDisplayMessage,
                               with action: InAppMessagingAction) {
    super.messageClicked(inAppMessage, with: action)
    verifyLabel.text = "message clicked!"
  }

  override func messageDismissed(_ inAppMessage: InAppMessagingDisplayMessage,
                                 dismissType: FIRInAppMessagingDismissType) {
    super.messageDismissed(inAppMessage, dismissType: dismissType)
    verifyLabel.text = "message dismissed!"
  }

  @IBAction func showRegularBannerTapped(_ sender: Any) {
    verifyLabel.text = "Verification Label"
    let imageRawData = produceImageOfSize(size: CGSize(width: 200, height: 200))
    let fiamImageData = InAppMessagingImageData(imageURL: "url not important",
                                                imageData: imageRawData!)

    let bannerMessage = TestableBannerMessage(
      titleText: normalMessageTitle,
      bodyText: normalMessageBody,
      textColor: UIColor.black,
      backgroundColor: UIColor.blue,
      imageData: fiamImageData,
      actionURL: URL(string: "http://firebase.com")
    )

    displayImpl.displayMessage(bannerMessage, displayDelegate: self)
  }

  @IBAction func showBannerViewWithoutImageTapped(_ sender: Any) {
    verifyLabel.text = "Verification Label"
    let bannerMessage = TestableBannerMessage(
      titleText: normalMessageTitle,
      bodyText: normalMessageBody,
      textColor: UIColor.black,
      backgroundColor: UIColor.blue,
      imageData: nil,
      actionURL: URL(string: "http://firebase.com")
    )

    displayImpl.displayMessage(bannerMessage, displayDelegate: self)
  }

  @IBAction func showBannerViewWithWideImageTapped(_ sender: Any) {
    verifyLabel.text = "Verification Label"
    let imageRawData = produceImageOfSize(size: CGSize(width: 800, height: 200))
    let fiamImageData = InAppMessagingImageData(imageURL: "url not important",
                                                imageData: imageRawData!)

    let bannerMessage = TestableBannerMessage(
      titleText: normalMessageTitle,
      bodyText: normalMessageBody,
      textColor: UIColor.black,
      backgroundColor: UIColor.blue,
      imageData: fiamImageData,
      actionURL: URL(string: "http://firebase.com")
    )

    displayImpl.displayMessage(bannerMessage, displayDelegate: self)
  }

  @IBAction func showBannerViewWithNarrowImageTapped(_ sender: Any) {
    verifyLabel.text = "Verification Label"
    let imageRawData = produceImageOfSize(size: CGSize(width: 200, height: 800))
    let fiamImageData = InAppMessagingImageData(imageURL: "url not important",
                                                imageData: imageRawData!)

    let bannerMessage = TestableBannerMessage(
      titleText: normalMessageTitle,
      bodyText: normalMessageBody,
      textColor: UIColor.black,
      backgroundColor: UIColor.blue,
      imageData: fiamImageData,
      actionURL: URL(string: "http://firebase.com")
    )

    displayImpl.displayMessage(bannerMessage, displayDelegate: self)
  }

  @IBAction func showBannerViewWithLargeBodyTextTapped(_ sender: Any) {
    verifyLabel.text = "Verification Label"
    let imageRawData = produceImageOfSize(size: CGSize(width: 200, height: 200))
    let fiamImageData = InAppMessagingImageData(imageURL: "url not important",
                                                imageData: imageRawData!)

    let bannerMessage = TestableBannerMessage(
      titleText: normalMessageTitle,
      bodyText: longBodyText,
      textColor: UIColor.black,
      backgroundColor: UIColor.blue,
      imageData: fiamImageData,
      actionURL: URL(string: "http://firebase.com")
    )

    displayImpl.displayMessage(bannerMessage, displayDelegate: self)
  }

  @IBAction func showBannerViewWithLongTitleTextTapped(_ sender: Any) {
    verifyLabel.text = "Verification Label"
    let imageRawData = produceImageOfSize(size: CGSize(width: 200, height: 200))
    let fiamImageData = InAppMessagingImageData(imageURL: "url not important",
                                                imageData: imageRawData!)

    let modalMessage = TestableBannerMessage(
      titleText: longTitleText,
      bodyText: normalMessageBody,
      textColor: UIColor.black,
      backgroundColor: UIColor.blue,
      imageData: fiamImageData,
      actionURL: URL(string: "http://firebase.com")
    )

    displayImpl.displayMessage(modalMessage, displayDelegate: self)
  }
}