Try to fix the scrolling issue

Signed-off-by: denis0001-dev <denis0001.dev@ya.ru>

app/src/main/java/ru/fromchat/ui/App.kt

@@ -5,12 +5,14 @@ import androidx.compose.animation.AnimatedContentTransitionScope.SlideDirection.
 import androidx.compose.animation.core.tween
 import androidx.compose.runtime.Composable
 import androidx.compose.runtime.CompositionLocalProvider
+import androidx.compose.runtime.LaunchedEffect
 import androidx.compose.runtime.compositionLocalOf
 import androidx.compose.ui.unit.IntOffset
 import androidx.navigation.NavController
 import androidx.navigation.compose.NavHost
 import androidx.navigation.compose.composable
 import androidx.navigation.compose.rememberNavController
+import ru.fromchat.api.WebSocketManager
 import ru.fromchat.ui.auth.LoginScreen
 import ru.fromchat.ui.auth.RegisterScreen
 import ru.fromchat.ui.main.MainScreen
@@ -19,6 +21,10 @@ val LocalNavController = compositionLocalOf<NavController> { error("") }
 
 @Composable
 fun App() {
+    LaunchedEffect(Unit) {
+        WebSocketManager.connect()
+    }
+
     FromChatTheme(dynamicColor = false) {
         val navController = rememberNavController()
         val animationSpec = tween<IntOffset>(400)

app/src/main/java/ru/fromchat/ui/PublicChatScreen.kt

@@ -21,13 +21,14 @@ import androidx.compose.foundation.layout.size
 import androidx.compose.foundation.layout.width
 import androidx.compose.foundation.layout.widthIn
 import androidx.compose.foundation.layout.windowInsetsPadding
-import androidx.compose.foundation.rememberScrollState
+import androidx.compose.foundation.lazy.LazyColumn
+import androidx.compose.foundation.lazy.items
+import androidx.compose.foundation.lazy.rememberLazyListState
 import androidx.compose.foundation.shape.RoundedCornerShape
 import androidx.compose.foundation.text.BasicTextField
 import androidx.compose.foundation.text.input.TextFieldLineLimits
 import androidx.compose.foundation.text.input.rememberTextFieldState
 import androidx.compose.foundation.text.input.setTextAndPlaceCursorAtEnd
-import androidx.compose.foundation.verticalScroll
 import androidx.compose.material.icons.Icons
 import androidx.compose.material.icons.automirrored.filled.ArrowBack
 import androidx.compose.material.icons.automirrored.filled.Send
@@ -61,11 +62,8 @@ import androidx.compose.ui.text.font.FontWeight
 import androidx.compose.ui.text.style.TextOverflow
 import androidx.compose.ui.unit.dp
 import androidx.compose.ui.unit.sp
-import kotlinx.coroutines.CoroutineScope
-import kotlinx.coroutines.Dispatchers
 import kotlinx.coroutines.delay
 import kotlinx.coroutines.launch
-import kotlinx.coroutines.withContext
 import kotlinx.datetime.TimeZone
 import kotlinx.datetime.format
 import kotlinx.datetime.toLocalDateTime
@@ -91,15 +89,21 @@ import kotlin.time.Instant
 fun PublicChatScreen() {
     val navController = LocalNavController.current
     val scope = rememberCoroutineScope()
-    val scrollState = rememberScrollState()
+    val listState = rememberLazyListState()
 
     val messages = remember { mutableStateListOf<Message>() }
     var loading by remember { mutableStateOf(true) }
 
-    LaunchedEffect(Unit) {
-        val coroutineContext = coroutineContext
+    suspend fun scrollDown(animated: Boolean = true) {
+        Log.d("PublicChatScreen", "Scrolling down")
+        if (animated) {
+            listState.animateScrollToItem(messages.lastIndex)
+        } else {
+            listState.scrollToItem(messages.lastIndex)
+        }
+    }
 
-        CoroutineScope(Dispatchers.IO).launch {
+    LaunchedEffect(Unit) {
         apiRequest {
             ApiClient.getMessages()
         }.onSuccess {
@@ -113,8 +117,7 @@ fun PublicChatScreen() {
                 if (msg.type == "newMessage") {
                     try {
                         messages += Json.decodeFromJsonElement<Message>(msg.data!!)
-                            delay(500)
-                            scrollState.animateScrollTo(scrollState.maxValue)
+                        scrollDown()
                     } catch (e: Exception) {
                         Log.e("PublicChatScreen", "Failed to process incoming message:", e)
                     }
@@ -123,11 +126,8 @@ fun PublicChatScreen() {
         }
 
         delay(500)
-            withContext(coroutineContext) {
-                scrollState.scrollTo(scrollState.maxValue)
-            }
+        scrollDown(false)
         loading = false
-        }.join()
     }
 
     Scaffold(
@@ -207,8 +207,7 @@ fun PublicChatScreen() {
                                 )
                             }.getOrNull()?.let {
                                 messages += it.message
-                                delay(500)
-                                scrollState.animateScrollTo(scrollState.maxValue)
+                                scrollDown()
                             }
 
                             message.setTextAndPlaceCursorAtEnd("")
@@ -224,10 +223,9 @@ fun PublicChatScreen() {
         }
     ) { innerPadding ->
         Box(Modifier.fillMaxSize()) {
-            Column(
-                Modifier
-                    .padding(innerPadding)
-                    .verticalScroll(scrollState)
+            LazyColumn(
+                state = listState,
+                modifier = Modifier.padding(innerPadding)
             ) {
                 @Composable
                 fun Message(
@@ -302,7 +300,7 @@ fun PublicChatScreen() {
                     }
                 }
 
-                messages.forEach {
+                items(messages, { it.id }) {
                     Message(
                         text = it.content,
                         isAuthor = it.username == ApiClient.user!!.username,