Persist Codex activity log toggle across Virtuoso remounts

Author: williamstein

src/packages/frontend/chat/codex-activity.tsx

@@ -107,23 +107,34 @@ export interface CodexActivityProps {
   generating?: boolean;
   fontSize?: number;
   durationLabel?: string;
+  persistKey?: string;
   canResolveApproval?: boolean;
   onResolveApproval?: (args: {
     approvalId: string;
     optionId?: string;
   }) => Promise<void> | void;
 }
 
+// Persist log visibility per chat message so Virtuoso remounts don’t reset it.
+const expandedState = new Map<string, boolean>();
+
 export function CodexActivity({
   events,
   generating,
   fontSize,
   durationLabel,
+  persistKey,
   canResolveApproval,
   onResolveApproval,
 }: CodexActivityProps): React.ReactElement | null {
   const entries = useMemo(() => normalizeEvents(events ?? []), [events]);
-  const [expanded, setExpanded] = useState<boolean>(!!generating);
+  const [expanded, setExpanded] = useState<boolean>(() => {
+    if (persistKey) {
+      const persisted = expandedState.get(persistKey);
+      if (persisted != null) return persisted;
+    }
+    return !!generating;
+  });
   const [hovered, setHovered] = useState(false);
 
   useEffect(() => {
@@ -132,6 +143,20 @@ export function CodexActivity({
     }
   }, [generating]);
 
+  useEffect(() => {
+    if (!persistKey) return;
+    expandedState.set(persistKey, expanded);
+  }, [persistKey, expanded]);
+
+  useEffect(() => {
+    if (!persistKey) return;
+    const persisted = expandedState.get(persistKey);
+    const next = persisted ?? (generating ? true : expanded);
+    if (next !== expanded) {
+      setExpanded(next);
+    }
+  }, [persistKey, generating]); // eslint-disable-line react-hooks/exhaustive-deps
+
   if (!entries.length) return null;
 
   const baseFontSize = fontSize ?? 13;

src/packages/frontend/chat/message.tsx

@@ -753,6 +753,9 @@ export default function Message({
             events={codexEvents}
             generating={generating === true}
             fontSize={font_size}
+            persistKey={`${(project_id ?? "no-project").slice(0, 8)}:${
+              path ?? ""
+            }:${date}`}
             durationLabel={
               generating === true
                 ? elapsedLabel