MessageKit/Example/Sources/Views/CustomTextMessageContentCell.swift

//
//  CustomTextMessageContentCell.swift
//  ChatExample
//
//  Created by Vignesh J on 01/05/21.
//  Copyright © 2021 MessageKit. All rights reserved.
//

import MessageKit
import UIKit

class CustomTextMessageContentCell: CustomMessageContentCell {
  /// The label used to display the message's text.
  var messageLabel: UILabel = {
    let label = UILabel()
    label.numberOfLines = 0
    label.font = UIFont.preferredFont(forTextStyle: .body)

    return label
  }()

  override func prepareForReuse() {
    super.prepareForReuse()

    messageLabel.attributedText = nil
    messageLabel.text = nil
  }

  override func setupSubviews() {
    super.setupSubviews()

    messageContainerView.addSubview(messageLabel)
  }

  override func configure(
    with message: MessageType,
    at indexPath: IndexPath,
    in messagesCollectionView: MessagesCollectionView,
    dataSource: MessagesDataSource,
    and sizeCalculator: CustomLayoutSizeCalculator)
  {
    super.configure(
      with: message,
      at: indexPath,
      in: messagesCollectionView,
      dataSource: dataSource,
      and: sizeCalculator)

    guard let displayDelegate = messagesCollectionView.messagesDisplayDelegate else {
      return
    }

    let calculator = sizeCalculator as? CustomTextLayoutSizeCalculator
    messageLabel.frame = calculator?.messageLabelFrame(
      for: message,
      at: indexPath) ?? .zero

    let textMessageKind = message.kind
    switch textMessageKind {
    case .text(let text), .emoji(let text):
      let textColor = displayDelegate.textColor(for: message, at: indexPath, in: messagesCollectionView)
      messageLabel.text = text
      messageLabel.textColor = textColor
    case .attributedText(let text):
      messageLabel.attributedText = text
    default:
      break
    }
  }
}