more refactoring to make my life easier

Author: anthonykim1

src/client/repl/nativeRepl.ts

@@ -170,14 +170,11 @@ export class NativeRepl implements Disposable {
                 // e.g. creation of untitled notebook without Python extension knowing.
                 if (getTabNameForUri(replTabBeforeReload) !== 'Python REPL') {
                     mementoUri = undefined;
-                    await this.context.globalState.update(NATIVE_REPL_URI_MEMENTO, undefined);
-                    this.notebookDocument = undefined;
+                    await this.cleanRepl();
                 }
             }
         } else {
-            mementoUri = undefined;
-            await this.context.globalState.update(NATIVE_REPL_URI_MEMENTO, undefined);
-            this.notebookDocument = undefined;
+            await this.cleanRepl();
         }
 
         const notebookEditor = await openInteractiveREPL(this.replController, this.notebookDocument, mementoUri);
@@ -193,6 +190,15 @@ export class NativeRepl implements Disposable {
             }
         }
     }
+
+    /**
+     * Properly clean up notebook document stored inside Native REPL.
+     * Also remove the Native REPL URI from memento to prepare for brand new REPL creation.
+     */
+    private async cleanRepl(): Promise<void> {
+        this.notebookDocument = undefined;
+        await this.context.globalState.update(NATIVE_REPL_URI_MEMENTO, undefined);
+    }
 }
 
 /**

src/client/repl/replUtils.ts

@@ -99,6 +99,7 @@ export function getExistingReplViewColumn(notebookDocument: NotebookDocument): V
     }
     return undefined;
 }
+
 /**
  * Function that will return tab name for before reloading VS Code
  * This is so we can make sure tab name is still 'Python REPL' after reloading VS Code,