From eee087fc81cbdfabaf8660b926f5bb8e04e2713a Mon Sep 17 00:00:00 2001
From: Will Chen <willchen90@gmail.com>
Date: Tue, 1 Jul 2025 11:12:40 -0700
Subject: [PATCH] Rename Undo button in version pane to Restore (#536)

---
 e2e-tests/switch_versions.spec.ts   |  6 ++--
 e2e-tests/version_integrity.spec.ts |  6 ++--
 src/components/chat/VersionPane.tsx | 48 +++++++++++++++++------------
 3 files changed, 37 insertions(+), 23 deletions(-)

diff --git a/e2e-tests/switch_versions.spec.ts b/e2e-tests/switch_versions.spec.ts
index f954d25..39b99d0 100644
--- a/e2e-tests/switch_versions.spec.ts
+++ b/e2e-tests/switch_versions.spec.ts
@@ -11,10 +11,12 @@ const runSwitchVersionTest = async (po: PageObject, nativeGit: boolean) => {
     await po.page.getByRole("button", { name: "Version" }).textContent(),
   ).toBe("Version 2");
   await po.page.getByRole("button", { name: "Version" }).click();
-  await po.page.getByText("Init Dyad app Undo").click();
+  await po.page.getByText("Init Dyad app Restore").click();
   await po.snapshotPreview({ name: `v1` });
 
-  await po.page.getByRole("button", { name: "Undo to latest version" }).click();
+  await po.page
+    .getByRole("button", { name: "Restore to this version" })
+    .click();
   // Should be same as the previous snapshot, but just to be sure.
   await po.snapshotPreview({ name: `v1` });
 
diff --git a/e2e-tests/version_integrity.spec.ts b/e2e-tests/version_integrity.spec.ts
index 97419ec..06e460d 100644
--- a/e2e-tests/version_integrity.spec.ts
+++ b/e2e-tests/version_integrity.spec.ts
@@ -34,10 +34,12 @@ const runVersionIntegrityTest = async (po: PageObject, nativeGit: boolean) => {
 
   // Open version pane
   await po.page.getByRole("button", { name: "Version 3" }).click();
-  await po.page.getByText("Init Dyad app Undo").click();
+  await po.page.getByText("Init Dyad app Restore").click();
   await po.snapshotAppFiles({ name: "v1" });
 
-  await po.page.getByRole("button", { name: "Undo to latest version" }).click();
+  await po.page
+    .getByRole("button", { name: "Restore to this version" })
+    .click();
   // Should be same as the previous snapshot, but just to be sure.
   await po.snapshotAppFiles({ name: "v1" });
 };
diff --git a/src/components/chat/VersionPane.tsx b/src/components/chat/VersionPane.tsx
index 67a591e..b063549 100644
--- a/src/components/chat/VersionPane.tsx
+++ b/src/components/chat/VersionPane.tsx
@@ -8,6 +8,11 @@ import { cn } from "@/lib/utils";
 import { useEffect, useRef, useState } from "react";
 import { useCheckoutVersion } from "@/hooks/useCheckoutVersion";
 import { useLoadApp } from "@/hooks/useLoadApp";
+import {
+  Tooltip,
+  TooltipContent,
+  TooltipTrigger,
+} from "@/components/ui/tooltip";
 
 interface VersionPaneProps {
   isVisible: boolean;
@@ -153,25 +158,30 @@ export function VersionPane({ isVisible, onClose }: VersionPaneProps) {
                     </p>
                   )}
 
-                  <button
-                    onClick={async (e) => {
-                      e.stopPropagation();
-                      setSelectedVersionId(null);
-                      await revertVersion({
-                        versionId: version.oid,
-                      });
-                      // Close the pane after revert to force a refresh on next open
-                      onClose();
-                    }}
-                    className={cn(
-                      "invisible mt-1 flex items-center gap-1 px-2 py-0.5 text-sm font-medium bg-(--primary) text-(--primary-foreground) hover:bg-background-lightest rounded-md transition-colors",
-                      selectedVersionId === version.oid && "visible",
-                    )}
-                    aria-label="Undo to latest version"
-                  >
-                    <RotateCcw size={12} />
-                    <span>Undo</span>
-                  </button>
+                  <Tooltip>
+                    <TooltipTrigger asChild>
+                      <button
+                        onClick={async (e) => {
+                          e.stopPropagation();
+                          setSelectedVersionId(null);
+                          await revertVersion({
+                            versionId: version.oid,
+                          });
+                          // Close the pane after revert to force a refresh on next open
+                          onClose();
+                        }}
+                        className={cn(
+                          "invisible mt-1 flex items-center gap-1 px-2 py-0.5 text-sm font-medium bg-(--primary) text-(--primary-foreground) hover:bg-background-lightest rounded-md transition-colors",
+                          selectedVersionId === version.oid && "visible",
+                        )}
+                        aria-label="Restore to this version"
+                      >
+                        <RotateCcw size={12} />
+                        <span>Restore</span>
+                      </button>
+                    </TooltipTrigger>
+                    <TooltipContent>Restore to this version</TooltipContent>
+                  </Tooltip>
                 </div>
               </div>
             ))}