Rewind chat or files from any user message (#1154)

* Add Pi extension-launch plumbing for in-process tree primitives * Add Rewind controls for agent sessions * Preserve SDK checkpoint env var so Claude file rewind works The SDK injects CLAUDE_CODE_ENABLE_SDK_FILE_CHECKPOINTING=true when file checkpointing is enabled. The earlier parent-env stripping from c4782ec71c was too broad and removed that child flag along with real parent-session markers. Keep stripping parent markers, but preserve the checkpoint flag so the SDK rewind path can run directly. With that boundary fixed, the JSONL file-rewind fallback is no longer needed. * Update OpenCode history replay tests for rewind state * Show Rewind actions in an adaptive sheet * Fix rewind semantics across providers * Refactor Pi IDs through extension capture * Restore user message to composer after rewind * Fix Claude rewind tool result mapping * Fix Claude rewind session switch roundtrip * Fix rewind and optimistic message regressions * Tighten rewind test discipline * Fix OpenCode optimistic message reconciliation * Reconcile optimistic user messages by marker * Reorder user message actions * Unify provider user message contract * Add Claude rewind flow Playwright contract * Add Codex rewind flow Playwright contract * Add OpenCode rewind flow Playwright contract * Add Pi rewind flow Playwright contract * Remove real-provider rewind Playwright contracts * Gate real-provider rewind specs from CI Playwright
2026-05-25 17:11:34 +08:00
parent e966f70322
commit 132e572d6e
99 changed files with 7106 additions and 3425 deletions
@@ -0,0 +1,80 @@
+import { useCallback } from "react";
+import { useMutation } from "@tanstack/react-query";
+import { useToast } from "@/contexts/toast-context";
+import type { DaemonClient } from "@server/client/daemon-client";
+import type { RewindMode } from "./use-rewind-capabilities";
+import { useRewindComposerRestore } from "./composer-restore";
+import { useSessionStore } from "@/stores/session-store";
+import { shouldRestoreComposerForRewindMode } from "./rewind-mode";
+import { clearOptimisticUserMessages } from "@/types/stream";
+
+interface UseRewindAgentMutationInput {
+  serverId?: string;
+  agentId?: string;
+  messageId?: string;
+  client?: DaemonClient | null;
+}
+
+interface RewindAgentInput {
+  mode: RewindMode;
+  rewoundText: string;
+}
+
+export function useRewindAgentMutation(input: UseRewindAgentMutationInput): {
+  rewindAgent: (input: RewindAgentInput) => Promise<void>;
+  isPending: boolean;
+} {
+  const toast = useToast();
+  const composerRestore = useRewindComposerRestore();
+  const { isPending, mutateAsync } = useMutation({
+    mutationFn: async ({ mode }: RewindAgentInput) => {
+      if (!input.client || !input.agentId || !input.messageId) {
+        throw new Error("Daemon client not available");
+      }
+      await input.client.rewindAgent(input.agentId, input.messageId, mode);
+      if (mode !== "files") {
+        if (input.serverId) {
+          const session = useSessionStore.getState().sessions[input.serverId];
+          useSessionStore.getState().setAgentStreamState(input.serverId, input.agentId, {
+            tail: clearOptimisticUserMessages(session?.agentStreamTail.get(input.agentId) ?? []),
+            head: clearOptimisticUserMessages(session?.agentStreamHead.get(input.agentId) ?? []),
+          });
+        }
+        const cursor = input.serverId
+          ? useSessionStore
+              .getState()
+              .sessions[input.serverId]?.agentTimelineCursor.get(input.agentId)
+          : undefined;
+        await input.client.fetchAgentTimeline(input.agentId, {
+          direction: "tail",
+          projection: "canonical",
+          ...(cursor ? { cursor: { epoch: cursor.epoch, seq: cursor.endSeq } } : {}),
+        });
+      }
+    },
+    onSuccess: (_data, variables) => {
+      if (!shouldRestoreComposerForRewindMode(variables.mode)) {
+        return;
+      }
+      composerRestore?.restoreTextIfComposerEmpty(variables.rewoundText);
+    },
+    onError: (error) => {
+      toast.error(error instanceof Error ? error.message : "Failed to rewind agent");
+    },
+  });
+
+  const rewindAgent = useCallback(
+    async (rewindInput: RewindAgentInput) => {
+      if (isPending) {
+        return;
+      }
+      await mutateAsync(rewindInput);
+    },
+    [isPending, mutateAsync],
+  );
+
+  return {
+    rewindAgent,
+    isPending,
+  };
+}