better check for registering

Author: TylerLeonhardt

src/features/PowerShellNotebooks.ts

@@ -181,7 +181,7 @@ export class PowerShellNotebooksFeature implements vscode.NotebookContentProvide
 
     async _save(document: vscode.NotebookDocument, targetResource: vscode.Uri, _token: vscode.CancellationToken): Promise<void> {
         const retArr: string[] = [];
-        document.cells.forEach((cell, index) => {
+        document.cells.forEach((cell) => {
             if (cell.cellKind === vscode.CellKind.Code) {
                 retArr.push(...cell.document.getText().split(/\r|\n|\r\n/));
             } else {

src/main.ts

@@ -135,15 +135,6 @@ export function activate(context: vscode.ExtensionContext): void {
             PackageJSON.version,
             telemetryReporter);
 
-    const powerShellNotebooksFeature = new PowerShellNotebooksFeature();
-
-    try {
-        context.subscriptions.push(vscode.notebook.registerNotebookContentProvider('PowerShellNotebookMode', powerShellNotebooksFeature));
-    } catch (e) {
-        // This would happen in VS Code changes their API.
-        logger.writeVerbose("Failed to register NotebookContentProvider", e);
-    }
-
     // Create features
     extensionFeatures = [
         new ConsoleFeature(logger),
@@ -166,10 +157,23 @@ export function activate(context: vscode.ExtensionContext): void {
         new SpecifyScriptArgsFeature(context),
         new HelpCompletionFeature(logger),
         new CustomViewsFeature(),
-        new PickRunspaceFeature(),
-        powerShellNotebooksFeature
+        new PickRunspaceFeature()
     ];
 
+    // Notebook UI is only supported in VS Code Insiders.
+    if(vscode.env.uriScheme === "vscode-insiders") {
+        const powerShellNotebooksFeature = new PowerShellNotebooksFeature();
+
+        try {
+            context.subscriptions.push(vscode.notebook.registerNotebookContentProvider('PowerShellNotebookMode', powerShellNotebooksFeature));
+            extensionFeatures.push(powerShellNotebooksFeature);
+        } catch (e) {
+            // This would happen in VS Code changes their API.
+            powerShellNotebooksFeature.dispose();
+            logger.writeVerbose("Failed to register NotebookContentProvider", e);
+        }
+    }
+
     sessionManager.setExtensionFeatures(extensionFeatures);
 
     if (extensionSettings.startAutomatically) {