HtmlStyle.kt

package com.swmansion.enriched.textinput.styles

import android.graphics.Color
import com.facebook.react.bridge.ColorPropConverter
import com.facebook.react.bridge.ReactContext
import com.facebook.react.bridge.ReadableMap
import com.facebook.react.views.text.ReactTypefaceUtils.parseFontWeight
import com.swmansion.enriched.common.EnrichedConstants
import com.swmansion.enriched.common.EnrichedStyle
import com.swmansion.enriched.common.MentionStyle
import com.swmansion.enriched.common.pixelFromSpOrDp
import com.swmansion.enriched.textinput.EnrichedTextInputView
import kotlin.Float
import kotlin.Int
import kotlin.String
import kotlin.math.ceil

class HtmlStyle : EnrichedStyle {
  private var style: ReadableMap? = null
  private var view: EnrichedTextInputView? = null

  // Default values are ignored as they are specified on the JS side.
  // They are specified only because they are required by the constructor.
  // JS passes them as a prop - so they are initialized after the constructor is called.
  override var h1FontSize: Int = 72
  override var h1Bold: Boolean = false

  override var h2FontSize: Int = 64
  override var h2Bold: Boolean = false

  override var h3FontSize: Int = 56
  override var h3Bold: Boolean = false

  override var h4FontSize: Int = 48
  override var h4Bold: Boolean = false

  override var h5FontSize: Int = 40
  override var h5Bold: Boolean = false

  override var h6FontSize: Int = 32
  override var h6Bold: Boolean = false

  override var blockquoteColor: Int? = null
  override var blockquoteBorderColor: Int = Color.BLACK
  override var blockquoteStripeWidth: Int = 2
  override var blockquoteGapWidth: Int = 16

  override var olGapWidth: Int = 16
  override var olMarginLeft: Int = 24
  override var olMarkerFontWeight: Int? = null
  override var olMarkerColor: Int? = null

  override var ulGapWidth: Int = 16
  override var ulMarginLeft: Int = 24
  override var ulBulletSize: Int = 8
  override var ulBulletColor: Int = Color.BLACK

  override var ulCheckboxBoxSize: Int = 50
  override var ulCheckboxGapWidth: Int = 16
  override var ulCheckboxMarginLeft: Int = 24
  override var ulCheckboxBoxColor: Int = Color.BLACK

  override var aColor: Int = Color.BLACK
  override var aUnderline: Boolean = true

  override var codeBlockColor: Int = Color.BLACK
  override var codeBlockBackgroundColor: Int = Color.BLACK
  override var codeBlockRadius: Float = 4f

  override var inlineCodeColor: Int = Color.BLACK
  override var inlineCodeBackgroundColor: Int = Color.BLACK

  override var mentionsStyle: MutableMap<String, MentionStyle> = mutableMapOf()

  constructor(view: EnrichedTextInputView?, style: ReadableMap?) {
    this.view = view
    this.style = style

    invalidateStyles()
  }

  fun invalidateStyles() {
    val style = this.style ?: return

    val h1Style = style.getMap("h1")
    h1FontSize = parseFloat(h1Style, "fontSize").toInt()
    h1Bold = h1Style?.getBoolean("bold") == true

    val h2Style = style.getMap("h2")
    h2FontSize = parseFloat(h2Style, "fontSize").toInt()
    h2Bold = h2Style?.getBoolean("bold") == true

    val h3Style = style.getMap("h3")
    h3FontSize = parseFloat(h3Style, "fontSize").toInt()
    h3Bold = h3Style?.getBoolean("bold") == true

    val h4Style = style.getMap("h4")
    h4FontSize = parseFloat(h4Style, "fontSize").toInt()
    h4Bold = h4Style?.getBoolean("bold") == true

    val h5Style = style.getMap("h5")
    h5FontSize = parseFloat(h5Style, "fontSize").toInt()
    h5Bold = h5Style?.getBoolean("bold") == true

    val h6Style = style.getMap("h6")
    h6FontSize = parseFloat(h6Style, "fontSize").toInt()
    h6Bold = h6Style?.getBoolean("bold") == true

    val blockquoteStyle = style.getMap("blockquote")
    blockquoteColor = parseOptionalColor(blockquoteStyle, "color")
    blockquoteBorderColor = parseColor(blockquoteStyle, "borderColor")
    blockquoteGapWidth = parseFloat(blockquoteStyle, "gapWidth").toInt()
    blockquoteStripeWidth = parseFloat(blockquoteStyle, "borderWidth").toInt()

    val olStyle = style.getMap("ol")
    olMarginLeft = parseFloat(olStyle, "marginLeft").toInt()
    olGapWidth = parseFloat(olStyle, "gapWidth").toInt()
    olMarkerColor = parseOptionalColor(olStyle, "markerColor")
    olMarkerFontWeight = parseOptionalFontWeight(olStyle, "markerFontWeight")

    val ulStyle = style.getMap("ul")
    ulBulletColor = parseColor(ulStyle, "bulletColor")
    ulGapWidth = parseFloat(ulStyle, "gapWidth").toInt()
    ulMarginLeft = parseFloat(ulStyle, "marginLeft").toInt()
    ulBulletSize = parseFloat(ulStyle, "bulletSize").toInt()

    val ulCheckboxStyle = style.getMap("ulCheckbox")
    ulCheckboxBoxSize = parseFloat(ulCheckboxStyle, "boxSize").toInt()
    ulCheckboxGapWidth = parseFloat(ulCheckboxStyle, "gapWidth").toInt()
    ulCheckboxMarginLeft = parseFloat(ulCheckboxStyle, "marginLeft").toInt()
    ulCheckboxBoxColor = parseColor(ulCheckboxStyle, "boxColor")

    val aStyle = style.getMap("a")
    aColor = parseColor(aStyle, "color")
    aUnderline = parseIsUnderline(aStyle)

    val codeBlockStyle = style.getMap("codeblock")
    codeBlockRadius = parseFloat(codeBlockStyle, "borderRadius")
    codeBlockColor = parseColor(codeBlockStyle, "color")
    codeBlockBackgroundColor = parseColorWithOpacity(codeBlockStyle, "backgroundColor", 80)

    val inlineCodeStyle = style.getMap("code")
    inlineCodeColor = parseColor(inlineCodeStyle, "color")
    inlineCodeBackgroundColor = parseColorWithOpacity(inlineCodeStyle, "backgroundColor", 80)

    val mentionStyle = style.getMap("mention")
    mentionsStyle = parseMentionsStyle(mentionStyle)
  }

  private fun parseFloat(
    map: ReadableMap?,
    key: String,
  ): Float {
    val safeMap = ensureValueIsSet(map, key)
    val value = safeMap.getDouble(key)
    return ceil(pixelFromSpOrDp(value, view?.allowFontScaling ?: EnrichedConstants.ALLOW_FONT_SCALING_DEFAULT))
  }

  private fun parseColorWithOpacity(
    map: ReadableMap?,
    key: String,
    opacity: Int,
  ): Int {
    val color = parseColor(map, key)
    return withOpacity(color, opacity)
  }

  private fun parseOptionalColor(
    map: ReadableMap?,
    key: String,
  ): Int? {
    if (map == null) return null
    if (!map.hasKey(key)) return null
    if (map.isNull(key)) return null

    return parseColor(map, key)
  }

  private fun parseColor(
    map: ReadableMap?,
    key: String,
  ): Int {
    val safeMap = ensureValueIsSet(map, key)

    val color = safeMap.getDouble(key)
    val parsedColor = ColorPropConverter.getColor(color, view?.context as ReactContext)
    if (parsedColor == null) {
      throw Error("Specified color value is not supported: $color")
    }

    return parsedColor
  }

  private fun withOpacity(
    color: Int,
    alpha: Int,
  ): Int {
    if (Color.alpha(color) != 255) return color
    val a = alpha.coerceIn(0, 255)
    return (color and 0x00FFFFFF) or (a shl 24)
  }

  private fun parseIsUnderline(map: ReadableMap?): Boolean {
    val underline = map?.getString("textDecorationLine")
    val isEnabled = underline == "underline"
    val isDisabled = underline == "none"

    if (isEnabled) return true
    if (isDisabled) return false

    throw Error("Specified textDecorationLine value is not supported: $underline. Supported values are 'underline' and 'none'.")
  }

  private fun ensureValueIsSet(
    map: ReadableMap?,
    key: String,
  ): ReadableMap {
    if (map == null) throw Error("Style map cannot be null")

    if (!map.hasKey(key)) throw Error("Style map must contain key: $key")

    if (map.isNull(key)) throw Error("Style map cannot contain null value for key: $key")

    return map
  }

  private fun parseMentionsStyle(mentionsStyle: ReadableMap?): MutableMap<String, MentionStyle> {
    if (mentionsStyle == null) throw Error("Mentions style cannot be null")

    val parsedMentionsStyle: MutableMap<String, MentionStyle> = mutableMapOf()

    val iterator = mentionsStyle.keySetIterator()
    while (iterator.hasNextKey()) {
      val key = iterator.nextKey()
      val value = mentionsStyle.getMap(key)

      if (value == null) throw Error("Mention style for key '$key' cannot be null")

      val color = parseColor(value, "color")
      val backgroundColor = parseColorWithOpacity(value, "backgroundColor", 80)
      val isUnderline = parseIsUnderline(value)
      val parsedStyle = MentionStyle(color, backgroundColor, isUnderline)
      parsedMentionsStyle.put(key, parsedStyle)
    }

    return parsedMentionsStyle
  }

  private fun parseOptionalFontWeight(
    map: ReadableMap?,
    key: String,
  ): Int? {
    if (map == null) return null
    if (!map.hasKey(key)) return null
    if (map.isNull(key)) return null

    val fontWeight = map.getString(key) ?: return null
    return parseFontWeight(fontWeight)
  }

  override fun equals(other: Any?): Boolean {
    if (this === other) return true
    if (other !is HtmlStyle) return false

    return h1FontSize == other.h1FontSize &&
      h1Bold == other.h1Bold &&
      h2FontSize == other.h2FontSize &&
      h2Bold == other.h2Bold &&
      h3FontSize == other.h3FontSize &&
      h3Bold == other.h3Bold &&
      h4FontSize == other.h4FontSize &&
      h4Bold == other.h4Bold &&
      h5FontSize == other.h5FontSize &&
      h5Bold == other.h5Bold &&
      h6FontSize == other.h6FontSize &&
      h6Bold == other.h6Bold &&

      blockquoteColor == other.blockquoteColor &&
      blockquoteBorderColor == other.blockquoteBorderColor &&
      blockquoteStripeWidth == other.blockquoteStripeWidth &&
      blockquoteGapWidth == other.blockquoteGapWidth &&

      olGapWidth == other.olGapWidth &&
      olMarginLeft == other.olMarginLeft &&
      olMarkerFontWeight == other.olMarkerFontWeight &&
      olMarkerColor == other.olMarkerColor &&

      ulGapWidth == other.ulGapWidth &&
      ulMarginLeft == other.ulMarginLeft &&
      ulBulletSize == other.ulBulletSize &&
      ulBulletColor == other.ulBulletColor &&

      ulCheckboxBoxSize == other.ulCheckboxBoxSize &&
      ulCheckboxGapWidth == other.ulCheckboxGapWidth &&
      ulCheckboxMarginLeft == other.ulCheckboxMarginLeft &&
      ulCheckboxBoxColor == other.ulCheckboxBoxColor &&

      aColor == other.aColor &&
      aUnderline == other.aUnderline &&

      codeBlockColor == other.codeBlockColor &&
      codeBlockBackgroundColor == other.codeBlockBackgroundColor &&
      codeBlockRadius == other.codeBlockRadius &&

      inlineCodeColor == other.inlineCodeColor &&
      inlineCodeBackgroundColor == other.inlineCodeBackgroundColor &&

      mentionsStyle == other.mentionsStyle
  }

  override fun hashCode(): Int {
    var result = h1FontSize.hashCode()
    result = 31 * result + h1Bold.hashCode()
    result = 31 * result + h2FontSize.hashCode()
    result = 31 * result + h2Bold.hashCode()
    result = 31 * result + h3FontSize.hashCode()
    result = 31 * result + h3Bold.hashCode()
    result = 31 * result + h4FontSize.hashCode()
    result = 31 * result + h4Bold.hashCode()
    result = 31 * result + h5FontSize.hashCode()
    result = 31 * result + h5Bold.hashCode()
    result = 31 * result + h6FontSize.hashCode()
    result = 31 * result + h6Bold.hashCode()

    result = 31 * result + (blockquoteColor ?: 0)
    result = 31 * result + blockquoteBorderColor.hashCode()
    result = 31 * result + blockquoteStripeWidth.hashCode()
    result = 31 * result + blockquoteGapWidth.hashCode()

    result = 31 * result + olGapWidth.hashCode()
    result = 31 * result + olMarginLeft.hashCode()
    result = 31 * result + (olMarkerFontWeight?.hashCode() ?: 0)
    result = 31 * result + (olMarkerColor ?: 0)

    result = 31 * result + ulGapWidth.hashCode()
    result = 31 * result + ulMarginLeft.hashCode()
    result = 31 * result + ulBulletSize.hashCode()
    result = 31 * result + ulBulletColor.hashCode()

    result = 31 * result + ulCheckboxBoxSize.hashCode()
    result = 31 * result + ulCheckboxGapWidth.hashCode()
    result = 31 * result + ulCheckboxMarginLeft.hashCode()
    result = 31 * result + ulCheckboxBoxColor.hashCode()

    result = 31 * result + aColor.hashCode()
    result = 31 * result + aUnderline.hashCode()

    result = 31 * result + codeBlockColor.hashCode()
    result = 31 * result + codeBlockBackgroundColor.hashCode()
    result = 31 * result + codeBlockRadius.hashCode()

    result = 31 * result + inlineCodeColor.hashCode()
    result = 31 * result + inlineCodeBackgroundColor.hashCode()

    result = 31 * result + mentionsStyle.hashCode()

    return result
  }
}