add logic to decode special media types

Author: DenisaCG

jupyter_drives/manager.py

@@ -7,6 +7,7 @@
 import tornado
 import httpx
 import traitlets
+import base64
 from jupyter_server.utils import url_path_join
 
 import obstore as obs
@@ -155,7 +156,7 @@ async def unmount_drive(self, drive_name: str):
 
         return
 
-    async def get_contents(self, drive_name, path):
+    async def get_contents(self, drive_name, path, special_type=False):
         """Get contents of a file or directory.
 
         Args:
@@ -191,9 +192,18 @@ async def get_contents(self, drive_name, path):
 
                 # retrieve metadata of object
                 metadata = await obs.head_async(self._content_managers[drive_name], path)
+
+                # for certain media type files, extracted contents need to be read as a byte array and decoded to base64 to be viewable in JupyterLab
+                # the following extesnions correspond to a base64 file format or are of type PDF
+                ext = os.path.splitext(path)[1]
+                if ext == '.pdf' or ext == '.svg' or ext == '.tif' or ext == '.tiff' or ext == '.jpg' or ext == '.jpeg' or ext == '.gif' or ext == '.png' or ext == '.bmp' or ext == '.webp':
+                    processed_content = base64.b64encode(content).decode("utf-8")
+                else:
+                    processed_content = content.decode("utf-8")
+
                 data = {
                     "path": path, 
-                    "content": content.decode("utf-8"),
+                    "content": processed_content,
                     "last_modified": metadata["last_modified"].isoformat(),
                     "size": metadata["size"]
                 }

src/contents.ts

@@ -195,7 +195,6 @@ export class Drive implements Contents.IDrive {
     options?: Contents.IFetchOptions
   ): Promise<Contents.IModel> {
     let relativePath = '';
-    console.log('GET localpath: ', localPath);
     if (localPath !== '') {
       // extract current drive name
       const currentDrive = this._drivesList.filter(
@@ -261,7 +260,6 @@ export class Drive implements Contents.IDrive {
         type: 'directory'
       };
     }
-    console.log('GET: ', relativePath);
 
     Contents.validateContentsModel(data);
     return data;