@theia/notebook 1.53.0-next.55 → 1.53.0-next.64

package/src/browser/service/notebook-context-manager.ts

@@ -1,162 +1,162 @@
-// *****************************************************************************
-// Copyright (C) 2024 TypeFox and others.
-//
-// This program and the accompanying materials are made available under the
-// terms of the Eclipse Public License v. 2.0 which is available at
-// http://www.eclipse.org/legal/epl-2.0.
-//
-// This Source Code may also be made available under the following Secondary
-// Licenses when the conditions for such availability set forth in the Eclipse
-// Public License v. 2.0 are satisfied: GNU General Public License, version 2
-// with the GNU Classpath Exception which is available at
-// https://www.gnu.org/software/classpath/license.html.
-//
-// SPDX-License-Identifier: EPL-2.0 OR GPL-2.0-only WITH Classpath-exception-2.0
-// *****************************************************************************
-
-import { inject, injectable } from '@theia/core/shared/inversify';
-import { ContextKeyChangeEvent, ContextKeyService, ContextMatcher, ScopedValueStore } from '@theia/core/lib/browser/context-key-service';
-import { DisposableCollection, Emitter } from '@theia/core';
-import { NotebookKernelService } from './notebook-kernel-service';
-import {
-    NOTEBOOK_CELL_EDITABLE,
-    NOTEBOOK_CELL_EXECUTING, NOTEBOOK_CELL_EXECUTION_STATE,
-    NOTEBOOK_CELL_FOCUSED, NOTEBOOK_CELL_LIST_FOCUSED, NOTEBOOK_CELL_MARKDOWN_EDIT_MODE,
-    NOTEBOOK_CELL_TYPE, NOTEBOOK_HAS_OUTPUTS, NOTEBOOK_KERNEL, NOTEBOOK_KERNEL_SELECTED,
-    NOTEBOOK_OUTPUT_INPUT_FOCUSED,
-    NOTEBOOK_VIEW_TYPE
-} from '../contributions/notebook-context-keys';
-import { NotebookEditorWidget } from '../notebook-editor-widget';
-import { NotebookCellModel } from '../view-model/notebook-cell-model';
-import { CellKind, NotebookCellsChangeType } from '../../common';
-import { NotebookExecutionStateService } from './notebook-execution-state-service';
-
-@injectable()
-export class NotebookContextManager {
-    @inject(ContextKeyService) protected contextKeyService: ContextKeyService;
-
-    @inject(NotebookKernelService)
-    protected readonly notebookKernelService: NotebookKernelService;
-
-    @inject(NotebookExecutionStateService)
-    protected readonly executionStateService: NotebookExecutionStateService;
-
-    protected readonly toDispose = new DisposableCollection();
-
-    protected readonly onDidChangeContextEmitter = new Emitter<ContextKeyChangeEvent>();
-    readonly onDidChangeContext = this.onDidChangeContextEmitter.event;
-
-    protected _context?: HTMLElement;
-
-    scopedStore: ScopedValueStore;
-
-    get context(): HTMLElement | undefined {
-        return this._context;
-    }
-
-    protected cellContexts: Map<number, Record<string, unknown>> = new Map();
-
-    init(widget: NotebookEditorWidget): void {
-        this._context = widget.node;
-        this.scopedStore = this.contextKeyService.createScoped(widget.node);
-
-        this.toDispose.dispose();
-
-        this.scopedStore.setContext(NOTEBOOK_VIEW_TYPE, widget?.notebookType);
-
-        // Kernel related keys
-        const kernel = widget?.model ? this.notebookKernelService.getSelectedNotebookKernel(widget.model) : undefined;
-        this.scopedStore.setContext(NOTEBOOK_KERNEL_SELECTED, !!kernel);
-        this.scopedStore.setContext(NOTEBOOK_KERNEL, kernel?.id);
-        this.toDispose.push(this.notebookKernelService.onDidChangeSelectedKernel(e => {
-            if (e.notebook.toString() === widget?.getResourceUri()?.toString()) {
-                this.scopedStore.setContext(NOTEBOOK_KERNEL_SELECTED, !!e.newKernel);
-                this.scopedStore.setContext(NOTEBOOK_KERNEL, e.newKernel);
-                this.onDidChangeContextEmitter.fire(this.createContextKeyChangedEvent([NOTEBOOK_KERNEL_SELECTED, NOTEBOOK_KERNEL]));
-            }
-        }));
-
-        widget.model?.onDidChangeContent(events => {
-            if (events.some(e => e.kind === NotebookCellsChangeType.ModelChange || e.kind === NotebookCellsChangeType.Output)) {
-                this.scopedStore.setContext(NOTEBOOK_HAS_OUTPUTS, widget.model?.cells.some(cell => cell.outputs.length > 0));
-                this.onDidChangeContextEmitter.fire(this.createContextKeyChangedEvent([NOTEBOOK_HAS_OUTPUTS]));
-            }
-        });
-
-        this.scopedStore.setContext(NOTEBOOK_HAS_OUTPUTS, !!widget.model?.cells.find(cell => cell.outputs.length > 0));
-
-        // Cell Selection related keys
-        this.scopedStore.setContext(NOTEBOOK_CELL_FOCUSED, !!widget.model?.selectedCell);
-        widget.model?.onDidChangeSelectedCell(e => {
-            this.selectedCellChanged(e.cell);
-            this.scopedStore.setContext(NOTEBOOK_CELL_FOCUSED, !!e);
-            this.onDidChangeContextEmitter.fire(this.createContextKeyChangedEvent([NOTEBOOK_CELL_FOCUSED]));
-        });
-
-        this.toDispose.push(this.executionStateService.onDidChangeExecution(e => {
-            if (e.notebook.toString() === widget.model?.uri.toString()) {
-                this.setCellContext(e.cellHandle, NOTEBOOK_CELL_EXECUTING, !!e.changed);
-                this.setCellContext(e.cellHandle, NOTEBOOK_CELL_EXECUTION_STATE, e.changed?.state ?? 'idle');
-                this.onDidChangeContextEmitter.fire(this.createContextKeyChangedEvent([NOTEBOOK_CELL_EXECUTING, NOTEBOOK_CELL_EXECUTION_STATE]));
-            }
-        }));
-
-        widget.onDidChangeOutputInputFocus(focus => {
-            this.scopedStore.setContext(NOTEBOOK_OUTPUT_INPUT_FOCUSED, focus);
-            this.onDidChangeContextEmitter.fire(this.createContextKeyChangedEvent([NOTEBOOK_OUTPUT_INPUT_FOCUSED]));
-        });
-
-        this.onDidChangeContextEmitter.fire(this.createContextKeyChangedEvent([NOTEBOOK_VIEW_TYPE, NOTEBOOK_KERNEL_SELECTED, NOTEBOOK_KERNEL]));
-    }
-
-    protected cellDisposables = new DisposableCollection();
-
-    selectedCellChanged(cell: NotebookCellModel | undefined): void {
-        this.cellDisposables.dispose();
-
-        this.scopedStore.setContext(NOTEBOOK_CELL_TYPE, cell ? cell.cellKind === CellKind.Code ? 'code' : 'markdown' : undefined);
-
-        if (cell) {
-            this.scopedStore.setContext(NOTEBOOK_CELL_MARKDOWN_EDIT_MODE, cell.editing);
-            this.scopedStore.setContext(NOTEBOOK_CELL_EDITABLE, cell.cellKind === CellKind.Markup && !cell.editing);
-            this.cellDisposables.push(cell.onDidRequestCellEditChange(cellEdit => {
-                this.scopedStore.setContext(NOTEBOOK_CELL_MARKDOWN_EDIT_MODE, cellEdit);
-                this.scopedStore.setContext(NOTEBOOK_CELL_EDITABLE, cell.cellKind === CellKind.Markup && !cellEdit);
-                this.onDidChangeContextEmitter.fire(this.createContextKeyChangedEvent([NOTEBOOK_CELL_MARKDOWN_EDIT_MODE]));
-            }));
-        }
-
-        this.onDidChangeContextEmitter.fire(this.createContextKeyChangedEvent([NOTEBOOK_CELL_TYPE]));
-
-    }
-
-    protected setCellContext(cellHandle: number, key: string, value: unknown): void {
-        let cellContext = this.cellContexts.get(cellHandle);
-        if (!cellContext) {
-            cellContext = {};
-            this.cellContexts.set(cellHandle, cellContext);
-        }
-
-        cellContext[key] = value;
-    }
-
-    getCellContext(cellHandle: number): ContextMatcher {
-        return this.contextKeyService.createOverlay(Object.entries(this.cellContexts.get(cellHandle) ?? {}));
-    }
-
-    changeCellFocus(focus: boolean): void {
-        this.scopedStore.setContext(NOTEBOOK_CELL_FOCUSED, focus);
-    }
-
-    changeCellListFocus(focus: boolean): void {
-        this.scopedStore.setContext(NOTEBOOK_CELL_LIST_FOCUSED, focus);
-    }
-
-    createContextKeyChangedEvent(affectedKeys: string[]): ContextKeyChangeEvent {
-        return { affects: keys => affectedKeys.some(key => keys.has(key)) };
-    }
-
-    dispose(): void {
-        this.toDispose.dispose();
-    }
-}
+// *****************************************************************************
+// Copyright (C) 2024 TypeFox and others.
+//
+// This program and the accompanying materials are made available under the
+// terms of the Eclipse Public License v. 2.0 which is available at
+// http://www.eclipse.org/legal/epl-2.0.
+//
+// This Source Code may also be made available under the following Secondary
+// Licenses when the conditions for such availability set forth in the Eclipse
+// Public License v. 2.0 are satisfied: GNU General Public License, version 2
+// with the GNU Classpath Exception which is available at
+// https://www.gnu.org/software/classpath/license.html.
+//
+// SPDX-License-Identifier: EPL-2.0 OR GPL-2.0-only WITH Classpath-exception-2.0
+// *****************************************************************************
+
+import { inject, injectable } from '@theia/core/shared/inversify';
+import { ContextKeyChangeEvent, ContextKeyService, ContextMatcher, ScopedValueStore } from '@theia/core/lib/browser/context-key-service';
+import { DisposableCollection, Emitter } from '@theia/core';
+import { NotebookKernelService } from './notebook-kernel-service';
+import {
+    NOTEBOOK_CELL_EDITABLE,
+    NOTEBOOK_CELL_EXECUTING, NOTEBOOK_CELL_EXECUTION_STATE,
+    NOTEBOOK_CELL_FOCUSED, NOTEBOOK_CELL_LIST_FOCUSED, NOTEBOOK_CELL_MARKDOWN_EDIT_MODE,
+    NOTEBOOK_CELL_TYPE, NOTEBOOK_HAS_OUTPUTS, NOTEBOOK_KERNEL, NOTEBOOK_KERNEL_SELECTED,
+    NOTEBOOK_OUTPUT_INPUT_FOCUSED,
+    NOTEBOOK_VIEW_TYPE
+} from '../contributions/notebook-context-keys';
+import { NotebookEditorWidget } from '../notebook-editor-widget';
+import { NotebookCellModel } from '../view-model/notebook-cell-model';
+import { CellKind, NotebookCellsChangeType } from '../../common';
+import { NotebookExecutionStateService } from './notebook-execution-state-service';
+
+@injectable()
+export class NotebookContextManager {
+    @inject(ContextKeyService) protected contextKeyService: ContextKeyService;
+
+    @inject(NotebookKernelService)
+    protected readonly notebookKernelService: NotebookKernelService;
+
+    @inject(NotebookExecutionStateService)
+    protected readonly executionStateService: NotebookExecutionStateService;
+
+    protected readonly toDispose = new DisposableCollection();
+
+    protected readonly onDidChangeContextEmitter = new Emitter<ContextKeyChangeEvent>();
+    readonly onDidChangeContext = this.onDidChangeContextEmitter.event;
+
+    protected _context?: HTMLElement;
+
+    scopedStore: ScopedValueStore;
+
+    get context(): HTMLElement | undefined {
+        return this._context;
+    }
+
+    protected cellContexts: Map<number, Record<string, unknown>> = new Map();
+
+    init(widget: NotebookEditorWidget): void {
+        this._context = widget.node;
+        this.scopedStore = this.contextKeyService.createScoped(widget.node);
+
+        this.toDispose.dispose();
+
+        this.scopedStore.setContext(NOTEBOOK_VIEW_TYPE, widget?.notebookType);
+
+        // Kernel related keys
+        const kernel = widget?.model ? this.notebookKernelService.getSelectedNotebookKernel(widget.model) : undefined;
+        this.scopedStore.setContext(NOTEBOOK_KERNEL_SELECTED, !!kernel);
+        this.scopedStore.setContext(NOTEBOOK_KERNEL, kernel?.id);
+        this.toDispose.push(this.notebookKernelService.onDidChangeSelectedKernel(e => {
+            if (e.notebook.toString() === widget?.getResourceUri()?.toString()) {
+                this.scopedStore.setContext(NOTEBOOK_KERNEL_SELECTED, !!e.newKernel);
+                this.scopedStore.setContext(NOTEBOOK_KERNEL, e.newKernel);
+                this.onDidChangeContextEmitter.fire(this.createContextKeyChangedEvent([NOTEBOOK_KERNEL_SELECTED, NOTEBOOK_KERNEL]));
+            }
+        }));
+
+        widget.model?.onDidChangeContent(events => {
+            if (events.some(e => e.kind === NotebookCellsChangeType.ModelChange || e.kind === NotebookCellsChangeType.Output)) {
+                this.scopedStore.setContext(NOTEBOOK_HAS_OUTPUTS, widget.model?.cells.some(cell => cell.outputs.length > 0));
+                this.onDidChangeContextEmitter.fire(this.createContextKeyChangedEvent([NOTEBOOK_HAS_OUTPUTS]));
+            }
+        });
+
+        this.scopedStore.setContext(NOTEBOOK_HAS_OUTPUTS, !!widget.model?.cells.find(cell => cell.outputs.length > 0));
+
+        // Cell Selection related keys
+        this.scopedStore.setContext(NOTEBOOK_CELL_FOCUSED, !!widget.model?.selectedCell);
+        widget.model?.onDidChangeSelectedCell(e => {
+            this.selectedCellChanged(e.cell);
+            this.scopedStore.setContext(NOTEBOOK_CELL_FOCUSED, !!e);
+            this.onDidChangeContextEmitter.fire(this.createContextKeyChangedEvent([NOTEBOOK_CELL_FOCUSED]));
+        });
+
+        this.toDispose.push(this.executionStateService.onDidChangeExecution(e => {
+            if (e.notebook.toString() === widget.model?.uri.toString()) {
+                this.setCellContext(e.cellHandle, NOTEBOOK_CELL_EXECUTING, !!e.changed);
+                this.setCellContext(e.cellHandle, NOTEBOOK_CELL_EXECUTION_STATE, e.changed?.state ?? 'idle');
+                this.onDidChangeContextEmitter.fire(this.createContextKeyChangedEvent([NOTEBOOK_CELL_EXECUTING, NOTEBOOK_CELL_EXECUTION_STATE]));
+            }
+        }));
+
+        widget.onDidChangeOutputInputFocus(focus => {
+            this.scopedStore.setContext(NOTEBOOK_OUTPUT_INPUT_FOCUSED, focus);
+            this.onDidChangeContextEmitter.fire(this.createContextKeyChangedEvent([NOTEBOOK_OUTPUT_INPUT_FOCUSED]));
+        });
+
+        this.onDidChangeContextEmitter.fire(this.createContextKeyChangedEvent([NOTEBOOK_VIEW_TYPE, NOTEBOOK_KERNEL_SELECTED, NOTEBOOK_KERNEL]));
+    }
+
+    protected cellDisposables = new DisposableCollection();
+
+    selectedCellChanged(cell: NotebookCellModel | undefined): void {
+        this.cellDisposables.dispose();
+
+        this.scopedStore.setContext(NOTEBOOK_CELL_TYPE, cell ? cell.cellKind === CellKind.Code ? 'code' : 'markdown' : undefined);
+
+        if (cell) {
+            this.scopedStore.setContext(NOTEBOOK_CELL_MARKDOWN_EDIT_MODE, cell.editing);
+            this.scopedStore.setContext(NOTEBOOK_CELL_EDITABLE, cell.cellKind === CellKind.Markup && !cell.editing);
+            this.cellDisposables.push(cell.onDidRequestCellEditChange(cellEdit => {
+                this.scopedStore.setContext(NOTEBOOK_CELL_MARKDOWN_EDIT_MODE, cellEdit);
+                this.scopedStore.setContext(NOTEBOOK_CELL_EDITABLE, cell.cellKind === CellKind.Markup && !cellEdit);
+                this.onDidChangeContextEmitter.fire(this.createContextKeyChangedEvent([NOTEBOOK_CELL_MARKDOWN_EDIT_MODE]));
+            }));
+        }
+
+        this.onDidChangeContextEmitter.fire(this.createContextKeyChangedEvent([NOTEBOOK_CELL_TYPE]));
+
+    }
+
+    protected setCellContext(cellHandle: number, key: string, value: unknown): void {
+        let cellContext = this.cellContexts.get(cellHandle);
+        if (!cellContext) {
+            cellContext = {};
+            this.cellContexts.set(cellHandle, cellContext);
+        }
+
+        cellContext[key] = value;
+    }
+
+    getCellContext(cellHandle: number): ContextMatcher {
+        return this.contextKeyService.createOverlay(Object.entries(this.cellContexts.get(cellHandle) ?? {}));
+    }
+
+    changeCellFocus(focus: boolean): void {
+        this.scopedStore.setContext(NOTEBOOK_CELL_FOCUSED, focus);
+    }
+
+    changeCellListFocus(focus: boolean): void {
+        this.scopedStore.setContext(NOTEBOOK_CELL_LIST_FOCUSED, focus);
+    }
+
+    createContextKeyChangedEvent(affectedKeys: string[]): ContextKeyChangeEvent {
+        return { affects: keys => affectedKeys.some(key => keys.has(key)) };
+    }
+
+    dispose(): void {
+        this.toDispose.dispose();
+    }
+}

package/src/browser/service/notebook-editor-widget-service.ts

@@ -1,101 +1,101 @@
-// *****************************************************************************
-// Copyright (C) 2023 TypeFox and others.
-//
-// This program and the accompanying materials are made available under the
-// terms of the Eclipse Public License v. 2.0 which is available at
-// http://www.eclipse.org/legal/epl-2.0.
-//
-// This Source Code may also be made available under the following Secondary
-// Licenses when the conditions for such availability set forth in the Eclipse
-// Public License v. 2.0 are satisfied: GNU General Public License, version 2
-// with the GNU Classpath Exception which is available at
-// https://www.gnu.org/software/classpath/license.html.
-//
-// SPDX-License-Identifier: EPL-2.0 OR GPL-2.0-only WITH Classpath-exception-2.0
-// *****************************************************************************
-
-/*---------------------------------------------------------------------------------------------
- *  Copyright (c) Microsoft Corporation. All rights reserved.
- *  Licensed under the MIT License. See License.txt in the project root for license information.
- *--------------------------------------------------------------------------------------------*/
-
-import { Emitter } from '@theia/core';
-import { inject, injectable, postConstruct } from '@theia/core/shared/inversify';
-import { ApplicationShell } from '@theia/core/lib/browser';
-import { NotebookEditorWidget } from '../notebook-editor-widget';
-import { ContextKeyService } from '@theia/core/lib/browser/context-key-service';
-import { NOTEBOOK_EDITOR_FOCUSED } from '../contributions/notebook-context-keys';
-
-@injectable()
-export class NotebookEditorWidgetService {
-
-    @inject(ApplicationShell)
-    protected applicationShell: ApplicationShell;
-
-    @inject(ContextKeyService)
-    protected contextKeyService: ContextKeyService;
-
-    protected readonly notebookEditors = new Map<string, NotebookEditorWidget>();
-
-    protected readonly onNotebookEditorAddEmitter = new Emitter<NotebookEditorWidget>();
-    protected readonly onNotebookEditorRemoveEmitter = new Emitter<NotebookEditorWidget>();
-    readonly onDidAddNotebookEditor = this.onNotebookEditorAddEmitter.event;
-    readonly onDidRemoveNotebookEditor = this.onNotebookEditorRemoveEmitter.event;
-
-    protected readonly onDidChangeFocusedEditorEmitter = new Emitter<NotebookEditorWidget | undefined>();
-    readonly onDidChangeFocusedEditor = this.onDidChangeFocusedEditorEmitter.event;
-
-    focusedEditor?: NotebookEditorWidget = undefined;
-
-    @postConstruct()
-    protected init(): void {
-        this.applicationShell.onDidChangeActiveWidget(event => {
-            this.notebookEditorFocusChanged(event.newValue as NotebookEditorWidget, event.newValue instanceof NotebookEditorWidget);
-        });
-    }
-
-    // --- editor management
-
-    addNotebookEditor(editor: NotebookEditorWidget): void {
-        if (this.notebookEditors.has(editor.id)) {
-            console.warn('Attempting to add duplicated notebook editor: ' + editor.id);
-        }
-        this.notebookEditors.set(editor.id, editor);
-        this.onNotebookEditorAddEmitter.fire(editor);
-        if (editor.isVisible) {
-            this.notebookEditorFocusChanged(editor, true);
-        }
-    }
-
-    removeNotebookEditor(editor: NotebookEditorWidget): void {
-        if (this.notebookEditors.has(editor.id)) {
-            this.notebookEditors.delete(editor.id);
-            this.onNotebookEditorRemoveEmitter.fire(editor);
-        } else {
-            console.warn('Attempting to remove not registered editor: ' + editor.id);
-        }
-    }
-
-    getNotebookEditor(editorId: string): NotebookEditorWidget | undefined {
-        return this.notebookEditors.get(editorId);
-    }
-
-    getNotebookEditors(): readonly NotebookEditorWidget[] {
-        return Array.from(this.notebookEditors.values());
-    }
-
-    notebookEditorFocusChanged(editor: NotebookEditorWidget, focus: boolean): void {
-        if (focus) {
-            if (editor !== this.focusedEditor) {
-                this.focusedEditor = editor;
-                this.contextKeyService.setContext(NOTEBOOK_EDITOR_FOCUSED, true);
-                this.onDidChangeFocusedEditorEmitter.fire(this.focusedEditor);
-            }
-        } else if (this.focusedEditor) {
-            this.focusedEditor = undefined;
-            this.contextKeyService.setContext(NOTEBOOK_EDITOR_FOCUSED, false);
-            this.onDidChangeFocusedEditorEmitter.fire(undefined);
-        }
-    }
-
-}
+// *****************************************************************************
+// Copyright (C) 2023 TypeFox and others.
+//
+// This program and the accompanying materials are made available under the
+// terms of the Eclipse Public License v. 2.0 which is available at
+// http://www.eclipse.org/legal/epl-2.0.
+//
+// This Source Code may also be made available under the following Secondary
+// Licenses when the conditions for such availability set forth in the Eclipse
+// Public License v. 2.0 are satisfied: GNU General Public License, version 2
+// with the GNU Classpath Exception which is available at
+// https://www.gnu.org/software/classpath/license.html.
+//
+// SPDX-License-Identifier: EPL-2.0 OR GPL-2.0-only WITH Classpath-exception-2.0
+// *****************************************************************************
+
+/*---------------------------------------------------------------------------------------------
+ *  Copyright (c) Microsoft Corporation. All rights reserved.
+ *  Licensed under the MIT License. See License.txt in the project root for license information.
+ *--------------------------------------------------------------------------------------------*/
+
+import { Emitter } from '@theia/core';
+import { inject, injectable, postConstruct } from '@theia/core/shared/inversify';
+import { ApplicationShell } from '@theia/core/lib/browser';
+import { NotebookEditorWidget } from '../notebook-editor-widget';
+import { ContextKeyService } from '@theia/core/lib/browser/context-key-service';
+import { NOTEBOOK_EDITOR_FOCUSED } from '../contributions/notebook-context-keys';
+
+@injectable()
+export class NotebookEditorWidgetService {
+
+    @inject(ApplicationShell)
+    protected applicationShell: ApplicationShell;
+
+    @inject(ContextKeyService)
+    protected contextKeyService: ContextKeyService;
+
+    protected readonly notebookEditors = new Map<string, NotebookEditorWidget>();
+
+    protected readonly onNotebookEditorAddEmitter = new Emitter<NotebookEditorWidget>();
+    protected readonly onNotebookEditorRemoveEmitter = new Emitter<NotebookEditorWidget>();
+    readonly onDidAddNotebookEditor = this.onNotebookEditorAddEmitter.event;
+    readonly onDidRemoveNotebookEditor = this.onNotebookEditorRemoveEmitter.event;
+
+    protected readonly onDidChangeFocusedEditorEmitter = new Emitter<NotebookEditorWidget | undefined>();
+    readonly onDidChangeFocusedEditor = this.onDidChangeFocusedEditorEmitter.event;
+
+    focusedEditor?: NotebookEditorWidget = undefined;
+
+    @postConstruct()
+    protected init(): void {
+        this.applicationShell.onDidChangeActiveWidget(event => {
+            this.notebookEditorFocusChanged(event.newValue as NotebookEditorWidget, event.newValue instanceof NotebookEditorWidget);
+        });
+    }
+
+    // --- editor management
+
+    addNotebookEditor(editor: NotebookEditorWidget): void {
+        if (this.notebookEditors.has(editor.id)) {
+            console.warn('Attempting to add duplicated notebook editor: ' + editor.id);
+        }
+        this.notebookEditors.set(editor.id, editor);
+        this.onNotebookEditorAddEmitter.fire(editor);
+        if (editor.isVisible) {
+            this.notebookEditorFocusChanged(editor, true);
+        }
+    }
+
+    removeNotebookEditor(editor: NotebookEditorWidget): void {
+        if (this.notebookEditors.has(editor.id)) {
+            this.notebookEditors.delete(editor.id);
+            this.onNotebookEditorRemoveEmitter.fire(editor);
+        } else {
+            console.warn('Attempting to remove not registered editor: ' + editor.id);
+        }
+    }
+
+    getNotebookEditor(editorId: string): NotebookEditorWidget | undefined {
+        return this.notebookEditors.get(editorId);
+    }
+
+    getNotebookEditors(): readonly NotebookEditorWidget[] {
+        return Array.from(this.notebookEditors.values());
+    }
+
+    notebookEditorFocusChanged(editor: NotebookEditorWidget, focus: boolean): void {
+        if (focus) {
+            if (editor !== this.focusedEditor) {
+                this.focusedEditor = editor;
+                this.contextKeyService.setContext(NOTEBOOK_EDITOR_FOCUSED, true);
+                this.onDidChangeFocusedEditorEmitter.fire(this.focusedEditor);
+            }
+        } else if (this.focusedEditor) {
+            this.focusedEditor = undefined;
+            this.contextKeyService.setContext(NOTEBOOK_EDITOR_FOCUSED, false);
+            this.onDidChangeFocusedEditorEmitter.fire(undefined);
+        }
+    }
+
+}