From ba5caa10c419e24e4a5999950e87f8e23c5028fc Mon Sep 17 00:00:00 2001
From: Michael Chung <michaelchung@michaels-mbp-2.home>
Date: Fri, 2 May 2025 12:52:09 -0400
Subject: [PATCH] Hotfix: Add onDownloadClicked to lightbox and gallery modals.

---
 .../components/gallery-modal/src/lib/gallery-modal.tsx     | 2 ++
 .../components/lightbox-modal/src/lib/lightbox-modal.tsx   | 7 ++++++-
 2 files changed, 8 insertions(+), 1 deletion(-)

diff --git a/frontend/libs/components/gallery-modal/src/lib/gallery-modal.tsx b/frontend/libs/components/gallery-modal/src/lib/gallery-modal.tsx
index 9bbebb13fe..83a439a523 100644
--- a/frontend/libs/components/gallery-modal/src/lib/gallery-modal.tsx
+++ b/frontend/libs/components/gallery-modal/src/lib/gallery-modal.tsx
@@ -39,6 +39,7 @@ interface GalleryModalProps {
   tabs: { id: string; label: string }[];
   activeTab: string;
   onTabChange: (tabId: string) => void;
+  onDownloadClicked?: () => void;
 }
 
 export const GalleryModal = React.memo(
@@ -53,6 +54,7 @@ export const GalleryModal = React.memo(
     tabs,
     activeTab,
     onTabChange,
+    onDownloadClicked,
   }: GalleryModalProps) => {
     const [groupedItems, setGroupedItems] = useState<GroupedItems>({});
     const [loading, setLoading] = useState(false);
diff --git a/frontend/libs/components/lightbox-modal/src/lib/lightbox-modal.tsx b/frontend/libs/components/lightbox-modal/src/lib/lightbox-modal.tsx
index d13a62039b..5bf3bf2d43 100644
--- a/frontend/libs/components/lightbox-modal/src/lib/lightbox-modal.tsx
+++ b/frontend/libs/components/lightbox-modal/src/lib/lightbox-modal.tsx
@@ -15,6 +15,7 @@ interface LightboxModalProps {
   createdAt?: string;
   additionalInfo?: React.ReactNode;
   downloadUrl?: string;
+  onDownloadClicked?: () => void;
 }
 
 export function LightboxModal({
@@ -27,6 +28,7 @@ export function LightboxModal({
   createdAt,
   additionalInfo,
   downloadUrl,
+  onDownloadClicked,
 }: LightboxModalProps) {
   return createPortal(
     <Transition appear show={isOpen}>
@@ -104,7 +106,10 @@ export function LightboxModal({
                       >
                         <Button
                           icon={faDownToLine}
-                          onClick={(e) => e.stopPropagation()}
+                          onClick={(e) => {
+                            e.stopPropagation();
+                            onDownloadClicked?.();
+                          }}
                         >
                           Download
                         </Button>