@@ -1,37 +1,57 @@
11package soy.iko.opencode.ui.screens
22
3+import android.content.ClipData
4+import android.content.ClipboardManager
5+import android.content.Context
6+import android.widget.Toast
37import androidx.compose.animation.AnimatedVisibility
8+import androidx.compose.animation.core.RepeatMode
9+import androidx.compose.animation.core.animateFloat
10+import androidx.compose.animation.core.infiniteRepeatable
11+import androidx.compose.animation.core.rememberInfiniteTransition
12+import androidx.compose.animation.core.tween
413import androidx.compose.animation.fadeIn
514import androidx.compose.animation.fadeOut
615import androidx.compose.animation.scaleIn
716import androidx.compose.animation.scaleOut
17+import androidx.compose.foundation.ExperimentalFoundationApi
818import androidx.compose.foundation.background
919import androidx.compose.foundation.clickable
20+import androidx.compose.foundation.combinedClickable
1021import androidx.compose.foundation.layout.Arrangement
1122import androidx.compose.foundation.layout.Box
1223import androidx.compose.foundation.layout.BoxScope
1324import androidx.compose.foundation.layout.Column
1425import androidx.compose.foundation.layout.Row
26+import androidx.compose.foundation.layout.Spacer
1527import androidx.compose.foundation.layout.fillMaxSize
1628import androidx.compose.foundation.layout.fillMaxWidth
29+import androidx.compose.foundation.layout.heightIn
1730import androidx.compose.foundation.layout.PaddingValues
31+import androidx.compose.foundation.layout.WindowInsets
32+import androidx.compose.foundation.layout.ime
1833import androidx.compose.foundation.layout.imePadding
1934import androidx.compose.foundation.layout.padding
2035import androidx.compose.foundation.layout.size
2136import androidx.compose.foundation.lazy.LazyColumn
22−import androidx.compose.foundation.lazy.items
37+import androidx.compose.foundation.lazy.itemsIndexed
2338import androidx.compose.foundation.lazy.rememberLazyListState
2439import androidx.compose.foundation.shape.CircleShape
40+import androidx.compose.foundation.shape.RoundedCornerShape
2541import androidx.compose.foundation.text.KeyboardActions
2642import androidx.compose.foundation.text.KeyboardOptions
2743import androidx.compose.material.icons.Icons
2844import androidx.compose.material.icons.automirrored.filled.ArrowBack
2945import androidx.compose.material.icons.automirrored.filled.Send
46+import androidx.compose.material.icons.filled.ChatBubbleOutline
47+import androidx.compose.material.icons.filled.Close
3048import androidx.compose.material.icons.filled.KeyboardArrowDown
49+import androidx.compose.material.icons.filled.KeyboardArrowUp
3150import androidx.compose.material.icons.filled.Stop
3251import androidx.compose.material3.CircularProgressIndicator
3352import androidx.compose.material3.ExperimentalMaterial3Api
3453import androidx.compose.material3.FilledIconButton
54+import androidx.compose.material3.FilledTonalButton
3555import androidx.compose.material3.FloatingActionButton
3656import androidx.compose.material3.Icon
3757import androidx.compose.material3.IconButton
@@ -41,6 +61,7 @@ import androidx.compose.material3.OutlinedTextField
4161import androidx.compose.material3.Scaffold
4262import androidx.compose.material3.Surface
4363import androidx.compose.material3.Text
64+import androidx.compose.material3.TextButton
4465import androidx.compose.material3.TopAppBar
4566import androidx.compose.runtime.Composable
4667import androidx.compose.runtime.LaunchedEffect
@@ -54,8 +75,13 @@ import androidx.compose.ui.Alignment
5475import androidx.compose.ui.Modifier
5576import androidx.compose.ui.draw.clip
5677import androidx.compose.ui.hapticfeedback.HapticFeedbackType
78+import androidx.compose.ui.platform.LocalClipboardManager
79+import androidx.compose.ui.platform.LocalContext
80+import androidx.compose.ui.platform.LocalDensity
5781import androidx.compose.ui.platform.LocalHapticFeedback
82+import androidx.compose.ui.text.font.FontStyle
5883import androidx.compose.ui.text.input.ImeAction
84+import androidx.compose.ui.text.style.TextAlign
5985import androidx.compose.ui.text.style.TextOverflow
6086import androidx.compose.ui.unit.dp
6187import kotlinx.coroutines.launch
@@ -63,11 +89,17 @@ import soy.iko.opencode.core.Event
6389import soy.iko.opencode.core.UiState
6490import soy.iko.opencode.ui.components.MessageBubble
6591
66−@OptIn(ExperimentalMaterial3Api::class)
92+@OptIn(ExperimentalMaterial3Api::class, ExperimentalFoundationApi::class)
6793@Composable
6894fun ChatScreen(state: UiState, dispatch: (Event) -> Unit) {
6995 val listState = rememberLazyListState()
7096 val coroutineScope = rememberCoroutineScope()
97+ val context = LocalContext.current
98+
99+ // Hide the jump-to-bottom FAB while the keyboard is up: it would otherwise
100+ // sit over the top of the IME. Reading the ime inset bottom during composition
101+ // observes snapshot state, so this flips as the keyboard opens/closes.
102+ val isImeVisible = WindowInsets.ime.getBottom(LocalDensity.current) > 0
71103
72104 // Show a jump-to-bottom button once the user scrolls up away from the latest
73105 // message. derivedStateOf keeps this from recomposing on every pixel of scroll —
@@ -81,6 +113,12 @@ fun ChatScreen(state: UiState, dispatch: (Event) -> Unit) {
81113 }
82114 }
83115
116+ // Offer a jump-to-top shortcut once the conversation is long and the user has
117+ // scrolled a few messages down — saves flick-scrolling back through history.
118+ val showScrollTop by remember {
119+ derivedStateOf { state.messages.size > 10 && listState.firstVisibleItemIndex > 3 }
120+ }
121+
84122 // Track how many messages the user has "seen" (i.e. was at the bottom for). While
85123 // they're scrolled up, any messages beyond that count are unseen — flag the FAB.
86124 var lastSeenCount by remember { mutableStateOf(0) }
@@ -110,11 +148,20 @@ fun ChatScreen(state: UiState, dispatch: (Event) -> Unit) {
110148 topBar = {
111149 TopAppBar(
112150 title = {
151+ val title = state.currentSessionTitle.ifEmpty { "Chat" }
113152 Text(
114− text = state.currentSessionTitle.ifEmpty { "Chat" },
153+ text = if (state.messages.isNotEmpty()) "$title (${state.messages.size})" else title,
115154 maxLines = 1,
116155 overflow = TextOverflow.Ellipsis,
117− modifier = Modifier.clickable { dispatch(Event.NavigateToSessions) },
156+ modifier = Modifier.combinedClickable(
157+ onClick = { dispatch(Event.NavigateToSessions) },
158+ onLongClick = {
159+ state.currentSessionId?.let { id ->
160+ copyToClipboard(context, id)
161+ Toast.makeText(context, "Session ID copied", Toast.LENGTH_SHORT).show()
162+ }
163+ },
164+ ),
118165 )
119166 },
120167 navigationIcon = {
@@ -136,6 +183,7 @@ fun ChatScreen(state: UiState, dispatch: (Event) -> Unit) {
136183 ChatInputBar(
137184 draft = state.draft,
138185 generating = state.generating,
186+ loading = state.loading && state.messages.isEmpty(),
139187 onDraftChange = { dispatch(Event.DraftChanged(it)) },
140188 onSend = { dispatch(Event.SendMessage(state.draft)) },
141189 onStop = { dispatch(Event.CancelGeneration) },
@@ -149,35 +197,86 @@ fun ChatScreen(state: UiState, dispatch: (Event) -> Unit) {
149197 if (state.loading && state.messages.isEmpty()) {
150198 LinearProgressIndicator(Modifier.fillMaxWidth())
151199 }
200+ if (state.error != null && state.messages.isNotEmpty()) {
201+ ErrorBanner(
202+ message = state.error,
203+ onDismiss = { dispatch(Event.DismissError) },
204+ )
205+ }
152206 Box(Modifier.fillMaxWidth().weight(1f)) {
153207 if (state.messages.isEmpty() && !state.loading) {
154− Text(
155− text = "Send a message to start the conversation.",
156− style = MaterialTheme.typography.bodyMedium,
157− color = MaterialTheme.colorScheme.onSurfaceVariant,
208+ Column(
158209 modifier = Modifier.align(Alignment.Center).padding(32.dp),
159− )
210+ horizontalAlignment = Alignment.CenterHorizontally,
211+ ) {
212+ Icon(
213+ Icons.Filled.ChatBubbleOutline,
214+ contentDescription = null,
215+ modifier = Modifier.size(48.dp),
216+ tint = MaterialTheme.colorScheme.primary.copy(alpha = 0.5f),
217+ )
218+ Spacer(Modifier.size(16.dp))
219+ Text(
220+ text = "Send a message to start the conversation.",
221+ style = MaterialTheme.typography.bodyMedium,
222+ color = MaterialTheme.colorScheme.onSurfaceVariant,
223+ textAlign = TextAlign.Center,
224+ )
225+ }
160226 }
161227 LazyColumn(
162228 state = listState,
163229 modifier = Modifier.fillMaxSize(),
164230 contentPadding = PaddingValues(vertical = 8.dp),
165231 ) {
166− items(state.messages, key = { it.id }) { message ->
167− MessageBubble(message)
232+ itemsIndexed(state.messages, key = { _, message -> message.id }) { index, message ->
233+ // Insert a date separator whenever a message falls on a different
234+ // calendar day than the one before it (and above the first message).
235+ val previous = state.messages.getOrNull(index - 1)
236+ if (shouldShowDateSeparator(previous, message)) {
237+ DateSeparator(message.time)
238+ }
239+ val sameSenderAsPrevious = index > 0 && state.messages[index - 1].isUser == message.isUser
240+ MessageBubble(
241+ message,
242+ onRetry = { dispatch(Event.SendMessage(message.text)) },
243+ compactSpacing = sameSenderAsPrevious,
244+ )
245+ }
246+ // While the assistant is spinning up but hasn't emitted a streaming
247+ // message yet (only the user's message is present), show a typing
248+ // indicator so the wait doesn't look like a stall.
249+ if (state.generating && state.messages.none { it.streaming }) {
250+ item(key = "typing-indicator") {
251+ TypingIndicator()
252+ }
168253 }
169254 }
170255 ScrollToBottomButton(
171− visible = showScrollButton,
256+ visible = showScrollButton && !isImeVisible,
172257 showBadge = hasUnseenMessages,
173258 onClick = {
174259 coroutineScope.launch {
175260 if (state.messages.isNotEmpty()) {
176− listState.animateScrollToItem(state.messages.size - 1)
261+ // When the typing indicator is showing it occupies the item
262+ // just past the last message, so scroll one further to reach it.
263+ val target = if (state.generating && state.messages.none { it.streaming })
264+ state.messages.size
265+ else
266+ state.messages.size - 1
267+ listState.animateScrollToItem(target)
177268 }
178269 }
179270 },
180271 )
272+ ScrollToTopButton(
273+ visible = showScrollTop,
274+ onClick = {
275+ coroutineScope.launch {
276+ listState.animateScrollToItem(0)
277+ }
278+ },
279+ )
181280 }
182281 }
183282 }
@@ -217,6 +316,123 @@ private fun BoxScope.ScrollToBottomButton(visible: Boolean, showBadge: Boolean,
217316 }
218317}
219318
319+// Mirror of ScrollToBottomButton, pinned to the top-start corner. Same BoxScope
320+// extension trick so the top-level AnimatedVisibility overload resolves cleanly.
321+@Composable
322+private fun BoxScope.ScrollToTopButton(visible: Boolean, onClick: () -> Unit) {
323+ AnimatedVisibility(
324+ visible = visible,
325+ modifier = Modifier.align(Alignment.TopStart),
326+ enter = scaleIn() + fadeIn(),
327+ exit = scaleOut() + fadeOut(),
328+ ) {
329+ Box(Modifier.padding(16.dp)) {
330+ FloatingActionButton(
331+ onClick = onClick,
332+ modifier = Modifier.size(44.dp),
333+ ) {
334+ Icon(
335+ Icons.Filled.KeyboardArrowUp,
336+ contentDescription = "Scroll to top",
337+ )
338+ }
339+ }
340+ }
341+}
342+
343+private fun copyToClipboard(context: Context, text: String) {
344+ val clipboard = context.getSystemService(Context.CLIPBOARD_SERVICE) as ClipboardManager
345+ clipboard.setPrimaryClip(ClipData.newPlainText("session", text))
346+}
347+
348+private val dateSeparatorFormat = java.time.format.DateTimeFormatter.ofPattern("MMM d")
349+
350+/** Convert an epoch-millis timestamp to the local calendar day it falls on. */
351+private fun localDayOf(epochMillis: Long): java.time.LocalDate =
352+ java.time.Instant.ofEpochMilli(epochMillis)
353+ .atZone(java.time.ZoneId.systemDefault())
354+ .toLocalDate()
355+
356+/**
357+ * A separator is shown above a message when it starts a new calendar day — i.e. the
358+ * first timestamped message, or one whose day differs from the previous message's.
359+ * Messages without a valid time (time <= 0) never get one.
360+ */
361+private fun shouldShowDateSeparator(previous: soy.iko.opencode.core.MessageView?, current: soy.iko.opencode.core.MessageView): Boolean {
362+ if (current.time <= 0) return false
363+ if (previous == null || previous.time <= 0) return true
364+ return localDayOf(previous.time) != localDayOf(current.time)
365+}
366+
367+/** Human-friendly day label: "Today", "Yesterday", or "MMM d" (e.g. "Jul 8"). */
368+private fun dateSeparatorLabel(day: java.time.LocalDate): String {
369+ val today = java.time.LocalDate.now()
370+ return when (day) {
371+ today -> "Today"
372+ today.minusDays(1) -> "Yesterday"
373+ else -> day.format(dateSeparatorFormat)
374+ }
375+}
376+
377+@Composable
378+private fun DateSeparator(epochMillis: Long) {
379+ Text(
380+ text = dateSeparatorLabel(localDayOf(epochMillis)),
381+ style = MaterialTheme.typography.labelSmall,
382+ color = MaterialTheme.colorScheme.onSurfaceVariant,
383+ textAlign = TextAlign.Center,
384+ modifier = Modifier.fillMaxWidth().padding(vertical = 8.dp),
385+ )
386+}
387+
388+// Assistant-style bubble with three sequentially-pulsing dots, shown as a stand-in
389+// until the real streaming reply starts. Mirrors MessageBubble's left-aligned,
390+// surfaceVariant look so it reads as an incoming message.
391+@Composable
392+private fun TypingIndicator() {
393+ val shape = RoundedCornerShape(topStart = 16.dp, topEnd = 16.dp, bottomStart = 4.dp, bottomEnd = 16.dp)
394+ val transition = rememberInfiniteTransition(label = "typing")
395+ Row(
396+ modifier = Modifier.fillMaxWidth().padding(horizontal = 12.dp, vertical = 4.dp),
397+ horizontalArrangement = Arrangement.Start,
398+ ) {
399+ Row(
400+ modifier = Modifier
401+ .clip(shape)
402+ .background(MaterialTheme.colorScheme.surfaceVariant)
403+ .padding(horizontal = 14.dp, vertical = 12.dp),
404+ verticalAlignment = Alignment.CenterVertically,
405+ horizontalArrangement = Arrangement.spacedBy(6.dp),
406+ ) {
407+ Text(
408+ text = "Thinking",
409+ style = MaterialTheme.typography.bodyMedium.copy(fontStyle = FontStyle.Italic),
410+ color = MaterialTheme.colorScheme.onSurfaceVariant,
411+ )
412+ Row(horizontalArrangement = Arrangement.spacedBy(4.dp)) {
413+ // Stagger the three dots so they fade in and out in sequence.
414+ listOf(0, 150, 300).forEach { staggerMs ->
415+ val alpha by transition.animateFloat(
416+ initialValue = 0.2f,
417+ targetValue = 1f,
418+ animationSpec = infiniteRepeatable(
419+ animation = tween(durationMillis = 600, delayMillis = staggerMs),
420+ repeatMode = RepeatMode.Reverse,
421+ ),
422+ label = "dot",
423+ )
424+ Box(
425+ modifier = Modifier
426+ .size(6.dp)
427+ .clip(CircleShape)
428+ .background(MaterialTheme.colorScheme.onSurfaceVariant.copy(alpha = alpha)),
429+ )
430+ }
431+ }
432+ }
433+ }
434+}
435+
220436@Composable
221437private fun ReconnectingBanner() {
222438 Surface(
@@ -242,35 +458,109 @@ private fun ReconnectingBanner() {
242458 }
243459}
244460
461+@Composable
462+private fun ErrorBanner(message: String, onDismiss: () -> Unit) {
463+ Surface(
464+ modifier = Modifier.fillMaxWidth(),
465+ color = MaterialTheme.colorScheme.errorContainer,
466+ ) {
467+ Row(
468+ modifier = Modifier.fillMaxWidth().padding(start = 16.dp, top = 6.dp, bottom = 6.dp, end = 4.dp),
469+ verticalAlignment = Alignment.CenterVertically,
470+ horizontalArrangement = Arrangement.spacedBy(8.dp),
471+ ) {
472+ Text(
473+ text = message,
474+ style = MaterialTheme.typography.bodySmall,
475+ color = MaterialTheme.colorScheme.onErrorContainer,
476+ modifier = Modifier.weight(1f),
477+ )
478+ IconButton(onClick = onDismiss, modifier = Modifier.size(32.dp)) {
479+ Icon(
480+ Icons.Filled.Close,
481+ contentDescription = "Dismiss error",
482+ tint = MaterialTheme.colorScheme.onErrorContainer,
483+ modifier = Modifier.size(18.dp),
484+ )
485+ }
486+ }
487+ }
488+}
489+
245490@Composable
246491private fun ChatInputBar(
247492 draft: String,
248493 generating: Boolean,
494+ loading: Boolean,
249495 onDraftChange: (String) -> Unit,
250496 onSend: () -> Unit,
251497 onStop: () -> Unit,
252498) {
253499 val haptic = LocalHapticFeedback.current
500+ val clipboardManager = LocalClipboardManager.current
254501 Surface(tonalElevation = 3.dp) {
255− Row(
502+ Column(
256503 modifier = Modifier.fillMaxWidth().imePadding().padding(horizontal = 12.dp, vertical = 8.dp),
504+ ) {
505+ Row(
506+ modifier = Modifier.fillMaxWidth(),
257507 verticalAlignment = Alignment.Bottom,
258508 horizontalArrangement = Arrangement.spacedBy(8.dp),
259509 ) {
260510 OutlinedTextField(
261511 value = draft,
262512 onValueChange = onDraftChange,
263− modifier = Modifier.weight(1f),
513+ modifier = Modifier.weight(1f).heightIn(min = 48.dp),
514+ enabled = !loading,
264515 placeholder = { Text("Message opencode…") },
265516 maxLines = 5,
517+ // Offer a quick way to wipe the draft once there's something to clear
518+ // and we're not mid-generation (where the field is a poor place to fuss).
519+ trailingIcon = if (draft.isNotEmpty() && !generating) {
520+ {
521+ IconButton(
522+ onClick = { onDraftChange("") },
523+ modifier = Modifier.size(20.dp),
524+ ) {
525+ Icon(
526+ Icons.Filled.Close,
527+ contentDescription = "Clear draft",
528+ modifier = Modifier.size(16.dp),
529+ )
530+ }
531+ }
532+ } else {
533+ null
534+ },
535+ // Surface a running count once the draft gets long, so the user has a
536+ // sense of scale before sending a wall of text.
537+ supportingText = if (draft.length > 500) {
538+ {
539+ Text(
540+ text = "${draft.length} characters",
541+ style = MaterialTheme.typography.labelSmall,
542+ color = MaterialTheme.colorScheme.onSurfaceVariant,
543+ textAlign = TextAlign.End,
544+ modifier = Modifier.fillMaxWidth(),
545+ )
546+ }
547+ } else {
548+ null
549+ },
266550 keyboardOptions = KeyboardOptions(imeAction = ImeAction.Send),
267551 keyboardActions = KeyboardActions(onSend = {
268− if (!generating && draft.isNotBlank()) onSend()
552+ if (!generating && !loading && draft.isNotBlank()) onSend()
269553 }),
270554 )
271555 if (generating) {
272− FilledIconButton(onClick = onStop) {
273− Icon(Icons.Filled.Stop, contentDescription = "Stop generating")
556+ FilledTonalButton(onClick = onStop) {
557+ Icon(
558+ Icons.Filled.Stop,
559+ contentDescription = null,
560+ modifier = Modifier.size(18.dp),
561+ )
562+ Spacer(Modifier.size(6.dp))
563+ Text("Stop")
274564 }
275565 } else {
276566 FilledIconButton(
@@ -278,11 +568,23 @@ private fun ChatInputBar(
278568 haptic.performHapticFeedback(HapticFeedbackType.LongPress)
279569 onSend()
280570 },
281− enabled = draft.isNotBlank(),
571+ enabled = draft.isNotBlank() && !loading,
282572 ) {
283573 Icon(Icons.AutoMirrored.Filled.Send, contentDescription = "Send")
284574 }
285575 }
286576 }
577+ // With nothing typed yet, offer a one-tap way to drop in clipboard contents.
578+ if (draft.isEmpty() && !generating && !loading) {
579+ TextButton(
580+ onClick = {
581+ clipboardManager.getText()?.text?.let { onDraftChange(it) }
582+ },
583+ contentPadding = PaddingValues(0.dp),
584+ ) {
585+ Text("Paste")
586+ }
587+ }
588+ }
287589 }
288590}