Send the command to enable Trace Inspector as soon as possible

Since it takes time to get vscode.debug.activeDebugSession, we use the session obtained in DebugAdapterDescriptorFactory#createDebugAdapterDescriptor.

Author: ono-max

src/inspector.ts

@@ -26,6 +26,9 @@ export function registerInspectorView(emitter: vscode.EventEmitter<any>, version
     const inlayHintsProvider = new RdbgCodeLensProvider(view);
     const disposables: vscode.Disposable[] = [];
     let traceInspectorEnabled: boolean | undefined;
+    // Since it takes time to get vscode.debug.activeDebugSession,
+    // we holds the session obtained in DebugAdapterDescriptorFactory#createDebugAdapterDescriptor.
+    let activeSession: vscode.DebugSession | undefined;
 
     disposables.push(
         vscode.languages.registerCodeLensProvider(
@@ -63,10 +66,7 @@ export function registerInspectorView(emitter: vscode.EventEmitter<any>, version
                     if (!traceInspectorEnabled) {
                         return;
                     }
-                    while (!vscode.debug.activeDebugSession) {
-                        await new Promise((resolve) => setTimeout(resolve, 10));
-                    }
-                    await treeProvider.toggleTreeItem.enable();
+                    await treeProvider.toggleTreeItem.enable(activeSession);
                     break;
             }
         }),
@@ -90,6 +90,7 @@ export function registerInspectorView(emitter: vscode.EventEmitter<any>, version
                 traceInspectorEnabled = false;
                 return;
             }
+            activeSession = session;
             const config = session.configuration as LaunchConfiguration;
             traceInspectorEnabled = await validVersion(config, versionChecker, treeProvider);
         }),

src/rdbgTreeItem.ts

@@ -183,18 +183,18 @@ export class ToggleTreeItem extends RdbgTreeItem {
 	}
 
 	async toggle() {
+        const session = vscode.debug.activeDebugSession;
 		if (this._enabled) {
-			this.disable();
+			this.disable(session);
 		} else {
-			this.enable();
+			this.enable(session);
 		}
 	}
 
-	async enable() {
+	async enable(session: vscode.DebugSession | undefined) {
 		this.iconPath = stopCircleIcon;
 		this._enabled = true;
 		this.label = "Disable Trace";
-		const session = vscode.debug.activeDebugSession;
 		if (session === undefined) {
 			return;
 		}
@@ -249,9 +249,8 @@ export class ToggleTreeItem extends RdbgTreeItem {
 		this.label = "Enable Trace";
 	}
 
-	async disable() {
+	async disable(session: vscode.DebugSession | undefined) {
 		this.resetView();
-		const session = vscode.debug.activeDebugSession;
 		if (session === undefined || this._enabledCommand === undefined) {
 			return;
 		}