feat: add fileSize to asset downloads

frontend/packages/data-portal/app/graphql/getDatasetByIdV2.server.ts

@@ -34,7 +34,7 @@ const GET_DATASET_BY_ID_QUERY_V2 = gql(`
       lastModifiedDate
       releaseDate
       depositionDate
-
+      fileSize
       # Dataset metadata
       id
       title

frontend/packages/data-portal/app/graphql/getRunByIdDiffer.ts

@@ -456,6 +456,15 @@ export function logIfHasDiff(
         },
       ],
     },
+    numTotalSizeAnnotationFiles: {
+      aggregate: [
+        {
+          sum: {
+            fileSize: v1.annotation_files_aggregate_for_total.aggregate?.count,
+          },
+        },
+      ],
+    },
     tomogramsAggregate: {
       aggregate: [
         {

frontend/packages/data-portal/app/graphql/getRunByIdV2.server.ts

@@ -69,6 +69,7 @@ const GET_RUN_BY_ID_QUERY_V2 = gql(`
         cellTypeId
         depositionDate
         description
+        fileSize
         gridPreparation
         id
         lastModifiedDate
@@ -223,6 +224,7 @@ const GET_RUN_BY_ID_QUERY_V2 = gql(`
             format
             httpsPath
             s3Path
+            fileSize
           }
         }
       }
@@ -294,6 +296,8 @@ const GET_RUN_BY_ID_QUERY_V2 = gql(`
       }
       ctfCorrected
       fiducialAlignmentStatus
+      fileSizeMrc
+      fileSizeOmezarr
       httpsMrcFile
       id
       isPortalStandard
@@ -404,6 +408,14 @@ const GET_RUN_BY_ID_QUERY_V2 = gql(`
       }
     }
 
+    numTotalSizeAnnotationFiles: annotationFilesAggregate(where: { annotationShape: { annotation: { runId: { _eq: $id }}}}) {
+      aggregate {
+        sum {
+          fileSize
+        }
+      }
+    }
+
     # Tomogram counts:
     tomogramsAggregate(where: { runId: { _eq: $id }}) {
       aggregate {

frontend/packages/data-portal/app/hooks/useRunById.ts

@@ -73,6 +73,8 @@ export function useRunById() {
     groundTruthCount:
       v2.numFilteredGroundTruthAnnotationRows.aggregate?.[0].count ?? 0,
     otherCount: v2.numFilteredOtherAnnotationRows.aggregate?.[0].count ?? 0,
+    totalSize:
+      v2.numTotalSizeAnnotationFiles?.aggregate?.[0].sum?.fileSize ?? 0,
   }
 
   const tomogramsCount = v2.tomogramsAggregate.aggregate?.[0].count ?? 0

frontend/packages/data-portal/app/routes/datasets.$id.tsx

@@ -116,6 +116,7 @@ export default function DatasetByIdPage() {
           datasetId={dataset.id}
           datasetTitle={dataset.title}
           s3Path={dataset.s3Prefix}
+          fileSize={dataset.fileSize ?? undefined}
           type="dataset"
         />
       }

frontend/packages/data-portal/app/routes/runs.$id.tsx

@@ -23,7 +23,6 @@ import { getRunById } from 'app/graphql/getRunById.server'
 import { logIfHasDiff } from 'app/graphql/getRunByIdDiffer'
 import { getRunByIdV2 } from 'app/graphql/getRunByIdV2.server'
 import { useDownloadModalQueryParamState } from 'app/hooks/useDownloadModalQueryParamState'
-import { useFileSize } from 'app/hooks/useFileSize'
 import { useI18n } from 'app/hooks/useI18n'
 import { useQueryParam } from 'app/hooks/useQueryParam'
 import { useRunById } from 'app/hooks/useRunById'
@@ -133,15 +132,28 @@ export default function RunByIdPage() {
       annotationShape.shapeType === objectShapeType,
   )
 
-  const httpsPath = activeAnnotationShape
-    ? activeAnnotationShape.annotationFiles.edges.find(
-        (file) => file.node.format === fileFormat,
-      )?.node.httpsPath
+  const activeFile = activeAnnotationShape?.annotationFiles.edges.find(
+    (file) => file.node.format === fileFormat,
+  )
+
+  const httpsPath = activeFile
+    ? activeFile.node.httpsPath
     : activeTomogram?.httpsMrcFile ?? undefined
 
-  const { data: fileSize } = useFileSize(httpsPath, {
-    enabled: fileFormat !== 'zarr',
-  })
+  const getFileSize = () => {
+    if (activeFile) {
+      return activeFile.node.fileSize ?? undefined
+    }
+    if (fileFormat === 'mrc') {
+      return activeTomogram?.fileSizeMrc ?? undefined
+    }
+    if (fileFormat === 'zarr') {
+      return activeTomogram?.fileSizeOmezarr ?? undefined
+    }
+    return annotationFilesAggregates.totalSize
+  }
+
+  const fileSize = getFileSize()
 
   const [depositionId] = useQueryParam<string>(QueryParams.DepositionId)