override new directory command for drive browser

Author: DenisaCG

schema/drives-file-browser.json

@@ -11,7 +11,7 @@
       },
       {
         "name": "new-directory",
-        "command": "filebrowser:create-new-directory",
+        "command": "drives:create-new-directory",
         "label": "",
         "rank": 10
       },
@@ -36,6 +36,21 @@
       }
     ]
   },
+  "jupyter.lab.menus": {
+    "context": [
+      {
+        "command": "filebrowser:create-new-directory",
+        "selector": ".jp-DirListing-content",
+        "rank": 55,
+        "disabled": true
+      },
+      {
+        "command": "drives:create-new-directory",
+        "selector": ".jp-DirListing-content",
+        "rank": 55
+      }
+    ]
+  },
   "jupyter.lab.setting-icon": "jupyter-drives:drive-browser",
   "jupyter.lab.setting-icon-label": "Drive Browser",
   "type": "object",

src/plugins/driveBrowserPlugin.ts

@@ -22,7 +22,8 @@ import { ISettingRegistry } from '@jupyterlab/settingregistry';
 import {
   filterIcon,
   FilenameSearcher,
-  IScore
+  IScore,
+  newFolderIcon
 } from '@jupyterlab/ui-components';
 import { CommandRegistry } from '@lumino/commands';
 import { Widget } from '@lumino/widgets';
@@ -148,8 +149,9 @@ export const driveFileBrowser: JupyterFrontEndPlugin<void> = {
     );
 
     const updateVisibility = () => {
-      // Visibility of command changed.
+      // Visibility of context menu and toolbar commands changed.
       app.commands.notifyCommandChanged(CommandIDs.createNewDrive);
+      app.commands.notifyCommandChanged(CommandIDs.createNewDirectory);
     };
 
     // Listen for path changes.
@@ -370,5 +372,17 @@ namespace Private {
       icon: filterIcon.bindprops({ stylesheet: 'menuItem' }),
       label: 'Toggle File Filter'
     });
+
+    app.commands.addCommand(CommandIDs.createNewDirectory, {
+      isVisible: () => {
+        return browser.model.path !== 's3:';
+      },
+      execute: () => {
+        console.log('NEW COMMAND DIR');
+        app.commands.execute('filebrowser:create-new-directory');
+      },
+      icon: newFolderIcon.bindprops({ stylesheet: 'menuItem' }),
+      label: 'New Folder'
+    });
   }
 }

src/token.ts

@@ -10,6 +10,7 @@ export namespace CommandIDs {
   export const createNewDrive = 'drives:create-new-drive';
   export const launcher = 'launcher:create';
   export const toggleFileFilter = 'drives:toggle-file-filter';
+  export const createNewDirectory = 'drives:create-new-directory';
 }
 
 /**