|
1 | 1 | import { useAtomValue } from "@effect/atom-react"; |
2 | | -import { useCallback, useEffect, useMemo } from "react"; |
| 2 | +import { useCallback, useEffect, useMemo, useRef, useState } from "react"; |
3 | 3 |
|
4 | 4 | import { |
5 | 5 | CommandId, |
6 | 6 | MessageId, |
7 | 7 | type EnvironmentId, |
8 | 8 | type ModelSelection, |
| 9 | + type OrchestrationThreadActivity, |
9 | 10 | type ProviderInteractionMode, |
10 | 11 | type RuntimeMode, |
11 | 12 | type ThreadId, |
@@ -36,11 +37,15 @@ import { |
36 | 37 | useComposerDraft, |
37 | 38 | } from "./use-composer-drafts"; |
38 | 39 | import { setPendingConnectionError } from "../state/use-remote-environment-registry"; |
| 40 | +import { orchestrationEnvironment } from "../state/orchestration"; |
39 | 41 | import { useSelectedThreadDetail } from "../state/use-thread-detail"; |
40 | 42 | import { useThreadSelection } from "../state/use-thread-selection"; |
| 43 | +import { useAtomCommand } from "./use-atom-command"; |
41 | 44 | import { enqueueThreadOutboxMessage } from "./thread-outbox"; |
42 | 45 | import { useThreadOutboxMessages } from "./use-thread-outbox"; |
43 | 46 |
|
| 47 | +const EMPTY_ACTIVITIES: ReadonlyArray<OrchestrationThreadActivity> = []; |
| 48 | + |
44 | 49 | export function appendReviewCommentToDraft(input: { |
45 | 50 | readonly environmentId: EnvironmentId; |
46 | 51 | readonly threadId: ThreadId; |
@@ -89,9 +94,113 @@ export function useThreadComposerState() { |
89 | 94 | () => (selectedThreadKey ? (queuedMessagesByThreadKey[selectedThreadKey] ?? []) : []), |
90 | 95 | [queuedMessagesByThreadKey, selectedThreadKey], |
91 | 96 | ); |
| 97 | + |
| 98 | + // ── Older-history lazy-load (mirrors web ChatView) ────────────────────────── |
| 99 | + // The detail snapshot windows activities to the most recent page (the server |
| 100 | + // sets `hasMoreActivities`); older pages are fetched on demand and prepended. |
| 101 | + const [olderActivities, setOlderActivities] = useState< |
| 102 | + ReadonlyArray<OrchestrationThreadActivity> |
| 103 | + >([]); |
| 104 | + const [olderLoaded, setOlderLoaded] = useState(false); |
| 105 | + const [olderHasMore, setOlderHasMore] = useState(false); |
| 106 | + const [loadingOlderActivities, setLoadingOlderActivities] = useState(false); |
| 107 | + const loadThreadActivities = useAtomCommand(orchestrationEnvironment.loadThreadActivities, { |
| 108 | + reportFailure: false, |
| 109 | + }); |
| 110 | + |
| 111 | + const activityRequestKey = selectedThreadShell |
| 112 | + ? `${selectedThreadShell.environmentId}\u0000${selectedThreadShell.id}` |
| 113 | + : null; |
| 114 | + const activityRequestKeyRef = useRef(activityRequestKey); |
| 115 | + activityRequestKeyRef.current = activityRequestKey; |
| 116 | + useEffect(() => { |
| 117 | + setOlderActivities([]); |
| 118 | + setOlderLoaded(false); |
| 119 | + setOlderHasMore(false); |
| 120 | + setLoadingOlderActivities(false); |
| 121 | + }, [activityRequestKey]); |
| 122 | + |
| 123 | + const liveActivities = selectedThreadDetail?.activities ?? EMPTY_ACTIVITIES; |
| 124 | + const mergedActivities = useMemo( |
| 125 | + () => (olderActivities.length > 0 ? [...olderActivities, ...liveActivities] : liveActivities), |
| 126 | + [olderActivities, liveActivities], |
| 127 | + ); |
| 128 | + // Before any page is loaded, the server tells us whether older history exists. |
| 129 | + const hasMoreOlderActivities = olderLoaded |
| 130 | + ? olderHasMore |
| 131 | + : (selectedThreadDetail?.hasMoreActivities ?? false); |
| 132 | + |
| 133 | + // Synchronous in-flight guard keyed by thread: the list fires onLoadOlder |
| 134 | + // repeatedly while pinned at the top, but loading *state* only updates next |
| 135 | + // render, so without this a fast scroll dispatches duplicate same-cursor calls. |
| 136 | + const inFlightOlderKeyRef = useRef<string | null>(null); |
| 137 | + const onLoadOlderActivities = useCallback(() => { |
| 138 | + if (!selectedThreadShell || !hasMoreOlderActivities) { |
| 139 | + return; |
| 140 | + } |
| 141 | + const oldestActivity = mergedActivities[0]; |
| 142 | + if (!oldestActivity || !activityRequestKey) { |
| 143 | + return; |
| 144 | + } |
| 145 | + if (inFlightOlderKeyRef.current === activityRequestKey) { |
| 146 | + return; |
| 147 | + } |
| 148 | + const cursorInput = |
| 149 | + oldestActivity.sequence !== undefined |
| 150 | + ? { beforeSequence: oldestActivity.sequence } |
| 151 | + : { beforeCreatedAt: oldestActivity.createdAt, beforeActivityId: oldestActivity.id }; |
| 152 | + const requestKey = activityRequestKey; |
| 153 | + inFlightOlderKeyRef.current = requestKey; |
| 154 | + setLoadingOlderActivities(true); |
| 155 | + void loadThreadActivities({ |
| 156 | + environmentId: selectedThreadShell.environmentId, |
| 157 | + input: { threadId: selectedThreadShell.id, ...cursorInput }, |
| 158 | + }) |
| 159 | + .then((result) => { |
| 160 | + if (activityRequestKeyRef.current !== requestKey) { |
| 161 | + return; |
| 162 | + } |
| 163 | + if (result._tag !== "Success") { |
| 164 | + return; |
| 165 | + } |
| 166 | + const page = result.value; |
| 167 | + setOlderActivities((prev) => { |
| 168 | + // Dedup against both already-loaded older pages and the live window, |
| 169 | + // since mobile merges everything into one array (duplicate ids would |
| 170 | + // produce duplicate React keys in the feed). |
| 171 | + const seen = new Set(prev.map((activity) => activity.id)); |
| 172 | + for (const activity of liveActivities) { |
| 173 | + seen.add(activity.id); |
| 174 | + } |
| 175 | + const fresh = page.activities.filter((activity) => !seen.has(activity.id)); |
| 176 | + return [...fresh, ...prev]; |
| 177 | + }); |
| 178 | + setOlderLoaded(true); |
| 179 | + setOlderHasMore(page.hasMore); |
| 180 | + }) |
| 181 | + .finally(() => { |
| 182 | + if (inFlightOlderKeyRef.current === requestKey) { |
| 183 | + inFlightOlderKeyRef.current = null; |
| 184 | + } |
| 185 | + if (activityRequestKeyRef.current === requestKey) { |
| 186 | + setLoadingOlderActivities(false); |
| 187 | + } |
| 188 | + }); |
| 189 | + }, [ |
| 190 | + selectedThreadShell, |
| 191 | + hasMoreOlderActivities, |
| 192 | + mergedActivities, |
| 193 | + activityRequestKey, |
| 194 | + liveActivities, |
| 195 | + loadThreadActivities, |
| 196 | + ]); |
| 197 | + |
92 | 198 | const selectedThreadFeed = useMemo( |
93 | | - () => (selectedThreadDetail ? buildThreadFeed(selectedThreadDetail) : []), |
94 | | - [selectedThreadDetail], |
| 199 | + () => |
| 200 | + selectedThreadDetail |
| 201 | + ? buildThreadFeed({ ...selectedThreadDetail, activities: mergedActivities }) |
| 202 | + : [], |
| 203 | + [selectedThreadDetail, mergedActivities], |
95 | 204 | ); |
96 | 205 |
|
97 | 206 | const selectedDraft = selectedThreadKey ? composerDrafts[selectedThreadKey] : null; |
@@ -299,6 +408,9 @@ export function useThreadComposerState() { |
299 | 408 | runtimeMode, |
300 | 409 | interactionMode, |
301 | 410 | activeThreadBusy, |
| 411 | + hasMoreOlderActivities, |
| 412 | + loadingOlderActivities, |
| 413 | + onLoadOlderActivities, |
302 | 414 | onChangeDraftMessage, |
303 | 415 | onPickDraftImages, |
304 | 416 | onPasteIntoDraft, |
|
0 commit comments