From ab6b7ca69076993cc42f84c7b78777991276a9a3 Mon Sep 17 00:00:00 2001
From: Marc-Aurele Besner
 <82244926+marc-aurele-besner@users.noreply.github.com>
Date: Fri, 13 Dec 2024 07:51:54 -0500
Subject: [PATCH] detect if file is encrypted

---
 .../components/Storage/Files/FilePreview.tsx  | 17 ++++++++-
 explorer/src/utils/file.ts                    | 36 ++++++++++++-------
 2 files changed, 39 insertions(+), 14 deletions(-)

diff --git a/explorer/src/components/Storage/Files/FilePreview.tsx b/explorer/src/components/Storage/Files/FilePreview.tsx
index 7b678bb9..0855e606 100644
--- a/explorer/src/components/Storage/Files/FilePreview.tsx
+++ b/explorer/src/components/Storage/Files/FilePreview.tsx
@@ -1,3 +1,4 @@
+import { LockClosedIcon } from '@heroicons/react/24/outline'
 import { Arguments } from 'components/common/Arguments'
 import { Spinner } from 'components/common/Spinner'
 import { GetCidQuery, GetCidQueryVariables } from 'gql/graphql'
@@ -16,6 +17,9 @@ type Props = {
 type PreviewData = {
   type: string
   data: ArrayBuffer
+  isEncrypted: boolean
+  // eslint-disable-next-line @typescript-eslint/no-explicit-any
+  uploadOptions: any
 }
 
 export const FilePreview: FC<Props> = ({ cid }) => {
@@ -31,11 +35,13 @@ export const FilePreview: FC<Props> = ({ cid }) => {
 
   const fetchData = useCallback(async () => {
     if (data) {
-      const { dataArrayBuffer } = extractFileData(data)
+      const { dataArrayBuffer, isEncrypted, uploadOptions } = extractFileData(data)
       const fileType = await detectFileType(dataArrayBuffer)
       setRawData({
         type: fileType,
         data: dataArrayBuffer,
+        isEncrypted,
+        uploadOptions,
       })
     }
   }, [data])
@@ -60,6 +66,15 @@ export const FilePreview: FC<Props> = ({ cid }) => {
     if (!imageSrc) return <></>
 
     switch (true) {
+      case rawData?.isEncrypted:
+        return (
+          <div className='flex items-center'>
+            <LockClosedIcon className={'ml-2 size-6 shrink-0'} stroke='#DE67E4' />
+            <p className='text-sm font-medium text-grayDarker dark:text-white md:text-xl'>
+              No preview available for encrypted files
+            </p>
+          </div>
+        )
       case rawData?.type.startsWith('image'):
         return (
           <Image
diff --git a/explorer/src/utils/file.ts b/explorer/src/utils/file.ts
index 85e44f8b..5c67b0cd 100644
--- a/explorer/src/utils/file.ts
+++ b/explorer/src/utils/file.ts
@@ -74,6 +74,19 @@ const extractFileDataByType = (data: any, type: 'file' | 'folder' | 'metadata')
   let rawData: string = ''
   let dataArrayBuffer: ArrayBuffer = new ArrayBuffer(0)
   let depth = 0
+  // eslint-disable-next-line @typescript-eslint/no-explicit-any
+  let uploadOptions: any
+
+  try {
+    if (data['files_' + type + 's'][0].chunk?.uploadOptions) {
+      uploadOptions = JSON.parse(data['files_' + type + 's'][0].chunk?.uploadOptions)
+      if (uploadOptions.encryption.algorithm)
+        return { rawData, dataArrayBuffer, isEncrypted: true, uploadOptions }
+    }
+  } catch (error) {
+    console.error('Error checking uploadOptions:', error)
+  }
+
   if (data['files_' + type + 's'][0][type + '_cids'].length === 0) {
     rawData = data['files_' + type + 's'][0].chunk?.data ?? ''
     dataArrayBuffer = Object.values(JSON.parse(rawData)) as unknown as ArrayBuffer
@@ -92,23 +105,20 @@ const extractFileDataByType = (data: any, type: 'file' | 'folder' | 'metadata')
     }
   }
   try {
-    if (data['files_' + type + 's'][0].chunk?.uploadOptions) {
-      const options = JSON.parse(data['files_' + type + 's'][0].chunk?.uploadOptions)
-      if (options.compression.algorithm === 'ZLIB') {
-        const inflate = new Zlib.Inflate(new Uint8Array(dataArrayBuffer), {
-          index: 0,
-          bufferSize: 1024,
-          bufferType: Zlib.Inflate.BufferType.BLOCK,
-          resize: true,
-          verify: true,
-        })
-        dataArrayBuffer = inflate.decompress()
-      }
+    if (uploadOptions.compression.algorithm === 'ZLIB') {
+      const inflate = new Zlib.Inflate(new Uint8Array(dataArrayBuffer), {
+        index: 0,
+        bufferSize: 1024,
+        bufferType: Zlib.Inflate.BufferType.BLOCK,
+        resize: true,
+        verify: true,
+      })
+      dataArrayBuffer = inflate.decompress()
     }
   } catch (error) {
     console.error('Error decompressing data:', error)
   }
-  return { rawData, dataArrayBuffer }
+  return { rawData, dataArrayBuffer, isEncrypted: false, uploadOptions }
 }
 
 // eslint-disable-next-line @typescript-eslint/no-explicit-any