@copilotkit/react-core 1.56.3 → 1.56.4-canary.1777529757

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@copilotkit/react-core",
-  "version": "1.56.3",
+  "version": "1.56.4-canary.1777529757",
   "private": false,
   "keywords": [
     "ai",
@@ -52,8 +52,8 @@
     "access": "public"
   },
   "dependencies": {
-    "@ag-ui/client": "0.0.52",
-    "@ag-ui/core": "0.0.52",
+    "@ag-ui/client": "0.0.53",
+    "@ag-ui/core": "0.0.53",
     "@jetbrains/websandbox": "^1.1.3",
     "@lit-labs/react": "^2.0.2",
     "@radix-ui/react-dropdown-menu": "^2.1.15",
@@ -73,11 +73,11 @@
     "untruncate-json": "^0.0.1",
     "use-stick-to-bottom": "^1.1.1",
     "zod-to-json-schema": "^3.24.5",
-    "@copilotkit/core": "1.56.3",
-    "@copilotkit/a2ui-renderer": "1.56.3",
-    "@copilotkit/runtime-client-gql": "1.56.3",
-    "@copilotkit/web-inspector": "1.56.3",
-    "@copilotkit/shared": "1.56.3"
+    "@copilotkit/a2ui-renderer": "1.56.4-canary.1777529757",
+    "@copilotkit/shared": "1.56.4-canary.1777529757",
+    "@copilotkit/web-inspector": "1.56.4-canary.1777529757",
+    "@copilotkit/runtime-client-gql": "1.56.4-canary.1777529757",
+    "@copilotkit/core": "1.56.4-canary.1777529757"
   },
   "devDependencies": {
     "@tailwindcss/cli": "^4.1.11",

package/src/v2/components/chat/CopilotChatAttachmentQueue.tsx

@@ -1,13 +1,13 @@
-import React, { useCallback, useEffect, useId, useRef, useState } from "react";
-import { createPortal, flushSync } from "react-dom";
+import React, { useEffect, useState } from "react";
 import type { Attachment } from "@copilotkit/shared";
 import {
   formatFileSize,
   getSourceUrl,
   getDocumentIcon,
 } from "@copilotkit/shared";
-import { Play, X } from "lucide-react";
+import { Play } from "lucide-react";
 import { cn } from "../../lib/utils";
+import { Lightbox, useLightbox } from "./Lightbox";
 
 interface CopilotChatAttachmentQueueProps {
   attachments: Attachment[];
@@ -21,7 +21,10 @@ export const CopilotChatAttachmentQueue: React.FC<
   if (attachments.length === 0) return null;
 
   return (
-    <div className={cn("cpk:flex cpk:flex-wrap cpk:gap-2 cpk:p-2", className)}>
+    <div
+      data-testid="copilot-attachment-queue"
+      className={cn("cpk:flex cpk:flex-wrap cpk:gap-2 cpk:p-2", className)}
+    >
       {attachments.map((attachment) => {
         const isMedia =
           attachment.type === "image" || attachment.type === "video";
@@ -85,116 +88,6 @@ function AttachmentPreview({ attachment }: { attachment: Attachment }) {
   }
 }
 
-// ---------------------------------------------------------------------------
-// Lightbox – fullscreen overlay for images and videos (portal to body)
-// Uses the View Transition API when available for a smooth thumbnail-to-
-// fullscreen morph; falls back to a simple opacity fade.
-// ---------------------------------------------------------------------------
-
-interface LightboxProps {
-  onClose: () => void;
-  children: React.ReactNode;
-}
-
-function Lightbox({ onClose, children }: LightboxProps) {
-  useEffect(() => {
-    const handleKey = (e: KeyboardEvent) => {
-      if (e.key === "Escape") onClose();
-    };
-    document.addEventListener("keydown", handleKey);
-    return () => document.removeEventListener("keydown", handleKey);
-  }, [onClose]);
-
-  if (typeof document === "undefined") return null;
-
-  return createPortal(
-    <div
-      className="cpk:fixed cpk:inset-0 cpk:z-[9999] cpk:flex cpk:items-center cpk:justify-center cpk:bg-black/80 cpk:animate-fade-in"
-      onClick={onClose}
-    >
-      <button
-        onClick={onClose}
-        className="cpk:absolute cpk:top-4 cpk:right-4 cpk:text-white cpk:bg-white/10 cpk:hover:bg-white/20 cpk:rounded-full cpk:w-10 cpk:h-10 cpk:flex cpk:items-center cpk:justify-center cpk:cursor-pointer cpk:border-none cpk:z-10"
-        aria-label="Close preview"
-      >
-        <X className="cpk:w-5 cpk:h-5" />
-      </button>
-
-      <div onClick={(e) => e.stopPropagation()}>{children}</div>
-    </div>,
-    document.body,
-  );
-}
-
-type DocWithVT = Document & {
-  startViewTransition?: (cb: () => void) => { finished: Promise<void> };
-};
-
-/**
- * Hook that manages lightbox open/close and uses the View Transition API to
- * morph the thumbnail into fullscreen content.
- *
- * The trick: `view-transition-name` must live on exactly ONE element at a time.
- * - Old state (thumbnail visible): name is on the thumbnail.
- * - New state (lightbox visible): name moves to the lightbox content.
- * `flushSync` ensures React commits the DOM change synchronously inside the
- * `startViewTransition` callback so the API can snapshot old → new correctly.
- */
-function useLightbox() {
-  const thumbnailRef = useRef<HTMLElement>(null);
-  const [open, setOpen] = useState(false);
-  const vtName = useId();
-
-  const openLightbox = useCallback(() => {
-    const thumb = thumbnailRef.current;
-    const doc = document as DocWithVT;
-
-    if (doc.startViewTransition && thumb) {
-      // Old snapshot: name on the thumbnail
-      thumb.style.viewTransitionName = vtName;
-
-      doc.startViewTransition(() => {
-        // New snapshot: remove from thumb (lightbox content will have it)
-        thumb.style.viewTransitionName = "";
-        flushSync(() => setOpen(true));
-      });
-    } else {
-      setOpen(true);
-    }
-  }, []);
-
-  const closeLightbox = useCallback(() => {
-    const thumb = thumbnailRef.current;
-    const doc = document as DocWithVT;
-
-    if (doc.startViewTransition && thumb) {
-      const transition = doc.startViewTransition(() => {
-        // New snapshot: name back on thumbnail
-        flushSync(() => setOpen(false));
-        thumb.style.viewTransitionName = vtName;
-      });
-      // Clean up the name after animation finishes (or fails)
-      transition.finished
-        .then(() => {
-          thumb.style.viewTransitionName = "";
-        })
-        .catch(() => {
-          thumb.style.viewTransitionName = "";
-        });
-    } else {
-      setOpen(false);
-    }
-  }, []);
-
-  return {
-    thumbnailRef,
-    vtName,
-    open,
-    openLightbox,
-    closeLightbox,
-  };
-}
-
 // ---------------------------------------------------------------------------
 // Image
 // ---------------------------------------------------------------------------

package/src/v2/components/chat/CopilotChatAttachmentRenderer.tsx

@@ -2,6 +2,7 @@ import React, { memo, useState } from "react";
 import type { InputContentSource } from "@copilotkit/shared";
 import { getSourceUrl, getDocumentIcon } from "@copilotkit/shared";
 import { cn } from "../../lib/utils";
+import { Lightbox, useLightbox } from "./Lightbox";
 
 interface CopilotChatAttachmentRendererProps {
   type: "image" | "audio" | "video" | "document";
@@ -18,6 +19,8 @@ const ImageAttachment = memo(function ImageAttachment({
   className?: string;
 }) {
   const [error, setError] = useState(false);
+  const { thumbnailRef, vtName, open, openLightbox, closeLightbox } =
+    useLightbox();
 
   if (error) {
     return (
@@ -33,12 +36,29 @@ const ImageAttachment = memo(function ImageAttachment({
   }
 
   return (
-    <img
-      src={src}
-      alt="Image attachment"
-      className={cn("cpk:max-w-full cpk:h-auto cpk:rounded-lg", className)}
-      onError={() => setError(true)}
-    />
+    <>
+      <img
+        ref={thumbnailRef as React.Ref<HTMLImageElement>}
+        src={src}
+        alt="Image attachment"
+        className={cn(
+          "cpk:max-w-[80px] cpk:max-h-[80px] cpk:w-auto cpk:h-auto cpk:rounded-xl cpk:object-cover cpk:cursor-pointer cpk:bg-muted",
+          className,
+        )}
+        onClick={openLightbox}
+        onError={() => setError(true)}
+      />
+      {open && (
+        <Lightbox onClose={closeLightbox}>
+          <img
+            style={{ viewTransitionName: vtName }}
+            src={src}
+            alt="Image attachment"
+            className="cpk:max-w-[90vw] cpk:max-h-[90vh] cpk:object-contain cpk:rounded-lg"
+          />
+        </Lightbox>
+      )}
+    </>
   );
 });
 

package/src/v2/components/chat/CopilotChatUserMessage.tsx

@@ -217,9 +217,8 @@ export function CopilotChatUserMessage({
       data-message-id={message.id}
       {...props}
     >
-      {BoundMessageRenderer}
       {mediaParts.length > 0 && (
-        <div className="cpk:flex cpk:flex-col cpk:items-end cpk:gap-2 cpk:mt-2">
+        <div className="cpk:flex cpk:flex-row cpk:flex-wrap cpk:justify-end cpk:gap-2 cpk:mb-2">
           {mediaParts.map((part, index) => (
             <CopilotChatAttachmentRenderer
               key={index}
@@ -230,6 +229,7 @@ export function CopilotChatUserMessage({
           ))}
         </div>
       )}
+      {BoundMessageRenderer}
       {BoundToolbar}
     </div>
   );

package/src/v2/components/chat/CopilotChatView.tsx

@@ -6,17 +6,19 @@ import React, {
   useLayoutEffect,
 } from "react";
 import { ScrollElementContext } from "./scroll-element-context";
-import { WithSlots, SlotValue, renderSlot } from "../../lib/slots";
+import type { WithSlots, SlotValue } from "../../lib/slots";
+import { renderSlot } from "../../lib/slots";
 import CopilotChatMessageView from "./CopilotChatMessageView";
-import CopilotChatInput, {
+import type {
   CopilotChatInputProps,
   CopilotChatInputMode,
 } from "./CopilotChatInput";
+import CopilotChatInput from "./CopilotChatInput";
 import CopilotChatSuggestionView, {
   CopilotChatSuggestionViewProps,
 } from "./CopilotChatSuggestionView";
-import { Suggestion } from "@copilotkit/core";
-import { Message } from "@ag-ui/core";
+import type { Suggestion } from "@copilotkit/core";
+import type { Message } from "@ag-ui/core";
 import type { Attachment } from "@copilotkit/shared";
 import { CopilotChatAttachmentQueue } from "./CopilotChatAttachmentQueue";
 import { twMerge } from "tailwind-merge";
@@ -37,29 +39,8 @@ import { normalizeAutoScroll } from "./normalize-auto-scroll";
 import type { AutoScrollMode } from "./normalize-auto-scroll";
 import { usePinToSend } from "../../hooks/use-pin-to-send";
 
-// Height of the feather gradient overlay (h-24 = 6rem = 96px)
-const FEATHER_HEIGHT = 96;
-
-// Pin-to-send uses a softer, shorter feather than pin-to-bottom so readable
-// content isn't obscured (h-12 = 3rem = 48px).
-const PIN_TO_SEND_FEATHER_HEIGHT = 48;
-
-const PinToSendSoftFeather: React.FC<React.HTMLAttributes<HTMLDivElement>> = ({
-  className,
-  style,
-  ...props
-}) => (
-  <div
-    className={cn(
-      "cpk:absolute cpk:bottom-0 cpk:left-0 cpk:right-4 cpk:h-12 cpk:pointer-events-none cpk:z-10 cpk:bg-gradient-to-t",
-      "cpk:from-white cpk:to-transparent",
-      "cpk:dark:from-[rgb(33,33,33)]",
-      className,
-    )}
-    style={style}
-    {...props}
-  />
-);
+// Vertical gap between the scroll-to-bottom button and the input container.
+const SCROLL_BUTTON_OFFSET = 16;
 
 // Forward declaration for WelcomeScreen component type
 export type WelcomeScreenProps = WithSlots<
@@ -186,7 +167,17 @@ export function CopilotChatView({
   className,
   ...props
 }: CopilotChatViewProps) {
-  const inputContainerRef = useRef<HTMLDivElement>(null);
+  // Element-as-state via callback ref. The overlay wrapper only renders on the
+  // chat-view branch (the welcome-screen branch omits it), so a plain
+  // useRef + `[]` useEffect would observe `null` on mount whenever the chat
+  // starts on the welcome screen and never re-attach after the user sends
+  // their first message — leaving inputContainerHeight at 0 and the scroll
+  // content's reserved bottom padding at 32px instead of ~input height. The
+  // result is the last messages scrolling underneath the absolute-positioned
+  // input pill. Subscribing to element state lets the observer attach (and
+  // detach) reactively as the overlay mounts/unmounts.
+  const [inputContainerEl, setInputContainerEl] =
+    useState<HTMLDivElement | null>(null);
   const [inputContainerHeight, setInputContainerHeight] = useState(0);
   const [isResizing, setIsResizing] = useState(false);
   const resizeTimeoutRef = useRef<NodeJS.Timeout | null>(null);
@@ -197,8 +188,14 @@ export function CopilotChatView({
 
   // Track input container height changes
   useEffect(() => {
-    const element = inputContainerRef.current;
-    if (!element) return;
+    const element = inputContainerEl;
+    if (!element) {
+      // Reset measured height so the scroll content's paddingBottom doesn't
+      // hold a stale value if the overlay unmounts (e.g. messages cleared
+      // and the welcome screen returns).
+      setInputContainerHeight(0);
+      return;
+    }
 
     const resizeObserver = new ResizeObserver((entries) => {
       for (const entry of entries) {
@@ -237,7 +234,7 @@ export function CopilotChatView({
         clearTimeout(resizeTimeoutRef.current);
       }
     };
-  }, []);
+  }, [inputContainerEl]);
 
   const BoundMessageView = renderSlot(messageView, CopilotChatMessageView, {
     messages,
@@ -258,24 +255,23 @@ export function CopilotChatView({
     onAddFile,
     positioning: "static",
     keyboardHeight: isKeyboardOpen ? keyboardHeight : 0,
-    containerRef: inputContainerRef,
     showDisclaimer: true,
-    // This input is the last flex child of the chat column, so it sits at
-    // the bottom where the license banner would overlap. The welcome-screen
-    // input (below) intentionally omits this flag.
+    // The parent overlay wrapper handles absolute bottom-0 positioning.
+    // `bottomAnchored` still triggers the license-banner offset padding
+    // inside CopilotChatInput. The welcome-screen input (below) intentionally
+    // omits this flag.
     bottomAnchored: true,
     ...(disclaimer !== undefined ? { disclaimer } : {}),
   } as CopilotChatInputProps);
 
-  // Hide suggestions while a thread is connecting or a run is in flight.
-  // Otherwise, mid-replay (bootstrap stream from /connect) or mid-run, the
-  // suggestions would render against a still-assembling message tree and
-  // visibly jump as each final text chunk reflows the layout.
+  // Hide suggestions while a thread is connecting (mid-replay would render
+  // against a still-assembling message tree and visibly jump as each final
+  // text chunk reflows the layout). Run-in-flight is handled by the
+  // SuggestionEngine: at run start, non-"always" suggestions are cleared and
+  // only "always" ones are restored — so a non-empty `suggestions` array
+  // here already means "this is something the user opted to keep visible."
   const hasSuggestions =
-    !isConnecting &&
-    !isRunning &&
-    Array.isArray(suggestions) &&
-    suggestions.length > 0;
+    !isConnecting && Array.isArray(suggestions) && suggestions.length > 0;
   const BoundSuggestionView = hasSuggestions
     ? renderSlot(suggestionView, CopilotChatSuggestionView, {
         suggestions,
@@ -291,8 +287,9 @@ export function CopilotChatView({
     isResizing,
     children: (
       <div
+        data-testid="copilot-scroll-content"
         style={{
-          paddingBottom: `${hasSuggestions ? 4 : 32}px`,
+          paddingBottom: `${inputContainerHeight + (hasSuggestions ? 4 : 32)}px`,
         }}
       >
         <div className="cpk:max-w-3xl cpk:mx-auto">
@@ -415,17 +412,22 @@ export function CopilotChatView({
       {dragOver && <DropOverlay />}
       {BoundScrollView}
 
-      <div className="cpk:max-w-3xl cpk:mx-auto cpk:w-full">
+      <div
+        ref={setInputContainerEl}
+        data-testid="copilot-input-overlay"
+        className="cpk:absolute cpk:bottom-0 cpk:left-0 cpk:right-0 cpk:z-20 cpk:pointer-events-none"
+      >
         {attachments && attachments.length > 0 && (
-          <CopilotChatAttachmentQueue
-            attachments={attachments}
-            onRemoveAttachment={(id) => onRemoveAttachment?.(id)}
-            className="cpk:px-4"
-          />
+          <div className="cpk:max-w-3xl cpk:mx-auto cpk:w-full cpk:pointer-events-auto">
+            <CopilotChatAttachmentQueue
+              attachments={attachments}
+              onRemoveAttachment={(id) => onRemoveAttachment?.(id)}
+              className="cpk:px-4"
+            />
+          </div>
         )}
+        {BoundInput}
       </div>
-
-      {BoundInput}
     </div>
   );
 }
@@ -476,7 +478,6 @@ export namespace CopilotChatView {
             </div>
           </StickToBottom.Content>
 
-          {/* Feather gradient overlay */}
           {BoundFeather}
 
           {/* Scroll to bottom button - hidden during resize */}
@@ -484,7 +485,7 @@ export namespace CopilotChatView {
             <div
               className="cpk:absolute cpk:inset-x-0 cpk:flex cpk:justify-center cpk:z-30 cpk:pointer-events-none"
               style={{
-                bottom: `${inputContainerHeight + FEATHER_HEIGHT + 16}px`,
+                bottom: `${inputContainerHeight + SCROLL_BUTTON_OFFSET}px`,
               }}
             >
               {renderSlot(
@@ -541,21 +542,13 @@ export namespace CopilotChatView {
       topOffset: 16,
     });
 
-    // Pin-to-send uses a SOFTER feather than pin-to-bottom:
-    //   - default: h-24 + from-white via-white to-transparent (fully opaque
-    //     bottom half, aggressive). Good for streaming-to-bottom where
-    //     the edge is always churning.
-    //   - pin-to-send: h-12 + from-white to-transparent (gradual fade,
-    //     no opaque midline). Gives a visual soft edge above the input
-    //     without obscuring otherwise-readable content.
-    // Consumers can still override with the `feather` slot.
-    const BoundFeather = renderSlot(feather, PinToSendSoftFeather, {});
-
-    // Feather and scroll-to-bottom button live OUTSIDE the scroll container.
-    // `position: absolute` children of an `overflow: auto` element are
-    // positioned relative to the scroll *content*, which means they scroll
-    // away with it. Placing them as siblings of the scroll container
+    // The feather and scroll-to-bottom button live OUTSIDE the scroll
+    // container. `position: absolute` children of an `overflow: auto` element
+    // are positioned relative to the scroll *content*, which means they
+    // scroll away with it. Placing them as siblings of the scroll container
     // (inside a `relative` wrapper) keeps them pinned to the viewport bottom.
+    const BoundFeather = renderSlot(feather, CopilotChatView.Feather, {});
+
     return (
       <ScrollElementContext.Provider value={nonAutoScrollEl}>
         <div
@@ -582,14 +575,13 @@ export namespace CopilotChatView {
               style={{ height: 0, flex: "0 0 auto" }}
             />
           </div>
-          {/* Soft feather — pinned to wrapper bottom */}
           {BoundFeather}
           {/* Scroll to bottom button */}
           {showScrollButton && !isResizing && (
             <div
               className="cpk:absolute cpk:inset-x-0 cpk:flex cpk:justify-center cpk:z-30 cpk:pointer-events-none"
               style={{
-                bottom: `${inputContainerHeight + PIN_TO_SEND_FEATHER_HEIGHT + 16}px`,
+                bottom: `${inputContainerHeight + SCROLL_BUTTON_OFFSET}px`,
               }}
             >
               {renderSlot(
@@ -722,7 +714,6 @@ export namespace CopilotChatView {
               {children}
             </div>
 
-            {/* Feather gradient overlay */}
             {BoundFeather}
 
             {/* Scroll to bottom button for manual mode */}
@@ -730,7 +721,7 @@ export namespace CopilotChatView {
               <div
                 className="cpk:absolute cpk:inset-x-0 cpk:flex cpk:justify-center cpk:z-30 cpk:pointer-events-none"
                 style={{
-                  bottom: `${inputContainerHeight + FEATHER_HEIGHT + 16}px`,
+                  bottom: `${inputContainerHeight + SCROLL_BUTTON_OFFSET}px`,
                 }}
               >
                 {renderSlot(
@@ -812,22 +803,14 @@ export namespace CopilotChatView {
     </Button>
   );
 
+  // Default renders an empty div — no visual, but the element is still in the
+  // tree so a slot override of the form `scrollView={{ feather: "my-class" }}`
+  // can apply classes (and any consumer with a full component override gets
+  // the className/style forwarding they expect).
   export const Feather: React.FC<React.HTMLAttributes<HTMLDivElement>> = ({
     className,
-    style,
     ...props
-  }) => (
-    <div
-      className={cn(
-        "cpk:absolute cpk:bottom-0 cpk:left-0 cpk:right-4 cpk:h-24 cpk:pointer-events-none cpk:z-10 cpk:bg-gradient-to-t",
-        "cpk:from-white cpk:via-white cpk:to-transparent",
-        "cpk:dark:from-[rgb(33,33,33)] cpk:dark:via-[rgb(33,33,33)]",
-        className,
-      )}
-      style={style}
-      {...props}
-    />
-  );
+  }) => <div className={className} {...props} />;
 
   export const WelcomeMessage: React.FC<
     React.HTMLAttributes<HTMLDivElement>

package/src/v2/components/chat/Lightbox.tsx

@@ -0,0 +1,103 @@
+import React, { useCallback, useEffect, useId, useRef, useState } from "react";
+import { createPortal, flushSync } from "react-dom";
+import { X } from "lucide-react";
+
+interface LightboxProps {
+  onClose: () => void;
+  children: React.ReactNode;
+}
+
+export function Lightbox({ onClose, children }: LightboxProps) {
+  useEffect(() => {
+    const handleKey = (e: KeyboardEvent) => {
+      if (e.key === "Escape") onClose();
+    };
+    document.addEventListener("keydown", handleKey);
+    return () => document.removeEventListener("keydown", handleKey);
+  }, [onClose]);
+
+  if (typeof document === "undefined") return null;
+
+  return createPortal(
+    <div
+      className="cpk:fixed cpk:inset-0 cpk:z-[9999] cpk:flex cpk:items-center cpk:justify-center cpk:bg-black/80 cpk:animate-fade-in"
+      onClick={onClose}
+    >
+      <button
+        onClick={onClose}
+        className="cpk:absolute cpk:top-4 cpk:right-4 cpk:text-white cpk:bg-white/10 cpk:hover:bg-white/20 cpk:rounded-full cpk:w-10 cpk:h-10 cpk:flex cpk:items-center cpk:justify-center cpk:cursor-pointer cpk:border-none cpk:z-10"
+        aria-label="Close preview"
+      >
+        <X className="cpk:w-5 cpk:h-5" />
+      </button>
+
+      <div onClick={(e) => e.stopPropagation()}>{children}</div>
+    </div>,
+    document.body,
+  );
+}
+
+type DocWithVT = Document & {
+  startViewTransition?: (cb: () => void) => { finished: Promise<void> };
+};
+
+/**
+ * Hook that manages lightbox open/close and uses the View Transition API to
+ * morph the thumbnail into fullscreen content.
+ *
+ * The trick: `view-transition-name` must live on exactly ONE element at a time.
+ * - Old state (thumbnail visible): name is on the thumbnail.
+ * - New state (lightbox visible): name moves to the lightbox content.
+ * `flushSync` ensures React commits the DOM change synchronously inside the
+ * `startViewTransition` callback so the API can snapshot old → new correctly.
+ */
+export function useLightbox() {
+  const thumbnailRef = useRef<HTMLElement>(null);
+  const [open, setOpen] = useState(false);
+  const vtName = useId();
+
+  const openLightbox = useCallback(() => {
+    const thumb = thumbnailRef.current;
+    const doc = document as DocWithVT;
+
+    if (doc.startViewTransition && thumb) {
+      thumb.style.viewTransitionName = vtName;
+
+      doc.startViewTransition(() => {
+        thumb.style.viewTransitionName = "";
+        flushSync(() => setOpen(true));
+      });
+    } else {
+      setOpen(true);
+    }
+  }, [vtName]);
+
+  const closeLightbox = useCallback(() => {
+    const thumb = thumbnailRef.current;
+    const doc = document as DocWithVT;
+
+    if (doc.startViewTransition && thumb) {
+      const transition = doc.startViewTransition(() => {
+        flushSync(() => setOpen(false));
+        thumb.style.viewTransitionName = vtName;
+      });
+      transition.finished
+        .then(() => {
+          thumb.style.viewTransitionName = "";
+        })
+        .catch(() => {
+          thumb.style.viewTransitionName = "";
+        });
+    } else {
+      setOpen(false);
+    }
+  }, [vtName]);
+
+  return {
+    thumbnailRef,
+    vtName,
+    open,
+    openLightbox,
+    closeLightbox,
+  };
+}