@mathers101/react-native-render-rich-text 0.1.9 → 0.1.14

package/android/src/main/java/com/renderrichtext/MeasurementStore.kt

@@ -1,644 +1,644 @@
-package com.renderrichtext
-
-import android.content.Context
-import android.graphics.Typeface
-import android.os.Build
-import android.text.Spannable
-import android.text.Spanned
-import android.text.StaticLayout
-import android.text.TextPaint
-import android.text.TextUtils.TruncateAt
-import android.util.Log
-import com.facebook.react.bridge.ReadableMap
-import com.facebook.react.uimanager.PixelUtil
-import com.facebook.yoga.YogaMeasureMode
-import com.facebook.yoga.YogaMeasureOutput
-import com.renderrichtext.parser.Md4cFlags
-import com.renderrichtext.parser.Parser
-import com.renderrichtext.renderer.Renderer
-import com.renderrichtext.spans.MarginBottomSpan
-import com.renderrichtext.styles.LinkPillContent
-import com.renderrichtext.styles.StyleConfig
-import com.renderrichtext.utils.common.BreakStrategyUtils
-import com.renderrichtext.utils.common.EllipsizeUtils
-import com.renderrichtext.utils.common.LinkRegexConfig
-import com.renderrichtext.utils.common.getArrayOrNull
-import com.renderrichtext.utils.common.getBooleanOrDefault
-import com.renderrichtext.utils.common.getIntOrDefault
-import com.renderrichtext.utils.common.getMapOrNull
-import com.renderrichtext.utils.common.getStringOrDefault
-import com.renderrichtext.utils.common.parseImageRequestHeaders
-import com.renderrichtext.utils.common.parseLinkPillContent
-import com.renderrichtext.utils.common.parseLinkRecognition
-import com.renderrichtext.utils.text.span.prepareWidthAwareSpans
-import java.util.concurrent.ConcurrentHashMap
-import kotlin.math.ceil
-
-/**
- * Manages text measurements for ShadowNode layout.
- * Parses and renders tiptap to Spannable at measure time for accurate height calculation.
- */
-object MeasurementStore {
-  private const val TAG = "MeasurementStore"
-
-  private data class PaintParams(
-    val typeface: Typeface,
-    val fontSize: Float,
-  )
-
-  private data class MeasurementParams(
-    val cachedWidth: Float,
-    val cachedSize: Long,
-    val spannable: CharSequence?,
-    val paintParams: PaintParams,
-    val tiptapHash: Int,
-  )
-
-  private val data = ConcurrentHashMap<Int, MeasurementParams>()
-
-  // Store font scaling settings per view ID
-  private data class FontScalingSettings(
-    val allowFontScaling: Boolean = true,
-    val maxFontSizeMultiplier: Float = 0f,
-  )
-
-  private val fontScalingSettings = ConcurrentHashMap<Int, FontScalingSettings>()
-
-  private val breakStrategies = ConcurrentHashMap<Int, String>()
-
-  // The measure path and the display TextView resolve the same clamp. 0 means unlimited.
-  private val numberOfLinesByViewId = ConcurrentHashMap<Int, Int>()
-
-  private val ellipsizeModeByViewId = ConcurrentHashMap<Int, String>()
-
-  // Code blocks the user wrapped, by document-order index across nesting (see CodeBlockWrapState).
-  private val wrappedCodeBlocks = ConcurrentHashMap<Int, Set<Int>>()
-
-  private fun resolveFontScalingSettings(
-    viewId: Int?,
-    props: ReadableMap?,
-  ): FontScalingSettings {
-    val stored = viewId?.let { fontScalingSettings[it] }
-    return FontScalingSettings(
-      allowFontScaling =
-        props?.takeIf { it.hasKey("allowFontScaling") }?.getBoolean("allowFontScaling")
-          ?: stored?.allowFontScaling
-          ?: true,
-      maxFontSizeMultiplier =
-        props?.takeIf { it.hasKey("maxFontSizeMultiplier") }?.getDouble("maxFontSizeMultiplier")?.toFloat()
-          ?: stored?.maxFontSizeMultiplier
-          ?: 0f,
-    )
-  }
-
-  private val measurePaint = TextPaint()
-  private val measureRenderer = Renderer()
-
-  @Volatile
-  private var lastKnownFontScale: Float = 1.0f
-
-  /** Updates measurement with rendered Spannable. Returns true if height changed. */
-  fun store(
-    id: Int,
-    spannable: CharSequence?,
-    paint: TextPaint,
-    trailingMarginBottomPx: Float = 0f,
-  ): Boolean {
-    val cached = data[id]
-    val width = cached?.cachedWidth ?: 0f
-    val oldSize = cached?.cachedSize ?: 0L
-    val existingHash = cached?.tiptapHash ?: 0
-    val paintParams = PaintParams(paint.typeface ?: Typeface.DEFAULT, paint.textSize)
-
-    var newSize = measure(width, spannable, paint, id)
-    if (trailingMarginBottomPx > 0f) {
-      newSize =
-        YogaMeasureOutput.make(
-          YogaMeasureOutput.getWidth(newSize),
-          YogaMeasureOutput.getHeight(newSize) + PixelUtil.toDIPFromPixel(trailingMarginBottomPx),
-        )
-    }
-    data[id] = MeasurementParams(width, newSize, spannable, paintParams, existingHash)
-    return oldSize != newSize
-  }
-
-  fun release(id: Int) {
-    data.remove(id)
-    numberOfLinesByViewId.remove(id)
-    ellipsizeModeByViewId.remove(id)
-    wrappedCodeBlocks.remove(id)
-  }
-
-  fun updateWrappedCodeBlocks(
-    viewId: Int,
-    indices: Set<Int>,
-  ) {
-    if (indices.isEmpty()) wrappedCodeBlocks.remove(viewId) else wrappedCodeBlocks[viewId] = indices
-    data.remove(viewId)
-  }
-
-  fun invalidate(id: Int) {
-    data.remove(id)
-  }
-
-  // Content-box width (px) the measure pass last wrapped at; the display view insets to match (#805).
-  fun contentBoxWidthPx(id: Int): Float? = data[id]?.cachedWidth
-
-  /** Main entry point for ShadowNode measurement. */
-  fun getMeasureById(
-    context: Context,
-    id: Int?,
-    width: Float,
-    widthMode: YogaMeasureMode?,
-    height: Float,
-    heightMode: YogaMeasureMode?,
-    props: ReadableMap?,
-  ): Long {
-    // Early exit for empty tiptap. Honor widthMode here too: empty content has
-    // zero intrinsic width, so only fill when the width is EXACTLY assigned -
-    // otherwise an empty view would claim all available horizontal space.
-    val tiptap = props.getStringOrDefault("document", "")
-    if (tiptap.isEmpty()) {
-      val emptyWidth = if (widthMode === YogaMeasureMode.EXACTLY) PixelUtil.toDIPFromPixel(width) else 0f
-      return YogaMeasureOutput.make(emptyWidth, 0f)
-    }
-
-    // Resolve the line clamp from props here so the (background) measure pass and
-    // the (main-thread) display TextView never race on the per-viewId cache. Only
-    // overwrite when the key is present so we never clobber the view's value.
-    if (id != null && props != null) {
-      if (props.hasKey("numberOfLines")) {
-        numberOfLinesByViewId[id] = props.getDouble("numberOfLines").toInt()
-      }
-      if (props.hasKey("ellipsizeMode")) {
-        ellipsizeModeByViewId[id] = props.getString("ellipsizeMode") ?: EllipsizeUtils.DEFAULT_MODE
-      }
-    }
-
-    val size = getMeasureByIdInternal(context, id, width, props)
-
-    // Honor the Yoga width contract. When the parent assigns an exact width
-    // (flex child, alignItems: stretch, explicit width / width: "100%"), fill it
-    // instead of reporting the content's widest line, which would make the node
-    // hug its content and wrap early. Mirrors RN's TextLayoutManager.calculateWidth;
-    // iOS already does this via ENRMClampMeasuredSize.
-    val resultWidth =
-      if (widthMode === YogaMeasureMode.EXACTLY) {
-        PixelUtil.toDIPFromPixel(width)
-      } else {
-        YogaMeasureOutput.getWidth(size)
-      }
-
-    var resultHeight = YogaMeasureOutput.getHeight(size)
-    if (heightMode === YogaMeasureMode.AT_MOST) {
-      resultHeight = resultHeight.coerceAtMost(PixelUtil.toDIPFromPixel(height))
-    }
-
-    return YogaMeasureOutput.make(resultWidth, resultHeight)
-  }
-
-  fun updateFontScalingSettings(
-    viewId: Int,
-    allowFontScaling: Boolean,
-    maxFontSizeMultiplier: Float,
-  ) {
-    val previous = fontScalingSettings[viewId]
-    fontScalingSettings[viewId] = FontScalingSettings(allowFontScaling, maxFontSizeMultiplier)
-    if (previous != null && (previous.allowFontScaling != allowFontScaling || previous.maxFontSizeMultiplier != maxFontSizeMultiplier)) {
-      data.remove(viewId)
-    }
-  }
-
-  fun clearFontScalingSettings(viewId: Int) {
-    fontScalingSettings.remove(viewId)
-  }
-
-  fun updateBreakStrategy(
-    viewId: Int,
-    strategy: String,
-  ) {
-    breakStrategies[viewId] = strategy
-  }
-
-  fun clearBreakStrategy(viewId: Int) {
-    breakStrategies.remove(viewId)
-  }
-
-  private fun resolveBreakStrategy(viewId: Int?): Int = BreakStrategyUtils.resolveBreakStrategy(viewId?.let { breakStrategies[it] })
-
-  fun updateNumberOfLines(
-    viewId: Int,
-    numberOfLines: Int,
-  ) {
-    numberOfLinesByViewId[viewId] = numberOfLines
-  }
-
-  fun updateEllipsizeMode(
-    viewId: Int,
-    mode: String,
-  ) {
-    ellipsizeModeByViewId[viewId] = mode
-  }
-
-  private fun resolveMaxLines(viewId: Int?): Int = viewId?.let { numberOfLinesByViewId[it] } ?: 0
-
-  private fun resolveEllipsize(viewId: Int?): TruncateAt? =
-    EllipsizeUtils.resolveTruncateAt(viewId?.let { ellipsizeModeByViewId[it] } ?: EllipsizeUtils.DEFAULT_MODE)
-
-  private fun getMeasureByIdInternal(
-    context: Context,
-    id: Int?,
-    width: Float,
-    props: ReadableMap?,
-  ): Long {
-    val (allowFontScaling, maxFontSizeMultiplier) = resolveFontScalingSettings(id, props)
-
-    val fontScale = checkAndUpdateFontScale(context, allowFontScaling, maxFontSizeMultiplier)
-
-    val safeId = id ?: return measureAndCache(context, null, width, props, allowFontScaling, fontScale, maxFontSizeMultiplier)
-    val cached = data[safeId] ?: return measureAndCache(context, safeId, width, props, allowFontScaling, fontScale, maxFontSizeMultiplier)
-
-    val currentHash = computePropsHash(props, allowFontScaling, fontScale, maxFontSizeMultiplier)
-
-    if (cached.tiptapHash != currentHash) {
-      return measureAndCache(context, safeId, width, props, allowFontScaling, fontScale, maxFontSizeMultiplier)
-    }
-
-    // Width changed - re-measure with cached spannable
-    if (cached.cachedWidth != width) {
-      val newSize = measure(width, cached.spannable, cached.paintParams, safeId)
-      data[safeId] = cached.copy(cachedWidth = width, cachedSize = newSize)
-      return newSize
-    }
-
-    return cached.cachedSize
-  }
-
-  private fun computePropsHash(
-    props: ReadableMap?,
-    allowFontScaling: Boolean,
-    fontScale: Float,
-    maxFontSizeMultiplier: Float,
-  ): Int {
-    val tiptap = props.getStringOrDefault("document", "")
-    return computePropsHashForTiptap(tiptap, props, allowFontScaling, fontScale, maxFontSizeMultiplier)
-  }
-
-  private fun computePropsHashForTiptap(
-    tiptap: String,
-    props: ReadableMap?,
-    allowFontScaling: Boolean,
-    fontScale: Float,
-    maxFontSizeMultiplier: Float,
-  ): Int {
-    val styleMap = props.getMapOrNull("tiptapStyle")
-    val md4cFlagsMap = props.getMapOrNull("md4cFlags")
-    val allowTrailingMargin = props.getBooleanOrDefault("allowTrailingMargin", false)
-    val imageRequestHeaders = parseImageRequestHeaders(props.getArrayOrNull("imageRequestHeaders"))
-    val linkPillContent = parseLinkPillContent(props.getArrayOrNull("linkPillContent"))
-    var result = tiptap.hashCode()
-    result = 31 * result + (styleMap?.hashCode() ?: 0)
-    result = 31 * result + (md4cFlagsMap?.hashCode() ?: 0)
-    result = 31 * result + parseLinkRecognition(props.getMapOrNull("linkRecognition")).hashCode()
-    result = 31 * result + fontScale.toBits()
-    result = 31 * result + allowFontScaling.hashCode()
-    result = 31 * result + maxFontSizeMultiplier.toBits()
-    result = 31 * result + allowTrailingMargin.hashCode()
-    result = 31 * result + imageRequestHeaders.hashCode()
-    result = 31 * result + linkPillContent.hashCode()
-    result = 31 * result + props.getIntOrDefault("numberOfLines", 0)
-    result = 31 * result + props.getStringOrDefault("ellipsizeMode", EllipsizeUtils.DEFAULT_MODE).hashCode()
-    return result
-  }
-
-  private fun checkAndUpdateFontScale(
-    context: Context,
-    allowFontScaling: Boolean,
-    maxFontSizeMultiplier: Float,
-  ): Float {
-    if (!allowFontScaling) {
-      // Clear cache if we switched from scaling to non-scaling
-      if (lastKnownFontScale != 1.0f) {
-        lastKnownFontScale = 1.0f
-        data.clear()
-      }
-      return 1.0f
-    }
-
-    var currentFontScale = context.resources.configuration.fontScale
-
-    if (maxFontSizeMultiplier >= 1.0f && currentFontScale > maxFontSizeMultiplier) {
-      currentFontScale = maxFontSizeMultiplier
-    }
-    if (currentFontScale != lastKnownFontScale) {
-      lastKnownFontScale = currentFontScale
-      data.clear()
-    }
-    return currentFontScale
-  }
-
-  private fun measureAndCache(
-    context: Context,
-    id: Int?,
-    width: Float,
-    props: ReadableMap?,
-    allowFontScaling: Boolean,
-    fontScale: Float,
-    maxFontSizeMultiplier: Float,
-  ): Long {
-    // 1. Extract Props & Setup
-    val tiptap = props.getStringOrDefault("document", "")
-    val styleMap = props.getMapOrNull("tiptapStyle")
-    val md4cFlags =
-      Md4cFlags(
-        underline = props.getMapOrNull("md4cFlags").getBooleanOrDefault("underline", false),
-        latexMath = false,
-        superscript = props.getMapOrNull("md4cFlags").getBooleanOrDefault("superscript", false),
-        subscript = props.getMapOrNull("md4cFlags").getBooleanOrDefault("subscript", false),
-        highlight = props.getMapOrNull("md4cFlags").getBooleanOrDefault("highlight", false),
-        hardSoftBreaks = props.getMapOrNull("md4cFlags").getBooleanOrDefault("hardSoftBreaks", false),
-        preserveBlankLines = props.getMapOrNull("md4cFlags").getBooleanOrDefault("preserveBlankLines", false),
-        admonitions = props.getMapOrNull("md4cFlags").getBooleanOrDefault("admonitions", true),
-      )
-    val isGFM = props.getBooleanOrDefault("isGFM", false)
-
-    val fontSize = getInitialFontSize(styleMap, context, allowFontScaling, fontScale, maxFontSizeMultiplier)
-    val propsHash = computePropsHash(props, allowFontScaling, fontScale, maxFontSizeMultiplier)
-
-    // 2. Render & Measure
-    measurePaint.textSize = fontSize
-    val imageRequestHeaders = parseImageRequestHeaders(props.getArrayOrNull("imageRequestHeaders"))
-    val linkRecognition = parseLinkRecognition(props.getMapOrNull("linkRecognition"))
-    val spannable =
-      tryRenderTiptap(
-        tiptap,
-        styleMap,
-        context,
-        md4cFlags,
-        isGFM,
-        allowFontScaling,
-        maxFontSizeMultiplier,
-        imageRequestHeaders,
-        parseLinkPillContent(props.getArrayOrNull("linkPillContent")),
-        linkRecognition.text,
-        linkRecognition.inlineCode,
-      )
-    val textToMeasure = spannable ?: tiptap
-    val (size, _) = measureWithLayout(width, textToMeasure, measurePaint, id)
-
-    // 3. Calculate Margin
-    val allowTrailingMargin = props.getBooleanOrDefault("allowTrailingMargin", false)
-    val marginBottom =
-      if (allowTrailingMargin && spannable != null) {
-        PixelUtil.toDIPFromPixel(measureRenderer.getLastElementMarginBottom())
-      } else {
-        0f
-      }
-
-    // 4. Finalize Height
-    val currentWidth = YogaMeasureOutput.getWidth(size)
-    val currentHeight = YogaMeasureOutput.getHeight(size)
-    val adjustedSize = YogaMeasureOutput.make(currentWidth, currentHeight + marginBottom)
-
-    if (id != null) {
-      data[id] = MeasurementParams(width, adjustedSize, textToMeasure, PaintParams(Typeface.DEFAULT, fontSize), propsHash)
-    }
-
-    return adjustedSize
-  }
-
-  private fun createStaticLayout(
-    text: CharSequence,
-    fontSize: Float,
-    widthPx: Int,
-    viewId: Int?,
-  ): StaticLayout {
-    measurePaint.textSize = fontSize
-    prepareWidthAwareSpans(text, widthPx)
-    return StaticLayout.Builder
-      .obtain(text, 0, text.length, measurePaint, widthPx)
-      .setIncludePad(false)
-      .setLineSpacing(0f, 1f)
-      .apply {
-        if (Build.VERSION.SDK_INT >= Build.VERSION_CODES.M) {
-          @Suppress("WrongConstant")
-          setBreakStrategy(resolveBreakStrategy(viewId))
-        }
-        if (Build.VERSION.SDK_INT >= Build.VERSION_CODES.P) {
-          setUseLineSpacingFromFallbacks(true)
-        }
-      }.build()
-  }
-
-  private fun tryRenderTiptap(
-    tiptap: String,
-    styleMap: ReadableMap?,
-    context: Context,
-    md4cFlags: Md4cFlags,
-    isGFM: Boolean,
-    allowFontScaling: Boolean,
-    maxFontSizeMultiplier: Float,
-    imageRequestHeaders: Map<String, String> = emptyMap(),
-    linkPillContent: Map<String, LinkPillContent> = emptyMap(),
-    linkRegex: LinkRegexConfig? = null,
-    inlineCodeLinkRegex: LinkRegexConfig? = null,
-  ): Spannable? {
-    if (styleMap == null) return null
-
-    return try {
-      val ast = Parser.shared.parseTiptap(tiptap, md4cFlags, isGFM, linkRegex, inlineCodeLinkRegex) ?: return null
-      val style = StyleConfig(styleMap, context, allowFontScaling, maxFontSizeMultiplier)
-      style.imageRequestHeaders = imageRequestHeaders
-      style.linkPillContent = linkPillContent
-      measureRenderer.configure(style, context)
-      measureRenderer.renderDocument(ast, null)
-    } catch (e: Exception) {
-      Log.w(TAG, "Failed to render tiptap for measurement, falling back to raw text", e)
-      null
-    }
-  }
-
-  private fun getInitialFontSize(
-    styleMap: ReadableMap?,
-    context: Context,
-    allowFontScaling: Boolean,
-    fontScale: Float,
-    maxFontSizeMultiplier: Float,
-  ): Float {
-    val fontSizeSp = styleMap?.getMap("paragraph")?.getDouble("fontSize")?.toFloat() ?: 16f
-    val density = context.resources.displayMetrics.density
-
-    if (!allowFontScaling) {
-      return ceil(fontSizeSp * density)
-    }
-
-    val cappedFontScale =
-      if (maxFontSizeMultiplier >= 1.0f && fontScale > maxFontSizeMultiplier) {
-        maxFontSizeMultiplier
-      } else {
-        fontScale
-      }
-    return ceil(fontSizeSp * cappedFontScale * density)
-  }
-
-  private fun measure(
-    maxWidth: Float,
-    text: CharSequence?,
-    paintParams: PaintParams,
-    viewId: Int?,
-  ): Long {
-    measurePaint.reset()
-    measurePaint.typeface = paintParams.typeface
-    measurePaint.textSize = paintParams.fontSize
-    return measure(maxWidth, text, measurePaint, viewId)
-  }
-
-  private fun measure(
-    maxWidth: Float,
-    text: CharSequence?,
-    paint: TextPaint,
-    viewId: Int?,
-  ): Long {
-    val content = text ?: ""
-    val safeWidth = ceil(maxWidth).toInt().coerceAtLeast(1)
-    prepareWidthAwareSpans(content, safeWidth)
-
-    val builder =
-      StaticLayout.Builder
-        .obtain(content, 0, content.length, paint, safeWidth)
-        .setIncludePad(false)
-        .setLineSpacing(0f, 1f)
-
-    if (Build.VERSION.SDK_INT >= Build.VERSION_CODES.M) {
-      @Suppress("WrongConstant")
-      builder.setBreakStrategy(resolveBreakStrategy(viewId))
-    }
-
-    if (Build.VERSION.SDK_INT >= Build.VERSION_CODES.P) {
-      builder.setUseLineSpacingFromFallbacks(true)
-    }
-
-    applyLineLimit(builder, viewId)
-
-    val layout = builder.build()
-    val visibleLineCount = visibleLineCount(layout, viewId)
-    val measuredHeight = truncatedHeight(layout, visibleLineCount, content)
-
-    // Calculate actual content width (widest visible line)
-    val measuredWidth = (0 until visibleLineCount).maxOfOrNull { layout.getLineWidth(it) } ?: 0f
-
-    return YogaMeasureOutput.make(
-      PixelUtil.toDIPFromPixel(ceil(measuredWidth)),
-      PixelUtil.toDIPFromPixel(measuredHeight),
-    )
-  }
-
-  // Applies the per-view numberOfLines / ellipsizeMode clamp to a builder, matching
-  // RN's TextLayoutManager.buildLayout: only when a line limit is set.
-  private fun applyLineLimit(
-    builder: StaticLayout.Builder,
-    viewId: Int?,
-  ) {
-    val maxLines = resolveMaxLines(viewId)
-    if (maxLines > 0) {
-      builder.setMaxLines(maxLines).setEllipsize(resolveEllipsize(viewId))
-    }
-  }
-
-  private fun visibleLineCount(
-    layout: StaticLayout,
-    viewId: Int?,
-  ): Int {
-    val maxLines = resolveMaxLines(viewId)
-    return if (maxLines > 0) minOf(maxLines, layout.lineCount) else layout.lineCount
-  }
-
-  private fun truncatedHeight(
-    layout: StaticLayout,
-    visibleLineCount: Int,
-    text: CharSequence,
-  ): Float {
-    if (visibleLineCount <= 0) return 0f
-    val lastLine = visibleLineCount - 1
-    val bottom = layout.getLineBottom(lastLine).toFloat()
-    return (bottom - measureLeakedTrailingMargin(layout, lastLine, text)).coerceAtLeast(0f)
-  }
-
-  // When we clamp the text mid-paragraph, drop the trailing block margin that Android
-  // bakes into the truncated last line (only when content follows, like MarginBottomSpan).
-  private fun measureLeakedTrailingMargin(
-    layout: StaticLayout,
-    lastLine: Int,
-    text: CharSequence,
-  ): Float {
-    val spanned = text as? Spanned ?: return 0f
-    val lineStart = layout.getLineStart(lastLine)
-    val lineEnd = layout.getLineEnd(lastLine)
-    if (lineEnd <= lineStart || lineEnd < 1 || text[lineEnd - 1] != '\n') return 0f
-    if (!hasContentAfter(text, lineEnd)) return 0f
-    val trailing =
-      spanned
-        .getSpans(lineStart, lineEnd, MarginBottomSpan::class.java)
-        .firstOrNull { spanned.getSpanEnd(it) == lineEnd } ?: return 0f
-    return trailing.marginBottom.toInt().toFloat()
-  }
-
-  private fun hasContentAfter(
-    text: CharSequence,
-    pos: Int,
-  ): Boolean {
-    if (pos >= text.length) return false
-    if (text[pos] == '\n') {
-      val nextPos = pos + 1
-      if (nextPos >= text.length) return false
-      return text[nextPos] != '\n'
-    }
-    return true
-  }
-
-  /**
-   * Measures text and returns both the size and the layout for calculating last line descent.
-   */
-  private fun measureWithLayout(
-    maxWidth: Float,
-    text: CharSequence?,
-    paint: TextPaint,
-    viewId: Int?,
-  ): Pair<Long, StaticLayout> {
-    val content = text ?: ""
-    val widthPx = ceil(maxWidth).toInt().coerceAtLeast(1)
-    prepareWidthAwareSpans(content, widthPx)
-
-    val layout =
-      StaticLayout.Builder
-        .obtain(content, 0, content.length, paint, widthPx)
-        .setIncludePad(false)
-        .setLineSpacing(0f, 1f)
-        .apply {
-          if (Build.VERSION.SDK_INT >= Build.VERSION_CODES.M) {
-            @Suppress("WrongConstant")
-            setBreakStrategy(resolveBreakStrategy(viewId))
-          }
-          if (Build.VERSION.SDK_INT >= Build.VERSION_CODES.P) {
-            setUseLineSpacingFromFallbacks(true)
-          }
-          applyLineLimit(this, viewId)
-        }.build()
-
-    val visibleLineCount = visibleLineCount(layout, viewId)
-
-    // Find the widest visible line to get the actual content width
-    val maxLineWidth =
-      (0 until visibleLineCount)
-        .maxOfOrNull { layout.getLineWidth(it) } ?: 0f
-
-    val size =
-      YogaMeasureOutput.make(
-        PixelUtil.toDIPFromPixel(ceil(maxLineWidth)),
-        PixelUtil.toDIPFromPixel(truncatedHeight(layout, visibleLineCount, content)),
-      )
-
-    return size to layout
-  }
-
-}
+package com.renderrichtext
+
+import android.content.Context
+import android.graphics.Typeface
+import android.os.Build
+import android.text.Spannable
+import android.text.Spanned
+import android.text.StaticLayout
+import android.text.TextPaint
+import android.text.TextUtils.TruncateAt
+import android.util.Log
+import com.facebook.react.bridge.ReadableMap
+import com.facebook.react.uimanager.PixelUtil
+import com.facebook.yoga.YogaMeasureMode
+import com.facebook.yoga.YogaMeasureOutput
+import com.renderrichtext.parser.Md4cFlags
+import com.renderrichtext.parser.Parser
+import com.renderrichtext.renderer.Renderer
+import com.renderrichtext.spans.MarginBottomSpan
+import com.renderrichtext.styles.LinkPillContent
+import com.renderrichtext.styles.StyleConfig
+import com.renderrichtext.utils.common.BreakStrategyUtils
+import com.renderrichtext.utils.common.EllipsizeUtils
+import com.renderrichtext.utils.common.LinkRegexConfig
+import com.renderrichtext.utils.common.getArrayOrNull
+import com.renderrichtext.utils.common.getBooleanOrDefault
+import com.renderrichtext.utils.common.getIntOrDefault
+import com.renderrichtext.utils.common.getMapOrNull
+import com.renderrichtext.utils.common.getStringOrDefault
+import com.renderrichtext.utils.common.parseImageRequestHeaders
+import com.renderrichtext.utils.common.parseLinkPillContent
+import com.renderrichtext.utils.common.parseLinkRecognition
+import com.renderrichtext.utils.text.span.prepareWidthAwareSpans
+import java.util.concurrent.ConcurrentHashMap
+import kotlin.math.ceil
+
+/**
+ * Manages text measurements for ShadowNode layout.
+ * Parses and renders tiptap to Spannable at measure time for accurate height calculation.
+ */
+object MeasurementStore {
+  private const val TAG = "MeasurementStore"
+
+  private data class PaintParams(
+    val typeface: Typeface,
+    val fontSize: Float,
+  )
+
+  private data class MeasurementParams(
+    val cachedWidth: Float,
+    val cachedSize: Long,
+    val spannable: CharSequence?,
+    val paintParams: PaintParams,
+    val tiptapHash: Int,
+  )
+
+  private val data = ConcurrentHashMap<Int, MeasurementParams>()
+
+  // Store font scaling settings per view ID
+  private data class FontScalingSettings(
+    val allowFontScaling: Boolean = true,
+    val maxFontSizeMultiplier: Float = 0f,
+  )
+
+  private val fontScalingSettings = ConcurrentHashMap<Int, FontScalingSettings>()
+
+  private val breakStrategies = ConcurrentHashMap<Int, String>()
+
+  // The measure path and the display TextView resolve the same clamp. 0 means unlimited.
+  private val numberOfLinesByViewId = ConcurrentHashMap<Int, Int>()
+
+  private val ellipsizeModeByViewId = ConcurrentHashMap<Int, String>()
+
+  // Code blocks the user wrapped, by document-order index across nesting (see CodeBlockWrapState).
+  private val wrappedCodeBlocks = ConcurrentHashMap<Int, Set<Int>>()
+
+  private fun resolveFontScalingSettings(
+    viewId: Int?,
+    props: ReadableMap?,
+  ): FontScalingSettings {
+    val stored = viewId?.let { fontScalingSettings[it] }
+    return FontScalingSettings(
+      allowFontScaling =
+        props?.takeIf { it.hasKey("allowFontScaling") }?.getBoolean("allowFontScaling")
+          ?: stored?.allowFontScaling
+          ?: true,
+      maxFontSizeMultiplier =
+        props?.takeIf { it.hasKey("maxFontSizeMultiplier") }?.getDouble("maxFontSizeMultiplier")?.toFloat()
+          ?: stored?.maxFontSizeMultiplier
+          ?: 0f,
+    )
+  }
+
+  private val measurePaint = TextPaint()
+  private val measureRenderer = Renderer()
+
+  @Volatile
+  private var lastKnownFontScale: Float = 1.0f
+
+  /** Updates measurement with rendered Spannable. Returns true if height changed. */
+  fun store(
+    id: Int,
+    spannable: CharSequence?,
+    paint: TextPaint,
+    trailingMarginBottomPx: Float = 0f,
+  ): Boolean {
+    val cached = data[id]
+    val width = cached?.cachedWidth ?: 0f
+    val oldSize = cached?.cachedSize ?: 0L
+    val existingHash = cached?.tiptapHash ?: 0
+    val paintParams = PaintParams(paint.typeface ?: Typeface.DEFAULT, paint.textSize)
+
+    var newSize = measure(width, spannable, paint, id)
+    if (trailingMarginBottomPx > 0f) {
+      newSize =
+        YogaMeasureOutput.make(
+          YogaMeasureOutput.getWidth(newSize),
+          YogaMeasureOutput.getHeight(newSize) + PixelUtil.toDIPFromPixel(trailingMarginBottomPx),
+        )
+    }
+    data[id] = MeasurementParams(width, newSize, spannable, paintParams, existingHash)
+    return oldSize != newSize
+  }
+
+  fun release(id: Int) {
+    data.remove(id)
+    numberOfLinesByViewId.remove(id)
+    ellipsizeModeByViewId.remove(id)
+    wrappedCodeBlocks.remove(id)
+  }
+
+  fun updateWrappedCodeBlocks(
+    viewId: Int,
+    indices: Set<Int>,
+  ) {
+    if (indices.isEmpty()) wrappedCodeBlocks.remove(viewId) else wrappedCodeBlocks[viewId] = indices
+    data.remove(viewId)
+  }
+
+  fun invalidate(id: Int) {
+    data.remove(id)
+  }
+
+  // Content-box width (px) the measure pass last wrapped at; the display view insets to match (#805).
+  fun contentBoxWidthPx(id: Int): Float? = data[id]?.cachedWidth
+
+  /** Main entry point for ShadowNode measurement. */
+  fun getMeasureById(
+    context: Context,
+    id: Int?,
+    width: Float,
+    widthMode: YogaMeasureMode?,
+    height: Float,
+    heightMode: YogaMeasureMode?,
+    props: ReadableMap?,
+  ): Long {
+    // Early exit for empty tiptap. Honor widthMode here too: empty content has
+    // zero intrinsic width, so only fill when the width is EXACTLY assigned -
+    // otherwise an empty view would claim all available horizontal space.
+    val tiptap = props.getStringOrDefault("document", "")
+    if (tiptap.isEmpty()) {
+      val emptyWidth = if (widthMode === YogaMeasureMode.EXACTLY) PixelUtil.toDIPFromPixel(width) else 0f
+      return YogaMeasureOutput.make(emptyWidth, 0f)
+    }
+
+    // Resolve the line clamp from props here so the (background) measure pass and
+    // the (main-thread) display TextView never race on the per-viewId cache. Only
+    // overwrite when the key is present so we never clobber the view's value.
+    if (id != null && props != null) {
+      if (props.hasKey("numberOfLines")) {
+        numberOfLinesByViewId[id] = props.getDouble("numberOfLines").toInt()
+      }
+      if (props.hasKey("ellipsizeMode")) {
+        ellipsizeModeByViewId[id] = props.getString("ellipsizeMode") ?: EllipsizeUtils.DEFAULT_MODE
+      }
+    }
+
+    val size = getMeasureByIdInternal(context, id, width, props)
+
+    // Honor the Yoga width contract. When the parent assigns an exact width
+    // (flex child, alignItems: stretch, explicit width / width: "100%"), fill it
+    // instead of reporting the content's widest line, which would make the node
+    // hug its content and wrap early. Mirrors RN's TextLayoutManager.calculateWidth;
+    // iOS already does this via ENRMClampMeasuredSize.
+    val resultWidth =
+      if (widthMode === YogaMeasureMode.EXACTLY) {
+        PixelUtil.toDIPFromPixel(width)
+      } else {
+        YogaMeasureOutput.getWidth(size)
+      }
+
+    var resultHeight = YogaMeasureOutput.getHeight(size)
+    if (heightMode === YogaMeasureMode.AT_MOST) {
+      resultHeight = resultHeight.coerceAtMost(PixelUtil.toDIPFromPixel(height))
+    }
+
+    return YogaMeasureOutput.make(resultWidth, resultHeight)
+  }
+
+  fun updateFontScalingSettings(
+    viewId: Int,
+    allowFontScaling: Boolean,
+    maxFontSizeMultiplier: Float,
+  ) {
+    val previous = fontScalingSettings[viewId]
+    fontScalingSettings[viewId] = FontScalingSettings(allowFontScaling, maxFontSizeMultiplier)
+    if (previous != null && (previous.allowFontScaling != allowFontScaling || previous.maxFontSizeMultiplier != maxFontSizeMultiplier)) {
+      data.remove(viewId)
+    }
+  }
+
+  fun clearFontScalingSettings(viewId: Int) {
+    fontScalingSettings.remove(viewId)
+  }
+
+  fun updateBreakStrategy(
+    viewId: Int,
+    strategy: String,
+  ) {
+    breakStrategies[viewId] = strategy
+  }
+
+  fun clearBreakStrategy(viewId: Int) {
+    breakStrategies.remove(viewId)
+  }
+
+  private fun resolveBreakStrategy(viewId: Int?): Int = BreakStrategyUtils.resolveBreakStrategy(viewId?.let { breakStrategies[it] })
+
+  fun updateNumberOfLines(
+    viewId: Int,
+    numberOfLines: Int,
+  ) {
+    numberOfLinesByViewId[viewId] = numberOfLines
+  }
+
+  fun updateEllipsizeMode(
+    viewId: Int,
+    mode: String,
+  ) {
+    ellipsizeModeByViewId[viewId] = mode
+  }
+
+  private fun resolveMaxLines(viewId: Int?): Int = viewId?.let { numberOfLinesByViewId[it] } ?: 0
+
+  private fun resolveEllipsize(viewId: Int?): TruncateAt? =
+    EllipsizeUtils.resolveTruncateAt(viewId?.let { ellipsizeModeByViewId[it] } ?: EllipsizeUtils.DEFAULT_MODE)
+
+  private fun getMeasureByIdInternal(
+    context: Context,
+    id: Int?,
+    width: Float,
+    props: ReadableMap?,
+  ): Long {
+    val (allowFontScaling, maxFontSizeMultiplier) = resolveFontScalingSettings(id, props)
+
+    val fontScale = checkAndUpdateFontScale(context, allowFontScaling, maxFontSizeMultiplier)
+
+    val safeId = id ?: return measureAndCache(context, null, width, props, allowFontScaling, fontScale, maxFontSizeMultiplier)
+    val cached = data[safeId] ?: return measureAndCache(context, safeId, width, props, allowFontScaling, fontScale, maxFontSizeMultiplier)
+
+    val currentHash = computePropsHash(props, allowFontScaling, fontScale, maxFontSizeMultiplier)
+
+    if (cached.tiptapHash != currentHash) {
+      return measureAndCache(context, safeId, width, props, allowFontScaling, fontScale, maxFontSizeMultiplier)
+    }
+
+    // Width changed - re-measure with cached spannable
+    if (cached.cachedWidth != width) {
+      val newSize = measure(width, cached.spannable, cached.paintParams, safeId)
+      data[safeId] = cached.copy(cachedWidth = width, cachedSize = newSize)
+      return newSize
+    }
+
+    return cached.cachedSize
+  }
+
+  private fun computePropsHash(
+    props: ReadableMap?,
+    allowFontScaling: Boolean,
+    fontScale: Float,
+    maxFontSizeMultiplier: Float,
+  ): Int {
+    val tiptap = props.getStringOrDefault("document", "")
+    return computePropsHashForTiptap(tiptap, props, allowFontScaling, fontScale, maxFontSizeMultiplier)
+  }
+
+  private fun computePropsHashForTiptap(
+    tiptap: String,
+    props: ReadableMap?,
+    allowFontScaling: Boolean,
+    fontScale: Float,
+    maxFontSizeMultiplier: Float,
+  ): Int {
+    val styleMap = props.getMapOrNull("tiptapStyle")
+    val md4cFlagsMap = props.getMapOrNull("md4cFlags")
+    val allowTrailingMargin = props.getBooleanOrDefault("allowTrailingMargin", false)
+    val imageRequestHeaders = parseImageRequestHeaders(props.getArrayOrNull("imageRequestHeaders"))
+    val linkPillContent = parseLinkPillContent(props.getArrayOrNull("linkPillContent"))
+    var result = tiptap.hashCode()
+    result = 31 * result + (styleMap?.hashCode() ?: 0)
+    result = 31 * result + (md4cFlagsMap?.hashCode() ?: 0)
+    result = 31 * result + parseLinkRecognition(props.getMapOrNull("linkRecognition")).hashCode()
+    result = 31 * result + fontScale.toBits()
+    result = 31 * result + allowFontScaling.hashCode()
+    result = 31 * result + maxFontSizeMultiplier.toBits()
+    result = 31 * result + allowTrailingMargin.hashCode()
+    result = 31 * result + imageRequestHeaders.hashCode()
+    result = 31 * result + linkPillContent.hashCode()
+    result = 31 * result + props.getIntOrDefault("numberOfLines", 0)
+    result = 31 * result + props.getStringOrDefault("ellipsizeMode", EllipsizeUtils.DEFAULT_MODE).hashCode()
+    return result
+  }
+
+  private fun checkAndUpdateFontScale(
+    context: Context,
+    allowFontScaling: Boolean,
+    maxFontSizeMultiplier: Float,
+  ): Float {
+    if (!allowFontScaling) {
+      // Clear cache if we switched from scaling to non-scaling
+      if (lastKnownFontScale != 1.0f) {
+        lastKnownFontScale = 1.0f
+        data.clear()
+      }
+      return 1.0f
+    }
+
+    var currentFontScale = context.resources.configuration.fontScale
+
+    if (maxFontSizeMultiplier >= 1.0f && currentFontScale > maxFontSizeMultiplier) {
+      currentFontScale = maxFontSizeMultiplier
+    }
+    if (currentFontScale != lastKnownFontScale) {
+      lastKnownFontScale = currentFontScale
+      data.clear()
+    }
+    return currentFontScale
+  }
+
+  private fun measureAndCache(
+    context: Context,
+    id: Int?,
+    width: Float,
+    props: ReadableMap?,
+    allowFontScaling: Boolean,
+    fontScale: Float,
+    maxFontSizeMultiplier: Float,
+  ): Long {
+    // 1. Extract Props & Setup
+    val tiptap = props.getStringOrDefault("document", "")
+    val styleMap = props.getMapOrNull("tiptapStyle")
+    val md4cFlags =
+      Md4cFlags(
+        underline = props.getMapOrNull("md4cFlags").getBooleanOrDefault("underline", false),
+        latexMath = false,
+        superscript = props.getMapOrNull("md4cFlags").getBooleanOrDefault("superscript", false),
+        subscript = props.getMapOrNull("md4cFlags").getBooleanOrDefault("subscript", false),
+        highlight = props.getMapOrNull("md4cFlags").getBooleanOrDefault("highlight", false),
+        hardSoftBreaks = props.getMapOrNull("md4cFlags").getBooleanOrDefault("hardSoftBreaks", false),
+        preserveBlankLines = props.getMapOrNull("md4cFlags").getBooleanOrDefault("preserveBlankLines", false),
+        admonitions = props.getMapOrNull("md4cFlags").getBooleanOrDefault("admonitions", true),
+      )
+    val isGFM = props.getBooleanOrDefault("isGFM", false)
+
+    val fontSize = getInitialFontSize(styleMap, context, allowFontScaling, fontScale, maxFontSizeMultiplier)
+    val propsHash = computePropsHash(props, allowFontScaling, fontScale, maxFontSizeMultiplier)
+
+    // 2. Render & Measure
+    measurePaint.textSize = fontSize
+    val imageRequestHeaders = parseImageRequestHeaders(props.getArrayOrNull("imageRequestHeaders"))
+    val linkRecognition = parseLinkRecognition(props.getMapOrNull("linkRecognition"))
+    val spannable =
+      tryRenderTiptap(
+        tiptap,
+        styleMap,
+        context,
+        md4cFlags,
+        isGFM,
+        allowFontScaling,
+        maxFontSizeMultiplier,
+        imageRequestHeaders,
+        parseLinkPillContent(props.getArrayOrNull("linkPillContent")),
+        linkRecognition.text,
+        linkRecognition.inlineCode,
+      )
+    val textToMeasure = spannable ?: tiptap
+    val (size, _) = measureWithLayout(width, textToMeasure, measurePaint, id)
+
+    // 3. Calculate Margin
+    val allowTrailingMargin = props.getBooleanOrDefault("allowTrailingMargin", false)
+    val marginBottom =
+      if (allowTrailingMargin && spannable != null) {
+        PixelUtil.toDIPFromPixel(measureRenderer.getLastElementMarginBottom())
+      } else {
+        0f
+      }
+
+    // 4. Finalize Height
+    val currentWidth = YogaMeasureOutput.getWidth(size)
+    val currentHeight = YogaMeasureOutput.getHeight(size)
+    val adjustedSize = YogaMeasureOutput.make(currentWidth, currentHeight + marginBottom)
+
+    if (id != null) {
+      data[id] = MeasurementParams(width, adjustedSize, textToMeasure, PaintParams(Typeface.DEFAULT, fontSize), propsHash)
+    }
+
+    return adjustedSize
+  }
+
+  private fun createStaticLayout(
+    text: CharSequence,
+    fontSize: Float,
+    widthPx: Int,
+    viewId: Int?,
+  ): StaticLayout {
+    measurePaint.textSize = fontSize
+    prepareWidthAwareSpans(text, widthPx)
+    return StaticLayout.Builder
+      .obtain(text, 0, text.length, measurePaint, widthPx)
+      .setIncludePad(false)
+      .setLineSpacing(0f, 1f)
+      .apply {
+        if (Build.VERSION.SDK_INT >= Build.VERSION_CODES.M) {
+          @Suppress("WrongConstant")
+          setBreakStrategy(resolveBreakStrategy(viewId))
+        }
+        if (Build.VERSION.SDK_INT >= Build.VERSION_CODES.P) {
+          setUseLineSpacingFromFallbacks(true)
+        }
+      }.build()
+  }
+
+  private fun tryRenderTiptap(
+    tiptap: String,
+    styleMap: ReadableMap?,
+    context: Context,
+    md4cFlags: Md4cFlags,
+    isGFM: Boolean,
+    allowFontScaling: Boolean,
+    maxFontSizeMultiplier: Float,
+    imageRequestHeaders: Map<String, String> = emptyMap(),
+    linkPillContent: Map<String, LinkPillContent> = emptyMap(),
+    linkRegex: LinkRegexConfig? = null,
+    inlineCodeLinkRegex: LinkRegexConfig? = null,
+  ): Spannable? {
+    if (styleMap == null) return null
+
+    return try {
+      val ast = Parser.shared.parseTiptap(tiptap, md4cFlags, isGFM, linkRegex, inlineCodeLinkRegex) ?: return null
+      val style = StyleConfig(styleMap, context, allowFontScaling, maxFontSizeMultiplier)
+      style.imageRequestHeaders = imageRequestHeaders
+      style.linkPillContent = linkPillContent
+      measureRenderer.configure(style, context)
+      measureRenderer.renderDocument(ast, null)
+    } catch (e: Exception) {
+      Log.w(TAG, "Failed to render tiptap for measurement, falling back to raw text", e)
+      null
+    }
+  }
+
+  private fun getInitialFontSize(
+    styleMap: ReadableMap?,
+    context: Context,
+    allowFontScaling: Boolean,
+    fontScale: Float,
+    maxFontSizeMultiplier: Float,
+  ): Float {
+    val fontSizeSp = styleMap?.getMap("paragraph")?.getDouble("fontSize")?.toFloat() ?: 16f
+    val density = context.resources.displayMetrics.density
+
+    if (!allowFontScaling) {
+      return ceil(fontSizeSp * density)
+    }
+
+    val cappedFontScale =
+      if (maxFontSizeMultiplier >= 1.0f && fontScale > maxFontSizeMultiplier) {
+        maxFontSizeMultiplier
+      } else {
+        fontScale
+      }
+    return ceil(fontSizeSp * cappedFontScale * density)
+  }
+
+  private fun measure(
+    maxWidth: Float,
+    text: CharSequence?,
+    paintParams: PaintParams,
+    viewId: Int?,
+  ): Long {
+    measurePaint.reset()
+    measurePaint.typeface = paintParams.typeface
+    measurePaint.textSize = paintParams.fontSize
+    return measure(maxWidth, text, measurePaint, viewId)
+  }
+
+  private fun measure(
+    maxWidth: Float,
+    text: CharSequence?,
+    paint: TextPaint,
+    viewId: Int?,
+  ): Long {
+    val content = text ?: ""
+    val safeWidth = ceil(maxWidth).toInt().coerceAtLeast(1)
+    prepareWidthAwareSpans(content, safeWidth)
+
+    val builder =
+      StaticLayout.Builder
+        .obtain(content, 0, content.length, paint, safeWidth)
+        .setIncludePad(false)
+        .setLineSpacing(0f, 1f)
+
+    if (Build.VERSION.SDK_INT >= Build.VERSION_CODES.M) {
+      @Suppress("WrongConstant")
+      builder.setBreakStrategy(resolveBreakStrategy(viewId))
+    }
+
+    if (Build.VERSION.SDK_INT >= Build.VERSION_CODES.P) {
+      builder.setUseLineSpacingFromFallbacks(true)
+    }
+
+    applyLineLimit(builder, viewId)
+
+    val layout = builder.build()
+    val visibleLineCount = visibleLineCount(layout, viewId)
+    val measuredHeight = truncatedHeight(layout, visibleLineCount, content)
+
+    // Calculate actual content width (widest visible line)
+    val measuredWidth = (0 until visibleLineCount).maxOfOrNull { layout.getLineWidth(it) } ?: 0f
+
+    return YogaMeasureOutput.make(
+      PixelUtil.toDIPFromPixel(ceil(measuredWidth)),
+      PixelUtil.toDIPFromPixel(measuredHeight),
+    )
+  }
+
+  // Applies the per-view numberOfLines / ellipsizeMode clamp to a builder, matching
+  // RN's TextLayoutManager.buildLayout: only when a line limit is set.
+  private fun applyLineLimit(
+    builder: StaticLayout.Builder,
+    viewId: Int?,
+  ) {
+    val maxLines = resolveMaxLines(viewId)
+    if (maxLines > 0) {
+      builder.setMaxLines(maxLines).setEllipsize(resolveEllipsize(viewId))
+    }
+  }
+
+  private fun visibleLineCount(
+    layout: StaticLayout,
+    viewId: Int?,
+  ): Int {
+    val maxLines = resolveMaxLines(viewId)
+    return if (maxLines > 0) minOf(maxLines, layout.lineCount) else layout.lineCount
+  }
+
+  private fun truncatedHeight(
+    layout: StaticLayout,
+    visibleLineCount: Int,
+    text: CharSequence,
+  ): Float {
+    if (visibleLineCount <= 0) return 0f
+    val lastLine = visibleLineCount - 1
+    val bottom = layout.getLineBottom(lastLine).toFloat()
+    return (bottom - measureLeakedTrailingMargin(layout, lastLine, text)).coerceAtLeast(0f)
+  }
+
+  // When we clamp the text mid-paragraph, drop the trailing block margin that Android
+  // bakes into the truncated last line (only when content follows, like MarginBottomSpan).
+  private fun measureLeakedTrailingMargin(
+    layout: StaticLayout,
+    lastLine: Int,
+    text: CharSequence,
+  ): Float {
+    val spanned = text as? Spanned ?: return 0f
+    val lineStart = layout.getLineStart(lastLine)
+    val lineEnd = layout.getLineEnd(lastLine)
+    if (lineEnd <= lineStart || lineEnd < 1 || text[lineEnd - 1] != '\n') return 0f
+    if (!hasContentAfter(text, lineEnd)) return 0f
+    val trailing =
+      spanned
+        .getSpans(lineStart, lineEnd, MarginBottomSpan::class.java)
+        .firstOrNull { spanned.getSpanEnd(it) == lineEnd } ?: return 0f
+    return trailing.marginBottom.toInt().toFloat()
+  }
+
+  private fun hasContentAfter(
+    text: CharSequence,
+    pos: Int,
+  ): Boolean {
+    if (pos >= text.length) return false
+    if (text[pos] == '\n') {
+      val nextPos = pos + 1
+      if (nextPos >= text.length) return false
+      return text[nextPos] != '\n'
+    }
+    return true
+  }
+
+  /**
+   * Measures text and returns both the size and the layout for calculating last line descent.
+   */
+  private fun measureWithLayout(
+    maxWidth: Float,
+    text: CharSequence?,
+    paint: TextPaint,
+    viewId: Int?,
+  ): Pair<Long, StaticLayout> {
+    val content = text ?: ""
+    val widthPx = ceil(maxWidth).toInt().coerceAtLeast(1)
+    prepareWidthAwareSpans(content, widthPx)
+
+    val layout =
+      StaticLayout.Builder
+        .obtain(content, 0, content.length, paint, widthPx)
+        .setIncludePad(false)
+        .setLineSpacing(0f, 1f)
+        .apply {
+          if (Build.VERSION.SDK_INT >= Build.VERSION_CODES.M) {
+            @Suppress("WrongConstant")
+            setBreakStrategy(resolveBreakStrategy(viewId))
+          }
+          if (Build.VERSION.SDK_INT >= Build.VERSION_CODES.P) {
+            setUseLineSpacingFromFallbacks(true)
+          }
+          applyLineLimit(this, viewId)
+        }.build()
+
+    val visibleLineCount = visibleLineCount(layout, viewId)
+
+    // Find the widest visible line to get the actual content width
+    val maxLineWidth =
+      (0 until visibleLineCount)
+        .maxOfOrNull { layout.getLineWidth(it) } ?: 0f
+
+    val size =
+      YogaMeasureOutput.make(
+        PixelUtil.toDIPFromPixel(ceil(maxLineWidth)),
+        PixelUtil.toDIPFromPixel(truncatedHeight(layout, visibleLineCount, content)),
+      )
+
+    return size to layout
+  }
+
+}