feat: add Pixi extension recommendation and support for Pixi environments

Author: eleanorjboyd

src/common/localize.ts

@@ -136,6 +136,14 @@ export namespace SysManagerStrings {
     export const packageRefreshError = l10n.t('Error refreshing packages');
 }
 
+export namespace PixiStrings {
+    export const pixiExtensionRecommendation = l10n.t(
+        'Pixi environments were detected. Install the Pixi extension for full support including activation and environment management.',
+    );
+    export const install = l10n.t('Install Pixi Extension');
+    export const dontAskAgain = l10n.t("Don't Ask Again");
+}
+
 export namespace CondaStrings {
     export const condaManager = l10n.t('Manages Conda environments');
     export const condaDiscovering = l10n.t('Discovering Conda environments');

src/common/utils/pythonPath.ts

@@ -16,17 +16,19 @@ const priorityOrder = [
     `${PYTHON_EXTENSION_ID}:system`,
 ];
 function sortManagersByPriority(managers: InternalEnvironmentManager[]): InternalEnvironmentManager[] {
+    const systemId = priorityOrder[priorityOrder.length - 1];
     return managers.sort((a, b) => {
         const aIndex = priorityOrder.indexOf(a.id);
         const bIndex = priorityOrder.indexOf(b.id);
         if (aIndex === -1 && bIndex === -1) {
             return 0;
         }
         if (aIndex === -1) {
-            return 1;
+            // Unknown managers should come before system (last resort) but after other known managers
+            return b.id === systemId ? -1 : 1;
         }
         if (bIndex === -1) {
-            return -1;
+            return a.id === systemId ? 1 : -1;
         }
         return aIndex - bIndex;
     });

src/managers/builtin/utils.ts

@@ -9,9 +9,12 @@ import {
     PythonEnvironmentInfo,
 } from '../../api';
 import { showErrorMessageWithLogs } from '../../common/errors/utils';
-import { SysManagerStrings } from '../../common/localize';
-import { traceVerbose } from '../../common/logging';
-import { withProgress } from '../../common/window.apis';
+import { getExtension } from '../../common/extension.apis';
+import { PixiStrings, SysManagerStrings } from '../../common/localize';
+import { traceInfo, traceVerbose } from '../../common/logging';
+import { getGlobalPersistentState } from '../../common/persistentState';
+import { showInformationMessage, withProgress } from '../../common/window.apis';
+import { installExtension } from '../../common/workbenchCommands';
 import {
     isNativeEnvInfo,
     NativeEnvInfo,
@@ -22,6 +25,10 @@ import { shortVersion, sortEnvironments } from '../common/utils';
 import { runPython, runUV, shouldUseUv } from './helpers';
 import { parsePipList, PipPackage } from './pipListUtils';
 
+const PIXI_EXTENSION_ID = 'renan-r-santos.pixi-code';
+const PIXI_RECOMMEND_DONT_ASK_KEY = 'pixi-extension-recommend-dont-ask';
+let pixiRecommendationShown = false;
+
 function asPackageQuickPickItem(name: string, version?: string): QuickPickItem {
     return {
         label: name,
@@ -99,6 +106,38 @@ function getPythonInfo(env: NativeEnvInfo): PythonEnvironmentInfo {
     }
 }
 
+async function recommendPixiExtension(): Promise<void> {
+    if (pixiRecommendationShown) {
+        return;
+    }
+    pixiRecommendationShown = true;
+
+    if (getExtension(PIXI_EXTENSION_ID)) {
+        return;
+    }
+
+    const state = await getGlobalPersistentState();
+    const dontAsk = await state.get<boolean>(PIXI_RECOMMEND_DONT_ASK_KEY);
+    if (dontAsk) {
+        traceInfo('Skipping Pixi extension recommendation: user selected "Don\'t ask again"');
+        return;
+    }
+
+    const result = await showInformationMessage(
+        PixiStrings.pixiExtensionRecommendation,
+        PixiStrings.install,
+        PixiStrings.dontAskAgain,
+    );
+
+    if (result === PixiStrings.install) {
+        traceInfo(`Installing extension: ${PIXI_EXTENSION_ID}`);
+        await installExtension(PIXI_EXTENSION_ID);
+    } else if (result === PixiStrings.dontAskAgain) {
+        await state.set(PIXI_RECOMMEND_DONT_ASK_KEY, true);
+        traceInfo('User selected "Don\'t ask again" for Pixi extension recommendation');
+    }
+}
+
 export async function refreshPythons(
     hardRefresh: boolean,
     nativeFinder: NativePythonFinder,
@@ -109,24 +148,28 @@ export async function refreshPythons(
 ): Promise<PythonEnvironment[]> {
     const collection: PythonEnvironment[] = [];
     const data = await nativeFinder.refresh(hardRefresh, uris);
-    const envs = data
-        .filter((e) => isNativeEnvInfo(e))
-        .map((e) => e as NativeEnvInfo)
-        .filter(
-            (e) =>
-                e.kind === undefined ||
-                (e.kind &&
-                    [
-                        NativePythonEnvironmentKind.globalPaths,
-                        NativePythonEnvironmentKind.homebrew,
-                        NativePythonEnvironmentKind.linuxGlobal,
-                        NativePythonEnvironmentKind.macCommandLineTools,
-                        NativePythonEnvironmentKind.macPythonOrg,
-                        NativePythonEnvironmentKind.macXCode,
-                        NativePythonEnvironmentKind.windowsRegistry,
-                        NativePythonEnvironmentKind.windowsStore,
-                    ].includes(e.kind)),
-        );
+    const allNativeEnvs = data.filter((e) => isNativeEnvInfo(e)).map((e) => e as NativeEnvInfo);
+
+    const hasPixiEnvs = allNativeEnvs.some((e) => e.kind === NativePythonEnvironmentKind.pixi);
+    if (hasPixiEnvs) {
+        recommendPixiExtension().catch((e) => log.error('Error recommending Pixi extension', e));
+    }
+
+    const envs = allNativeEnvs.filter(
+        (e) =>
+            e.kind === undefined ||
+            (e.kind &&
+                [
+                    NativePythonEnvironmentKind.globalPaths,
+                    NativePythonEnvironmentKind.homebrew,
+                    NativePythonEnvironmentKind.linuxGlobal,
+                    NativePythonEnvironmentKind.macCommandLineTools,
+                    NativePythonEnvironmentKind.macPythonOrg,
+                    NativePythonEnvironmentKind.macXCode,
+                    NativePythonEnvironmentKind.windowsRegistry,
+                    NativePythonEnvironmentKind.windowsStore,
+                ].includes(e.kind)),
+    );
     envs.forEach((env) => {
         try {
             const envInfo = getPythonInfo(env);

src/managers/common/nativePythonFinder.ts

@@ -124,6 +124,7 @@ export function isNativeEnvInfo(info: NativeInfo): info is NativeEnvInfo {
 export enum NativePythonEnvironmentKind {
     conda = 'Conda',
     homebrew = 'Homebrew',
+    pixi = 'Pixi',
     pyenv = 'Pyenv',
     globalPaths = 'GlobalPaths',
     pyenvVirtualEnv = 'PyenvVirtualEnv',