Handle single images

beyondessential · Sep 18, 2024 · 1181ce4 · 1181ce4
1 parent caff163
commit 1181ce4
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/packages/tupaia-web/src/features/Visuals/View/MultiPhotograph/useDownloadImages.ts b/packages/tupaia-web/src/features/Visuals/View/MultiPhotograph/useDownloadImages.ts
@@ -33,6 +33,17 @@ export const useDownloadImages = (
     },
     {
       onSuccess: async fetchedImages => {
+        if (fetchedImages.length === 1) {
+          const { blob } = fetchedImages[0];
+          const { type } = blob;
+          const fileExtension = type.split('/')[1];
+          download(
+            blob,
+            `${projectCode}_${entityCode}_${dashboardCode}_${reportCode}.${fileExtension}`,
+          );
+          return;
+        }
+
         const blobsWithNames: {
           blob: Blob;
           name: string;