Implement the changes necessary to use the new custom editor

.vscode/launch.json

@@ -25,7 +25,7 @@
                 // Enable this to turn on redux logging during debugging
                 "XVSC_PYTHON_FORCE_LOGGING": "1",
                 // Enable this to try out new experiments locally
-                "XVSC_PYTHON_LOAD_EXPERIMENTS_FROM_FILE": "1",
+                "VSC_PYTHON_LOAD_EXPERIMENTS_FROM_FILE": "1",
                 // Enable this to log telemetry to the output during debugging
                 "XVSC_PYTHON_LOG_TELEMETRY": "1",
                 // Enable this to log debugger output. Directory must exist ahead of time

customEditor.json

@@ -0,0 +1,19 @@
+{
+    "activationEvents": [
+        "onCustomEditor:ms-python.python.notebook.ipynb"        
+    ],
+    "contributes": {
+        "customEditors": [
+            {
+                "viewType": "ms-python.python.notebook.ipynb",
+                "displayName": "Jupyter Notebook",
+                "selector": [
+                    {
+                        "filenamePattern": "*.ipynb"
+                    }
+                ],
+                "priority": "default"
+            }
+        ]
+    }
+}

experiments.json

@@ -196,6 +196,18 @@
         "salt": "RunByLine",
         "max": 0,
         "min": 0
+    },
+    {
+        "name": "CustomEditorSupport - control",
+        "salt": "CustomEditorSupport",
+        "min": 0,
+        "max": 100
+    },
+    {
+        "name": "CustomEditorSupport - experiment",
+        "salt": "CustomEditorSupport",
+        "max": 0,
+        "min": 0
     }
 
 ]

news/1 Enhancements/10744.md

@@ -0,0 +1 @@
+Enable the use of the custom editor for native notebooks.

package.json

@@ -91,7 +91,7 @@
         "onCommand:python.datascience.selectJupyterInterpreter",
         "onCommand:python.datascience.selectjupytercommandline",
         "onCommand:python.enableSourceMapSupport",
-        "onCustomEditor:NativeEditorProvider.ipynb",
+
         "onNotebookEditor:jupyter-notebook",
         "workspaceContains:**/mspythonconfig.json"
     ],
@@ -3162,7 +3162,7 @@
         "@types/tmp": "0.0.33",
         "@types/untildify": "^3.0.0",
         "@types/uuid": "^3.4.3",
-        "@types/vscode": "^1.43.0",
+        "@types/vscode": "^1.45.0",
         "@types/webpack-bundle-analyzer": "^2.13.0",
         "@types/winreg": "^1.2.30",
         "@types/ws": "^6.0.1",

package.nls.json

@@ -510,5 +510,6 @@
     "DataScience.continueRunByLine": "Stop",
     "DataScience.couldNotInstallLibrary": "Could not install {0}. If pip is not available, please use the package manager of your choice to manually install this library into your Python environment.",
     "DataScience.rawKernelSessionFailed": "Unable to start session for kernel {0}. Select another kernel to launch with.",
-    "DataScience.rawKernelConnectingSession": "Connecting to kernel."
+    "DataScience.rawKernelConnectingSession": "Connecting to kernel.",
+    "DataScience.reloadCustomEditor": "Please reload VS Code to use the custom editor API"
 }

src/client/common/application/customEditorService.ts

@@ -2,23 +2,37 @@
 // Licensed under the MIT License.
 'use strict';
 import { inject, injectable } from 'inversify';
+import * as path from 'path';
 import * as vscode from 'vscode';
+import { DataScience } from '../../common/utils/localize';
 
-import { UseCustomEditorApi } from '../constants';
+import { traceError } from '../../logging';
+import { EXTENSION_ROOT_DIR, UseCustomEditorApi } from '../constants';
+import { IFileSystem } from '../platform/types';
 import { noop } from '../utils/misc';
-import { CustomEditorProvider, ICommandManager, ICustomEditorService } from './types';
+import { CustomEditorProvider, IApplicationEnvironment, ICommandManager, ICustomEditorService } from './types';
 
 @injectable()
 export class CustomEditorService implements ICustomEditorService {
     constructor(
         @inject(ICommandManager) private commandManager: ICommandManager,
-        @inject(UseCustomEditorApi) private readonly useCustomEditorApi: boolean
-    ) {}
+        @inject(UseCustomEditorApi) private readonly useCustomEditorApi: boolean,
+        @inject(IApplicationEnvironment) private readonly appEnvironment: IApplicationEnvironment,
+        @inject(IFileSystem) private readonly fileSystem: IFileSystem
+    ) {
+        // Double check the package json has the necessary entries for contributing a custom editor
+        if (this.useCustomEditorApi && !appEnvironment.packageJson.contributes?.customEditors) {
+            this.rewritePackageJson().catch((e) => traceError(`Error rewriting package json: `, e));
+        }
+    }
 
     public registerCustomEditorProvider(
         viewType: string,
         provider: CustomEditorProvider,
-        options?: vscode.WebviewPanelOptions
+        options?: {
+            readonly webviewOptions?: vscode.WebviewPanelOptions;
+            readonly supportsMultipleEditorsPerDocument?: boolean;
+        }
     ): vscode.Disposable {
         if (this.useCustomEditorApi) {
             // tslint:disable-next-line: no-any
@@ -33,4 +47,21 @@ export class CustomEditorService implements ICustomEditorService {
             await this.commandManager.executeCommand('vscode.openWith', file, viewType);
         }
     }
+
+    private async rewritePackageJson() {
+        // tslint:disable-next-line:no-require-imports no-var-requires
+        const _mergeWith = require('lodash/mergeWith') as typeof import('lodash/mergeWith');
+        const current = this.appEnvironment.packageJson;
+        const improvedContents = await this.fileSystem.readFile(path.join(EXTENSION_ROOT_DIR, 'customEditor.json'));
+        const improved = _mergeWith({ ...current }, JSON.parse(improvedContents), (l, r) => {
+            if (Array.isArray(l) && Array.isArray(r)) {
+                return [...l, ...r];
+            }
+        });
+        await this.fileSystem.writeFile(
+            path.join(EXTENSION_ROOT_DIR, 'package.json'),
+            JSON.stringify(improved, null, 4)
+        );
+        this.commandManager.executeCommand('python.reloadVSCode', DataScience.reloadCustomEditor());
+    }
 }

src/client/common/application/notebook.ts

@@ -14,7 +14,8 @@ import type {
     NotebookOutputSelector
 } from 'vscode-proposed';
 import { UseProposedApi } from '../constants';
-import { IDisposableRegistry } from '../types';
+import { NativeNotebook } from '../experiments/groups';
+import { IDisposableRegistry, IExperimentsManager } from '../types';
 import {
     IVSCodeNotebook,
     NotebookCellLanguageChangeEvent,
@@ -62,9 +63,10 @@ export class VSCodeNotebook implements IVSCodeNotebook {
     private readonly handledCellChanges = new WeakSet<VSCNotebookCellsChangeEvent>();
     constructor(
         @inject(UseProposedApi) private readonly useProposedApi: boolean,
-        @inject(IDisposableRegistry) private readonly disposables: IDisposableRegistry
+        @inject(IDisposableRegistry) private readonly disposables: IDisposableRegistry,
+        @inject(IExperimentsManager) readonly experimentManager: IExperimentsManager
     ) {
-        if (this.useProposedApi) {
+        if (this.useProposedApi && experimentManager.inExperiment(NativeNotebook.experiment)) {
             this.addEventHandlers();
         }
     }