Update FocusContainer

Author: egorikftp

sdk/compose/highlights-core/src/commonMain/kotlin/io/github/composegears/valkyrie/sdk/compose/highlights/core/Common.kt

@@ -9,6 +9,7 @@ import androidx.compose.ui.text.font.FontWeight
 import androidx.compose.ui.text.style.TextDecoration
 import dev.snipme.highlights.Highlights
 import dev.snipme.highlights.model.BoldHighlight
+import dev.snipme.highlights.model.CodeHighlight
 import dev.snipme.highlights.model.ColorHighlight
 import dev.snipme.highlights.model.PhraseLocation
 import dev.snipme.highlights.model.SyntaxLanguage
@@ -61,6 +62,27 @@ fun Highlights.buildAnnotatedString() = buildAnnotatedString {
         }
 }
 
+fun List<CodeHighlight>.generateAnnotatedString(code: String) =
+    buildAnnotatedString {
+        append(code)
+
+        forEach {
+            when (it) {
+                is BoldHighlight -> addStyle(
+                    SpanStyle(fontWeight = FontWeight.Bold),
+                    start = it.location.start,
+                    end = it.location.end,
+                )
+
+                is ColorHighlight -> addStyle(
+                    SpanStyle(color = Color(it.rgb).copy(alpha = 1f)),
+                    start = it.location.start,
+                    end = it.location.end,
+                )
+            }
+        }
+    }
+
 fun getEmphasisLocations(
     codeBlock: String,
     highlightText: String,

tools/idea-plugin/src/main/kotlin/io/github/composegears/valkyrie/jewel/editor/CodeEditor.kt

@@ -13,24 +13,28 @@ import androidx.compose.foundation.verticalScroll
 import androidx.compose.runtime.Composable
 import androidx.compose.runtime.LaunchedEffect
 import androidx.compose.runtime.getValue
+import androidx.compose.runtime.remember
 import androidx.compose.runtime.setValue
 import androidx.compose.ui.Alignment
 import androidx.compose.ui.Modifier
 import androidx.compose.ui.graphics.SolidColor
+import androidx.compose.ui.text.AnnotatedString
 import androidx.compose.ui.text.TextStyle
 import androidx.compose.ui.text.input.OffsetMapping
 import androidx.compose.ui.text.input.TransformedText
 import androidx.compose.ui.text.style.TextAlign
 import androidx.compose.ui.unit.dp
+import dev.snipme.highlights.DefaultHighlightsResultListener
 import dev.snipme.highlights.Highlights
+import dev.snipme.highlights.model.CodeHighlight
 import dev.snipme.highlights.model.SyntaxLanguage
 import dev.snipme.highlights.model.SyntaxThemes
 import io.github.composegears.valkyrie.jewel.VerticalDivider
 import io.github.composegears.valkyrie.jewel.tooling.PreviewTheme
 import io.github.composegears.valkyrie.sdk.compose.foundation.layout.Spacer
 import io.github.composegears.valkyrie.sdk.compose.foundation.rememberMutableIntState
 import io.github.composegears.valkyrie.sdk.compose.foundation.rememberMutableState
-import io.github.composegears.valkyrie.sdk.compose.highlights.core.buildAnnotatedString
+import io.github.composegears.valkyrie.sdk.compose.highlights.core.generateAnnotatedString
 import org.jetbrains.compose.ui.tooling.preview.Preview
 import org.jetbrains.jewel.foundation.theme.JewelTheme
 import org.jetbrains.jewel.ui.component.HorizontalScrollbar
@@ -54,6 +58,27 @@ fun CodeEditor(
 
     val isDark = JewelTheme.isDark
 
+    val highlights = remember(syntaxLanguage, isDark) {
+        Highlights.Builder()
+            .language(syntaxLanguage)
+            .theme(SyntaxThemes.darcula(darkMode = isDark))
+            .build()
+    }
+
+    var transformText by rememberMutableState { AnnotatedString(text) }
+    var highlightedText by rememberMutableState { AnnotatedString(text) }
+
+    LaunchedEffect(transformText) {
+        highlights.getBuilder()
+            .code(transformText.text)
+            .build()
+            .getHighlightsAsync(object : DefaultHighlightsResultListener() {
+                override fun onSuccess(result: List<CodeHighlight>) {
+                    highlightedText = result.generateAnnotatedString(text)
+                }
+            })
+    }
+
     val linesCount by rememberMutableIntState(text) { text.count { it == '\n' } + 1 }
 
     LaunchedEffect(linesTextScroll.value) {
@@ -91,17 +116,19 @@ fun CodeEditor(
                     value = text,
                     onValueChange = { onValueChange(it.updateTabs(translateTabToSpaces)) },
                     visualTransformation = { text ->
-                        val code = Highlights.Builder()
-                            .code(text.text)
-                            .language(syntaxLanguage)
-                            .theme(SyntaxThemes.darcula(darkMode = isDark))
-                            .build()
-                            .buildAnnotatedString()
-
-                        TransformedText(
-                            text = code,
-                            offsetMapping = OffsetMapping.Identity,
-                        )
+                        if (transformText != text) {
+                            transformText = text
+
+                            TransformedText(
+                                text = text,
+                                offsetMapping = OffsetMapping.Identity,
+                            )
+                        }else {
+                            TransformedText(
+                                text = highlightedText,
+                                offsetMapping = OffsetMapping.Identity,
+                            )
+                        }
                     },
                     cursorBrush = SolidColor(JewelTheme.textFieldStyle.colors.caret),
                     textStyle = textStyle,