@@ -1,57 +1,34 @@
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
73import 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
134import androidx.compose.animation.fadeIn
145import androidx.compose.animation.fadeOut
156import androidx.compose.animation.scaleIn
167import androidx.compose.animation.scaleOut
17−import androidx.compose.foundation.ExperimentalFoundationApi
18−import androidx.compose.foundation.background
19−import androidx.compose.foundation.clickable
20−import androidx.compose.foundation.combinedClickable
218import androidx.compose.foundation.layout.Arrangement
229import androidx.compose.foundation.layout.Box
2310import androidx.compose.foundation.layout.BoxScope
2411import androidx.compose.foundation.layout.Column
2512import androidx.compose.foundation.layout.Row
26−import androidx.compose.foundation.layout.Spacer
2713import androidx.compose.foundation.layout.fillMaxSize
2814import androidx.compose.foundation.layout.fillMaxWidth
29−import androidx.compose.foundation.layout.heightIn
3015import androidx.compose.foundation.layout.PaddingValues
31−import androidx.compose.foundation.layout.WindowInsets
32−import androidx.compose.foundation.layout.ime
3316import androidx.compose.foundation.layout.imePadding
3417import androidx.compose.foundation.layout.padding
3518import androidx.compose.foundation.layout.size
3619import androidx.compose.foundation.lazy.LazyColumn
37−import androidx.compose.foundation.lazy.itemsIndexed
20+import androidx.compose.foundation.lazy.items
3821import androidx.compose.foundation.lazy.rememberLazyListState
39−import androidx.compose.foundation.shape.CircleShape
40−import androidx.compose.foundation.shape.RoundedCornerShape
4122import androidx.compose.foundation.text.KeyboardActions
4223import androidx.compose.foundation.text.KeyboardOptions
4324import androidx.compose.material.icons.Icons
4425import androidx.compose.material.icons.automirrored.filled.ArrowBack
4526import androidx.compose.material.icons.automirrored.filled.Send
46−import androidx.compose.material.icons.filled.ChatBubbleOutline
47−import androidx.compose.material.icons.filled.Close
4827import androidx.compose.material.icons.filled.KeyboardArrowDown
49−import androidx.compose.material.icons.filled.KeyboardArrowUp
5028import androidx.compose.material.icons.filled.Stop
5129import androidx.compose.material3.CircularProgressIndicator
5230import androidx.compose.material3.ExperimentalMaterial3Api
5331import androidx.compose.material3.FilledIconButton
54−import androidx.compose.material3.FilledTonalButton
5532import androidx.compose.material3.FloatingActionButton
5633import androidx.compose.material3.Icon
5734import androidx.compose.material3.IconButton
@@ -61,27 +38,16 @@ import androidx.compose.material3.OutlinedTextField
6138import androidx.compose.material3.Scaffold
6239import androidx.compose.material3.Surface
6340import androidx.compose.material3.Text
64−import androidx.compose.material3.TextButton
6541import androidx.compose.material3.TopAppBar
6642import androidx.compose.runtime.Composable
6743import androidx.compose.runtime.LaunchedEffect
6844import androidx.compose.runtime.derivedStateOf
6945import androidx.compose.runtime.getValue
70−import androidx.compose.runtime.mutableStateOf
7146import androidx.compose.runtime.remember
7247import androidx.compose.runtime.rememberCoroutineScope
73−import androidx.compose.runtime.setValue
7448import androidx.compose.ui.Alignment
7549import androidx.compose.ui.Modifier
76−import androidx.compose.ui.draw.clip
77−import 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
81−import androidx.compose.ui.platform.LocalHapticFeedback
82−import androidx.compose.ui.text.font.FontStyle
8350import androidx.compose.ui.text.input.ImeAction
84−import androidx.compose.ui.text.style.TextAlign
8551import androidx.compose.ui.text.style.TextOverflow
8652import androidx.compose.ui.unit.dp
8753import kotlinx.coroutines.launch
@@ -89,17 +55,11 @@ import soy.iko.opencode.core.Event
8955import soy.iko.opencode.core.UiState
9056import soy.iko.opencode.ui.components.MessageBubble
9157
92−@OptIn(ExperimentalMaterial3Api::class, ExperimentalFoundationApi::class)
58+@OptIn(ExperimentalMaterial3Api::class)
9359@Composable
9460fun ChatScreen(state: UiState, dispatch: (Event) -> Unit) {
9561 val listState = rememberLazyListState()
9662 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
10363
10464 // Show a jump-to-bottom button once the user scrolls up away from the latest
10565 // message. derivedStateOf keeps this from recomposing on every pixel of scroll —
@@ -113,22 +73,6 @@ fun ChatScreen(state: UiState, dispatch: (Event) -> Unit) {
11373 }
11474 }
11575
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−
122− // Track how many messages the user has "seen" (i.e. was at the bottom for). While
123− // they're scrolled up, any messages beyond that count are unseen — flag the FAB.
124− var lastSeenCount by remember { mutableStateOf(0) }
125− LaunchedEffect(showScrollButton, state.messages.size) {
126− if (!showScrollButton) {
127− lastSeenCount = state.messages.size
128− }
129− }
130− val hasUnseenMessages = showScrollButton && state.messages.size > lastSeenCount
131−
13276 // Keep the newest content in view as it streams in, but only when the user is
13377 // already near the bottom. If they've scrolled up to read history, leave it alone.
13478 val lastLen = state.messages.lastOrNull()?.text?.length ?: 0
@@ -148,20 +92,10 @@ fun ChatScreen(state: UiState, dispatch: (Event) -> Unit) {
14892 topBar = {
14993 TopAppBar(
15094 title = {
151− val title = state.currentSessionTitle.ifEmpty { "Chat" }
15295 Text(
153− text = if (state.messages.isNotEmpty()) "$title (${state.messages.size})" else title,
96+ text = state.currentSessionTitle.ifEmpty { "Chat" },
15497 maxLines = 1,
15598 overflow = TextOverflow.Ellipsis,
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− ),
16599 )
166100 },
167101 navigationIcon = {
@@ -183,7 +117,6 @@ fun ChatScreen(state: UiState, dispatch: (Event) -> Unit) {
183117 ChatInputBar(
184118 draft = state.draft,
185119 generating = state.generating,
186− loading = state.loading && state.messages.isEmpty(),
187120 onDraftChange = { dispatch(Event.DraftChanged(it)) },
188121 onSend = { dispatch(Event.SendMessage(state.draft)) },
189122 onStop = { dispatch(Event.CancelGeneration) },
@@ -197,86 +130,34 @@ fun ChatScreen(state: UiState, dispatch: (Event) -> Unit) {
197130 if (state.loading && state.messages.isEmpty()) {
198131 LinearProgressIndicator(Modifier.fillMaxWidth())
199132 }
200− if (state.error != null && state.messages.isNotEmpty()) {
201− ErrorBanner(
202− message = state.error,
203− onDismiss = { dispatch(Event.DismissError) },
204− )
205− }
206133 Box(Modifier.fillMaxWidth().weight(1f)) {
207134 if (state.messages.isEmpty() && !state.loading) {
208− Column(
135+ Text(
136+ text = "Send a message to start the conversation.",
137+ style = MaterialTheme.typography.bodyMedium,
138+ color = MaterialTheme.colorScheme.onSurfaceVariant,
209139 modifier = Modifier.align(Alignment.Center).padding(32.dp),
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− }
140+ )
226141 }
227142 LazyColumn(
228143 state = listState,
229144 modifier = Modifier.fillMaxSize(),
230145 contentPadding = PaddingValues(vertical = 8.dp),
231146 ) {
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− }
147+ items(state.messages, key = { it.id }) { message ->
148+ MessageBubble(message)
253149 }
254150 }
255151 ScrollToBottomButton(
256− visible = showScrollButton && !isImeVisible,
257− showBadge = hasUnseenMessages,
152+ visible = showScrollButton,
258153 onClick = {
259154 coroutineScope.launch {
260155 if (state.messages.isNotEmpty()) {
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)
156+ listState.animateScrollToItem(state.messages.size - 1)
268157 }
269158 }
270159 },
271160 )
272− ScrollToTopButton(
273− visible = showScrollTop,
274− onClick = {
275− coroutineScope.launch {
276− listState.animateScrollToItem(0)
277− }
278− },
279− )
280161 }
281162 }
282163 }
@@ -286,149 +167,23 @@ fun ChatScreen(state: UiState, dispatch: (Event) -> Unit) {
286167// resolves cleanly. Called inline inside a Column, the enclosing ColumnScope receiver
287168// would otherwise capture the ColumnScope.AnimatedVisibility overload and fail to compile.
288169@Composable
289−private fun BoxScope.ScrollToBottomButton(visible: Boolean, showBadge: Boolean, onClick: () -> Unit) {
170+private fun BoxScope.ScrollToBottomButton(visible: Boolean, onClick: () -> Unit) {
290171 AnimatedVisibility(
291172 visible = visible,
292173 modifier = Modifier.align(Alignment.BottomEnd),
293174 enter = scaleIn() + fadeIn(),
294175 exit = scaleOut() + fadeOut(),
295176 ) {
296− Box(Modifier.padding(16.dp)) {
297− FloatingActionButton(
298− onClick = onClick,
299− modifier = Modifier.size(44.dp),
300− ) {
301− Icon(
302− Icons.Filled.KeyboardArrowDown,
303− contentDescription = "Scroll to latest message",
304− )
305− }
306− if (showBadge) {
307− Box(
308− modifier = Modifier
309− .align(Alignment.TopEnd)
310− .size(8.dp)
311− .clip(CircleShape)
312− .background(MaterialTheme.colorScheme.error),
313− )
314− }
315− }
316− }
317−}
318−
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(
177+ FloatingActionButton(
178+ onClick = onClick,
400179 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),
180+ .padding(16.dp)
181+ .size(44.dp),
406182 ) {
407− Text(
408− text = "Thinking",
409− style = MaterialTheme.typography.bodyMedium.copy(fontStyle = FontStyle.Italic),
410− color = MaterialTheme.colorScheme.onSurfaceVariant,
183+ Icon(
184+ Icons.Filled.KeyboardArrowDown,
185+ contentDescription = "Scroll to latest message",
411186 )
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− }
432187 }
433188 }
434189}
@@ -458,133 +213,43 @@ private fun ReconnectingBanner() {
458213 }
459214}
460215
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−
490216@Composable
491217private fun ChatInputBar(
492218 draft: String,
493219 generating: Boolean,
494− loading: Boolean,
495220 onDraftChange: (String) -> Unit,
496221 onSend: () -> Unit,
497222 onStop: () -> Unit,
498223) {
499− val haptic = LocalHapticFeedback.current
500− val clipboardManager = LocalClipboardManager.current
501224 Surface(tonalElevation = 3.dp) {
502− Column(
503− modifier = Modifier.fillMaxWidth().imePadding().padding(horizontal = 12.dp, vertical = 8.dp),
504− ) {
505225 Row(
506− modifier = Modifier.fillMaxWidth(),
226+ modifier = Modifier.fillMaxWidth().imePadding().padding(horizontal = 12.dp, vertical = 8.dp),
507227 verticalAlignment = Alignment.Bottom,
508228 horizontalArrangement = Arrangement.spacedBy(8.dp),
509229 ) {
510230 OutlinedTextField(
511231 value = draft,
512232 onValueChange = onDraftChange,
513− modifier = Modifier.weight(1f).heightIn(min = 48.dp),
514− enabled = !loading,
233+ modifier = Modifier.weight(1f),
515234 placeholder = { Text("Message opencode…") },
516235 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− },
550236 keyboardOptions = KeyboardOptions(imeAction = ImeAction.Send),
551237 keyboardActions = KeyboardActions(onSend = {
552− if (!generating && !loading && draft.isNotBlank()) onSend()
238+ if (!generating && draft.isNotBlank()) onSend()
553239 }),
554240 )
555241 if (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")
242+ FilledIconButton(onClick = onStop) {
243+ Icon(Icons.Filled.Stop, contentDescription = "Stop generating")
564244 }
565245 } else {
566246 FilledIconButton(
567− onClick = {
568− haptic.performHapticFeedback(HapticFeedbackType.LongPress)
569− onSend()
570− },
571− enabled = draft.isNotBlank() && !loading,
247+ onClick = onSend,
248+ enabled = draft.isNotBlank(),
572249 ) {
573250 Icon(Icons.AutoMirrored.Filled.Send, contentDescription = "Send")
574251 }
575252 }
576253 }
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− }
589254 }
590255}