add command for external drives

Author: DenisaCG

src/contents.ts

@@ -24,7 +24,8 @@ import {
   createDrive,
   getDrivesList,
   excludeDrive,
-  includeDrive
+  includeDrive,
+  addExternalDrive
 } from './requests';
 
 export class Drive implements Contents.IDrive {
@@ -850,6 +851,42 @@ export class Drive implements Contents.IDrive {
     return data;
   }
 
+  /**
+   * Add external drive.
+   *
+   * @param options: The options used to add the external drive.
+   *
+   * @returns A promise which resolves with the contents model.
+   */
+  async addExternalDrive(
+    driveUrl: string,
+    location: string
+  ): Promise<Contents.IModel> {
+    await addExternalDrive(driveUrl, location);
+
+    const data: Contents.IModel = {
+      name: driveUrl,
+      path: driveUrl,
+      last_modified: '',
+      created: '',
+      content: [],
+      format: 'json',
+      mimetype: '',
+      size: 0,
+      writable: true,
+      type: 'directory'
+    };
+
+    Contents.validateContentsModel(data);
+    this._fileChanged.emit({
+      type: 'new',
+      oldValue: null,
+      newValue: data
+    });
+
+    return data;
+  }
+
   /**
    * Exclude drive from browser.
    *

src/plugins/driveBrowserPlugin.ts

@@ -454,6 +454,38 @@ namespace Private {
       rank: 110
     });
 
+    app.commands.addCommand(CommandIDs.addExternalDrive, {
+      isVisible: () => {
+        return browser.model.path === 's3:';
+      },
+      execute: async () => {
+        return showDialog({
+          title: 'Add External Drive',
+          body: new Private.CreateDriveHandler(drive.name),
+          focusNodeSelector: 'input',
+          buttons: [
+            Dialog.cancelButton(),
+            Dialog.okButton({
+              label: 'Add',
+              ariaLabel: 'Add Drive'
+            })
+          ]
+        }).then(result => {
+          if (result.value) {
+            drive.addExternalDrive(result.value[0], result.value[1]);
+          }
+        });
+      },
+      label: 'Add External Drive',
+      icon: driveBrowserIcon.bindprops({ stylesheet: 'menuItem' })
+    });
+
+    app.contextMenu.addItem({
+      command: CommandIDs.addExternalDrive,
+      selector: '#drive-file-browser.jp-SidePanel .jp-DirListing-content',
+      rank: 110
+    });
+
     app.commands.addCommand(CommandIDs.toggleFileFilter, {
       execute: () => {
         // Update toggled state, then let the toolbar button update

src/requests.ts

@@ -509,7 +509,22 @@ export async function createDrive(
  */
 export async function addPublicDrive(driveUrl: string) {
   return await requestAPI<any>('drives/' + driveUrl + '/', 'POST', {
-    public: true
+    is_public: true
+  });
+}
+
+/**
+ * Add external drive.
+ *
+ * @param driveUrl The drive URL.
+ * @param location The drive region.
+ *
+ * @returns A promise which resolves with the contents model.
+ */
+export async function addExternalDrive(driveUrl: string, location: string) {
+  return await requestAPI<any>('drives/' + driveUrl + '/', 'POST', {
+    is_public: false,
+    region: location
   });
 }
 

src/token.ts

@@ -9,6 +9,7 @@ export namespace CommandIDs {
   export const toggleBrowser = 'drives:toggle-main';
   export const createNewDrive = 'drives:create-new-drive';
   export const addPublicDrive = 'drives:add-public-drive';
+  export const addExternalDrive = 'drives:add-external-drive';
   export const launcher = 'launcher:create';
   export const toggleFileFilter = 'drives:toggle-file-filter';
   export const createNewDirectory = 'drives:create-new-directory';