Fix race condition in terminal quick fix provider registration

Co-authored-by: Tyriar <[email protected]>

Author: Copilot

src/vs/workbench/contrib/terminalContrib/quickFix/browser/terminalQuickFixService.ts

@@ -20,6 +20,8 @@ export class TerminalQuickFixService implements ITerminalQuickFixService {
 	private _providers: Map<string, ITerminalQuickFixProvider> = new Map();
 	get providers(): Map<string, ITerminalQuickFixProvider> { return this._providers; }
 
+	private _pendingProviders: Map<string, ITerminalQuickFixProvider> = new Map();
+
 	private readonly _onDidRegisterProvider = new Emitter<ITerminalQuickFixProviderSelector>();
 	readonly onDidRegisterProvider = this._onDidRegisterProvider.event;
 	private readonly _onDidRegisterCommandSelector = new Emitter<ITerminalCommandSelector>();
@@ -50,6 +52,14 @@ export class TerminalQuickFixService implements ITerminalQuickFixService {
 	registerCommandSelector(selector: ITerminalCommandSelector): void {
 		this._selectors.set(selector.id, selector);
 		this._onDidRegisterCommandSelector.fire(selector);
+		
+		// Check if there's a pending provider for this selector
+		const pendingProvider = this._pendingProviders.get(selector.id);
+		if (pendingProvider) {
+			this._pendingProviders.delete(selector.id);
+			this._providers.set(selector.id, pendingProvider);
+			this._onDidRegisterProvider.fire({ selector, provider: pendingProvider });
+		}
 	}
 
 	registerQuickFixProvider(id: string, provider: ITerminalQuickFixProvider): IDisposable {
@@ -61,17 +71,20 @@ export class TerminalQuickFixService implements ITerminalQuickFixService {
 			if (disposed) {
 				return;
 			}
-			this._providers.set(id, provider);
 			const selector = this._selectors.get(id);
-			if (!selector) {
-				this._logService.error(`No registered selector for ID: ${id}`);
-				return;
+			if (selector) {
+				// Selector is already available, register immediately
+				this._providers.set(id, provider);
+				this._onDidRegisterProvider.fire({ selector, provider });
+			} else {
+				// Selector not yet available, store provider as pending
+				this._pendingProviders.set(id, provider);
 			}
-			this._onDidRegisterProvider.fire({ selector, provider });
 		});
 		return toDisposable(() => {
 			disposed = true;
 			this._providers.delete(id);
+			this._pendingProviders.delete(id);
 			const selector = this._selectors.get(id);
 			if (selector) {
 				this._selectors.delete(id);