feat: expose registered package manager

Author: edvilme

api/CHANGELOG.md

@@ -8,3 +8,4 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 ## [1.37.0]
 
 - Aligned the API package version with the Python Environments extension version.
+- Added `getPackageManager` to retrieve the registered package manager for an environment.

src/api.ts

@@ -1091,6 +1091,14 @@ export interface PythonPackageManagerRegistrationApi {
 }
 
 export interface PythonPackageGetterApi {
+    /**
+     * Get the registered package manager associated with a Python Environment.
+     *
+     * @param environment The Python Environment whose package manager is required.
+     * @returns The registered package manager, or undefined if no package manager is available.
+     */
+    getPackageManager(environment: PythonEnvironment): Promise<PackageManager | undefined>;
+
     /**
      * Refresh the list of packages in a Python Environment.
      *

src/features/pythonApi.ts

@@ -303,6 +303,10 @@ export class PythonEnvironmentApiImpl implements PythonEnvironmentApi {
         }
         return manager.manage(context, options);
     }
+    async getPackageManager(context: PythonEnvironment): Promise<PackageManager | undefined> {
+        await waitForEnvManagerId([context.envId.managerId]);
+        return this.envManagers.getPackageManager(context);
+    }
     async refreshPackages(context: PythonEnvironment): Promise<Package[] | undefined> {
         await waitForEnvManagerId([context.envId.managerId]);
         const manager = this.envManagers.getPackageManager(context);

src/test/integration/packageManager.integration.test.ts

@@ -94,6 +94,16 @@ for (const profile of profiles) {
             assert.ok(!packages?.some((pkg) => pkg.name === 'requests'), 'Package not uninstalled');
         });
 
+        test(`${profile.name} Package Manager should list available package versions`, async () => {
+            const packageManager = await api.getPackageManager(environment!);
+            assert.ok(packageManager, 'Package manager not available');
+            assert.ok(packageManager.getPackageAvailableVersions, 'Available versions method not available');
+
+            const versions = await packageManager.getPackageAvailableVersions(environment!, 'requests');
+            assert.ok(versions, 'Package versions not available');
+            assert.ok(versions.length > 0, 'No package versions available');
+        });
+
         suiteTeardown(async () => {
             try {
                 await deleteEnvironmentDirectory(vscode.Uri.joinPath(workspaceUri, profile.environmentDirectory));