react-native-enriched-markdown-scaffold 1.0.2-scaffold.2 → 1.0.2-scaffold.4

package/android/src/main/java/com/swmansion/enriched/markdown/EnrichedMarkdown.kt

@@ -25,9 +25,12 @@ import com.swmansion.enriched.markdown.utils.common.RenderedSegment
 import com.swmansion.enriched.markdown.utils.common.SegmentReconciler
 import com.swmansion.enriched.markdown.utils.common.StreamingMarkdownFilter
 import com.swmansion.enriched.markdown.utils.common.TableStreamingMode
+import com.swmansion.enriched.markdown.utils.common.emitLinkLayout
 import com.swmansion.enriched.markdown.utils.common.isReducedMotionEnabled
 import com.swmansion.enriched.markdown.utils.common.splitASTIntoSegments
 import com.swmansion.enriched.markdown.utils.text.TailFadeInAnimator
+import com.swmansion.enriched.markdown.utils.text.view.LinkLayoutCollector
+import com.swmansion.enriched.markdown.utils.text.view.LinkLayoutFragment
 import com.swmansion.enriched.markdown.utils.text.view.SelectionMenuConfig
 import com.swmansion.enriched.markdown.utils.text.view.applySelectionColors
 import com.swmansion.enriched.markdown.views.BlockMenuOptions
@@ -99,6 +102,8 @@ class EnrichedMarkdown
     private var selectionMenuConfig = SelectionMenuConfig()
     private var accessibilityLabels = AccessibilityLabels()
     private var textBreakStrategy: String = BreakStrategyUtils.DEFAULT_STRATEGY
+    private var linkLayoutEnabled = false
+    private var lastLinkLayout: List<LinkLayoutFragment>? = null
 
     private var onLinkPressCallback: ((String) -> Unit)? = null
     private var onLinkLongPressCallback: ((String) -> Unit)? = null
@@ -106,6 +111,7 @@ class EnrichedMarkdown
     private var onCopyPressCallback: ((String, String) -> Unit)? = null
     private var contextMenuItemTexts: List<String> = emptyList()
     var onContextMenuItemPressCallback: ((itemText: String, selectedText: String, selectionStart: Int, selectionEnd: Int) -> Unit)? = null
+    var onCopyRequestCallback: ((source: String, markdown: String, text: String, whole: Boolean) -> Unit)? = null
     var spoilerOverlay: SpoilerOverlay = SpoilerOverlay.PARTICLES
       set(value) {
         if (field == value) return
@@ -318,19 +324,18 @@ class EnrichedMarkdown
     private fun pushCopyLabelsToBlockSegments() {
       val copyLabel = selectionMenuConfig.copyLabel
       val copyAsMarkdownLabel = selectionMenuConfig.copyAsMarkdownLabel
-      val menuOptions = BlockMenuOptions.from(selectionMenuConfig)
       segmentViews.forEach { view ->
         when {
           view is TableContainerView -> {
             view.copyLabel = copyLabel
             view.copyAsMarkdownLabel = copyAsMarkdownLabel
-            view.menuOptions = menuOptions
+            view.menuOptions = blockMenuOptions("table")
           }
 
           view is CodeBlockContainerView -> {
             view.copyLabel = copyLabel
             view.copyAsMarkdownLabel = copyAsMarkdownLabel
-            view.menuOptions = menuOptions
+            view.menuOptions = blockMenuOptions("code")
           }
 
           isMathContainerView(view) -> {
@@ -341,7 +346,7 @@ class EnrichedMarkdown
                 .invoke(view, copyAsMarkdownLabel)
               view.javaClass
                 .getMethod("setMenuOptions", BlockMenuOptions::class.java)
-                .invoke(view, menuOptions)
+                .invoke(view, blockMenuOptions("math"))
             }
           }
         }
@@ -368,6 +373,22 @@ class EnrichedMarkdown
       }
     }
 
+    private fun blockMenuOptions(source: String) =
+      BlockMenuOptions.from(selectionMenuConfig) { markdown, text ->
+        onCopyRequestCallback?.invoke(source, markdown, text, true)
+      }
+
+    // Segments only see their own text, so a selection is the whole component
+    // only when it spans the sole segment; the markdown is then the original.
+    private fun forwardSegmentCopyRequest(
+      markdown: String,
+      text: String,
+      fullSelection: Boolean,
+    ) {
+      val whole = fullSelection && segmentViews.size == 1 && currentMarkdown.isNotEmpty()
+      onCopyRequestCallback?.invoke("selection", if (whole) currentMarkdown else markdown, text, whole)
+    }
+
     private fun forwardContextMenuItemPress(
       itemText: String,
       selectedText: String,
@@ -615,6 +636,7 @@ class EnrichedMarkdown
         onTaskListItemPressCallback = { taskIndex, checked, itemText ->
           this@EnrichedMarkdown.onTaskListItemPressCallback?.invoke(taskIndex, checked, itemText)
         }
+        onCopyRequest = ::forwardSegmentCopyRequest
 
         if (contextMenuItemTexts.isNotEmpty()) {
           setContextMenuItems(contextMenuItemTexts, ::forwardContextMenuItemPress)
@@ -635,7 +657,7 @@ class EnrichedMarkdown
       copyLabel = this@EnrichedMarkdown.selectionMenuConfig.copyLabel
       copyAsMarkdownLabel = this@EnrichedMarkdown.selectionMenuConfig.copyAsMarkdownLabel
       contextMenuHidden = this@EnrichedMarkdown.contextMenuHidden
-      menuOptions = BlockMenuOptions.from(this@EnrichedMarkdown.selectionMenuConfig)
+      menuOptions = blockMenuOptions("table")
       applyTableNode(segment.node)
     }
 
@@ -646,7 +668,7 @@ class EnrichedMarkdown
       copyLabel = this@EnrichedMarkdown.selectionMenuConfig.copyLabel
       copyAsMarkdownLabel = this@EnrichedMarkdown.selectionMenuConfig.copyAsMarkdownLabel
       contextMenuHidden = this@EnrichedMarkdown.contextMenuHidden
-      menuOptions = BlockMenuOptions.from(this@EnrichedMarkdown.selectionMenuConfig)
+      menuOptions = blockMenuOptions("code")
       onCopyPress = { code, language ->
         this@EnrichedMarkdown.onCopyPressCallback?.invoke(code, language)
       }
@@ -678,7 +700,7 @@ class EnrichedMarkdown
           .invoke(view, selectionMenuConfig.copyAsMarkdownLabel)
         resolvedClass
           .getMethod("setMenuOptions", BlockMenuOptions::class.java)
-          .invoke(view, BlockMenuOptions.from(selectionMenuConfig))
+          .invoke(view, blockMenuOptions("math"))
         setMathContextMenuHidden(view)
         resolvedClass.getMethod("applyLatex", String::class.java).invoke(view, segment.latex)
         view
@@ -765,6 +787,39 @@ class EnrichedMarkdown
           currentY += segment?.segmentMarginBottom ?: 0
         }
       }
+
+      emitLinkLayoutIfNeeded()
+    }
+
+    fun setEmitLinkLayout(enabled: Boolean) {
+      if (linkLayoutEnabled == enabled) return
+      linkLayoutEnabled = enabled
+      lastLinkLayout = null
+      emitLinkLayoutIfNeeded()
+    }
+
+    /**
+     * Reports link rects of every text segment (offset by the segment's position) to JS for
+     * `renderLinkDecoration`. Tables are skipped: their cells scroll horizontally, so rects would
+     * go stale. Runs after each segment layout pass; deduplicated so an unchanged layout sends nothing.
+     */
+    private fun emitLinkLayoutIfNeeded() {
+      if (!linkLayoutEnabled || width <= 0) return
+      val fragments = mutableListOf<LinkLayoutFragment>()
+      var linkCount = 0
+      segmentViews.forEach { view ->
+        if (view is TableContainerView) {
+          linkCount += view.linkCount
+          return@forEach
+        }
+        if (view !is EnrichedMarkdownInternalText) return@forEach
+        val (segmentFragments, segmentLinks) = LinkLayoutCollector.collect(view, view.left, view.top, linkCount)
+        fragments += segmentFragments
+        linkCount += segmentLinks
+      }
+      if (fragments == lastLinkLayout) return
+      lastLinkLayout = fragments
+      emitLinkLayout(this, fragments)
     }
 
     private fun computeSegmentsTotalHeight(): Int {

package/android/src/main/java/com/swmansion/enriched/markdown/EnrichedMarkdownInternalText.kt

@@ -56,6 +56,7 @@ class EnrichedMarkdownInternalText
     private var contextMenuItemTexts: List<String> = emptyList()
     private var onContextMenuItemPress: ((itemText: String, selectedText: String, selectionStart: Int, selectionEnd: Int) -> Unit)? = null
     var selectionMenuConfig: SelectionMenuConfig = SelectionMenuConfig()
+    var onCopyRequest: ((markdown: String, text: String, fullSelection: Boolean) -> Unit)? = null
     private var activeActionMode: ActionMode? = null
     var contextMenuHidden: Boolean = false
       set(value) {
@@ -82,6 +83,9 @@ class EnrichedMarkdownInternalText
           onCustomItemPress = { itemText, selectedText, start, end ->
             onContextMenuItemPress?.invoke(itemText, selectedText, start, end)
           },
+          onCopyRequest = { markdown, text, fullSelection ->
+            onCopyRequest?.invoke(markdown, text, fullSelection)
+          },
         )
     }
 

package/android/src/main/java/com/swmansion/enriched/markdown/EnrichedMarkdownManager.kt

@@ -18,6 +18,7 @@ import com.swmansion.enriched.markdown.utils.common.CodeBlockStreamingMode
 import com.swmansion.enriched.markdown.utils.common.TableStreamingMode
 import com.swmansion.enriched.markdown.utils.common.emitContextMenuItemPress
 import com.swmansion.enriched.markdown.utils.common.emitCopyPress
+import com.swmansion.enriched.markdown.utils.common.emitCopyRequest
 import com.swmansion.enriched.markdown.utils.common.emitLinkLongPress
 import com.swmansion.enriched.markdown.utils.common.emitLinkPress
 import com.swmansion.enriched.markdown.utils.common.emitTaskListItemPress
@@ -65,6 +66,10 @@ class EnrichedMarkdownManager :
       emitCopyPress(view, code, language)
     }
 
+    view.onCopyRequestCallback = { source, markdown, text, whole ->
+      emitCopyRequest(view, source, markdown, text, whole)
+    }
+
     return view
   }
 
@@ -172,6 +177,14 @@ class EnrichedMarkdownManager :
     view?.setContextMenuHidden(contextMenuHidden)
   }
 
+  @ReactProp(name = "emitLinkLayout", defaultBoolean = false)
+  override fun setEmitLinkLayout(
+    view: EnrichedMarkdown?,
+    emitLinkLayout: Boolean,
+  ) {
+    view?.setEmitLinkLayout(emitLinkLayout)
+  }
+
   @ReactProp(name = "enableLinkPreview", defaultBoolean = true)
   override fun setEnableLinkPreview(
     view: EnrichedMarkdown?,

package/android/src/main/java/com/swmansion/enriched/markdown/EnrichedMarkdownText.kt

@@ -24,8 +24,11 @@ import com.swmansion.enriched.markdown.spoiler.SpoilerOverlay
 import com.swmansion.enriched.markdown.spoiler.SpoilerOverlayDrawer
 import com.swmansion.enriched.markdown.styles.StyleConfig
 import com.swmansion.enriched.markdown.utils.common.BreakStrategyUtils
+import com.swmansion.enriched.markdown.utils.common.emitLinkLayout
 import com.swmansion.enriched.markdown.utils.text.TailFadeInAnimator
 import com.swmansion.enriched.markdown.utils.text.interaction.CheckboxTouchHelper
+import com.swmansion.enriched.markdown.utils.text.view.LinkLayoutCollector
+import com.swmansion.enriched.markdown.utils.text.view.LinkLayoutFragment
 import com.swmansion.enriched.markdown.utils.text.view.LinkLongPressMovementMethod
 import com.swmansion.enriched.markdown.utils.text.view.SelectionMenuConfig
 import com.swmansion.enriched.markdown.utils.text.view.applySelectableState
@@ -68,6 +71,7 @@ class EnrichedMarkdownText
 
     private var contextMenuItemTexts: List<String> = emptyList()
     var onContextMenuItemPressCallback: ((itemText: String, selectedText: String, selectionStart: Int, selectionEnd: Int) -> Unit)? = null
+    var onCopyRequestCallback: ((markdown: String, text: String, whole: Boolean) -> Unit)? = null
 
     var markdownStyle: StyleConfig? = null
       private set
@@ -101,6 +105,9 @@ class EnrichedMarkdownText
     private var contextMenuHidden = false
     private var activeActionMode: ActionMode? = null
 
+    private var linkLayoutEnabled = false
+    private var lastLinkLayout: List<LinkLayoutFragment>? = null
+
     init {
       setupAsMarkdownTextView()
       customSelectionActionModeCallback =
@@ -113,6 +120,10 @@ class EnrichedMarkdownText
           onCustomItemPress = { itemText, selectedText, start, end ->
             onContextMenuItemPressCallback?.invoke(itemText, selectedText, start, end)
           },
+          // A full selection already carries the original markdown.
+          onCopyRequest = { markdown, text, fullSelection ->
+            onCopyRequestCallback?.invoke(markdown, text, fullSelection && currentMarkdown.isNotEmpty())
+          },
         )
     }
 
@@ -305,6 +316,37 @@ class EnrichedMarkdownText
       }
 
       applySelectionColors(selectionColor, selectionHandleColor)
+      emitLinkLayoutIfNeeded()
+    }
+
+    fun setEmitLinkLayout(enabled: Boolean) {
+      if (linkLayoutEnabled == enabled) return
+      linkLayoutEnabled = enabled
+      lastLinkLayout = null
+      emitLinkLayoutIfNeeded()
+    }
+
+    /**
+     * Reports link rects to JS (for `renderLinkDecoration`) after every render and layout;
+     * deduplicated so an unchanged layout — the common case while streaming — sends nothing.
+     */
+    private fun emitLinkLayoutIfNeeded() {
+      if (!linkLayoutEnabled || layout == null || width <= 0) return
+      val (fragments, _) = LinkLayoutCollector.collect(this)
+      if (fragments == lastLinkLayout) return
+      lastLinkLayout = fragments
+      emitLinkLayout(this, fragments)
+    }
+
+    override fun onLayout(
+      changed: Boolean,
+      left: Int,
+      top: Int,
+      right: Int,
+      bottom: Int,
+    ) {
+      super.onLayout(changed, left, top, right, bottom)
+      emitLinkLayoutIfNeeded()
     }
 
     // Trailing bottom margin included in the shadow-node measurement — must be

package/android/src/main/java/com/swmansion/enriched/markdown/EnrichedMarkdownTextManager.kt

@@ -15,6 +15,7 @@ import com.facebook.react.viewmanagers.EnrichedMarkdownTextManagerInterface
 import com.facebook.yoga.YogaMeasureMode
 import com.swmansion.enriched.markdown.spoiler.SpoilerOverlay
 import com.swmansion.enriched.markdown.utils.common.emitContextMenuItemPress
+import com.swmansion.enriched.markdown.utils.common.emitCopyRequest
 import com.swmansion.enriched.markdown.utils.common.emitLinkLongPress
 import com.swmansion.enriched.markdown.utils.common.emitLinkPress
 import com.swmansion.enriched.markdown.utils.common.emitTaskListItemPress
@@ -42,6 +43,9 @@ class EnrichedMarkdownTextManager :
     view.onContextMenuItemPressCallback = { itemText, selectedText, selectionStart, selectionEnd ->
       emitContextMenuItemPress(view, itemText, selectedText, selectionStart, selectionEnd)
     }
+    view.onCopyRequestCallback = { markdown, text, whole ->
+      emitCopyRequest(view, "selection", markdown, text, whole)
+    }
 
     view.setOnLinkPressCallback { url ->
       emitLinkPress(view, url)
@@ -173,6 +177,14 @@ class EnrichedMarkdownTextManager :
     view?.setContextMenuHidden(contextMenuHidden)
   }
 
+  @ReactProp(name = "emitLinkLayout", defaultBoolean = false)
+  override fun setEmitLinkLayout(
+    view: EnrichedMarkdownText?,
+    emitLinkLayout: Boolean,
+  ) {
+    view?.setEmitLinkLayout(emitLinkLayout)
+  }
+
   @ReactProp(name = "enableLinkPreview", defaultBoolean = true)
   override fun setEnableLinkPreview(
     view: EnrichedMarkdownText?,

package/android/src/main/java/com/swmansion/enriched/markdown/events/CopyRequestEvent.kt

@@ -0,0 +1,28 @@
+package com.swmansion.enriched.markdown.events
+
+import com.facebook.react.bridge.Arguments
+import com.facebook.react.bridge.WritableMap
+import com.facebook.react.uimanager.events.Event
+
+class CopyRequestEvent(
+  surfaceId: Int,
+  viewId: Int,
+  private val source: String,
+  private val markdown: String,
+  private val text: String,
+  private val whole: Boolean,
+) : Event<CopyRequestEvent>(surfaceId, viewId) {
+  override fun getEventName(): String = EVENT_NAME
+
+  override fun getEventData(): WritableMap =
+    Arguments.createMap().apply {
+      putString("source", source)
+      putString("markdown", markdown)
+      putString("text", text)
+      putBoolean("whole", whole)
+    }
+
+  companion object {
+    const val EVENT_NAME: String = "onCopyRequest"
+  }
+}

package/android/src/main/java/com/swmansion/enriched/markdown/events/LinkLayoutEvent.kt

@@ -0,0 +1,36 @@
+package com.swmansion.enriched.markdown.events
+
+import com.facebook.react.bridge.Arguments
+import com.facebook.react.bridge.WritableMap
+import com.facebook.react.uimanager.events.Event
+import com.swmansion.enriched.markdown.utils.text.view.LinkLayoutFragment
+
+class LinkLayoutEvent(
+  surfaceId: Int,
+  viewId: Int,
+  private val fragments: List<LinkLayoutFragment>,
+) : Event<LinkLayoutEvent>(surfaceId, viewId) {
+  override fun getEventName(): String = EVENT_NAME
+
+  override fun getEventData(): WritableMap {
+    val array = Arguments.createArray()
+    fragments.forEach { fragment ->
+      array.pushMap(
+        Arguments.createMap().apply {
+          putInt("linkIndex", fragment.linkIndex)
+          putString("url", fragment.url)
+          putString("text", fragment.text)
+          putDouble("x", fragment.x.toDouble())
+          putDouble("y", fragment.y.toDouble())
+          putDouble("width", fragment.width.toDouble())
+          putDouble("height", fragment.height.toDouble())
+        },
+      )
+    }
+    return Arguments.createMap().apply { putArray("fragments", array) }
+  }
+
+  companion object {
+    const val EVENT_NAME: String = "onLinkLayout"
+  }
+}

package/android/src/main/java/com/swmansion/enriched/markdown/renderer/LinkRenderer.kt

@@ -2,6 +2,9 @@ package com.swmansion.enriched.markdown.renderer
 
 import android.text.SpannableStringBuilder
 import com.swmansion.enriched.markdown.parser.MarkdownASTNode
+import com.swmansion.enriched.markdown.spans.LinkChipBackgroundSpan
+import com.swmansion.enriched.markdown.spans.LinkSpacerSpan
+import com.swmansion.enriched.markdown.spans.LinkSpacers
 import com.swmansion.enriched.markdown.spans.LinkSpan
 import com.swmansion.enriched.markdown.utils.text.span.SPAN_FLAGS_EXCLUSIVE_EXCLUSIVE
 
@@ -16,14 +19,41 @@ class LinkRenderer(
     factory: RendererFactory,
   ) {
     val url = node.getAttribute("url") ?: return
+    // Chip ("box") properties never fall back to the base link style.
+    val chip = factory.styleCache.linkChipForUrl(url)
+    val padding = chip.paddingHorizontal
 
-    factory.renderWithSpan(builder, { factory.renderChildren(node, builder, onLinkPress, onLinkLongPress) }) { start, end, blockStyle ->
-      builder.setSpan(
-        LinkSpan(url, onLinkPress, onLinkLongPress, factory.styleCache, blockStyle, factory.context),
-        start,
-        end,
-        SPAN_FLAGS_EXCLUSIVE_EXCLUSIVE,
-      )
+    val start = builder.length
+    // The leading spacer goes in before the children so ranges they register stay valid.
+    if (padding > 0f) appendSpacer(builder, padding)
+    val contentStart = builder.length
+
+    factory.renderChildren(node, builder, onLinkPress, onLinkLongPress)
+
+    if (builder.length == contentStart) {
+      if (contentStart > start) builder.delete(start, contentStart)
+      return
     }
+    if (padding > 0f) appendSpacer(builder, padding)
+    val end = builder.length
+
+    val blockStyle = factory.blockStyleContext.requireBlockStyle()
+    builder.setSpan(
+      LinkSpan(url, onLinkPress, onLinkLongPress, factory.styleCache, blockStyle, factory.context),
+      start,
+      end,
+      SPAN_FLAGS_EXCLUSIVE_EXCLUSIVE,
+    )
+    // One instance per link: also identifies the link for link-layout reporting.
+    builder.setSpan(LinkChipBackgroundSpan(chip), start, end, SPAN_FLAGS_EXCLUSIVE_EXCLUSIVE)
+  }
+
+  private fun appendSpacer(
+    builder: SpannableStringBuilder,
+    width: Float,
+  ) {
+    val position = builder.length
+    builder.append(LinkSpacers.CHAR)
+    builder.setSpan(LinkSpacerSpan(width), position, position + 1, SPAN_FLAGS_EXCLUSIVE_EXCLUSIVE)
   }
 }

package/android/src/main/java/com/swmansion/enriched/markdown/renderer/SpanStyleCache.kt

@@ -4,6 +4,7 @@ import android.content.res.AssetManager
 import android.graphics.Typeface
 import com.facebook.react.common.ReactConstants
 import com.facebook.react.views.text.ReactTypefaceUtils.applyStyles
+import com.swmansion.enriched.markdown.styles.LinkChipStyle
 import com.swmansion.enriched.markdown.styles.LinkVariantEntry
 import com.swmansion.enriched.markdown.styles.StyleConfig
 
@@ -26,7 +27,7 @@ class SpanStyleCache(
   val linkFontFamily: String = style.linkStyle.fontFamily
   val linkColor: Int = style.linkStyle.color
   val linkUnderline: Boolean = style.linkStyle.underline
-  val linkBackgroundColor: Int = style.linkStyle.backgroundColor
+  val linkChip: LinkChipStyle = style.linkStyle.chip
   val linkVariants: List<LinkVariantEntry> = style.linkVariants
 
   private val compiledVariantPatterns: List<Pair<Regex, LinkVariantEntry>> =
@@ -100,6 +101,9 @@ class SpanStyleCache(
     return compiledVariantPatterns.firstOrNull { (regex, _) -> regex.containsMatchIn(url) }?.second
   }
 
+  /** Chip style for [url]: the matching variant's own, else the base link's (never merged). */
+  fun linkChipForUrl(url: String): LinkChipStyle = resolvedVariantForUrl(url)?.chip ?: linkChip
+
   /**
    * Cached typeface for font family + style (BOLD, ITALIC, BOLD_ITALIC).
    *

package/android/src/main/java/com/swmansion/enriched/markdown/spans/CodeBackgroundSpan.kt

@@ -5,9 +5,6 @@ import android.graphics.Paint
 import android.graphics.Path
 import android.graphics.RectF
 import android.text.Spanned
-import android.text.StaticLayout
-import android.text.TextPaint
-import android.text.style.LeadingMarginSpan
 import android.text.style.LineBackgroundSpan
 import com.swmansion.enriched.markdown.styles.StyleConfig
 import kotlin.math.max
@@ -73,14 +70,6 @@ class CodeBackgroundSpan(
     drawShapes(canvas, isFirst, isLast)
   }
 
-  /**
-   * Returns the x position of [index] relative to the line's left edge, including any
-   * leading margin. The measuring StaticLayout is built from a subSequence that keeps
-   * all spans, so LeadingMarginSpans (lists, blockquotes) are already applied to
-   * getPrimaryHorizontal; adding the margin again on top would shift the background
-   * right by the indent. The margin is only added explicitly in the early-return case,
-   * where no layout is built.
-   */
   private fun getHorizontalOffset(
     text: CharSequence,
     lineStart: Int,
@@ -88,105 +77,30 @@ class CodeBackgroundSpan(
     index: Int,
     paint: Paint,
     leadingMargin: Int,
-  ): Float {
-    if (index <= lineStart) return leadingMargin.toFloat()
-    val lineText = text.subSequence(lineStart, lineEnd)
-    val textPaint = paint as? TextPaint ?: TextPaint(paint)
-    val layout = StaticLayout.Builder.obtain(lineText, 0, lineText.length, textPaint, 10000).build()
-    return layout.getPrimaryHorizontal(index - lineStart)
-  }
+  ): Float = InlineChipPainter.horizontalOffset(text, lineStart, lineEnd, index, paint, leadingMargin)
 
   private fun drawShapes(
     canvas: Canvas,
     isFirst: Boolean,
     isLast: Boolean,
   ) {
-    val radii = createRadii(isFirst, isLast)
-
-    path.reset()
-    path.addRoundRect(rect, radii, Path.Direction.CW)
-    canvas.drawPath(path, sharedBackgroundPaint)
-
-    if (isFirst && isLast) {
-      canvas.drawPath(path, sharedBorderPaint)
-    } else {
-      drawOpenBorders(canvas, isFirst, isLast)
-    }
-  }
-
-  private fun drawOpenBorders(
-    canvas: Canvas,
-    isFirst: Boolean,
-    isLast: Boolean,
-  ) {
-    val r = CORNER_RADIUS
-    path.reset()
-
-    if (isFirst) {
-      path.moveTo(rect.right, rect.top)
-      path.lineTo(rect.left + r, rect.top)
-      path.quadTo(rect.left, rect.top, rect.left, rect.top + r)
-      path.lineTo(rect.left, rect.bottom - r)
-      path.quadTo(rect.left, rect.bottom, rect.left + r, rect.bottom)
-      path.lineTo(rect.right, rect.bottom)
-    } else if (isLast) {
-      path.moveTo(rect.left, rect.top)
-      path.lineTo(rect.right - r, rect.top)
-      path.quadTo(rect.right, rect.top, rect.right, rect.top + r)
-      path.lineTo(rect.right, rect.bottom - r)
-      path.quadTo(rect.right, rect.bottom, rect.right - r, rect.bottom)
-      path.lineTo(rect.left, rect.bottom)
-    } else {
-      path.moveTo(rect.left, rect.top)
-      path.lineTo(rect.right, rect.top)
-      path.moveTo(rect.left, rect.bottom)
-      path.lineTo(rect.right, rect.bottom)
-    }
-    canvas.drawPath(path, sharedBorderPaint)
-  }
-
-  private fun createRadii(
-    isFirst: Boolean,
-    isLast: Boolean,
-  ) = when {
-    isFirst && isLast -> {
-      floatArrayOf(
-        CORNER_RADIUS,
-        CORNER_RADIUS,
-        CORNER_RADIUS,
-        CORNER_RADIUS,
-        CORNER_RADIUS,
-        CORNER_RADIUS,
-        CORNER_RADIUS,
-        CORNER_RADIUS,
-      )
-    }
-
-    isFirst -> {
-      floatArrayOf(CORNER_RADIUS, CORNER_RADIUS, 0f, 0f, 0f, 0f, CORNER_RADIUS, CORNER_RADIUS)
-    }
-
-    isLast -> {
-      floatArrayOf(0f, 0f, CORNER_RADIUS, CORNER_RADIUS, CORNER_RADIUS, CORNER_RADIUS, 0f, 0f)
-    }
-
-    else -> {
-      floatArrayOf(0f, 0f, 0f, 0f, 0f, 0f, 0f, 0f)
-    }
+    InlineChipPainter.draw(
+      canvas,
+      rect,
+      path,
+      CORNER_RADIUS,
+      sharedBackgroundPaint,
+      sharedBorderPaint,
+      isFirst,
+      isLast,
+      strokeInset = false,
+    )
   }
 
   private fun leadingMarginAt(
     text: Spanned,
     lineStart: Int,
-  ): Int {
-    if (lineStart >= text.length) return 0
-    val spans = text.getSpans(lineStart, lineStart + 1, LeadingMarginSpan::class.java)
-    var margin = 0
-    for (span in spans) {
-      margin += span.getLeadingMargin(false)
-    }
-    return margin
-  }
+  ): Int = InlineChipPainter.leadingMarginAt(text, lineStart)
 
   private fun adjustBottomForMargin(
     text: Spanned,