cancel checkpoint restore limit

Author: NaccOll

webview-ui/src/components/chat/checkpoints/CheckpointMenu.tsx

@@ -24,7 +24,6 @@ export const CheckpointMenu = ({ ts, commitHash, currentHash, checkpoint }: Chec
 	const isCurrent = currentHash === commitHash
 	const isFirst = checkpoint.isFirst
 	const isDiffAvailable = !isFirst
-	const isRestoreAvailable = !isFirst || !isCurrent
 
 	const previousCommitHash = checkpoint?.from
 
@@ -54,71 +53,67 @@ export const CheckpointMenu = ({ ts, commitHash, currentHash, checkpoint }: Chec
 					</Button>
 				</StandardTooltip>
 			)}
-			{isRestoreAvailable && (
-				<Popover
-					open={isOpen}
-					onOpenChange={(open) => {
-						setIsOpen(open)
-						setIsConfirming(false)
-					}}>
-					<StandardTooltip content={t("chat:checkpoint.menu.restore")}>
-						<PopoverTrigger asChild>
-							<Button variant="ghost" size="icon">
-								<span className="codicon codicon-history" />
-							</Button>
-						</PopoverTrigger>
-					</StandardTooltip>
-					<PopoverContent align="end" container={portalContainer}>
-						<div className="flex flex-col gap-2">
-							{!isCurrent && (
-								<div className="flex flex-col gap-1 group hover:text-foreground">
-									<Button variant="secondary" onClick={onPreview}>
-										{t("chat:checkpoint.menu.restoreFiles")}
-									</Button>
-									<div className="text-muted transition-colors group-hover:text-foreground">
-										{t("chat:checkpoint.menu.restoreFilesDescription")}
-									</div>
+			<Popover
+				open={isOpen}
+				onOpenChange={(open) => {
+					setIsOpen(open)
+					setIsConfirming(false)
+				}}>
+				<StandardTooltip content={t("chat:checkpoint.menu.restore")}>
+					<PopoverTrigger asChild>
+						<Button variant="ghost" size="icon">
+							<span className="codicon codicon-history" />
+						</Button>
+					</PopoverTrigger>
+				</StandardTooltip>
+				<PopoverContent align="end" container={portalContainer}>
+					<div className="flex flex-col gap-2">
+						{!isCurrent && (
+							<div className="flex flex-col gap-1 group hover:text-foreground">
+								<Button variant="secondary" onClick={onPreview}>
+									{t("chat:checkpoint.menu.restoreFiles")}
+								</Button>
+								<div className="text-muted transition-colors group-hover:text-foreground">
+									{t("chat:checkpoint.menu.restoreFilesDescription")}
 								</div>
-							)}
-							{!isFirst && (
-								<div className="flex flex-col gap-1 group hover:text-foreground">
-									<div className="flex flex-col gap-1 group hover:text-foreground">
-										{!isConfirming ? (
-											<Button variant="secondary" onClick={() => setIsConfirming(true)}>
-												{t("chat:checkpoint.menu.restoreFilesAndTask")}
-											</Button>
-										) : (
-											<>
-												<Button variant="default" onClick={onRestore} className="grow">
-													<div className="flex flex-row gap-1">
-														<CheckIcon />
-														<div>{t("chat:checkpoint.menu.confirm")}</div>
-													</div>
-												</Button>
-												<Button variant="secondary" onClick={() => setIsConfirming(false)}>
-													<div className="flex flex-row gap-1">
-														<Cross2Icon />
-														<div>{t("chat:checkpoint.menu.cancel")}</div>
-													</div>
-												</Button>
-											</>
-										)}
-										{isConfirming ? (
-											<div className="text-destructive font-bold">
-												{t("chat:checkpoint.menu.cannotUndo")}
+							</div>
+						)}
+						<div className="flex flex-col gap-1 group hover:text-foreground">
+							<div className="flex flex-col gap-1 group hover:text-foreground">
+								{!isConfirming ? (
+									<Button variant="secondary" onClick={() => setIsConfirming(true)}>
+										{t("chat:checkpoint.menu.restoreFilesAndTask")}
+									</Button>
+								) : (
+									<>
+										<Button variant="default" onClick={onRestore} className="grow">
+											<div className="flex flex-row gap-1">
+												<CheckIcon />
+												<div>{t("chat:checkpoint.menu.confirm")}</div>
 											</div>
-										) : (
-											<div className="text-muted transition-colors group-hover:text-foreground">
-												{t("chat:checkpoint.menu.restoreFilesAndTaskDescription")}
+										</Button>
+										<Button variant="secondary" onClick={() => setIsConfirming(false)}>
+											<div className="flex flex-row gap-1">
+												<Cross2Icon />
+												<div>{t("chat:checkpoint.menu.cancel")}</div>
 											</div>
-										)}
+										</Button>
+									</>
+								)}
+								{isConfirming ? (
+									<div className="text-destructive font-bold">
+										{t("chat:checkpoint.menu.cannotUndo")}
 									</div>
-								</div>
-							)}
+								) : (
+									<div className="text-muted transition-colors group-hover:text-foreground">
+										{t("chat:checkpoint.menu.restoreFilesAndTaskDescription")}
+									</div>
+								)}
+							</div>
 						</div>
-					</PopoverContent>
-				</Popover>
-			)}
+					</div>
+				</PopoverContent>
+			</Popover>
 		</div>
 	)
 }