Add custom commands that use "memory" terminology instead of the word "kernel" (#156)

* Add a `jupytereverywhere:restart-memory` command

* Add `jupytereverywhere:restart-and-run-all`

* Add comments to describe both commands

* "Kernel restart failed" ➡️ "Memory restart failed"

* Fix path to `fast-forward.svg`

* Add tests for both commands

* Update Playwright Snapshots

---------

Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>

Author: agriyakhetarpal

schema/plugin.json

@@ -58,11 +58,14 @@
       },
       {
         "name": "restart",
-        "caption": "Restart notebook"
+        "command": "jupytereverywhere:restart-memory",
+        "caption": "Restart notebook",
+        "label": ""
       },
       {
         "name": "restart-and-run",
-        "caption": "Restart and run all cells"
+        "caption": "Restart and run all cells",
+        "command": "jupytereverywhere:restart-and-run-all"
       },
       {
         "name": "share",

src/commands.ts

@@ -6,4 +6,6 @@ export namespace Commands {
   export const downloadPDFCommand = 'jupytereverywhere:download-pdf';
   export const shareNotebookCommand = 'jupytereverywhere:share-notebook';
   export const createCopyNotebookCommand = 'jupytereverywhere:create-copy-notebook';
+  export const restartMemoryCommand = 'jupytereverywhere:restart-memory';
+  export const restartMemoryAndRunAllCommand = 'jupytereverywhere:restart-and-run-all';
 }

src/index.ts

@@ -9,6 +9,10 @@ import { SharingService } from './sharing-service';
 
 import { createSuccessDialog, createErrorDialog } from './ui-components/share-dialog';
 
+import { LabIcon } from '@jupyterlab/ui-components';
+import refreshIcon from '../style/icons/refresh.svg';
+import fastForwardSvg from '../style/icons/fast-forward.svg';
+
 import { exportNotebookAsPDF } from './pdf';
 import { files } from './pages/files';
 import { Commands } from './commands';
@@ -211,6 +215,75 @@ const plugin: JupyterFrontEndPlugin<void> = {
       }
     });
 
+    /**
+     * Add a command to restart the notebook kernel, terming it as "memory"
+     */
+    const RefreshLabIcon = new LabIcon({
+      name: 'jupytereverywhere:refresh',
+      svgstr: refreshIcon
+    });
+
+    commands.addCommand(Commands.restartMemoryCommand, {
+      label: 'Restart Notebook Memory',
+      icon: RefreshLabIcon,
+      execute: async () => {
+        const panel = tracker.currentWidget;
+        if (!panel) {
+          console.warn('No active notebook to restart.');
+          return;
+        }
+
+        const result = await showDialog({
+          title: 'Would you like to restart the notebook’s memory?',
+          buttons: [Dialog.cancelButton({ label: 'Cancel' }), Dialog.okButton({ label: 'Restart' })]
+        });
+
+        if (result.button.accept) {
+          try {
+            await panel.sessionContext.restartKernel();
+          } catch (err) {
+            console.error('Memory restart failed', err);
+          }
+        }
+      }
+    });
+
+    /**
+     * Add a command to restart the notebook kernel, terming it as "memory",
+     * and run all cells after the restart.
+     */
+    const customFastForwardIcon = new LabIcon({
+      name: 'jupytereverywhere:restart-run',
+      svgstr: fastForwardSvg
+    });
+
+    commands.addCommand(Commands.restartMemoryAndRunAllCommand, {
+      label: 'Restart Notebook Memory and Run All Cells',
+      icon: customFastForwardIcon,
+      isEnabled: () => !!tracker.currentWidget,
+      execute: async () => {
+        const panel = tracker.currentWidget;
+        if (!panel) {
+          console.warn('No active notebook to restart and run.');
+          return;
+        }
+
+        const result = await showDialog({
+          title: 'Would you like to restart the notebook’s memory and rerun all cells?',
+          buttons: [Dialog.cancelButton({ label: 'Cancel' }), Dialog.okButton({ label: 'Restart' })]
+        });
+
+        if (result.button.accept) {
+          try {
+            await panel.sessionContext.restartKernel();
+            await commands.execute('notebook:run-all-cells');
+          } catch (err) {
+            console.error('Restarting and running all cells failed', err);
+          }
+        }
+      }
+    });
+
     /**
      * Add custom Share notebook command
      */

ui-tests/tests/jupytereverywhere.spec.ts

@@ -559,3 +559,27 @@ test.describe('Title of the pages should be "Jupyter Everywhere"', () => {
     expect(title).toBe('Jupyter Everywhere');
   });
 });
+
+test.describe('Kernel commands should use memory terminology', () => {
+  test('Restart memory command', async ({ page }) => {
+    const promise = runCommand(page, 'jupytereverywhere:restart-memory');
+    const dialog = page.locator('.jp-Dialog-content');
+
+    await expect(dialog).toBeVisible();
+    expect(await dialog.screenshot()).toMatchSnapshot('restart-memory-dialog.png');
+
+    await dialog.press('Escape');
+    await promise;
+  });
+
+  test('Restart memory and run all cells command', async ({ page }) => {
+    const promise = runCommand(page, 'jupytereverywhere:restart-and-run-all');
+    const dialog = page.locator('.jp-Dialog-content');
+
+    await expect(dialog).toBeVisible();
+    expect(await dialog.screenshot()).toMatchSnapshot('restart-memory-run-all-dialog.png');
+
+    await dialog.press('Escape');
+    await promise;
+  });
+});