Retrigger discovery in background if something has changed (microsoft#14944)

Author: Kartik Raj

src/client/pythonEnvironments/index.ts

@@ -55,6 +55,7 @@ export function initialize(ext: ExtensionState): PythonEnvironments {
  ext.legacyIOC.serviceManager,
  api,
  environmentsSecurity,
+ ext.disposables
  );
 
  return api;

src/client/pythonEnvironments/legacyIOC.ts

@@ -5,6 +5,7 @@ import { injectable } from 'inversify';
 import * as vscode from 'vscode';
 import { IExtensionSingleActivationService } from '../activation/types';
 import { DiscoveryVariants } from '../common/experiments/groups';
+import { IDisposableRegistry } from '../common/types';
 import { getVersionString, parseVersion } from '../common/utils/version';
 import {
  CONDA_ENV_FILE_SERVICE,
@@ -152,6 +153,7 @@ class ComponentAdapter implements IComponentAdapter, IExtensionSingleActivationS
  // The adapter only wraps one thing: the component API.
  private readonly api: IPythonEnvironments,
  private readonly environmentsSecurity: IEnvironmentsSecurity,
+ private readonly disposables: IDisposableRegistry,
  ) {}
 
  public async activate(): Promise<void> {
@@ -161,6 +163,14 @@ class ComponentAdapter implements IComponentAdapter, IExtensionSingleActivationS
  inExperiment(DiscoveryVariants.discoveryWithoutFileWatching),
  ],
  )).includes(true);
+ this.disposables.push(this.api.onChanged((e) => {
+ const query = {
+ kinds: e.kind ? [e.kind] : undefined,
+ searchLocations: e.searchLocation ? { roots: [e.searchLocation] } : undefined
+ };
+ // Trigger a background refresh of the environments.
+ getEnvs(this.api.iterEnvs(query)).ignoreErrors();
+ }));
  }
 
  // IInterpreterLocatorProgressHandler
@@ -409,10 +419,11 @@ export function registerNewDiscoveryForIOC(
  serviceManager: IServiceManager,
  api: IPythonEnvironments,
  environmentsSecurity: EnvironmentsSecurity,
+ disposables: IDisposableRegistry
 ): void {
  serviceManager.addSingletonInstance<IComponentAdapter>(
  IComponentAdapter,
- new ComponentAdapter(api, environmentsSecurity),
+ new ComponentAdapter(api, environmentsSecurity, disposables),
  );
  serviceManager.addBinding(IComponentAdapter, IExtensionSingleActivationService);
 }

src/test/pythonEnvironments/legacyIOC.ts

@@ -23,5 +23,6 @@ export function registerForIOC(
  serviceManager,
  instance(mock(PythonEnvironments)),
  instance(mock(EnvironmentsSecurity)),
+ []
  );
 }