Merge pull request #6895 from Queuecumber/master

Running Sidebar Extension API

jupyterlab/staging/yarn.lock

@@ -1605,6 +1605,11 @@ ansi-regex@^2.0.0:
   resolved "https://registry.yarnpkg.com/ansi-regex/-/ansi-regex-2.1.1.tgz#c3b33ab5ee360d86e0e628f0468ae7ef27d654df"
   integrity sha1-w7M6te42DYbg5ijwRorn7yfWVN8=
 
+ansi-regex@^3.0.0:
+  version "3.0.0"
+  resolved "https://registry.yarnpkg.com/ansi-regex/-/ansi-regex-3.0.0.tgz#ed0317c322064f79466c02966bddb605ab37d998"
+  integrity sha1-7QMXwyIGT3lGbAKWa922Bas32Zg=
+
 ansi-regex@^4.1.0:
   version "4.1.0"
   resolved "https://registry.yarnpkg.com/ansi-regex/-/ansi-regex-4.1.0.tgz#8b9f8f08cf1acb843756a839ca8c7e3168c51997"
@@ -5273,6 +5278,11 @@ safe-buffer@5.1.2, safe-buffer@^5.0.1, safe-buffer@^5.1.0, safe-buffer@^5.1.1, s
   resolved "https://registry.yarnpkg.com/safe-buffer/-/safe-buffer-5.1.2.tgz#991ec69d296e0313747d59bdfd2b745c35f8828d"
   integrity sha512-Gd2UZBJDkXlY7GbJxfsE8/nvKkUEU1G38c1siN6QP6a9PT9MmHB8GnpscSmMJSoF8LOIrt8ud/wPtojys4G6+g==
 
+safe-buffer@~5.2.0:
+  version "5.2.0"
+  resolved "https://registry.yarnpkg.com/safe-buffer/-/safe-buffer-5.2.0.tgz#b74daec49b1148f88c64b68d49b1e815c1f2f519"
+  integrity sha512-fZEwUGbVl7kouZs1jCdMLdt95hdIv0ZeHg6L7qPeciMZhZ+/gdesW4wgTARkrFWEpspjEATAzUGPG8N2jJiwbg==
+
 safe-regex@^1.1.0:
   version "1.1.0"
   resolved "https://registry.yarnpkg.com/safe-regex/-/safe-regex-1.1.0.tgz#40a3669f3b077d1e943d44629e157dd48023bf2e"
@@ -5652,6 +5662,13 @@ strip-ansi@^3.0.0, strip-ansi@^3.0.1:
   dependencies:
     ansi-regex "^2.0.0"
 
+strip-ansi@^4.0.0:
+  version "4.0.0"
+  resolved "https://registry.yarnpkg.com/strip-ansi/-/strip-ansi-4.0.0.tgz#a8479022eb1ac368a871389b635262c505ee368f"
+  integrity sha1-qEeQIusaw2iocTibY1JixQXuNo8=
+  dependencies:
+    ansi-regex "^3.0.0"
+
 strip-ansi@^5.0.0, strip-ansi@^5.1.0, strip-ansi@^5.2.0:
   version "5.2.0"
   resolved "https://registry.yarnpkg.com/strip-ansi/-/strip-ansi-5.2.0.tgz#8c9a536feb6afc962bdfa5b104a5091c1ad9c0ae"

packages/running-extension/package.json

@@ -36,7 +36,10 @@
   },
   "dependencies": {
     "@jupyterlab/application": "^1.2.0-alpha.0",
-    "@jupyterlab/running": "^1.2.0-alpha.0"
+    "@jupyterlab/coreutils": "^3.2.0-alpha.0",
+    "@jupyterlab/running": "^1.2.0-alpha.0",
+    "@jupyterlab/services": "^4.2.0-alpha.0",
+    "@phosphor/algorithm": "^1.2.0"
   },
   "devDependencies": {
     "rimraf": "~2.6.2",

packages/running-extension/src/index.ts

@@ -6,15 +6,39 @@ import {
   JupyterFrontEnd,
   JupyterFrontEndPlugin
 } from '@jupyterlab/application';
+import {
+  IRunningSessions,
+  IRunningSessionManagers,
+  RunningSessionManagers,
+  RunningSessions
+} from '@jupyterlab/running';
+
+import { Session } from '@jupyterlab/services';
+import { PathExt } from '@jupyterlab/coreutils';
+import { toArray } from '@phosphor/algorithm';
+
+/**
+ * The class name added to a notebook icon.
+ */
+const NOTEBOOK_ICON_CLASS = 'jp-mod-notebook';
+
+/**
+ * The class name added to a console icon.
+ */
+const CONSOLE_ICON_CLASS = 'jp-mod-console';
 
-import { RunningSessions } from '@jupyterlab/running';
+/**
+ * The class name added to a file icon.
+ */
+const FILE_ICON_CLASS = 'jp-mod-file';
 
 /**
  * The default running sessions extension.
  */
-const plugin: JupyterFrontEndPlugin<void> = {
+const plugin: JupyterFrontEndPlugin<IRunningSessionManagers> = {
   activate,
   id: '@jupyterlab/running-extension:plugin',
+  provides: IRunningSessionManagers,
   optional: [ILayoutRestorer],
   autoStart: true
 };
@@ -30,8 +54,9 @@ export default plugin;
 function activate(
   app: JupyterFrontEnd,
   restorer: ILayoutRestorer | null
-): void {
-  let running = new RunningSessions({ manager: app.serviceManager });
+): IRunningSessionManagers {
+  let runningSessionManagers = new RunningSessionManagers();
+  let running = new RunningSessions(runningSessionManagers);
   running.id = 'jp-running-sessions';
   running.title.iconClass = 'jp-RunningIcon jp-SideBar-tabIcon';
   running.title.caption = 'Running Terminals and Kernels';
@@ -42,21 +67,77 @@ function activate(
   if (restorer) {
     restorer.add(running, 'running-sessions');
   }
+  addKernelRunningSessionManager(runningSessionManagers, app);
+  // Rank has been chosen somewhat arbitrarily to give priority to the running
+  // sessions widget in the sidebar.
+  app.shell.add(running, 'left', { rank: 200 });
 
-  running.sessionOpenRequested.connect((sender, model) => {
-    let path = model.path;
-    if (model.type.toLowerCase() === 'console') {
-      void app.commands.execute('console:open', { path });
-    } else {
-      void app.commands.execute('docmanager:open', { path });
-    }
-  });
+  return runningSessionManagers;
+}
 
-  running.terminalOpenRequested.connect((sender, model) => {
-    void app.commands.execute('terminal:open', { name: model.name });
+/**
+ * Add the running kernel manager (notebooks & consoles) to the running panel.
+ */
+function addKernelRunningSessionManager(
+  managers: IRunningSessionManagers,
+  app: JupyterFrontEnd
+) {
+  let manager = app.serviceManager.sessions;
+  function filterSessions(m: Session.IModel) {
+    return !!(
+      (m.name || PathExt.basename(m.path)).indexOf('.') !== -1 || m.name
+    );
+  }
+
+  managers.add({
+    name: 'Kernel',
+    running: () => {
+      return toArray(manager.running())
+        .filter(filterSessions)
+        .map(model => new RunningKernel(model));
+    },
+    shutdownAll: () => manager.shutdownAll(),
+    refreshRunning: () => manager.refreshRunning(),
+    runningChanged: manager.runningChanged
   });
 
-  // Rank has been chosen somewhat arbitrarily to give priority to the running
-  // sessions widget in the sidebar.
-  app.shell.add(running, 'left', { rank: 200 });
+  class RunningKernel implements IRunningSessions.IRunningItem {
+    constructor(model: Session.IModel) {
+      this._model = model;
+    }
+    open() {
+      let { path, type } = this._model;
+      if (type.toLowerCase() === 'console') {
+        void app.commands.execute('console:open', { path });
+      } else {
+        void app.commands.execute('docmanager:open', { path });
+      }
+    }
+    shutdown() {
+      return manager.shutdown(this._model.id);
+    }
+    iconClass() {
+      let { name, path, type } = this._model;
+      if ((name || PathExt.basename(path)).indexOf('.ipynb') !== -1) {
+        return NOTEBOOK_ICON_CLASS;
+      } else if (type.toLowerCase() === 'console') {
+        return CONSOLE_ICON_CLASS;
+      }
+      return FILE_ICON_CLASS;
+    }
+    label() {
+      return this._model.name || PathExt.basename(this._model.path);
+    }
+    labelTitle() {
+      let { kernel, path } = this._model;
+      let kernelName = kernel.name;
+      if (manager.specs) {
+        const spec = manager.specs.kernelspecs[kernelName];
+        kernelName = spec ? spec.display_name : 'unknown';
+      }
+      return `Path: ${path}\nKernel: ${kernelName}`;
+    }
+
+    private _model: Session.IModel;
+  }
 }

packages/running-extension/tsconfig.json

@@ -9,8 +9,14 @@
     {
       "path": "../application"
     },
+    {
+      "path": "../coreutils"
+    },
     {
       "path": "../running"
+    },
+    {
+      "path": "../services"
     }
   ]
 }

packages/running/package.json

@@ -36,9 +36,8 @@
   },
   "dependencies": {
     "@jupyterlab/apputils": "^1.2.0-alpha.0",
-    "@jupyterlab/coreutils": "^3.2.0-alpha.0",
-    "@jupyterlab/services": "^4.2.0-alpha.0",
-    "@phosphor/algorithm": "^1.2.0",
+    "@phosphor/coreutils": "^1.3.1",
+    "@phosphor/disposable": "^1.3.0",
     "@phosphor/signaling": "^1.3.0",
     "react": "~16.8.4"
   },