fix: setting a mention now replaces whole query

Author: hejsztynx

android/src/main/java/com/swmansion/enriched/textinput/styles/ParametrizedStyles.kt

@@ -1,6 +1,7 @@
 package com.swmansion.enriched.textinput.styles
 
 import android.text.Editable
+import android.text.Selection
 import android.text.Spannable
 import android.text.SpannableStringBuilder
 import android.text.Spanned
@@ -18,6 +19,7 @@ class ParametrizedStyles(
   private val view: EnrichedTextInputView,
 ) {
   private var mentionStart: Int? = null
+  private var mentionEnd: Int? = null
   private var isSettingLinkSpan = false
 
   var mentionIndicators: Array<String> = emptyArray<String>()
@@ -297,6 +299,9 @@ class ParametrizedStyles(
       indicator = mentionIndicatorRegex.find(currentWord.text)?.value ?: ""
     }
 
+    mentionStart = finalStart
+    mentionEnd = finalEnd
+
     // Mirror iOS conflicting-styles behaviour: check the full candidate range for
     // a finalized mention span. If the span's stored text still matches what is in
     // the buffer the mention is intact — block the event (covers HTML-loaded
@@ -314,6 +319,7 @@ class ParametrizedStyles(
       }
       spannable.removeSpan(span)
       mentionStart = spanStart
+      mentionEnd = spanEnd
     }
 
     // Extract text without indicator
@@ -322,6 +328,7 @@ class ParametrizedStyles(
     // Means we are starting mention
     if (text.isEmpty()) {
       mentionStart = finalStart
+      mentionEnd = finalEnd
     }
 
     mentionHandler.onMention(indicator, text)
@@ -383,9 +390,12 @@ class ParametrizedStyles(
     }
 
     val start = mentionStart ?: selectionStart
+    val end = mentionEnd ?: selectionEnd
 
     view.runAsATransaction {
-      spannable.replace(start, selectionEnd, text)
+      Selection.setSelection(spannable, end)
+
+      spannable.replace(start, end, text)
 
       val span = EnrichedInputMentionSpan(text, indicator, attributes, view.htmlStyle)
       val spanEnd = start + text.length
@@ -401,6 +411,7 @@ class ParametrizedStyles(
     view.mentionHandler?.reset()
     view.selection.validateStyles()
     mentionStart = null
+    mentionEnd = null
   }
 
   fun getStyleRange(): Pair<Int, Int> = view.selection?.getInlineSelection() ?: Pair(0, 0)