Add session context patch to include drive name in path

src/index.ts

@@ -2,13 +2,15 @@ import { JupyterFrontEndPlugin } from '@jupyterlab/application';
 import {
   driveFileBrowser,
   openDriveDialogPlugin,
-  launcherPlugin
+  launcherPlugin,
+  sessionContextPatch
 } from './plugins';
 
 const plugins: JupyterFrontEndPlugin<any>[] = [
   driveFileBrowser,
   openDriveDialogPlugin,
-  launcherPlugin
+  launcherPlugin,
+  sessionContextPatch
 ];
 
 export default plugins;

src/plugins/index.ts

@@ -1,3 +1,4 @@
 export * from './driveBrowserPlugin';
 export * from './launcherPlugin';
 export * from './driveDialogPlugin';
+export * from './sessionContextPatch';

src/plugins/sessionContextPatch.ts

@@ -0,0 +1,39 @@
+import {
+  JupyterFrontEndPlugin,
+  JupyterFrontEnd
+} from '@jupyterlab/application';
+import { SessionContext } from '@jupyterlab/apputils';
+import {
+  IDocumentManager,
+  IDocumentWidgetOpener
+} from '@jupyterlab/docmanager';
+
+/**
+ * A plugin to patch the session context path so it includes the drive name.
+ * associated with.
+ */
+export const sessionContextPatch: JupyterFrontEndPlugin<void> = {
+  id: '@jupyterlite/application-extension:session-context-patch',
+  autoStart: true,
+  requires: [IDocumentManager, IDocumentWidgetOpener],
+  activate: (
+    app: JupyterFrontEnd,
+    docManager: IDocumentManager,
+    widgetOpener: IDocumentWidgetOpener
+  ) => {
+    const contents = app.serviceManager.contents;
+
+    widgetOpener.opened.connect((_, widget) => {
+      const context = docManager.contextForWidget(widget);
+      const driveName = contents.driveName(context?.path ?? '');
+      if (driveName === '') {
+        // do nothing if this is the default drive
+        return;
+      }
+      const sessionContext = widget.context.sessionContext as SessionContext;
+
+      // Path the session context to include the drive name
+      sessionContext['_path'] = context?.path;
+    });
+  }
+};