Install dependencies

Author: DonJayamanne

src/extension.ts

@@ -1,11 +1,10 @@
-import {
-  commands,
-  ExtensionContext,
-  OutputChannel,
-  window,
-} from "vscode";
+import { commands, ExtensionContext, OutputChannel, window } from "vscode";
 import { installTslab } from "./installer";
-import { optIntoNativeNotebooks, registerWithJupyter } from "./jupyterExtension";
+import {
+  configureEditor,
+  optIntoNativeNotebooks,
+  registerWithJupyter,
+} from "./jupyterExtension";
 import { installKernelSpec } from "./kernel";
 import { logError, setOutputWindow, showLog } from "./logger";
 import { noop } from "./utils";
@@ -15,11 +14,13 @@ export async function activate(context: ExtensionContext) {
   setOutputWindow(outputChannel);
   // Install kernel (silently) as soon as extension activates.
   Promise.all([
-	  installTslab(outputChannel),
-	  installKernelSpec(),
-	  registerWithJupyter(),
-	  optIntoNativeNotebooks(),
-	  installKernelSpec()]).catch(noop);
+    installTslab(outputChannel),
+    installKernelSpec(),
+    registerWithJupyter(),
+    optIntoNativeNotebooks(),
+    configureEditor(),
+    installKernelSpec(),
+  ]).catch(noop);
   registerCommands(context, outputChannel);
 }
 

src/jupyterExtension.ts

@@ -3,30 +3,65 @@ import { logMessage } from "./logger";
 import { IJupyterExtensionApi } from "./types";
 
 export async function registerWithJupyter() {
-	const jupyter = extensions.getExtension<IJupyterExtensionApi>(
-	  "ms-toolsai.jupyter"
-	);
-	if (!jupyter) {
-	  return;
-	}
-	if (!jupyter.isActive) {
-	  await jupyter.activate();
-	}
-	if (jupyter.exports.registerNewNotebookContent) {
-	  jupyter.exports.registerNewNotebookContent({
-		defaultCellLanguage: "typescript",
-	  });
-	}
+  const jupyter = extensions.getExtension<IJupyterExtensionApi>(
+    "ms-toolsai.jupyter"
+  );
+  if (!jupyter) {
+    return;
   }
+  if (!jupyter.isActive) {
+    await jupyter.activate();
+  }
+  if (jupyter.exports.registerNewNotebookContent) {
+    jupyter.exports.registerNewNotebookContent({
+      defaultCellLanguage: "typescript",
+    });
+  }
+}
+
+export async function optIntoNativeNotebooks() {
+  const settings = workspace.getConfiguration("jupyter", undefined);
+  const optInto = settings.get<string[]>("experiments.optInto");
+  if (
+    !Array.isArray(optInto) ||
+    optInto.includes("All") ||
+    optInto.includes("__NativeNotebookEditor__")
+  ) {
+    logMessage("Native Notebook already setup");
+    return;
+  }
+  optInto.push("__NativeNotebookEditor__");
+  logMessage("Setting up Native Notebooks");
+  await settings.update(
+    "experiments.optInto",
+    optInto,
+    ConfigurationTarget.Global
+  );
+}
 
-export async function optIntoNativeNotebooks(){
-	const settings = workspace.getConfiguration('jupyter', undefined);
-	const optInto = settings.get<string[]>('experiments.optInto');
-	if (!Array.isArray(optInto) || optInto.includes('All') || optInto.includes('__NativeNotebookEditor__')){
-		logMessage('Native Notebook already setup');
-		return;
-	}
-	optInto.push('__NativeNotebookEditor__');
-	logMessage('Setting up Native Notebooks');
-	await settings.update('experiments.optInto', optInto, ConfigurationTarget.Global);
+type EditorAssociation = {
+  viewType: string;
+  filenamePattern: string;
+};
+
+export async function configureEditor() {
+  const settings = workspace.getConfiguration("workbench", undefined);
+  const associations = settings.get<EditorAssociation[]>("editorAssociations");
+  if (
+    !Array.isArray(associations) ||
+    associations.find((item) => item.viewType === "jupyter-notebook")
+  ) {
+    logMessage("Native Notebook Editor already setup");
+    return;
+  }
+  associations.push({
+    viewType: "jupyter-notebook",
+    filenamePattern: "*.ipynb",
+  });
+  logMessage("Setting up Native Notebook Editor");
+  await settings.update(
+    "editorAssociations",
+    associations,
+    ConfigurationTarget.Global
+  );
 }