pass through param to set focus on the REPL editor

Author: amunger

src/client/repl/nativeRepl.ts

@@ -145,12 +145,12 @@ export class NativeRepl implements Disposable {
     /**
      * Function that opens interactive repl, selects kernel, and send/execute code to the native repl.
      */
-    public async sendToNativeRepl(code?: string): Promise<void> {
-        const notebookEditor = await openInteractiveREPL(this.replController, this.notebookDocument);
+    public async sendToNativeRepl(code?: string | undefined, preserveFocus: boolean = true): Promise<void> {
+        const notebookEditor = await openInteractiveREPL(this.replController, this.notebookDocument, preserveFocus);
         this.notebookDocument = notebookEditor.notebook;
 
         if (this.notebookDocument) {
-            this.replController.updateNotebookAffinity(this.notebookDocument, NotebookControllerAffinity.Default);
+            this.replController.updateNotebookAffinity(this.notebookDocument, NotebookControllerAffinity.Preferred);
             await selectNotebookKernel(notebookEditor, this.replController.id, PVSC_EXTENSION_ID);
             if (code) {
                 await executeNotebookCell(notebookEditor, code);

src/client/repl/replCommandHandler.ts

@@ -20,6 +20,7 @@ import { PVSC_EXTENSION_ID } from '../common/constants';
 export async function openInteractiveREPL(
     notebookController: NotebookController,
     notebookDocument: NotebookDocument | undefined,
+    preserveFocus: boolean = true,
 ): Promise<NotebookEditor> {
     let viewColumn = ViewColumn.Beside;
 
@@ -34,7 +35,7 @@ export async function openInteractiveREPL(
     const editor = window.showNotebookDocument(notebookDocument!, {
         viewColumn,
         asRepl: 'Python REPL',
-        preserveFocus: true,
+        preserveFocus,
     });
     await commands.executeCommand('notebook.selectKernel', {
         editor,

src/client/repl/replCommands.ts

@@ -31,10 +31,8 @@ export async function registerStartNativeReplCommand(
             sendTelemetryEvent(EventName.REPL, undefined, { replType: 'Native' });
             const interpreter = await getActiveInterpreter(uri, interpreterService);
             if (interpreter) {
-                if (interpreter) {
-                    const nativeRepl = await getNativeRepl(interpreter, disposables);
-                    await nativeRepl.sendToNativeRepl();
-                }
+                const nativeRepl = await getNativeRepl(interpreter, disposables);
+                await nativeRepl.sendToNativeRepl(undefined, false);
             }
         }),
     );