jupyterlab_sparkmonitor/js/index.js

Jupyter Lab extension to monitor Apache Spark Jobs

/**
 * Entrypoint module for the SparkMonitor frontend extension.
 *
 * @module module
 */
import 'jquery-ui-bundle';
import 'jquery-ui-bundle/jquery-ui.css';
import { INotebookTracker } from '@jupyterlab/notebook';
import { IMainMenu } from '@jupyterlab/mainmenu';
import { Menu } from '@phosphor/widgets';
import SparkMonitor from './SparkMonitor';

/** Entrypoint: Called when the extension is loaded by jupyter. */
const extension = {
    id: 'jupyterlab_sparkmonitor',
    autoStart: true,
    requires: [INotebookTracker, IMainMenu],
    activate(app, notebooks, mainMenu) {
        let monitor;
        console.log('JupyterLab SparkMonitor is activated!');
        notebooks.widgetAdded.connect(async (sender, nbPanel) => {
            console.log('Notebook added!');

            // JupyterLab 1.0 backwards compatibility
            let kernel;
            let info;
            if (nbPanel.session) {
                await nbPanel.session.ready;
                kernel = nbPanel.session.kernel;
                await kernel.ready;
                info = kernel.info;
            } else {
                // JupyterLab 2.0
                const { sessionContext } = nbPanel;
                await sessionContext.ready;
                kernel = sessionContext.session.kernel;
                info = await kernel.info;
            }

            if (info.language_info.name === 'python') {
                monitor = new SparkMonitor(nbPanel);
                console.log('Notebook kernel ready');
                monitor.startComm(kernel);
            }
        });

        const commandID = 'toggle-monitor';
        let toggled = false;
        let toggleLabel = "Toggle"

        app.commands.addCommand(commandID, {
            label: toggleLabel + ' Monitors',
            isEnabled: () => true,
            isVisible: () => true,
            isToggled: () => toggled,
            execute: () => {
                console.log(`Executed ${commandID}`);
                toggled = !toggled;
                monitor.toggleAll();
            },
        });

        const menu = new Menu({ commands: app.commands });
        menu.title.label = 'SparkMonitor';
        menu.addItem({
            command: commandID,
            args: {},
        });

        mainMenu.addMenu(menu, { rank: 40 });
    },
};

export default extension;