Extension function for simpler annotated string creation with decoration in Jetpack Compose Compose

AnnotatedStringExt.kt

import androidx.compose.ui.text.AnnotatedString
import androidx.compose.ui.text.ParagraphStyle
import androidx.compose.ui.text.SpanStyle
import androidx.compose.ui.text.TtsAnnotation

fun AnnotatedString.Builder.appendDecorated(
    text: String,
    spanStyle: SpanStyle? = null,
    paragraphStyle: ParagraphStyle? = null,
    textAnnotation: TextAnnotation? = null,
    ttsAnnotation: TtsAnnotation? = null,
) = withDecoration(spanStyle, paragraphStyle, textAnnotation, ttsAnnotation) { append(text) }

fun AnnotatedString.Builder.withDecoration(
    spanStyle: SpanStyle? = null,
    paragraphStyle: ParagraphStyle? = null,
    textAnnotation: TextAnnotation? = null,
    ttsAnnotation: TtsAnnotation? = null,
    content: AnnotatedString.Builder.() -> Unit
) {
    var popCount = 0

    spanStyle?.let {
        pushStyle(it)
        popCount += 1
    }

    paragraphStyle?.let {
        pushStyle(it)
        popCount += 1
    }

    textAnnotation?.let {
        pushStringAnnotation(tag = it.tag, annotation = it.annotation)
        popCount += 1
    }

    ttsAnnotation?.let {
        pushTtsAnnotation(it)
        popCount += 1
    }

    this.content()

    repeat(popCount) {
        pop()
    }
}

class TextAnnotation(val tag: String, val annotation: String)

Sample usage

@Composable
fun SampleUsage() {
    val text1 = buildAnnotatedString {
        withDecoration(
            spanStyle = SpanStyle(Color.Black, textDecoration = TextDecoration.Underline),
            textAnnotation = TextAnnotation("tag", "annotation")
        ) {
            append("Decorated text")
        }
    }

    val text2 = buildAnnotatedString {
        appendDecorated(
            text = "Decorated text",
            spanStyle = SpanStyle(Color.Blue, textDecoration = TextDecoration.Underline),
            textAnnotation = TextAnnotation("tag", "annotation")
        )
    }
}