plugin.ts

import {
  ILabShell,
  JupyterFrontEnd,
  JupyterFrontEndPlugin
} from '@jupyterlab/application';
import {
  IThemeManager,
  IWidgetTracker,
  WidgetTracker
} from '@jupyterlab/apputils';
import { IEditorServices } from '@jupyterlab/codeeditor';
import { PathExt } from '@jupyterlab/coreutils';
import { IDocumentManager } from '@jupyterlab/docmanager';
import { IDefaultFileBrowser } from '@jupyterlab/filebrowser';
import { INotebookTracker, NotebookPanel } from '@jupyterlab/notebook';
import { IRenderMimeRegistry } from '@jupyterlab/rendermime';
import { IKernelSpecManager } from '@jupyterlab/services';
import { Widget } from '@lumino/widgets';

import {
  ISpectaDocTracker,
  ISpectaLayoutRegistry,
  ISpectaShell
} from '../token';
import {
  configLabLayout,
  createFileBrowser,
  hideAppLoadingIndicator,
  isSpectaApp,
  readSpectaConfig,
  registerDocumentFactory
} from '../tool';

const activate = (
  app: JupyterFrontEnd<ISpectaShell>,
  rendermime: IRenderMimeRegistry,
  tracker: INotebookTracker,
  editorServices: IEditorServices,
  contentFactory: NotebookPanel.IContentFactory,
  spectaLayoutRegistry: ISpectaLayoutRegistry,
  themeManager: IThemeManager
): IWidgetTracker => {
  const namespace = 'specta';
  const spectaTracker = new WidgetTracker<Widget>({ namespace });

  registerDocumentFactory({
    factoryName: 'specta',
    app,
    rendermime,
    tracker,
    editorServices,
    contentFactory,
    spectaTracker,
    spectaLayoutRegistry,
    themeManager
  });

  return spectaTracker;
};

export const spectaDocument: JupyterFrontEndPlugin<
  IWidgetTracker,
  ISpectaShell
> = {
  id: 'specta:notebook-doc',
  autoStart: true,
  requires: [
    IRenderMimeRegistry,
    INotebookTracker,
    IEditorServices,
    NotebookPanel.IContentFactory,
    ISpectaLayoutRegistry,
    IThemeManager
  ],
  activate,
  provides: ISpectaDocTracker
};

export const spectaOpener: JupyterFrontEndPlugin<void, ILabShell> = {
  id: 'specta/application-extension:opener',
  autoStart: true,
  requires: [
    IDocumentManager,
    IDefaultFileBrowser,
    ISpectaDocTracker,
    IKernelSpecManager
  ],
  activate: async (
    app: JupyterFrontEnd<ILabShell>,
    docManager: IDocumentManager,
    defaultBrowser: IDefaultFileBrowser
  ): Promise<void> => {
    const urlParams = new URLSearchParams(window.location.search);
    if (!isSpectaApp()) {
      // Not a specta app
      const path = urlParams.get('specta-path');

      if (!path) {
        return;
      }
      app.restored.then(async () => {
        const labShell = app.shell;

        if (PathExt.extname(path) === '.ipynb') {
          const commands = app.commands;
          const spectaConfig = readSpectaConfig({});
          console.log('spectaConfig', spectaConfig);
          await configLabLayout({
            config: spectaConfig.labConfig,
            labShell,
            commands
          });
          const widget = docManager.openOrReveal(path, 'specta');
          if (widget) {
            app.shell.add(widget, 'main');
          }
        }
      });
      return;
    } else {
      //  Specta app
      const path = urlParams.get('path');
      if (!path) {
        const browser = createFileBrowser({ defaultBrowser });
        app.shell.add(browser, 'main', { rank: 100 });
        hideAppLoadingIndicator();
      } else {
        if (PathExt.extname(path) === '.ipynb') {
          app.shell.addClass('specta-document-viewer');
          const widget = docManager.openOrReveal(path, 'specta');
          if (widget) {
            app.shell.add(widget, 'main');
          }
        } else {
          let count = 0;
          const tryOpen = () => {
            const widget = docManager.openOrReveal(path, 'default');
            if (widget) {
              app.shell.add(widget, 'main');
              hideAppLoadingIndicator();
            } else {
              count++;
              if (count > 10) {
                console.error('Failed to open file', path);
                const widget = new Widget();
                widget.node.innerHTML = `<h2 style="text-align: center; margin-top: 200px;">Failed to open file ${path}</h2>`;
                app.shell.add(widget, 'main');
                hideAppLoadingIndicator();
                return;
              }
              setTimeout(tryOpen, 100);
            }
          };
          tryOpen();
        }
      }
    }
  }
};