fix: add on.download callback

igordanchenko · Apr 24, 2023 · 5e43170 · 5e43170
1 parent 265b3d2
commit 5e43170
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 3 deletions.
diff --git a/src/plugins/download/DownloadButton.tsx b/src/plugins/download/DownloadButton.tsx
@@ -9,8 +9,8 @@ const DownloadIcon = createIcon(
 );
 
 export function DownloadButton() {
-    const { render } = useLightboxProps();
-    const { currentSlide } = useLightboxState();
+    const { render, on } = useLightboxProps();
+    const { currentSlide, currentIndex } = useLightboxState();
 
     if (render.buttonDownload) {
         return <>{render.buttonDownload()}</>;
@@ -28,6 +28,8 @@ export function DownloadButton() {
             onClick={() => {
                 if (downloadUrl) {
                     saveAs(downloadUrl, currentSlide?.downloadFilename);
+
+                    on.download?.({ index: currentIndex });
                 }
             }}
         />

diff --git a/src/plugins/download/index.ts b/src/plugins/download/index.ts
@@ -1,4 +1,4 @@
-import { RenderFunction } from "../../types.js";
+import { Callback, RenderFunction } from "../../types.js";
 import { PLUGIN_DOWNLOAD } from "../../core/index.js";
 import { Download } from "./Download.js";
 
@@ -17,6 +17,14 @@ declare module "../../types.js" {
         iconDownload?: RenderFunction;
     }
 
+    interface Callbacks {
+        download?: Callback<DownloadCallbackProps>;
+    }
+
+    export interface DownloadCallbackProps {
+        index: number;
+    }
+
     // noinspection JSUnusedGlobalSymbols
     interface ToolbarButtonKeys {
         [PLUGIN_DOWNLOAD]: null;