@@ -1,1531 +0,0 @@
1−package soy.iko.opencode.ui.chat
3−import android.app.Activity
4−import android.content.Intent
6−import android.speech.RecognizerIntent
7−import androidx.activity.compose.BackHandler
8−import androidx.activity.compose.rememberLauncherForActivityResult
9−import androidx.activity.result.PickVisualMediaRequest
10−import androidx.activity.result.contract.ActivityResultContracts
11−import androidx.compose.animation.AnimatedVisibility
12−import androidx.compose.foundation.clickable
13−import androidx.compose.foundation.layout.Arrangement
14−import androidx.compose.foundation.layout.Box
15−import androidx.compose.foundation.layout.Column
16−import androidx.compose.foundation.layout.FlowRow
17−import androidx.compose.foundation.layout.PaddingValues
18−import androidx.compose.foundation.layout.Row
19−import androidx.compose.foundation.layout.Spacer
20−import androidx.compose.foundation.layout.WindowInsets
21−import androidx.compose.foundation.layout.fillMaxSize
22−import androidx.compose.foundation.layout.fillMaxWidth
23−import androidx.compose.foundation.layout.imePadding
24−import androidx.compose.foundation.layout.navigationBars
25−import androidx.compose.foundation.layout.padding
26−import androidx.compose.foundation.layout.size
27−import androidx.compose.foundation.layout.union
28−import androidx.compose.foundation.layout.windowInsetsPadding
29−import androidx.compose.foundation.lazy.LazyColumn
30−import androidx.compose.foundation.lazy.items
31−import androidx.compose.foundation.lazy.rememberLazyListState
32−import androidx.compose.material.icons.Icons
33−import androidx.compose.material.icons.automirrored.filled.ArrowBack
34−import androidx.compose.material.icons.automirrored.filled.Send
35−import androidx.compose.material.icons.filled.Add
36−import androidx.compose.material.icons.filled.AutoAwesome
37−import androidx.compose.material.icons.filled.Delete
38−import androidx.compose.material.icons.filled.Description
39−import androidx.compose.material.icons.filled.Edit
40−import androidx.compose.material.icons.filled.KeyboardArrowDown
41−import androidx.compose.material.icons.filled.Mic
42−import androidx.compose.material.icons.filled.MoreVert
43−import androidx.compose.material.icons.filled.PhotoCamera
44−import androidx.compose.material.icons.filled.PhotoLibrary
45−import androidx.compose.material.icons.filled.Refresh
46−import androidx.compose.material.icons.filled.Restore
47−import androidx.compose.material.icons.filled.Stop
48−import androidx.compose.material3.AlertDialog
49−import androidx.compose.material3.Button
50−import androidx.compose.material3.CircularProgressIndicator
51−import androidx.compose.material3.DropdownMenu
52−import androidx.compose.material3.DropdownMenuItem
53−import androidx.compose.material3.ExperimentalMaterial3Api
54−import androidx.compose.material3.ExtendedFloatingActionButton
55−import androidx.compose.material3.FloatingActionButton
56−import androidx.compose.material3.Icon
57−import androidx.compose.material3.IconButton
58−import androidx.compose.material3.MaterialTheme
59−import androidx.compose.material3.OutlinedTextField
60−import androidx.compose.material3.Scaffold
61−import androidx.compose.material3.SnackbarHost
62−import androidx.compose.material3.SnackbarHostState
63−import androidx.compose.material3.Surface
64−import androidx.compose.material3.Text
65−import androidx.compose.material3.TextButton
66−import androidx.compose.material3.TopAppBar
67−import androidx.compose.material3.pulltorefresh.PullToRefreshBox
68−import androidx.compose.runtime.Composable
69−import androidx.compose.runtime.CompositionLocalProvider
70−import androidx.compose.runtime.DisposableEffect
71−import androidx.compose.runtime.LaunchedEffect
72−import androidx.compose.runtime.derivedStateOf
73−import androidx.compose.runtime.getValue
74−import androidx.compose.runtime.mutableStateOf
75−import androidx.compose.runtime.remember
76−import androidx.compose.runtime.rememberCoroutineScope
77−import androidx.compose.runtime.rememberUpdatedState
78−import androidx.compose.runtime.saveable.rememberSaveable
79−import androidx.compose.runtime.setValue
80−import androidx.compose.runtime.snapshotFlow
81−import androidx.compose.foundation.text.KeyboardActions
82−import androidx.compose.foundation.text.KeyboardOptions
83−import androidx.compose.ui.text.input.KeyboardCapitalization
84−import androidx.compose.ui.Alignment
85−import androidx.compose.ui.Modifier
86−import androidx.compose.ui.focus.focusRequester
87−import androidx.compose.ui.input.key.Key
88−import androidx.compose.ui.input.key.KeyEventType
89−import androidx.compose.ui.input.key.isCtrlPressed
90−import androidx.compose.ui.input.key.isShiftPressed
91−import androidx.compose.ui.input.key.key
92−import androidx.compose.ui.input.key.onPreviewKeyEvent
93−import androidx.compose.ui.input.key.type
94−import androidx.compose.ui.platform.LocalContext
95−import androidx.compose.ui.platform.LocalHapticFeedback
96−import androidx.compose.ui.platform.LocalView
97−import androidx.compose.ui.platform.testTag
98−import androidx.compose.ui.res.stringResource
99−import androidx.compose.ui.semantics.Role
100−import androidx.compose.ui.semantics.contentDescription
101−import androidx.compose.ui.semantics.semantics
102−import androidx.compose.ui.text.input.ImeAction
103−import androidx.compose.ui.text.style.TextOverflow
104−import androidx.compose.ui.unit.dp
105−import androidx.lifecycle.compose.collectAsStateWithLifecycle
106−import androidx.lifecycle.viewmodel.compose.viewModel
107−import kotlinx.coroutines.Dispatchers
108−import kotlinx.coroutines.launch
109−import kotlinx.coroutines.withContext
110−import soy.iko.opencode.data.model.Part
111−import soy.iko.opencode.data.model.MessageWithParts
112−import soy.iko.opencode.data.model.ReasoningPart
113−import soy.iko.opencode.data.model.TextPart
114−import soy.iko.opencode.data.model.ToolCompleted
115−import soy.iko.opencode.data.model.ToolError
116−import soy.iko.opencode.data.model.ToolPart
117−import soy.iko.opencode.data.model.ToolRunning
118−import soy.iko.opencode.data.network.EventStreamClient
119−import soy.iko.opencode.data.network.NetworkConfig
120−import soy.iko.opencode.di.AppContainer
121−import soy.iko.opencode.R
122−import soy.iko.opencode.ui.components.ConnectionBanner
123−import soy.iko.opencode.ui.components.copyToClipboard
124−import soy.iko.opencode.ui.components.LocalRelativeTimeTick
125−import soy.iko.opencode.ui.components.rememberRelativeTimeTick
126−import soy.iko.opencode.ui.components.showToast
127−import soy.iko.opencode.ui.components.toImageContext
128−import soy.iko.opencode.ui.vmFactory
129−import soy.iko.opencode.util.runCatchingCancellable
131−@OptIn(ExperimentalMaterial3Api::class)
134− container: AppContainer,
137− onOpenFile: ((String) -> Unit)? = null,
139− val vm: ChatViewModel = viewModel(key = sessionId, factory = vmFactory { ChatViewModel(container, sessionId) })
140− val hasMessages by vm.hasMessages.collectAsStateWithLifecycle()
141− val running by vm.running.collectAsStateWithLifecycle()
142− val aborting by vm.aborting.collectAsStateWithLifecycle()
143− val loading by vm.loading.collectAsStateWithLifecycle()
144− val loadError by vm.loadError.collectAsStateWithLifecycle()
145− val refreshing by vm.refreshing.collectAsStateWithLifecycle()
146− val models by vm.models.collectAsStateWithLifecycle()
147− val modelsLoading by vm.modelsLoading.collectAsStateWithLifecycle()
148− val modelsError by vm.modelsError.collectAsStateWithLifecycle()
149− val selectedModel by vm.selectedModel.collectAsStateWithLifecycle()
150− val connectionState by vm.connectionState.collectAsStateWithLifecycle()
151− val pendingPermission by vm.pendingPermission.collectAsStateWithLifecycle()
152− val agents by vm.agents.collectAsStateWithLifecycle()
153− val agentsLoading by vm.agentsLoading.collectAsStateWithLifecycle()
154− val agentsError by vm.agentsError.collectAsStateWithLifecycle()
155− val selectedAgent by vm.selectedAgent.collectAsStateWithLifecycle()
156− val commands by vm.commands.collectAsStateWithLifecycle()
157− val commandsLoading by vm.commandsLoading.collectAsStateWithLifecycle()
158− val commandsError by vm.commandsError.collectAsStateWithLifecycle()
159− val sessionTitle by vm.sessionTitle.collectAsStateWithLifecycle()
160− val sessionDeleted by vm.sessionDeleted.collectAsStateWithLifecycle()
161− val draft by vm.draft.collectAsStateWithLifecycle()
162− val attachments by vm.attachments.collectAsStateWithLifecycle()
163− val reverted by vm.reverted.collectAsStateWithLifecycle()
164− val shareUrl by vm.shareUrl.collectAsStateWithLifecycle()
165− val reconnecting by vm.reconnecting.collectAsStateWithLifecycle()
166− val sendOnEnter by container.settingsStore.sendOnEnter.collectAsStateWithLifecycle(initialValue = true)
167− val isOnline by container.isOnline.collectAsStateWithLifecycle()
168− val haptics = LocalHapticFeedback.current
169− val scope = rememberCoroutineScope()
170− val shareContext = LocalContext.current
171− val defaultShareSubject = stringResource(R.string.share_subject)
172− val sessionLabel = stringResource(R.string.session)
173− val defaultModelLabel = stringResource(R.string.default_model)
174− val subtitle = remember(selectedModel?.modelLabel, selectedAgent, defaultModelLabel) {
176− append(selectedModel?.modelLabel ?: defaultModelLabel)
177− selectedAgent?.let { append(" · $it") }
181− // Connection profile (with auth) for rendering inline image attachments.
182− // Collect as state so a server switch recomposes the image context.
183− val activeConnection by container.activeConnection.collectAsStateWithLifecycle()
184− val imageContext = remember(activeConnection?.profile) {
185− activeConnection?.profile?.toImageContext()
188− // Mark this session as read while the user is viewing it; clear on leave so new
189− // background activity can badge it again.
190− DisposableEffect(sessionId) {
191− container.setCurrentSession(sessionId)
192− onDispose { if (container.currentSession.value == sessionId) container.setCurrentSession(null) }
195− // Navigate away when the session is deleted via SSE so the user isn't left
196− // on a zombie screen showing a conversation that no longer exists. Surface a
197− // toast (not a snackbar) because the screen is about to be popped — a snackbar
198− // would be torn down before the user could read it, while a toast survives.
199− // Covers both SSE-driven deletion (deleted on another device) and a user-
200− // initiated delete from the chat overflow menu (deleteSession in ChatViewModel
201− // sets the same flag on success).
202− val deletedContext = LocalContext.current
203− LaunchedEffect(sessionDeleted) {
204− if (sessionDeleted) {
205− showToast(deletedContext, deletedContext.getString(R.string.session_deleted_chat))
210− val snackbar = remember { SnackbarHostState() }
211− val retryLabel = stringResource(R.string.retry)
212− val inputFocusRequester = remember { androidx.compose.ui.focus.FocusRequester() }
213− var showModelPicker by rememberSaveable { mutableStateOf(false) }
214− var showAgentPicker by rememberSaveable { mutableStateOf(false) }
215− var showCommandPicker by rememberSaveable { mutableStateOf(false) }
216− var showExitConfirm by rememberSaveable { mutableStateOf(false) }
217− var showStopConfirm by rememberSaveable { mutableStateOf(false) }
218− var showOverflowMenu by rememberSaveable { mutableStateOf(false) }
219− var showRenameDialog by rememberSaveable { mutableStateOf(false) }
220− var showDeleteDialog by rememberSaveable { mutableStateOf(false) }
221− var showShellDialog by rememberSaveable { mutableStateOf(false) }
223− // Keep the screen awake while the agent is working in *this* session. The foreground
224− // service that holds process priority during a run is managed app-wide off
225− // container.anyRunActive (see OpencodeApp) so it survives navigating away from the chat
226− // mid-run — otherwise leaving the screen would drop priority and Doze could choke the
227− // SSE stream before the completion notification fires.
228− val currentView = LocalView.current
229− val appContext = LocalContext.current.applicationContext
230− DisposableEffect(running) {
231− currentView.keepScreenOn = running
232− onDispose { currentView.keepScreenOn = false }
235− LaunchedEffect(Unit) {
236− vm.errorEvents.collect { event ->
237− // Only a failed *send* is retryable; attaching Retry to any other error (e.g. a
238− // message-load failure) would silently re-submit the last prompt on tap.
239− val result = if (event.retryable) {
240− snackbar.showSnackbar(message = event.message, actionLabel = retryLabel)
242− snackbar.showSnackbar(event.message)
244− if (result == androidx.compose.material3.SnackbarResult.ActionPerformed) vm.retryFailed()
248− // --- Attachments, voice, and share-link plumbing ---
249− val attachTooLargeMsg = stringResource(R.string.attachment_too_large)
250− val attachFailedMsg = stringResource(R.string.attachment_failed)
251− val attachLimitMsg = stringResource(R.string.attachment_limit)
252− val noCameraMsg = stringResource(R.string.no_camera_app)
253− val noVoiceMsg = stringResource(R.string.no_voice_app)
254− val voicePrompt = stringResource(R.string.voice_prompt)
255− val linkCopiedMsg = stringResource(R.string.link_copied)
257− // Convert each picked Uri to a base64 attachment off the main thread, honoring the
258− // per-prompt count cap and surfacing per-file errors without aborting the batch.
259− fun stageUris(uris: List<Uri>) {
260− if (uris.isEmpty()) return
263− if (vm.attachments.value.size >= NetworkConfig.maxAttachments) {
264− snackbar.showSnackbar(attachLimitMsg)
267− when (val result = uri.toAttachmentResult(appContext)) {
268− is AttachmentResult.Ok -> vm.addAttachment(result.attachment)
269− AttachmentResult.TooLarge -> snackbar.showSnackbar(attachTooLargeMsg)
270− AttachmentResult.Failed -> snackbar.showSnackbar(attachFailedMsg)
276− val photoPicker = rememberLauncherForActivityResult(
277− ActivityResultContracts.PickMultipleVisualMedia(NetworkConfig.maxAttachments),
278− ) { uris -> stageUris(uris) }
280− val filePicker = rememberLauncherForActivityResult(
281− ActivityResultContracts.GetMultipleContents(),
282− ) { uris -> stageUris(uris) }
284− // Uri survives a config change mid-capture so the TakePicture result still resolves.
285− var pendingCaptureUri by rememberSaveable { mutableStateOf<Uri?>(null) }
286− val cameraLauncher = rememberLauncherForActivityResult(
287− ActivityResultContracts.TakePicture(),
289− val uri = pendingCaptureUri
290− pendingCaptureUri = null
291− if (success && uri != null) stageUris(listOf(uri))
294− val voiceLauncher = rememberLauncherForActivityResult(
295− ActivityResultContracts.StartActivityForResult(),
297− if (result.resultCode == Activity.RESULT_OK) {
298− val spoken = result.data
299− ?.getStringArrayListExtra(RecognizerIntent.EXTRA_RESULTS)
301− ?.takeIf { it.isNotBlank() }
302− if (spoken != null) {
303− // Append to the existing draft rather than replacing, so dictation adds to
304− // whatever the user already typed.
305− val current = vm.draft.value
306− val combined = if (current.isBlank()) spoken else "$current $spoken"
307− vm.updateDraft(combined.take(NetworkConfig.maxDraftLengthChars))
312− fun launchCamera() {
313− val uri = newCameraCaptureUri(appContext)
315− scope.launch { snackbar.showSnackbar(attachFailedMsg) }
318− pendingCaptureUri = uri
319− runCatching { cameraLauncher.launch(uri) }
320− .onFailure { pendingCaptureUri = null; scope.launch { snackbar.showSnackbar(noCameraMsg) } }
324− val intent = Intent(RecognizerIntent.ACTION_RECOGNIZE_SPEECH).apply {
325− putExtra(RecognizerIntent.EXTRA_LANGUAGE_MODEL, RecognizerIntent.LANGUAGE_MODEL_FREE_FORM)
326− putExtra(RecognizerIntent.EXTRA_PROMPT, voicePrompt)
328− runCatching { voiceLauncher.launch(intent) }
329− .onFailure { scope.launch { snackbar.showSnackbar(noVoiceMsg) } }
332− // A freshly-created share link is copied to the clipboard so the user can paste it right away.
333− LaunchedEffect(Unit) {
334− vm.shareLinkEvents.collect { url ->
335− copyToClipboard(shareContext, "share", url)
336− snackbar.showSnackbar(linkCopiedMsg)
340− // Stage images shared into the app (via the system share sheet) as attachments, once.
341− LaunchedEffect(Unit) {
342− val shared = container.consumePendingSharedMedia()
343− if (shared.isNotEmpty()) stageUris(shared.map { Uri.parse(it) })
347− if (vm.send(draft)) {
348− // TextHandleMove (a light tick) is semantically right for a send commit;
349− // LongPress is reserved for long-press/swipe gestures.
350− haptics.performHapticFeedback(androidx.compose.ui.hapticfeedback.HapticFeedbackType.TextHandleMove)
354− BackHandler(enabled = running && !showModelPicker && !showAgentPicker && !showCommandPicker) { showExitConfirm = true }
360− val changeLabel = stringResource(R.string.change_model_agent)
363− .clickable(enabled = models.isNotEmpty(), role = Role.Button) { showModelPicker = true }
364− .semantics(mergeDescendants = true) { contentDescription = changeLabel },
365− verticalAlignment = Alignment.CenterVertically,
367− Column(modifier = Modifier.weight(1f, fill = false)) {
368− // maxLines=1 + Ellipsis so a long session title can't wrap and
369− // push the top bar to multiple lines, shifting the whole layout.
371− sessionTitle ?: sessionLabel,
373− overflow = TextOverflow.Ellipsis,
377− style = MaterialTheme.typography.labelSmall,
378− color = MaterialTheme.colorScheme.onSurfaceVariant,
380− overflow = TextOverflow.Ellipsis,
383− // Visual affordance that the title opens the model picker;
384− // without it the tappable title is indistinguishable from plain
385− // text and the picker is effectively undiscoverable.
386− if (models.isNotEmpty()) {
388− Icons.Filled.KeyboardArrowDown,
389− contentDescription = null,
390− tint = MaterialTheme.colorScheme.onSurfaceVariant,
391− modifier = Modifier.size(20.dp),
397− IconButton(onClick = {
398− if (running) showExitConfirm = true else onBack()
400− Icon(Icons.AutoMirrored.Filled.ArrowBack, contentDescription = stringResource(R.string.back))
404− // At-a-glance run status in the top bar. When the user is scrolled
405− // up (jump-to-latest FAB visible), the input-bar Stop button and the
406− // trailing "working…" row are off-screen, so a top-bar spinner is the
407− // only signal the agent is still working.
409− val workingLabel = stringResource(R.string.working)
410− CircularProgressIndicator(
411− Modifier.size(18.dp).semantics { contentDescription = workingLabel },
415− // Manual refresh: forces an SSE reconnect, which re-seeds messages
416− // from REST. A recovery path when the stream silently drops and the
417− // auto-reconnect re-seed is slow or fails. Shows a brief spinner as
418− // immediate tap feedback — the SSE reconnect may not visibly change
419− // the connection state when already Connected, so without it the tap
420− // appears to do nothing.
421− val refreshLabel = stringResource(R.string.refresh)
423− onClick = { vm.refreshMessages() },
424− enabled = activeConnection != null && !refreshing,
427− CircularProgressIndicator(
428− Modifier.size(18.dp).semantics { contentDescription = refreshLabel },
432− Icon(Icons.Filled.Refresh, contentDescription = refreshLabel)
435− // Overflow menu consolidates the less-frequent actions (share,
436− // commands, agent, model, rename, delete) so the top bar stays
437− // scannable on narrow phones — previously 5 icons crowded the bar
438− // alongside the back button and tappable title. Catalog-loading
439− // spinners stay inline (next to the overflow) so a load state is
440− // visible without opening the menu.
441− val moreLabel = stringResource(R.string.more)
442− val loadingLabel = stringResource(R.string.loading)
443− val shareLabel = stringResource(R.string.share_conversation)
444− val commandsLabel = stringResource(R.string.commands)
445− val agentLabel = stringResource(R.string.choose_agent)
446− val modelLabel = stringResource(R.string.choose_model)
447− val renameLabel = stringResource(R.string.rename_session_chat)
448− val deleteLabel = stringResource(R.string.delete_session_chat)
449− val createShareLabel = stringResource(R.string.create_share_link)
450− val copyShareLabel = stringResource(R.string.copy_share_link)
451− val stopShareLabel = stringResource(R.string.stop_sharing)
452− val summarizeLabel = stringResource(R.string.summarize_conversation)
453− val initLabel = stringResource(R.string.generate_agents_md)
454− val shellLabel = stringResource(R.string.run_shell_command)
456− IconButton(onClick = { showOverflowMenu = true }) {
457− Icon(Icons.Filled.MoreVert, contentDescription = moreLabel)
460− expanded = showOverflowMenu,
461− onDismissRequest = { showOverflowMenu = false },
464− text = { Text(shareLabel) },
465− enabled = hasMessages,
467− showOverflowMenu = false
469− val md = withContext(Dispatchers.Default) {
470− buildConversationMarkdown(vm.messages.value, sessionTitle)
472− val send = android.content.Intent(android.content.Intent.ACTION_SEND).apply {
473− type = "text/markdown"
474− putExtra(android.content.Intent.EXTRA_SUBJECT, sessionTitle ?: defaultShareSubject)
475− putExtra(android.content.Intent.EXTRA_TEXT, md)
477− runCatchingCancellable { shareContext.startActivity(android.content.Intent.createChooser(send, shareLabel)) }
478− .onFailure { showToast(shareContext, shareContext.getString(R.string.no_share_app)) }
483− text = { Text(commandsLabel) },
484− enabled = !commandsLoading || commands.isNotEmpty(),
485− onClick = { showOverflowMenu = false; showCommandPicker = true },
488− text = { Text(agentLabel) },
489− enabled = !agentsLoading || agents.isNotEmpty(),
490− onClick = { showOverflowMenu = false; showAgentPicker = true },
493− text = { Text(modelLabel) },
494− enabled = !modelsLoading || models.isNotEmpty(),
495− onClick = { showOverflowMenu = false; showModelPicker = true },
497− androidx.compose.material3.HorizontalDivider()
498− // Session sharing: create a public link, copy an existing one, or revoke it.
499− if (shareUrl == null) {
501− text = { Text(createShareLabel) },
502− enabled = hasMessages,
503− onClick = { showOverflowMenu = false; vm.shareSession() },
507− text = { Text(copyShareLabel) },
509− showOverflowMenu = false
511− copyToClipboard(shareContext, "share", it)
512− scope.launch { snackbar.showSnackbar(linkCopiedMsg) }
517− text = { Text(stopShareLabel) },
518− onClick = { showOverflowMenu = false; vm.unshareSession() },
522− text = { Text(summarizeLabel) },
523− enabled = hasMessages && !running,
524− onClick = { showOverflowMenu = false; vm.summarize() },
527− text = { Text(initLabel) },
529− onClick = { showOverflowMenu = false; vm.initProject() },
532− text = { Text(shellLabel) },
534− onClick = { showOverflowMenu = false; showShellDialog = true },
536− androidx.compose.material3.HorizontalDivider()
538− text = { Text(renameLabel) },
539− onClick = { showOverflowMenu = false; showRenameDialog = true },
542− text = { Text(deleteLabel, color = MaterialTheme.colorScheme.error) },
543− onClick = { showOverflowMenu = false; showDeleteDialog = true },
546− // While a catalog is initially loading (empty list + loading flag),
547− // show a tiny inline spinner beside the overflow so the user can see
548− // a load is in progress without opening the menu. Once loaded the
549− // spinner disappears; the menu items are always enabled (tapping a
550− // picker with no items shows the "no models/agents/commands" state).
551− val catalogLoading = (commandsLoading && commands.isEmpty()) ||
552− (agentsLoading && agents.isEmpty()) ||
553− (modelsLoading && models.isEmpty())
554− if (catalogLoading) {
555− CircularProgressIndicator(
557− .align(Alignment.TopEnd)
559− .semantics { contentDescription = loadingLabel },
567− snackbarHost = { SnackbarHost(snackbar) },
569− val queuedFollowUp by vm.queuedFollowUp.collectAsStateWithLifecycle()
571− AnimatedVisibility(visible = reverted) {
572− RevertBanner(onUndo = { vm.unrevert() })
576− onValueChange = vm::updateDraft,
578− aborting = aborting,
579− enabled = activeConnection != null,
580− sendOnEnter = sendOnEnter,
582− onAbort = { showStopConfirm = true },
583− queuedFollowUp = queuedFollowUp,
584− onQueueFollowUp = vm::queueFollowUp,
585− onCancelQueue = { vm.queueFollowUp("") },
586− focusRequester = inputFocusRequester,
587− attachments = attachments,
588− onRemoveAttachment = vm::removeAttachment,
591− PickVisualMediaRequest(ActivityResultContracts.PickVisualMedia.ImageOnly),
594− onPickFile = { filePicker.launch("*/*") },
595− onCamera = { launchCamera() },
596− onVoice = { launchVoice() },
601− // Collect messages inside the content lambda so streaming token updates
602− // recompose only this subtree, not the top bar / input bar / 20+ other
603− // state reads in the ChatScreen body.
604− val messages by vm.messages.collectAsStateWithLifecycle()
605− val listState = rememberLazyListState()
606− val contentScope = rememberCoroutineScope()
607− val timeTick = rememberRelativeTimeTick()
608− // Group messages by day and interleave date separators so a long conversation
609− // has visual "Today"/"Yesterday"/date breaks. Computed once per messages
610− // emission (memoized) so a scroll-induced recomposition doesn't re-scan. Lives
611− // outside the PullToRefreshBox so the auto-scroll effects below can reference
612− // the same list the LazyColumn renders (separators shift the indices vs. the
613− // raw messages list, so scroll targets must be in listItems space).
614− val todayLabel = stringResource(R.string.today)
615− val yesterdayLabel = stringResource(R.string.yesterday)
616− val listItems = remember(messages, todayLabel, yesterdayLabel) {
617− buildMessageListItems(messages, todayLabel, yesterdayLabel)
619− // listItems is a plain (non-snapshot) local, rebuilt each recomposition. The
620− // derivedStateOf and snapshotFlow lambdas below are created once (keyless
621− // remember / LaunchedEffect(Unit)); reading `listItems` directly inside them
622− // would capture the first composition's list — which is empty, since `messages`
623− // starts empty — freezing it forever. That pinned isPinnedToBottom at true and
624− // AutoScrollSignal.size at 0, breaking streaming auto-scroll and the
625− // jump-to-latest FAB. rememberUpdatedState hands those lambdas a stable State
626− // whose value tracks the latest list.
627− val currentListItems by rememberUpdatedState(listItems)
629− val isPinnedToBottom by remember {
631− val lastVisible = listState.layoutInfo.visibleItemsInfo.lastOrNull()?.index ?: -1
632− // When a run is active, the LazyColumn has one trailing "__typing"
633− // item after the last list item. Account for it so "pinned to bottom"
634− // includes the working indicator — otherwise the pin check targets
635− // listItems.lastIndex, the typing row sits just below the viewport,
636− // and the user never sees the progress indicator even when pinned.
637− val items = currentListItems
638− val effectiveLast = if (running) items.size else items.lastIndex
639− lastVisible >= effectiveLast || items.isEmpty()
643− // Jump to the newest message once, when the conversation first loads. Guarded by a
644− // saveable one-shot flag so it doesn't re-fire on recomposition after a config change
645− // (rotation) — rememberLazyListState restores the user's scroll offset, and an
646− // unconditional scrollToItem here would clobber it, snapping them back to the bottom.
647− var didInitialScroll by rememberSaveable { mutableStateOf(false) }
648− LaunchedEffect(listItems.isNotEmpty()) {
649− if (listItems.isNotEmpty() && !didInitialScroll) {
650− // When a run is active the LazyColumn has a trailing "__typing" row at
651− // listItems.size; target it (not listItems.lastIndex) so deep-linking into a
652− // running session lands truly pinned to bottom and streaming auto-scroll engages.
653− listState.scrollToItem(if (running) listItems.size else listItems.lastIndex)
654− didInitialScroll = true
658− // When a run starts, the LazyColumn gains a trailing "__typing" row, so
659− // effectiveLast jumps from listItems.lastIndex to listItems.size and the
660− // isPinnedToBottom check flips to false the instant you send — freezing
661− // auto-scroll. If the user was pinned to the last item just before the run
662− // began, bring the typing row into view so the pin (and streaming follow) is
663− // preserved. Guard on the raw last-visible index (isPinnedToBottom has already
664− // recomputed to false by now) so we don't scroll when the user had scrolled up.
665− LaunchedEffect(running) {
666− if (running && listItems.isNotEmpty()) {
667− val lastVisible = listState.layoutInfo.visibleItemsInfo.lastOrNull()?.index ?: -1
668− if (lastVisible >= listItems.lastIndex) {
669− runCatchingCancellable { listState.animateScrollToItem(listItems.size) }
674− LaunchedEffect(Unit) {
676− // Track list size + the streaming length of the last part so we auto-scroll
677− // as content arrives, plus the pinned flag. A small data class with primitive
678− // fields avoids the per-frame boxing that Triple<Int,Int,Boolean> pays
679− // (snapshotFlow re-evaluates this lambda every snapshot), and reading only the
680− // last part's length is O(1) vs. a sumOf over every part. Covering reasoning
681− // and tool output — not just TextPart — keeps a pinned view following a long
682− // reasoning/tool block while it streams (previously it stalled at 0).
683− val lastLen = streamingContentLength(messages.lastOrNull()?.parts?.lastOrNull())
684− AutoScrollSignal(currentListItems.size, lastLen, isPinnedToBottom)
685− }.collect { signal ->
686− if (signal.size > 0 && signal.pinned) {
687− // Scroll to the effective last index, including the trailing
688− // "__typing" row when a run is active so the working indicator
689− // is brought into view (not just the last message).
690− val items = currentListItems
691− val target = if (running) items.size else items.lastIndex
692− listState.scrollToItem(target)
697− CompositionLocalProvider(LocalRelativeTimeTick provides timeTick) {
698− // The bottomBar (ChatInputBar) already has imePadding, and the Scaffold's
699− // content padding accounts for the bottomBar's raised height, so the message
700− // list is already above the keyboard. Adding imePadding here would double-apply
701− // the IME inset and push messages too far up.
702− Box(modifier = Modifier.fillMaxSize().padding(padding)) {
703− if (activeConnection == null) {
705− modifier = Modifier.align(Alignment.Center).padding(24.dp),
706− horizontalAlignment = Alignment.CenterHorizontally,
709− stringResource(R.string.not_connected),
710− color = MaterialTheme.colorScheme.error,
712− Spacer(Modifier.size(12.dp))
713− Button(onClick = { vm.reconnect() }, enabled = !reconnecting) {
715− CircularProgressIndicator(
716− Modifier.size(20.dp),
717− color = MaterialTheme.colorScheme.onPrimary,
721− Text(stringResource(R.string.reconnect))
727− state = connectionState,
728− modifier = Modifier.align(Alignment.TopCenter),
729− isOnline = isOnline,
730− // On a hard failure, retry by forcing an SSE reconnect (which
731− // re-seeds from REST). refreshMessages is the right recovery path
732− // when the connection is present but the stream died; reconnect()
733− // is the path when the whole connection is gone (handled by the
734− // separate "Not connected" state below).
735− onRetry = { vm.refreshMessages() },
737− if (loading && messages.isEmpty()) {
738− val loadingLabel = stringResource(R.string.loading)
739− CircularProgressIndicator(
741− .align(Alignment.Center)
742− .semantics { contentDescription = loadingLabel },
744− } else if (loadError && messages.isEmpty()) {
745− // A failed load with nothing to show. Distinct from the empty
746− // conversation state below — the conversation isn't empty, it
747− // failed to load, and offering "Start a conversation" here is
748− // misleading. Offer a Retry instead, mirroring SessionListScreen.
750− modifier = Modifier.align(Alignment.Center).padding(24.dp),
751− horizontalAlignment = Alignment.CenterHorizontally,
754− stringResource(R.string.load_messages_failed),
755− color = MaterialTheme.colorScheme.error,
757− Spacer(Modifier.size(12.dp))
758− Button(onClick = { vm.refreshMessages() }) {
759− Text(stringResource(R.string.retry))
762− } else if (messages.isEmpty() && running) {
763− // A run just started but no parts have arrived yet — the empty
764− // list with only the trailing "working…" row looks broken, so
765− // surface a clear starting state until the first part streams in.
766− val workingText = stringResource(R.string.working)
768− modifier = Modifier.align(Alignment.Center).padding(24.dp),
769− horizontalAlignment = Alignment.CenterHorizontally,
771− CircularProgressIndicator(Modifier.size(28.dp), strokeWidth = 3.dp)
772− Spacer(Modifier.size(12.dp))
773− Text(workingText, style = MaterialTheme.typography.bodyMedium, color = MaterialTheme.colorScheme.onSurfaceVariant)
775− } else if (messages.isEmpty()) {
779− // Focus the input so the user can edit the suggestion before
780− // sending, instead of having to manually tap the field.
781− runCatching { inputFocusRequester.requestFocus() }
783− modifier = Modifier.align(Alignment.Center),
786− // Pull-to-refresh wraps the message list so the user can force an SSE
787− // reconnect + re-seed with the same gesture they use on the session and
788− // file lists, instead of hunting for the top-bar refresh icon. The empty
789− // / loading / error states above stay outside the PTR box since there's
790− // no list content to pull against.
792− isRefreshing = refreshing,
793− onRefresh = { vm.refreshMessages() },
794− modifier = Modifier.fillMaxSize(),
796− val lastMessageId = messages.lastOrNull()?.info?.id
797− // Text-to-speech for reading assistant replies aloud; shuts down when the
798− // message list leaves composition.
799− val tts = rememberTtsController()
800− val speakingMessageId by tts.speakingId
803− modifier = Modifier.fillMaxSize(),
804− contentPadding = PaddingValues(16.dp),
805− verticalArrangement = Arrangement.spacedBy(16.dp),
810− contentType = { it.contentType },
813− is MessageListItem.Separator -> DateSeparator(item.label)
814− is MessageListItem.Message -> {
815− val message = item.message
816− // Only the last (streaming) message needs isRunning — it drives the
817− // reasoning-block spinner. Passing the live flag to every bubble
818− // makes all visible messages recompose whenever a run starts or stops.
819− val modelLabel = remember(message.info, models) {
820− (message.info as? soy.iko.opencode.data.model.AssistantMessage)
821− ?.let { resolveModelLabel(it, models) }
825− isRunning = running && message.info.id == lastMessageId,
826− imageContext = imageContext,
827− modelLabel = modelLabel,
828− onOpenFile = onOpenFile,
829− onRevert = { vm.revertTo(message.info.id) },
830− onEdit = { text -> vm.editMessage(message.info.id, text) },
831− onSpeak = { text -> tts.toggle(message.info.id, text) },
832− isSpeaking = message.info.id == speakingMessageId,
838− item(key = "__typing") {
839− val workingText = stringResource(R.string.working)
841− verticalAlignment = Alignment.CenterVertically,
842− modifier = Modifier.semantics {
843− contentDescription = workingText
846− CircularProgressIndicator(Modifier.size(16.dp), strokeWidth = 2.dp)
847− Text(workingText, style = MaterialTheme.typography.labelMedium, modifier = Modifier.padding(start = 6.dp))
853− // Jump-to-latest affordance when the user has scrolled away during a stream.
855− visible = !isPinnedToBottom && listItems.isNotEmpty(),
856− modifier = Modifier.align(Alignment.BottomEnd),
858− ExtendedFloatingActionButton(
860− if (listItems.isNotEmpty()) {
861− // Scroll to the effective last index, including the
862− // trailing "__typing" row when a run is active.
863− val target = if (running) listItems.size else listItems.lastIndex
864− contentScope.launch { runCatchingCancellable { listState.animateScrollToItem(target) } }
867− icon = { Icon(Icons.Filled.KeyboardArrowDown, contentDescription = stringResource(R.string.latest)) },
868− text = { Text(stringResource(R.string.latest)) },
869− modifier = Modifier.padding(end = 16.dp, bottom = 16.dp),
878− if (showExitConfirm) {
880− onDismissRequest = { showExitConfirm = false },
881− title = { Text(stringResource(R.string.stop_and_exit_title)) },
882− text = { Text(stringResource(R.string.stop_and_exit_text)) },
884− TextButton(onClick = {
885− showExitConfirm = false
886− haptics.performHapticFeedback(androidx.compose.ui.hapticfeedback.HapticFeedbackType.TextHandleMove)
889− }) { Text(stringResource(R.string.stop_and_exit)) }
892− TextButton(onClick = { showExitConfirm = false }) { Text(stringResource(R.string.stay)) }
897− if (showStopConfirm) {
899− onDismissRequest = { showStopConfirm = false },
900− title = { Text(stringResource(R.string.stop_run_title)) },
901− text = { Text(stringResource(R.string.stop_run_text)) },
903− TextButton(onClick = {
904− showStopConfirm = false
905− haptics.performHapticFeedback(androidx.compose.ui.hapticfeedback.HapticFeedbackType.TextHandleMove)
907− }) { Text(stringResource(R.string.stop), color = MaterialTheme.colorScheme.error) }
910− TextButton(onClick = { showStopConfirm = false }) { Text(stringResource(R.string.cancel)) }
915− if (showRenameDialog) {
916− RenameSessionChatDialog(
917− initialTitle = sessionTitle ?: "",
918− onDismiss = { showRenameDialog = false },
919− onConfirm = { newName ->
920− showRenameDialog = false
921− vm.renameSession(newName)
926− if (showDeleteDialog) {
928− onDismissRequest = { showDeleteDialog = false },
929− title = { Text(stringResource(R.string.delete_session_chat_title)) },
930− text = { Text(stringResource(R.string.delete_session_chat_text, sessionTitle ?: sessionLabel)) },
932− TextButton(onClick = {
933− showDeleteDialog = false
934− haptics.performHapticFeedback(androidx.compose.ui.hapticfeedback.HapticFeedbackType.LongPress)
936− }) { Text(stringResource(R.string.delete), color = MaterialTheme.colorScheme.error) }
939− TextButton(onClick = { showDeleteDialog = false }) { Text(stringResource(R.string.cancel)) }
944− if (showShellDialog) {
946− onDismiss = { showShellDialog = false },
948− showShellDialog = false
954− if (showModelPicker) {
957− selected = selectedModel,
958− loading = modelsLoading,
959− error = modelsError,
960− onSelect = { vm.selectModel(it) },
961− onRetry = { vm.reloadModels() },
962− onDismiss = { showModelPicker = false },
966− if (showAgentPicker) {
969− selected = selectedAgent,
970− loading = agentsLoading,
971− error = agentsError,
972− onSelect = { vm.selectAgent(it?.name) },
973− onRetry = { vm.reloadAgents() },
974− onDismiss = { showAgentPicker = false },
978− if (showCommandPicker) {
980− commands = commands,
981− loading = commandsLoading,
982− error = commandsError,
983− onSelect = { vm.runCommand(it) },
984− onRetry = { vm.reloadCommands() },
985− onDismiss = { showCommandPicker = false },
989− pendingPermission?.let { permission ->
991− permission = permission,
992− onRespond = { response -> vm.respondPermission(permission, response) },
998−private fun ChatInputBar(
1000− onValueChange: (String) -> Unit,
1004− sendOnEnter: Boolean,
1005− onSend: () -> Unit,
1006− onAbort: () -> Unit,
1007− queuedFollowUp: String?,
1008− onQueueFollowUp: (String) -> Unit,
1009− onCancelQueue: () -> Unit,
1010− focusRequester: androidx.compose.ui.focus.FocusRequester,
1011− attachments: List<PendingAttachment>,
1012− onRemoveAttachment: (String) -> Unit,
1013− onPickPhoto: () -> Unit,
1014− onPickFile: () -> Unit,
1015− onCamera: () -> Unit,
1016− onVoice: () -> Unit,
1018− // Sendable when there's text OR at least one attachment (an image-only prompt is valid).
1019− val hasContent = value.isNotBlank() || attachments.isNotEmpty()
1020− Surface(tonalElevation = 3.dp, modifier = Modifier.imePadding()) {
1021− Column(modifier = Modifier.fillMaxWidth().windowInsetsPadding(WindowInsets.navigationBars)) {
1022− // A queued follow-up replaces the Stop button with a "queued" chip so the
1023− // user sees their message will be sent when the run finishes, and can cancel.
1024− if (queuedFollowUp != null) {
1026− modifier = Modifier
1028− .padding(start = 12.dp, top = 6.dp, end = 4.dp),
1029− verticalAlignment = Alignment.CenterVertically,
1031− Column(modifier = Modifier.weight(1f)) {
1033− stringResource(R.string.queued),
1034− style = MaterialTheme.typography.labelSmall,
1035− color = MaterialTheme.colorScheme.onSurfaceVariant,
1039− style = MaterialTheme.typography.bodySmall,
1040− color = MaterialTheme.colorScheme.onSurfaceVariant,
1042− overflow = TextOverflow.Ellipsis,
1045− TextButton(onClick = onCancelQueue, contentPadding = androidx.compose.foundation.layout.PaddingValues(horizontal = 8.dp)) {
1046− Text(stringResource(R.string.cancel))
1050− // Staged attachments: horizontally-scrollable thumbnails/chips, each removable.
1051− AttachmentStrip(attachments, onRemove = onRemoveAttachment)
1053− modifier = Modifier.fillMaxWidth().padding(8.dp),
1054− verticalAlignment = Alignment.Bottom,
1056− // Attach menu: photo picker, camera, or any file.
1057− var showAttachMenu by remember { mutableStateOf(false) }
1059− IconButton(onClick = { showAttachMenu = true }, enabled = enabled) {
1060− Icon(Icons.Filled.Add, contentDescription = stringResource(R.string.add_attachment))
1062− DropdownMenu(expanded = showAttachMenu, onDismissRequest = { showAttachMenu = false }) {
1064− leadingIcon = { Icon(Icons.Filled.PhotoLibrary, contentDescription = null) },
1065− text = { Text(stringResource(R.string.attach_photo)) },
1066− onClick = { showAttachMenu = false; onPickPhoto() },
1069− leadingIcon = { Icon(Icons.Filled.PhotoCamera, contentDescription = null) },
1070− text = { Text(stringResource(R.string.attach_camera)) },
1071− onClick = { showAttachMenu = false; onCamera() },
1074− leadingIcon = { Icon(Icons.Filled.Description, contentDescription = null) },
1075− text = { Text(stringResource(R.string.attach_file)) },
1076− onClick = { showAttachMenu = false; onPickFile() },
1080− // Voice dictation: appends recognized speech to the draft.
1081− IconButton(onClick = onVoice, enabled = enabled) {
1082− Icon(Icons.Filled.Mic, contentDescription = stringResource(R.string.voice_input))
1086− onValueChange = { v ->
1087− // Cap input length so a huge paste can't stall the UI. Shares are
1088− // capped separately in MainActivity; this guards typed/pasted drafts.
1089− onValueChange(v.take(NetworkConfig.maxDraftLengthChars))
1091− modifier = Modifier
1093− .focusRequester(focusRequester)
1094− .testTag("chat_input")
1095− .onPreviewKeyEvent { event ->
1096− if (event.type != KeyEventType.KeyDown || event.key != Key.Enter) return@onPreviewKeyEvent false
1097− // With "Send on Enter" on, Enter sends (Shift+Enter newlines).
1098− // With it off, Enter inserts a newline and Ctrl+Enter sends.
1099− val send = enterShouldSend(
1101− hasContent = hasContent,
1102− sendOnEnter = sendOnEnter,
1103− shift = event.isShiftPressed,
1104− ctrl = event.isCtrlPressed,
1107− if (running) onQueueFollowUp(value) else onSend()
1113− placeholder = { Text(stringResource(R.string.message_placeholder)) },
1116− // Show a "N / max" countdown once the draft crosses a high fraction of
1117− // the cap, so the user knows a paste is about to be truncated instead
1118− // of being silently cut off. Hidden for normal short prompts to avoid
1119− // clutter under a typical one-line message.
1121− val threshold = (NetworkConfig.maxDraftLengthChars * NetworkConfig.draftCountdownThresholdFraction).toInt()
1122− if (value.length >= threshold) {
1125− R.string.draft_chars_remaining,
1126− NetworkConfig.maxDraftLengthChars - value.length,
1127− NetworkConfig.maxDraftLengthChars,
1129− style = MaterialTheme.typography.labelSmall,
1130− color = if (value.length >= NetworkConfig.maxDraftLengthChars) MaterialTheme.colorScheme.error
1131− else MaterialTheme.colorScheme.onSurfaceVariant,
1135− keyboardOptions = KeyboardOptions(
1136− imeAction = ImeAction.Send,
1137− capitalization = KeyboardCapitalization.Sentences,
1139− keyboardActions = KeyboardActions(onSend = {
1140− if (enabled && hasContent) {
1141− if (running) onQueueFollowUp(value) else onSend()
1145− ComposerTrailingButton(
1147− aborting = aborting,
1148− canSend = enabled && hasContent,
1157−/** Whether a hardware Enter keypress should send: never when disabled/empty; otherwise
1158− * Enter sends (unless Shift) when send-on-Enter is on, else only Ctrl+Enter sends. */
1159−private fun enterShouldSend(
1161− hasContent: Boolean,
1162− sendOnEnter: Boolean,
1166− !enabled || !hasContent -> false
1167− sendOnEnter -> !shift
1171−/** The composer's trailing button: a Stop (with in-flight spinner) while a run is active,
1172− * otherwise Send. Extracted so [ChatInputBar] stays under the complexity threshold. */
1174−private fun ComposerTrailingButton(
1178− onSend: () -> Unit,
1179− onAbort: () -> Unit,
1182− // Show a spinner while the abort REST call is in flight so the user sees the stop
1183− // was sent, and disable to prevent a double-tap.
1186− enabled = !aborting,
1187− modifier = Modifier.padding(start = 4.dp).testTag("stop_button"),
1190− val stopLabel = stringResource(R.string.stop)
1191− CircularProgressIndicator(
1192− Modifier.size(18.dp).semantics { contentDescription = stopLabel },
1193− strokeWidth = 2.dp,
1196− Icon(Icons.Filled.Stop, contentDescription = stringResource(R.string.stop))
1203− modifier = Modifier.padding(start = 4.dp).testTag("send_button"),
1205− Icon(Icons.AutoMirrored.Filled.Send, contentDescription = stringResource(R.string.send))
1210−@OptIn(androidx.compose.foundation.layout.ExperimentalLayoutApi::class)
1212−private fun EmptyConversation(
1213− onSuggestion: (String) -> Unit,
1214− modifier: Modifier = Modifier,
1216− // A few example prompts so a first-time user has a one-tap path to their first
1217− // message instead of staring at a blank input. Tapping a chip fills the draft
1218− // (without sending) so the user can edit it before sending.
1219− val suggestions = remember {
1221− R.string.empty_chat_suggest_1,
1222− R.string.empty_chat_suggest_2,
1223− R.string.empty_chat_suggest_3,
1227− modifier = modifier.padding(32.dp),
1228− horizontalAlignment = Alignment.CenterHorizontally,
1231− Icons.Filled.AutoAwesome,
1232− contentDescription = null,
1233− modifier = Modifier.size(48.dp),
1234− tint = MaterialTheme.colorScheme.onSurfaceVariant,
1236− Spacer(Modifier.size(16.dp))
1238− stringResource(R.string.empty_chat_title),
1239− style = MaterialTheme.typography.titleMedium,
1241− Spacer(Modifier.size(8.dp))
1243− stringResource(R.string.empty_chat_hint),
1244− style = MaterialTheme.typography.bodyMedium,
1245− color = MaterialTheme.colorScheme.onSurfaceVariant,
1247− Spacer(Modifier.size(20.dp))
1248− // Suggestion chips wrap across rows on narrow screens instead of stacking full-width,
1249− // so a longer localized prompt doesn't push the chips off-screen on a phone.
1251− modifier = Modifier.fillMaxWidth(),
1252− horizontalArrangement = Arrangement.spacedBy(8.dp),
1253− verticalArrangement = Arrangement.spacedBy(8.dp),
1255− suggestions.forEach { resId ->
1256− val text = stringResource(resId)
1257− androidx.compose.material3.AssistChip(
1258− onClick = { onSuggestion(text) },
1259− label = { Text(text, style = MaterialTheme.typography.bodyMedium) },
1266−/** A banner shown while a revert checkpoint is active, offering an Undo (unrevert). */
1268−private fun RevertBanner(onUndo: () -> Unit) {
1270− color = MaterialTheme.colorScheme.tertiaryContainer,
1271− tonalElevation = 2.dp,
1272− modifier = Modifier.fillMaxWidth(),
1275− modifier = Modifier.fillMaxWidth().padding(start = 12.dp, end = 4.dp, top = 4.dp, bottom = 4.dp),
1276− verticalAlignment = Alignment.CenterVertically,
1277− horizontalArrangement = Arrangement.SpaceBetween,
1279− Row(verticalAlignment = Alignment.CenterVertically, modifier = Modifier.weight(1f)) {
1281− Icons.Filled.Restore,
1282− contentDescription = null,
1283− tint = MaterialTheme.colorScheme.onTertiaryContainer,
1284− modifier = Modifier.size(18.dp),
1287− stringResource(R.string.reverted_banner),
1288− style = MaterialTheme.typography.bodySmall,
1289− color = MaterialTheme.colorScheme.onTertiaryContainer,
1290− modifier = Modifier.padding(start = 8.dp),
1293− TextButton(onClick = onUndo) { Text(stringResource(R.string.undo)) }
1298−/** Prompt for a one-off shell command to run in the session's worktree. */
1300−private fun ShellCommandDialog(onDismiss: () -> Unit, onRun: (String) -> Unit) {
1301− var command by rememberSaveable { mutableStateOf("") }
1303− onDismissRequest = onDismiss,
1304− title = { Text(stringResource(R.string.run_shell_command)) },
1308− onValueChange = { command = it },
1309− modifier = Modifier.fillMaxWidth(),
1310− placeholder = { Text(stringResource(R.string.shell_command_hint)) },
1311− label = { Text(stringResource(R.string.shell_command_label)) },
1313− keyboardOptions = KeyboardOptions(imeAction = ImeAction.Done),
1314− keyboardActions = KeyboardActions(onDone = { if (command.isNotBlank()) onRun(command.trim()) }),
1319− onClick = { if (command.isNotBlank()) onRun(command.trim()) },
1320− enabled = command.isNotBlank(),
1321− ) { Text(stringResource(R.string.run)) }
1323− dismissButton = { TextButton(onClick = onDismiss) { Text(stringResource(R.string.cancel)) } },
1328− * Change signal for the auto-scroll-on-stream watcher. Primitive fields avoid the
1329− * boxing `Triple<Int, Int, Boolean>` would impose on every snapshot evaluation, and
1330− * equality is structural so `snapshotFlow` emits only when something relevant changes.
1332−private data class AutoScrollSignal(val size: Int, val lastTextLength: Int, val pinned: Boolean)
1334−/** Length of the streaming content carried by [part], driving auto-scroll as it grows.
1335− * Covers text, reasoning, and tool output so a pinned view keeps following non-text
1336− * streaming, not just plain assistant text. O(1). */
1337−private fun streamingContentLength(part: Part?): Int = when (part) {
1338− is TextPart -> part.text.length
1339− is ReasoningPart -> part.text.length
1340− is ToolPart -> when (val st = part.state) {
1341− is ToolCompleted -> st.output?.length ?: 0
1342− is ToolError -> st.error?.length ?: 0
1343− is ToolRunning -> st.title?.length ?: 0
1349−/** Rename dialog for the currently-open session. Mirrors the SessionListScreen rename
1350− * dialog but lives in the chat screen so the user can rename without backing out to
1351− * the list. Pre-filled with the current title; the cap matches the list dialog. */
1353−private fun RenameSessionChatDialog(
1354− initialTitle: String,
1355− onDismiss: () -> Unit,
1356− onConfirm: (String) -> Unit,
1358− var title by rememberSaveable(initialTitle) { mutableStateOf(initialTitle) }
1360− onDismissRequest = onDismiss,
1361− title = { Text(stringResource(R.string.rename_session_chat)) },
1365− onValueChange = { v -> title = v.take(NetworkConfig.maxSessionTitleChars) },
1366− modifier = Modifier.fillMaxWidth(),
1367− placeholder = { Text(stringResource(R.string.session_title_hint)) },
1368− label = { Text(stringResource(R.string.rename_session_chat)) },
1370− keyboardOptions = KeyboardOptions(imeAction = ImeAction.Done),
1371− keyboardActions = KeyboardActions(onDone = { if (title.isNotBlank()) onConfirm(title.trim()) }),
1375− TextButton(onClick = { if (title.isNotBlank()) onConfirm(title.trim()) }, enabled = title.isNotBlank()) {
1376− Text(stringResource(R.string.save))
1380− TextButton(onClick = onDismiss) { Text(stringResource(R.string.cancel)) }
1386− * A single item in the chat message list — either a [Message] bubble or a day
1387− * [Separator] inserted before the first message of a new day. The separator shifts
1388− * the LazyColumn indices vs. the raw messages list, so scroll-target math must use
1389− * the [List] of these items (not `messages.size`/`lastIndex`).
1391−private sealed interface MessageListItem {
1393− val contentType: Any
1395− data class Message(val message: MessageWithParts) : MessageListItem {
1396− override val key: Any get() = message.info.id
1397− override val contentType: Any get() = message.info::class
1400− data class Separator(val label: String, val ordinal: Int) : MessageListItem {
1401− // Key on the day-occurrence ordinal, not the label: two non-contiguous groups
1402− // can share a label (e.g. an untimestamped message — empty label — between two
1403− // same-day messages, or two untimestamped groups), and a label-only key would
1404− // then collide, which makes LazyColumn throw on duplicate keys. The ordinal is
1405− // stable for a given message ordering so slots aren't needlessly recreated.
1406− override val key: Any get() = "sep_${ordinal}_$label"
1407− // Separators share a contentType so the LazyColumn can recycle their slots.
1408− override val contentType: Any get() = "separator"
1413− * Build the interleaved list of [MessageListItem]s for the message list, inserting a
1414− * [MessageListItem.Separator] before the first message of each new calendar day. Uses
1415− * the message's `time.created` (falling back to `updated` then `completed`) to bucket.
1416− * Messages with no timestamp are grouped under an empty-label separator only if they
1417− * start the list, so a server that omits timestamps doesn't suppress the first divider.
1419− * [todayLabel]/[yesterdayLabel] are resolved by the caller (a @Composable can't call
1420− * stringResource from inside this plain function) and used for the "Today"/"Yesterday"
1421− * labels; older days fall back to a locale-stable medium-date format.
1423−private fun buildMessageListItems(
1424− messages: List<MessageWithParts>,
1425− todayLabel: String,
1426− yesterdayLabel: String,
1427−): List<MessageListItem> {
1428− if (messages.isEmpty()) return emptyList()
1429− val today = java.time.LocalDate.now(java.time.ZoneId.systemDefault())
1430− val dateFmt = mediumDateFormat()
1431− val result = ArrayList<MessageListItem>(messages.size + 4)
1432− var lastDayKey: String? = null
1434− for (message in messages) {
1435− val ts = message.info.time?.created ?: message.info.time?.updated ?: message.info.time?.completed
1436− val dayKey = ts?.let { dayKey(it) } ?: ""
1437− if (dayKey != lastDayKey) {
1438− result.add(MessageListItem.Separator(dayLabel(dayKey, ts ?: 0L, today, todayLabel, yesterdayLabel, dateFmt), sepOrdinal++))
1439− lastDayKey = dayKey
1441− result.add(MessageListItem.Message(message))
1447− * A stable bucket key for a timestamp's calendar day (epoch-days as a string).
1449− * [buildMessageListItems] re-runs on every streaming snapshot (the memo is keyed on the
1450− * whole `messages` list, which must stay so the live message's fresh parts render), so a
1451− * naive implementation would re-allocate an `Instant`/`ZonedDateTime`/`LocalDate` for
1452− * every message on every token. Message timestamps are immutable once assigned, so we
1453− * memoize the parse per epoch-millis: after the first snapshot, each unchanged message is
1454− * a cheap hash lookup. Bounded LRU (access-order) caps memory; only touched on the main
1455− * thread during composition, so the coarse lock is uncontended.
1457−private val dayKeyCache = object : LinkedHashMap<Long, String>(128, 0.75f, true) {
1458− override fun removeEldestEntry(eldest: MutableMap.MutableEntry<Long, String>): Boolean = size > 1024
1461−private fun dayKey(epochMillis: Long): String = synchronized(dayKeyCache) {
1462− dayKeyCache.getOrPut(epochMillis) {
1463− java.time.Instant.ofEpochMilli(epochMillis)
1464− .atZone(java.time.ZoneId.systemDefault()).toLocalDate().toEpochDay().toString()
1468−// The medium-date formatter is an ICU locale lookup; cache one instance instead of
1469−// allocating a fresh one per buildMessageListItems call, rebuilding only if the default
1470−// locale changes at runtime. Main-thread-only, so no synchronization is needed.
1471−private var cachedDateFmt: java.text.DateFormat? = null
1472−private var cachedDateFmtLocale: java.util.Locale? = null
1474−private fun mediumDateFormat(): java.text.DateFormat {
1475− val locale = java.util.Locale.getDefault()
1476− val existing = cachedDateFmt
1477− if (existing != null && cachedDateFmtLocale == locale) return existing
1478− return java.text.DateFormat.getDateInstance(java.text.DateFormat.MEDIUM).also {
1480− cachedDateFmtLocale = locale
1485− * Resolve a day-key back to a human-readable label. "Today" / "Yesterday" for the
1486− * recent days, otherwise a locale-stable medium-date format. Returns an empty string
1487− * when [dayKey] is empty (no timestamp) so no separator is rendered for un-timestamped
1490−private fun dayLabel(
1493− today: java.time.LocalDate,
1494− todayLabel: String,
1495− yesterdayLabel: String,
1496− dateFmt: java.text.DateFormat,
1498− if (dayKey.isEmpty()) return ""
1499− val epochDay = dayKey.toLongOrNull() ?: return ""
1500− val date = java.time.LocalDate.ofEpochDay(epochDay)
1502− date == today -> todayLabel
1503− date == today.minusDays(1) -> yesterdayLabel
1504− else -> dateFmt.format(java.util.Date(ts))
1509− * A centered day divider in the message list. Renders nothing for an empty label
1510− * (the no-timestamp case) so un-timestamped messages don't get a stray blank divider.
1513−private fun DateSeparator(label: String) {
1514− if (label.isBlank()) return
1516− modifier = Modifier.fillMaxWidth(),
1517− horizontalArrangement = Arrangement.Center,
1520− color = MaterialTheme.colorScheme.surfaceVariant,
1521− shape = MaterialTheme.shapes.small,
1525− style = MaterialTheme.typography.labelSmall,
1526− color = MaterialTheme.colorScheme.onSurfaceVariant,
1527− modifier = Modifier.padding(horizontal = 12.dp, vertical = 4.dp),