git.iko.soy/opencode-android
on 06fd41b help sign in
opencode-android/app/src/main/java/soy/iko/opencode/ui/chat/MessageBubble.kt
888 lines 43 kB history raw
1package soy.iko.opencode.ui.chat
3import androidx.compose.foundation.background
4import androidx.compose.foundation.clickable
5import androidx.compose.foundation.gestures.detectTapGestures
6import androidx.compose.foundation.layout.Arrangement
7import androidx.compose.foundation.layout.Box
8import androidx.compose.foundation.layout.BoxWithConstraints
9import androidx.compose.foundation.layout.Column
10import androidx.compose.foundation.layout.Row
11import androidx.compose.foundation.layout.fillMaxWidth
12import androidx.compose.foundation.layout.padding
13import androidx.compose.foundation.layout.size
14import androidx.compose.foundation.layout.widthIn
15import androidx.compose.material.icons.Icons
16import androidx.compose.material.icons.automirrored.filled.HelpOutline
17import androidx.compose.material.icons.automirrored.filled.PlaylistAdd
18import androidx.compose.material.icons.automirrored.filled.VolumeUp
19import androidx.compose.material.icons.filled.Close
20import androidx.compose.material.icons.filled.ContentCopy
21import androidx.compose.material.icons.filled.Edit
22import androidx.compose.material.icons.filled.Error
23import androidx.compose.material.icons.filled.FormatQuote
24import androidx.compose.material.icons.filled.MoreVert
25import androidx.compose.material.icons.filled.Pause
26import androidx.compose.material.icons.filled.PlayArrow
27import androidx.compose.material.icons.filled.Refresh
28import androidx.compose.material.icons.filled.Restore
29import androidx.compose.material.icons.filled.Share
30import androidx.compose.material.icons.filled.SmartToy
31import androidx.compose.material3.DropdownMenu
32import androidx.compose.material3.DropdownMenuItem
33import androidx.compose.material3.CircularProgressIndicator
34import androidx.compose.material3.Icon
35import androidx.compose.material3.IconButton
36import androidx.compose.material3.MaterialTheme
37import androidx.compose.material3.ripple
38import androidx.compose.material3.Text
39import androidx.compose.runtime.Composable
40import androidx.compose.runtime.getValue
41import androidx.compose.runtime.key
42import androidx.compose.runtime.mutableStateOf
43import androidx.compose.runtime.remember
44import androidx.compose.runtime.setValue
45import androidx.compose.ui.Alignment
46import androidx.compose.ui.Modifier
47import androidx.compose.ui.draw.clip
48import androidx.compose.ui.hapticfeedback.HapticFeedbackType
49import androidx.compose.ui.input.pointer.pointerInput
50import androidx.compose.ui.platform.LocalContext
51import androidx.compose.ui.platform.LocalHapticFeedback
52import androidx.compose.ui.res.stringResource
53import androidx.compose.ui.semantics.Role
54import androidx.compose.ui.semantics.contentDescription
55import androidx.compose.ui.semantics.semantics
56import androidx.compose.ui.text.style.TextOverflow
57import androidx.compose.ui.unit.dp
58import androidx.compose.foundation.interaction.MutableInteractionSource
59import soy.iko.opencode.R
60import soy.iko.opencode.data.model.AssistantMessage
61import soy.iko.opencode.data.model.MessageWithParts
62import soy.iko.opencode.data.model.ModelOption
63import soy.iko.opencode.data.model.FilePart
64import soy.iko.opencode.data.model.TextPart
65import soy.iko.opencode.data.model.UnknownMessage
66import soy.iko.opencode.data.model.UserMessage
67import soy.iko.opencode.data.network.NetworkConfig
68import soy.iko.opencode.ui.components.ImageLoadContext
69import soy.iko.opencode.ui.components.copyToClipboard
70import soy.iko.opencode.ui.components.showToast
71import soy.iko.opencode.ui.components.RelativeTimeText
73/**
74 * Resolve an assistant message's model id to a friendly label from the loaded catalog,
75 * falling back to the raw id (so a model not in the catalog still shows something
76 * meaningful instead of being hidden). Returns null only when the message has no model.
77 */
78fun resolveModelLabel(info: AssistantMessage, models: List<ModelOption>): String? {
79 val id = info.modelID ?: return null
80 if (models.isEmpty()) return id
81 val byPair = models.firstOrNull { it.providerID == info.providerID && it.modelID == id }
82 return byPair?.modelLabel
83 ?: models.firstOrNull { it.modelID == id }?.modelLabel
84 ?: id
87/** Status of an outgoing message, shown as a small indicator on the user bubble. */
88enum class MessageSendStatus { SENDING, FAILED }
90/** A single message: user prompts right-aligned in a bubble, assistant output full-width. */
91@OptIn(androidx.compose.foundation.ExperimentalFoundationApi::class)
92@Composable
93fun MessageBubble(
94 message: MessageWithParts,
95 modifier: Modifier = Modifier,
96 isRunning: Boolean = false,
97 imageContext: ImageLoadContext? = null,
98 modelLabel: String? = null,
99 onOpenFile: ((String) -> Unit)? = null,
100 onRevert: (() -> Unit)? = null,
101 onEdit: ((String) -> Unit)? = null,
102 onSpeak: ((String) -> Unit)? = null,
103 isSpeaking: Boolean = false,
104 ttsState: TtsState = TtsState.IDLE,
105 onPause: (() -> Unit)? = null,
106 onResume: (() -> Unit)? = null,
107 onQuote: ((String) -> Unit)? = null,
108 onBranch: ((String) -> Unit)? = null,
109 onRegenerate: (() -> Unit)? = null,
110 sendStatus: MessageSendStatus? = null,
111 isEdited: Boolean = false,
112 onRetry: (() -> Unit)? = null,
113 onDismiss: (() -> Unit)? = null,
114 onShare: (() -> Unit)? = null,
115) {
116 when (message.info) {
117 is UserMessage -> UserBubble(message, imageContext, modifier, onOpenFile, onRevert, onEdit, onQuote, onBranch, sendStatus, isEdited, onRetry, onDismiss, onShare)
118 is UnknownMessage -> UnknownMessageBlock(message, imageContext, modifier, onOpenFile)
119 else -> AssistantBlock(message, isRunning, imageContext, modifier, modelLabel, onOpenFile, onRevert, onSpeak, isSpeaking, ttsState, onPause, onResume, onQuote, onBranch, onRegenerate, isEdited, onShare)
120 }
123/** A long-press context menu consolidating all per-message actions (copy, quote, branch,
124 * revert). Surfacing these via long-press matches the conventional Android pattern so the
125 * actions are discoverable without spotting the small 18dp inline icon row. The menu is
126 * anchored to the long-press position via [offset], and only items with non-null callbacks
127 * (and text to act on) are shown. */
128@Composable
129private fun MessageLongPressMenu(
130 expanded: Boolean,
131 onDismiss: () -> Unit,
132 offset: androidx.compose.ui.unit.DpOffset,
133 text: String?,
134 onCopy: (() -> Unit)?,
135 onQuote: ((String) -> Unit)?,
136 onBranch: ((String) -> Unit)?,
137 onRevert: (() -> Unit)?,
138 onShare: (() -> Unit)?,
139) {
140 DropdownMenu(expanded = expanded, onDismissRequest = onDismiss, offset = offset) {
141 if (text != null && onCopy != null) {
142 DropdownMenuItem(
143 leadingIcon = { Icon(Icons.Filled.ContentCopy, contentDescription = null) },
144 text = { Text(stringResource(R.string.copy)) },
145 onClick = { onDismiss(); onCopy() },
146 )
147 }
148 // Per-message share: fire an ACTION_SEND with this message's Markdown, so a user can
149 // forward a single reply (e.g. a code snippet) without exporting the whole transcript.
150 // Gated on text so an image-only prompt doesn't show a no-op Share entry.
151 if (text != null && onShare != null) {
152 DropdownMenuItem(
153 leadingIcon = { Icon(Icons.Filled.Share, contentDescription = null) },
154 text = { Text(stringResource(R.string.share_message)) },
155 onClick = { onDismiss(); onShare() },
156 )
157 }
158 if (text != null && onQuote != null) {
159 DropdownMenuItem(
160 leadingIcon = { Icon(Icons.Filled.FormatQuote, contentDescription = null) },
161 text = { Text(stringResource(R.string.quote_reply)) },
162 onClick = { onDismiss(); onQuote(text) },
163 )
164 }
165 if (text != null && onBranch != null) {
166 DropdownMenuItem(
167 leadingIcon = { Icon(Icons.AutoMirrored.Filled.PlaylistAdd, contentDescription = null) },
168 text = { Text(stringResource(R.string.branch_session)) },
169 onClick = { onDismiss(); onBranch(text) },
170 )
171 }
172 if (onRevert != null) {
173 DropdownMenuItem(
174 leadingIcon = { Icon(Icons.Filled.Restore, contentDescription = null) },
175 text = { Text(stringResource(R.string.revert_to_here)) },
176 onClick = { onDismiss(); onRevert() },
177 )
178 }
179 }
182/** Overflow menu of secondary per-message actions (quote-reply, branch-a-new-session).
183 * Shown when at least one action is available and there's text to act on. Keeps the
184 * inline action row (copy/edit/revert/speak) uncluttered while still surfacing the extras. */
185@Composable
186private fun MessageOverflow(
187 text: String,
188 onQuote: ((String) -> Unit)?,
189 onBranch: ((String) -> Unit)?,
190) {
191 if (onQuote == null && onBranch == null) return
192 var expanded by remember { mutableStateOf(false) }
193 val moreLabel = stringResource(R.string.message_actions)
194 val haptics = LocalHapticFeedback.current
195 androidx.compose.foundation.layout.Box {
196 IconButton(onClick = { expanded = true }) {
197 Icon(
198 Icons.Filled.MoreVert,
199 contentDescription = moreLabel,
200 modifier = Modifier.size(18.dp),
201 tint = MaterialTheme.colorScheme.onSurfaceVariant,
202 )
203 }
204 DropdownMenu(expanded = expanded, onDismissRequest = { expanded = false }) {
205 onQuote?.let { quote ->
206 DropdownMenuItem(
207 leadingIcon = { Icon(Icons.Filled.FormatQuote, contentDescription = null) },
208 text = { Text(stringResource(R.string.quote_reply)) },
209 onClick = {
210 expanded = false
211 haptics.performHapticFeedback(HapticFeedbackType.TextHandleMove)
212 quote(text)
213 },
214 )
215 }
216 onBranch?.let { branch ->
217 DropdownMenuItem(
218 leadingIcon = { Icon(Icons.AutoMirrored.Filled.PlaylistAdd, contentDescription = null) },
219 text = { Text(stringResource(R.string.branch_session)) },
220 onClick = {
221 expanded = false
222 haptics.performHapticFeedback(HapticFeedbackType.TextHandleMove)
223 branch(text)
224 },
225 )
226 }
227 }
228 }
231/** A small "revert to before this message" icon button, shown next to Copy when a revert
232 * handler is supplied. Reverting is undoable (see the chat's revert banner), so it acts
233 * immediately without a confirmation dialog. */
234@Composable
235private fun RevertButton(onRevert: () -> Unit) {
236 val label = stringResource(R.string.revert_to_here)
237 val haptics = LocalHapticFeedback.current
238 IconButton(onClick = {
239 haptics.performHapticFeedback(HapticFeedbackType.TextHandleMove)
240 onRevert()
241 }) {
242 Icon(
243 Icons.Filled.Restore,
244 contentDescription = label,
245 modifier = Modifier.size(18.dp),
246 tint = MaterialTheme.colorScheme.onSurfaceVariant,
247 )
248 }
251@Composable
252private fun UnknownMessageBlock(
253 message: MessageWithParts,
254 imageContext: ImageLoadContext?,
255 modifier: Modifier,
256 onOpenFile: ((String) -> Unit)? = null,
257) {
258 // Forward-compat: a role the client doesn't model. Render a muted note so the user
259 // sees *something* rather than an unlabeled block, plus any parts (e.g. text) the
260 // server attached, so content isn't silently dropped.
261 Column(
262 modifier = modifier.fillMaxWidth(),
263 verticalArrangement = Arrangement.spacedBy(8.dp),
264 ) {
265 Row(verticalAlignment = Alignment.CenterVertically) {
266 Icon(
267 Icons.AutoMirrored.Filled.HelpOutline,
268 contentDescription = null,
269 modifier = Modifier.size(14.dp),
270 tint = MaterialTheme.colorScheme.onSurfaceVariant,
271 )
272 Text(
273 stringResource(R.string.unknown_message),
274 style = MaterialTheme.typography.labelSmall,
275 color = MaterialTheme.colorScheme.onSurfaceVariant,
276 modifier = Modifier.padding(start = 6.dp),
277 )
278 }
279 // key() per part so each PartView gets a distinct saveable registry slot: rememberSaveable
280 // inside PartView keys its RESET on part.id but its registry key is the positional
281 // compositeKeyHash, which collides for sibling parts in a keyless loop — mis-restoring
282 // expand/collapse state across parts after process death. (index disambiguates blank ids.)
283 message.parts.forEachIndexed { index, part ->
284 key(part.id, index) {
285 PartView(part, imageContext = imageContext, onOpenFile = onOpenFile)
286 }
287 }
288 MessageTimestampText(message.info)
289 }
292@OptIn(androidx.compose.foundation.ExperimentalFoundationApi::class)
293@Composable
294private fun UserBubble(
295 message: MessageWithParts,
296 imageContext: ImageLoadContext?,
297 modifier: Modifier,
298 onOpenFile: ((String) -> Unit)? = null,
299 onRevert: (() -> Unit)? = null,
300 onEdit: ((String) -> Unit)? = null,
301 onQuote: ((String) -> Unit)? = null,
302 onBranch: ((String) -> Unit)? = null,
303 sendStatus: MessageSendStatus? = null,
304 isEdited: Boolean = false,
305 onRetry: (() -> Unit)? = null,
306 onDismiss: (() -> Unit)? = null,
307 onShare: (() -> Unit)? = null,
308) {
309 val context = LocalContext.current
310 val haptics = LocalHapticFeedback.current
311 val density = androidx.compose.ui.platform.LocalDensity.current
312 val copyLabel = stringResource(R.string.copy)
313 val editLabel = stringResource(R.string.edit_message)
314 // Long-press context menu state: consolidates copy/quote/branch/revert into a standard
315 // Android long-press menu so the actions are discoverable without spotting the 18dp icons.
316 // The menu anchors at the actual touch point rather than a fixed corner.
317 var longPressMenu by remember { mutableStateOf(false) }
318 var longPressOffset by remember { mutableStateOf(androidx.compose.ui.unit.DpOffset(0.dp, 0.dp)) }
319 // Collect text from all TextParts for copying, so a user can reuse/repost their
320 // own prompt. Memoized so a scroll-induced recomposition doesn't re-scan the list.
321 val textToCopy = remember(message.parts) {
322 message.parts
323 .filterIsInstance<TextPart>()
324 .joinToString("\n\n") { it.text }
325 .takeIf { it.isNotBlank() }
326 }
327 val onCopy: (() -> Unit)? = textToCopy?.let { text ->
328 {
329 haptics.performHapticFeedback(HapticFeedbackType.TextHandleMove)
330 copyToClipboard(context, context.getString(R.string.clip_label_message), text)
331 }
332 }
333 // Wrap-content bubble capped at the width fraction (end-aligned), so a one-word prompt is
334 // snug instead of a wide empty box while long prompts still cap at the readable fraction.
335 BoxWithConstraints(modifier = modifier.fillMaxWidth()) {
336 Column(
337 modifier = Modifier
338 .align(Alignment.TopEnd)
339 .widthIn(max = maxWidth * NetworkConfig.userBubbleWidthFraction)
340 .clip(MaterialTheme.shapes.medium)
341 .background(MaterialTheme.colorScheme.primaryContainer)
342 .padding(12.dp),
343 verticalArrangement = Arrangement.spacedBy(6.dp),
344 ) {
345 // key() per part so each PartView gets its own saveable registry slot (see AssistantBlock).
346 message.parts.forEachIndexed { index, part ->
347 key(part.id, index) { PartView(part, imageContext = imageContext, onOpenFile = onOpenFile) }
348 }
349 // Footer row with timestamp + inline actions. Long-press opens a context menu
350 // consolidating copy/quote/branch/revert — the conventional Android pattern — so
351 // the actions are discoverable without spotting the 18dp inline icons. Uses
352 // pointerInput/detectTapGestures (not combinedClickable) so no misleading "Button"
353 // semantics are announced to TalkBack for an area whose tap does nothing, and the
354 // menu anchors at the actual long-press position instead of a fixed corner.
355 Box {
356 Row(
357 modifier = Modifier
358 .fillMaxWidth()
359 .pointerInput(Unit) {
360 detectTapGestures(onLongPress = { offset ->
361 haptics.performHapticFeedback(HapticFeedbackType.LongPress)
362 longPressOffset = androidx.compose.ui.unit.DpOffset(
363 with(density) { offset.x.toDp() },
364 with(density) { offset.y.toDp() },
365 )
366 longPressMenu = true
367 })
368 },
369 horizontalArrangement = Arrangement.SpaceBetween,
370 verticalAlignment = Alignment.CenterVertically,
371 ) {
372 Row(verticalAlignment = Alignment.CenterVertically, horizontalArrangement = Arrangement.spacedBy(6.dp)) {
373 when (sendStatus) {
374 MessageSendStatus.SENDING -> {
375 CircularProgressIndicator(
376 Modifier.size(12.dp),
377 strokeWidth = 1.5.dp,
378 color = MaterialTheme.colorScheme.onSurfaceVariant,
379 )
380 Text(
381 stringResource(R.string.message_sending),
382 style = MaterialTheme.typography.labelSmall,
383 color = MaterialTheme.colorScheme.onSurfaceVariant,
384 )
385 }
386 MessageSendStatus.FAILED -> {
387 // Tap-to-retry: the whole failed label is tappable to re-send (far more
388 // discoverable than relying on the transient snackbar). A dismiss (×) button
389 // removes the abandoned message entirely, since its text will never match a
390 // real server message and would otherwise linger forever.
391 val retryModifier = if (onRetry != null) {
392 Modifier.clickable(
393 interactionSource = remember { MutableInteractionSource() },
394 indication = ripple(),
395 role = Role.Button,
396 onClick = {
397 haptics.performHapticFeedback(HapticFeedbackType.TextHandleMove)
398 onRetry()
399 },
400 ).padding(vertical = 2.dp)
401 } else {
402 Modifier.padding(vertical = 2.dp)
403 }
404 Row(
405 verticalAlignment = Alignment.CenterVertically,
406 modifier = retryModifier.semantics(mergeDescendants = true) {
407 contentDescription = context.getString(R.string.message_send_failed) +
408 ". " + context.getString(R.string.tap_to_retry)
409 },
410 ) {
411 Icon(
412 Icons.Filled.Error,
413 contentDescription = null,
414 modifier = Modifier.size(12.dp),
415 tint = MaterialTheme.colorScheme.error,
416 )
417 Text(
418 stringResource(R.string.message_send_failed),
419 style = MaterialTheme.typography.labelSmall,
420 color = MaterialTheme.colorScheme.error,
421 modifier = Modifier.padding(start = 4.dp),
422 )
423 }
424 if (onDismiss != null) {
425 // No size override on the IconButton: it keeps the default 48dp
426 // touch target (only the inner Icon is shrunk to 14dp). An earlier
427 // `.size(20.dp)` here clamped the target to 20dp — well under the
428 // M3 minimum — which made the discard action hard to hit. Matches
429 // the pattern used by every other inline IconButton in this file.
430 IconButton(
431 onClick = {
432 haptics.performHapticFeedback(HapticFeedbackType.TextHandleMove)
433 onDismiss()
434 },
435 ) {
436 Icon(
437 Icons.Filled.Close,
438 contentDescription = stringResource(R.string.outbox_discard),
439 modifier = Modifier.size(14.dp),
440 tint = MaterialTheme.colorScheme.onSurfaceVariant,
441 )
442 }
443 }
444 }
445 null -> {}
446 }
447 if (isEdited && sendStatus == null) {
448 Text(
449 stringResource(R.string.message_edited),
450 style = MaterialTheme.typography.labelSmall,
451 color = MaterialTheme.colorScheme.onSurfaceVariant,
452 )
453 }
454 MessageTimestampText(message.info)
455 }
456 Row(verticalAlignment = Alignment.CenterVertically) {
457 // Edit affordance: reload this prompt into the composer, reverting the
458 // conversation to before it so a Send replaces it. Only when there's text
459 // to edit (an image-only prompt has nothing to reload).
460 if (onEdit != null && textToCopy != null) {
461 val dropsAttachmentsMsg = stringResource(R.string.edit_drops_attachments)
462 val hasAttachments = message.parts.any { it is FilePart }
463 IconButton(onClick = {
464 onEdit(textToCopy)
465 // Editing reloads only the text; if this prompt carried images or
466 // files, tell the user they won't be reattached rather than
467 // silently dropping them.
468 if (hasAttachments) showToast(context, dropsAttachmentsMsg)
469 }) {
470 Icon(
471 Icons.Filled.Edit,
472 contentDescription = editLabel,
473 modifier = Modifier.size(18.dp),
474 tint = MaterialTheme.colorScheme.onSurfaceVariant,
475 )
476 }
477 }
478 onRevert?.let { RevertButton(it) }
479 // Copy affordance for user prompts, mirroring the assistant block.
480 // Without it, reusing a prior prompt requires discovering the
481 // long-press on the markdown text (and only TextParts support that).
482 if (textToCopy != null) {
483 IconButton(
484 onClick = {
485 haptics.performHapticFeedback(HapticFeedbackType.TextHandleMove)
486 copyToClipboard(context, context.getString(R.string.clip_label_message), textToCopy)
487 },
488 ) {
489 Icon(
490 Icons.Filled.ContentCopy,
491 contentDescription = copyLabel,
492 modifier = Modifier.size(18.dp),
493 tint = MaterialTheme.colorScheme.onSurfaceVariant,
494 )
495 }
496 }
497 // Quote-reply / branch-a-new-session, tucked behind an overflow so the
498 // inline row stays compact. Only when there's prompt text to act on.
499 if (textToCopy != null) {
500 MessageOverflow(text = textToCopy, onQuote = onQuote, onBranch = onBranch)
501 }
502 }
503 }
504 MessageLongPressMenu(
505 expanded = longPressMenu,
506 onDismiss = { longPressMenu = false },
507 offset = longPressOffset,
508 text = textToCopy,
509 onCopy = onCopy,
510 onQuote = onQuote,
511 onBranch = onBranch,
512 onRevert = onRevert,
513 onShare = onShare,
514 )
515 }
516 }
517 }
520@OptIn(androidx.compose.foundation.ExperimentalFoundationApi::class)
521@Composable
522private fun AssistantBlock(
523 message: MessageWithParts,
524 isRunning: Boolean,
525 imageContext: ImageLoadContext?,
526 modifier: Modifier,
527 modelLabel: String? = null,
528 onOpenFile: ((String) -> Unit)? = null,
529 onRevert: (() -> Unit)? = null,
530 onSpeak: ((String) -> Unit)? = null,
531 isSpeaking: Boolean = false,
532 ttsState: TtsState = TtsState.IDLE,
533 onPause: (() -> Unit)? = null,
534 onResume: (() -> Unit)? = null,
535 onQuote: ((String) -> Unit)? = null,
536 onBranch: ((String) -> Unit)? = null,
537 onRegenerate: (() -> Unit)? = null,
538 isEdited: Boolean = false,
539 onShare: (() -> Unit)? = null,
540) {
541 Column(
542 modifier = modifier.fillMaxWidth(),
543 verticalArrangement = Arrangement.spacedBy(8.dp),
544 ) {
545 val info = message.info
546 if (info is AssistantMessage) {
547 // Avatar + model label row: the robot icon gives the assistant a consistent visual
548 // identity (left-aligned, mirroring the user's right-aligned bubble) so on a long
549 // scroll the speaker is unambiguous without reading the label.
550 AssistantHeader(label = modelLabel ?: info.modelID)
551 }
552 message.parts.forEachIndexed { index, part ->
553 // Only the final part of a running message is actively streaming; earlier parts
554 // (a finished reasoning block, a completed tool call) are already done, so passing
555 // isRunning to all of them would keep a completed reasoning block showing the
556 // "Thinking…" spinner until the whole message finishes.
557 val partStreaming = isRunning && index == message.parts.lastIndex
558 // key() per part so each PartView gets a distinct saveable registry slot: without it,
559 // rememberSaveable expand/collapse state (keyed only positionally) is shared across
560 // sibling parts and mis-restored after process death. (index disambiguates blank ids.)
561 key(part.id, index) {
562 PartView(part, isRunning = partStreaming, modifier = Modifier.fillMaxWidth(), imageContext = imageContext, onOpenFile = onOpenFile)
563 }
564 }
565 // Agent-reported error: the server sets AssistantMessage.error when the run finished
566 // with a model/agent error (distinct from a transport failure, which surfaces as a
567 // snackbar). Render it as an error-tinted note above the footer so a reader can tell
568 // the reply is incomplete/failed by glancing at the bubble, not just from the absence
569 // of content. Folded into the same `info is AssistantMessage` branch as the cost
570 // summary below to avoid an extra conditional raising this function's complexity.
571 if (info is AssistantMessage) {
572 info.error?.let { err -> AgentErrorNote(err) }
573 val cost = info.cost
574 val tokens = info.tokens
575 val tokenFormat = stringResource(R.string.tokens_in_out)
576 val reasoningFormat = stringResource(R.string.tokens_reasoning_format)
577 val cacheReadFormat = stringResource(R.string.tokens_cache_read_format)
578 val cacheWriteFormat = stringResource(R.string.tokens_cache_write_format)
579 val costShort = stringResource(R.string.cost_format_short)
580 val costLong = stringResource(R.string.cost_format_long)
581 // Memoize the formatted cost/tokens line so a scroll-induced or unrelated
582 // state-flip recomposition doesn't re-run NumberFormat + buildList +
583 // joinToString for every visible assistant bubble. The assembly logic lives in
584 // [buildCostSummary] (extracted to keep this function's complexity in check).
585 //
586 // Reasoning and cache figures are only added when non-zero, so a model that
587 // doesn't report them (or a request that didn't use them) shows the same compact
588 // "1.2k in · 540 out • $0.012" line as before; a reasoning-model reply gains a
589 // "3.1k reasoning" and/or "8.0k cache read" segment. The single-line cap and
590 // ellipsis below keep the line tidy at narrow widths even with all four segments.
591 val costSummary = remember(info.isComplete, tokens, cost, tokenFormat, reasoningFormat, cacheReadFormat, cacheWriteFormat, costShort, costLong) {
592 buildCostSummary(info.isComplete, tokens, cost, tokenFormat, reasoningFormat, cacheReadFormat, cacheWriteFormat, costShort, costLong)
593 }
594 // Long-press context menu state (see UserBubble for rationale).
595 var longPressMenu by remember { mutableStateOf(false) }
596 var longPressOffset by remember { mutableStateOf(androidx.compose.ui.unit.DpOffset(0.dp, 0.dp)) }
597 // Collect text from all TextParts for copy/read-aloud. Memoized so a
598 // scroll-induced recomposition doesn't re-scan the parts list.
599 val textToCopy = remember(message.parts) {
600 message.parts
601 .filterIsInstance<TextPart>()
602 .joinToString("\n\n") { it.text }
603 .takeIf { it.isNotBlank() }
604 }
605 val assistantHaptics = LocalHapticFeedback.current
606 val assistantContext = LocalContext.current
607 val assistantDensity = androidx.compose.ui.platform.LocalDensity.current
608 val onCopy: (() -> Unit)? = textToCopy?.let { text ->
609 {
610 assistantHaptics.performHapticFeedback(HapticFeedbackType.TextHandleMove)
611 copyToClipboard(assistantContext, assistantContext.getString(R.string.clip_label_message), text)
612 }
613 }
614 Box {
615 Row(
616 modifier = Modifier
617 .fillMaxWidth()
618 .pointerInput(Unit) {
619 detectTapGestures(onLongPress = { offset ->
620 assistantHaptics.performHapticFeedback(HapticFeedbackType.LongPress)
621 longPressOffset = androidx.compose.ui.unit.DpOffset(
622 with(assistantDensity) { offset.x.toDp() },
623 with(assistantDensity) { offset.y.toDp() },
624 )
625 longPressMenu = true
626 })
627 },
628 horizontalArrangement = Arrangement.SpaceBetween,
629 verticalAlignment = Alignment.CenterVertically,
630 ) {
631 Column(verticalArrangement = Arrangement.spacedBy(2.dp)) {
632 costSummary?.let {
633 Text(
634 it,
635 style = MaterialTheme.typography.labelSmall,
636 color = MaterialTheme.colorScheme.onSurfaceVariant,
637 // Cap to a single line so a long "1.2k in · 540 out • $0.012"
638 // summary doesn't wrap the "•" awkwardly at narrow widths.
639 maxLines = 1,
640 overflow = TextOverflow.Ellipsis,
641 )
642 }
643 if (isEdited) {
644 Text(
645 stringResource(R.string.message_edited),
646 style = MaterialTheme.typography.labelSmall,
647 color = MaterialTheme.colorScheme.onSurfaceVariant,
648 )
649 }
650 MessageTimestampText(message.info)
651 }
652 AssistantActions(
653 textToCopy = textToCopy,
654 isSpeaking = isSpeaking,
655 ttsState = ttsState,
656 onSpeak = onSpeak,
657 onPause = onPause,
658 onResume = onResume,
659 onRevert = onRevert,
660 onQuote = onQuote,
661 onBranch = onBranch,
662 onRegenerate = onRegenerate,
663 )
664 }
665 MessageLongPressMenu(
666 expanded = longPressMenu,
667 onDismiss = { longPressMenu = false },
668 offset = longPressOffset,
669 text = textToCopy,
670 onCopy = onCopy,
671 onQuote = onQuote,
672 onBranch = onBranch,
673 onRevert = onRevert,
674 onShare = onShare,
675 )
676 }
677 } else {
678 MessageTimestampText(message.info)
679 }
680 }
683/** Avatar + model label row for an assistant message. Extracted from [AssistantBlock] to keep
684 * that function's cyclomatic complexity under the detekt threshold. */
685@Composable
686private fun AssistantHeader(label: String?) {
687 Row(verticalAlignment = Alignment.CenterVertically) {
688 Box(
689 modifier = Modifier
690 .size(28.dp)
691 .clip(MaterialTheme.shapes.extraSmall)
692 .background(MaterialTheme.colorScheme.secondaryContainer),
693 contentAlignment = Alignment.Center,
694 ) {
695 Icon(
696 Icons.Filled.SmartToy,
697 contentDescription = null,
698 modifier = Modifier.size(18.dp),
699 tint = MaterialTheme.colorScheme.onSecondaryContainer,
700 )
701 }
702 if (label != null) {
703 Text(
704 label,
705 style = MaterialTheme.typography.labelSmall,
706 color = MaterialTheme.colorScheme.onSurfaceVariant,
707 maxLines = 1,
708 overflow = TextOverflow.Ellipsis,
709 modifier = Modifier
710 .padding(start = 6.dp)
711 .weight(1f, fill = false),
712 )
713 }
714 }
717/** Trailing action buttons for an assistant message: read-aloud (TTS), revert, copy.
718 * Extracted from [AssistantBlock] to keep that function under the complexity threshold. */
719@Composable
720private fun AssistantActions(
721 textToCopy: String?,
722 isSpeaking: Boolean,
723 ttsState: TtsState,
724 onSpeak: ((String) -> Unit)?,
725 onPause: (() -> Unit)?,
726 onResume: (() -> Unit)?,
727 onRevert: (() -> Unit)?,
728 onQuote: ((String) -> Unit)? = null,
729 onBranch: ((String) -> Unit)? = null,
730 onRegenerate: (() -> Unit)? = null,
731) {
732 val context = LocalContext.current
733 val haptics = LocalHapticFeedback.current
734 val copyLabel = stringResource(R.string.copy)
735 Row(verticalAlignment = Alignment.CenterVertically) {
736 // Regenerate: re-run the preceding user prompt to get a fresh reply. Only meaningful
737 // for a completed, non-streaming message, so the caller gates it on !isRunning.
738 if (onRegenerate != null) {
739 val regenerateLabel = stringResource(R.string.regenerate)
740 IconButton(onClick = {
741 haptics.performHapticFeedback(HapticFeedbackType.TextHandleMove)
742 onRegenerate()
743 }) {
744 Icon(
745 Icons.Filled.Refresh,
746 contentDescription = regenerateLabel,
747 modifier = Modifier.size(18.dp),
748 tint = MaterialTheme.colorScheme.onSurfaceVariant,
749 )
750 }
751 }
752 // Read-aloud control with pause/resume. Tapping cycles through play → pause → resume,
753 // so a user listening to a long reply can step away and continue from roughly where
754 // they stopped without re-playing the whole message. Extracted to [TtsButton] to keep
755 // this function's cyclomatic complexity under the detekt threshold.
756 if (onSpeak != null && textToCopy != null) {
757 TtsButton(
758 text = textToCopy,
759 isSpeaking = isSpeaking,
760 ttsState = ttsState,
761 onSpeak = onSpeak,
762 onPause = onPause,
763 onResume = onResume,
764 )
765 }
766 onRevert?.let { RevertButton(it) }
767 if (textToCopy != null) {
768 IconButton(onClick = {
769 haptics.performHapticFeedback(HapticFeedbackType.TextHandleMove)
770 copyToClipboard(context, context.getString(R.string.clip_label_message), textToCopy)
771 }) {
772 Icon(
773 Icons.Filled.ContentCopy,
774 contentDescription = copyLabel,
775 modifier = Modifier.size(18.dp),
776 tint = MaterialTheme.colorScheme.onSurfaceVariant,
777 )
778 }
779 }
780 // Assistant replies can be quoted into the composer or branched into a new session.
781 if (textToCopy != null && (onQuote != null || onBranch != null)) {
782 MessageOverflow(text = textToCopy, onQuote = onQuote, onBranch = onBranch)
783 }
784 }
787/** Read-aloud button with play/pause/resume states. Extracted from [AssistantActions] to keep
788 * that function's cyclomatic complexity under the detekt threshold. The icon reflects the
789 * next action: VolumeUp (play) when idle, Pause when playing, PlayArrow (resume) when paused. */
790@Composable
791private fun TtsButton(
792 text: String,
793 isSpeaking: Boolean,
794 ttsState: TtsState,
795 onSpeak: (String) -> Unit,
796 onPause: (() -> Unit)?,
797 onResume: (() -> Unit)?,
798) {
799 val haptics = LocalHapticFeedback.current
800 val label = when {
801 isSpeaking && ttsState == TtsState.PAUSED -> stringResource(R.string.resume_reading)
802 isSpeaking -> stringResource(R.string.pause_reading)
803 else -> stringResource(R.string.read_aloud)
804 }
805 IconButton(onClick = {
806 haptics.performHapticFeedback(HapticFeedbackType.TextHandleMove)
807 when {
808 isSpeaking && ttsState == TtsState.PAUSED -> onResume?.invoke()
809 isSpeaking -> onPause?.invoke()
810 else -> onSpeak(text)
811 }
812 }) {
813 val icon = when {
814 isSpeaking && ttsState == TtsState.PAUSED -> Icons.Filled.PlayArrow
815 isSpeaking -> Icons.Filled.Pause
816 else -> Icons.AutoMirrored.Filled.VolumeUp
817 }
818 val tint = if (isSpeaking) MaterialTheme.colorScheme.primary
819 else MaterialTheme.colorScheme.onSurfaceVariant
820 Icon(
821 icon,
822 contentDescription = label,
823 modifier = Modifier.size(18.dp),
824 tint = tint,
825 )
826 }
829@Composable
830private fun MessageTimestampText(info: soy.iko.opencode.data.model.MessageInfo) {
831 // Compact relative label by default; long-press reveals the full absolute timestamp
832 // (RelativeTimeText renders nothing when the time is absent, so no isNotEmpty guard).
833 RelativeTimeText(info.time?.created)
836/**
837 * Render an [AssistantMessage.error] as an error-tinted note on the bubble. The error is a
838 * tolerant-decoded [JsonElement]; we extract a readable string from it (a primitive's content
839 * for a bare string, a pretty-printed object otherwise, trimmed to a reasonable length so a
840 * verbose error object doesn't dominate the bubble).
841 */
842@Composable
843private fun AgentErrorNote(error: kotlinx.serialization.json.JsonElement) {
844 // Extract a readable string from the tolerant-decoded JsonElement: a bare string
845 // primitive returns its content (the common `{"error": "..."}` case); an object/array
846 // is pretty-printed and trimmed so a verbose error doesn't dominate the bubble.
847 val text = remember(error) {
848 val raw = if (error is kotlinx.serialization.json.JsonPrimitive) {
849 error.content
850 } else {
851 runCatching {
852 soy.iko.opencode.data.network.OpencodeJson.encodeToString(
853 kotlinx.serialization.json.JsonElement.serializer(),
854 error,
855 )
856 }.getOrDefault(error.toString())
857 }
858 if (raw.length <= 500) raw else raw.take(500) + "…"
859 }.takeIf { it.isNotBlank() } ?: return
860 val label = stringResource(R.string.agent_error_label)
861 Row(
862 modifier = Modifier
863 .fillMaxWidth()
864 .clip(MaterialTheme.shapes.small)
865 .background(MaterialTheme.colorScheme.errorContainer)
866 .padding(10.dp),
867 verticalAlignment = Alignment.Top,
868 ) {
869 Icon(
870 Icons.Filled.Error,
871 contentDescription = null,
872 tint = MaterialTheme.colorScheme.onErrorContainer,
873 modifier = Modifier.size(16.dp),
874 )
875 Column(modifier = Modifier.padding(start = 8.dp)) {
876 Text(
877 label,
878 style = MaterialTheme.typography.labelMedium,
879 color = MaterialTheme.colorScheme.onErrorContainer,
880 )
881 Text(
882 text,
883 style = MaterialTheme.typography.bodySmall,
884 color = MaterialTheme.colorScheme.onErrorContainer,
885 )
886 }
887 }