Add support for opening sub-directories

Author: martinRenou

src/drive.ts

@@ -1,7 +1,21 @@
 import { Contents, ServerConnection } from '@jupyterlab/services';
 
+import { PathExt } from '@jupyterlab/coreutils';
+
 import { ISignal, Signal } from '@lumino/signaling';
 
+async function toArray<T>(
+  asyncIterator: AsyncIterableIterator<T>
+): Promise<T[]> {
+  const arr = [];
+
+  for await (const i of asyncIterator) {
+    arr.push(i);
+  }
+
+  return arr;
+}
+
 export class FileSystemDrive implements Contents.IDrive {
   get isDisposed(): boolean {
     return this._isDisposed;
@@ -36,7 +50,7 @@ export class FileSystemDrive implements Contents.IDrive {
   }
 
   async get(
-    localPath: string,
+    path: string,
     options?: Contents.IFetchOptions
   ): Promise<Contents.IModel> {
     const root = this._rootHandle;
@@ -55,12 +69,33 @@ export class FileSystemDrive implements Contents.IDrive {
       };
     }
 
+    let parentHandle = root;
+    // If requesting a file/directory that is not under root, we need the right directory handle
+    for (const subPath of path.split('/').slice(0, -1)) {
+      parentHandle = await parentHandle.getDirectoryHandle(subPath);
+    }
+
+    const parentPath = PathExt.dirname(path);
+    const localPath = PathExt.basename(path);
+
+    let localHandle: FileSystemDirectoryHandle | FileSystemFileHandle;
+
+    const currentContent = await toArray(parentHandle.values());
+
     if (localPath) {
-      const handle = await root.getFileHandle(localPath);
-      const file = await handle.getFile();
+      localHandle = currentContent.filter(
+        element => element.name === localPath
+      )[0];
+    } else {
+      localHandle = parentHandle;
+    }
+
+    if (localHandle.kind === 'file') {
+      const file = await localHandle.getFile();
+
       return {
         name: file.name,
-        path: file.name,
+        path: PathExt.join(parentPath, localPath),
         created: new Date(file.lastModified).toISOString(),
         last_modified: new Date(file.lastModified).toISOString(),
         format: 'text',
@@ -69,52 +104,51 @@ export class FileSystemDrive implements Contents.IDrive {
         type: 'file',
         mimetype: 'text/plain'
       };
-    }
-
-    const content: Contents.IModel[] = [];
-
-    for await (const value of root.values()) {
-      if (value.kind === 'file') {
-        const file = await value.getFile();
-        content.push({
-          name: file.name,
-          path: file.name,
-          created: new Date(file.lastModified).toISOString(),
-          last_modified: new Date(file.lastModified).toISOString(),
-          format: 'text',
-          content: await file.text(),
-          writable: true,
-          type: 'file',
-          mimetype: 'text/plain'
-        });
-      } else {
-        content.push({
-          name: value.name,
-          path: value.name,
-          created: new Date().toISOString(),
-          last_modified: new Date().toISOString(),
-          format: 'json',
-          content: null,
-          writable: true,
-          type: 'directory',
-          mimetype: 'application/json'
-        });
+    } else {
+      const content: Contents.IModel[] = [];
+
+      for await (const value of localHandle.values()) {
+        if (value.kind === 'file') {
+          const file = await value.getFile();
+          content.push({
+            name: file.name,
+            path: PathExt.join(parentPath, localPath, file.name),
+            created: new Date(file.lastModified).toISOString(),
+            last_modified: new Date(file.lastModified).toISOString(),
+            format: 'text',
+            content: await file.text(),
+            writable: true,
+            type: 'file',
+            mimetype: 'text/plain'
+          });
+        } else {
+          content.push({
+            name: value.name,
+            path: PathExt.join(parentPath, localPath, value.name),
+            created: '',
+            last_modified: '',
+            format: 'json',
+            content: null,
+            writable: true,
+            type: 'directory',
+            mimetype: 'application/json'
+          });
+        }
       }
-    }
 
-    const date = new Date();
-    return {
-      name: 'root',
-      path: '',
-      last_modified: date.toISOString(),
-      created: date.toISOString(),
-      format: 'json',
-      mimetype: 'application/json',
-      content,
-      size: undefined,
-      writable: true,
-      type: 'directory'
-    };
+      return {
+        name: PathExt.basename(parentPath),
+        path: PathExt.join(parentPath, localPath),
+        last_modified: '',
+        created: '',
+        format: 'json',
+        mimetype: 'application/json',
+        content,
+        size: undefined,
+        writable: true,
+        type: 'directory'
+      };
+    }
   }
 
   getDownloadUrl(localPath: string): Promise<string> {