add select and selected feedback

Author: karthiknadig

package.json

@@ -193,6 +193,12 @@
                 "category": "Python",
                 "icon": "$(check)"
             },
+            {
+                "command": "python-envs.setEnvSelected",
+                "title": "%python-envs.setEnvSelected.title%",
+                "category": "Python",
+                "icon": "$(pass-filled)"
+            },
             {
                 "command": "python-envs.remove",
                 "title": "%python-envs.remove.title%",
@@ -333,6 +339,10 @@
                     "command": "python-envs.setEnv",
                     "when": "false"
                 },
+                {
+                    "command": "python-envs.setEnvSelected",
+                    "when": "false"
+                },
                 {
                     "command": "python-envs.remove",
                     "when": "false"
@@ -439,7 +449,12 @@
                 {
                     "command": "python-envs.setEnv",
                     "group": "inline",
-                    "when": "view == env-managers && viewItem =~ /.*pythonEnvironment.*/"
+                    "when": "view == env-managers && viewItem =~ /.*pythonEnvironment.*/ && viewItem =~ /^((?!selected).)*$/"
+                },
+                {
+                    "command": "python-envs.setEnvSelected",
+                    "group": "inline",
+                    "when": "view == env-managers && viewItem =~ /.*pythonEnvironment.*/ && viewItem =~ /.*selected.*/"
                 },
                 {
                     "command": "python-envs.createTerminal",

package.nls.json

@@ -28,6 +28,7 @@
     "python-envs.createAny.title": "Create Environment",
     "python-envs.set.title": "Set Project Environment",
     "python-envs.setEnv.title": "Set As Project Environment",
+    "python-envs.setEnvSelected.title": "Set!",
     "python-envs.remove.title": "Delete Environment",
     "python-envs.refreshAllManagers.title": "Refresh All Environment Managers",
     "python-envs.refreshPackages.title": "Refresh Packages List",

src/extension.ts

@@ -66,7 +66,7 @@ import { ProjectView } from './features/views/projectView';
 import { PythonStatusBarImpl } from './features/views/pythonStatusBar';
 import { updateViewsAndStatus } from './features/views/revealHandler';
 import { TemporaryStateManager } from './features/views/temporaryStateManager';
-import { ProjectItem } from './features/views/treeViewItems';
+import { ProjectItem, PythonEnvTreeItem } from './features/views/treeViewItems';
 import {
     collectEnvironmentInfo,
     getEnvManagerAndPackageManagerConfigLevels,
@@ -229,6 +229,12 @@ export async function activate(context: ExtensionContext): Promise<PythonEnviron
         }),
         commands.registerCommand('python-envs.setEnv', async (item) => {
             await setEnvironmentCommand(item, envManagers, projectManager);
+            if (item instanceof PythonEnvTreeItem) {
+                temporaryStateManager.setState(item.environment.envId.id, 'selected');
+            }
+        }),
+        commands.registerCommand('python-envs.setEnvSelected', async () => {
+            // No-op: This command is just for showing the feedback icon
         }),
         commands.registerCommand('python-envs.setEnvManager', async () => {
             await setEnvManagerCommand(envManagers, projectManager);

src/features/views/envManagersView.ts

@@ -24,6 +24,7 @@ import {
 } from './treeViewItems';
 
 const COPIED_STATE = 'copied';
+const SELECTED_STATE = 'selected';
 
 export class EnvManagerView implements TreeDataProvider<EnvTreeItem>, Disposable {
     private treeView: TreeView<EnvTreeItem>;
@@ -91,14 +92,27 @@ export class EnvManagerView implements TreeDataProvider<EnvTreeItem>, Disposable
     getTreeItem(element: EnvTreeItem): TreeItem | Thenable<TreeItem> {
         if (element.kind === EnvTreeItemKind.environment && element instanceof PythonEnvTreeItem) {
             const itemId = element.environment.envId.id;
-            const currentContext = element.treeItem.contextValue ?? '';
-            if (this.stateManager?.hasState(itemId, COPIED_STATE)) {
+            let currentContext = element.treeItem.contextValue ?? '';
+
+            // Handle copied state
+            if (this.stateManager.hasState(itemId, COPIED_STATE)) {
                 if (!currentContext.includes(COPIED_STATE)) {
-                    element.treeItem.contextValue = currentContext + COPIED_STATE + ';';
+                    currentContext = currentContext + COPIED_STATE + ';';
                 }
             } else if (currentContext.includes(COPIED_STATE)) {
-                element.treeItem.contextValue = currentContext.replace(COPIED_STATE + ';', '');
+                currentContext = currentContext.replace(COPIED_STATE + ';', '');
             }
+
+            // Handle selected state
+            if (this.stateManager.hasState(itemId, SELECTED_STATE)) {
+                if (!currentContext.includes(SELECTED_STATE)) {
+                    currentContext = currentContext + SELECTED_STATE + ';';
+                }
+            } else if (currentContext.includes(SELECTED_STATE)) {
+                currentContext = currentContext.replace(SELECTED_STATE + ';', '');
+            }
+
+            element.treeItem.contextValue = currentContext;
         }
         return element.treeItem;
     }

src/features/views/projectView.ts

@@ -27,6 +27,7 @@ import {
 } from './treeViewItems';
 
 const COPIED_STATE = 'copied';
+const SELECTED_STATE = 'selected';
 
 export class ProjectView implements TreeDataProvider<ProjectTreeItem> {
     private treeView: TreeView<ProjectTreeItem>;
@@ -145,7 +146,7 @@ export class ProjectView implements TreeDataProvider<ProjectTreeItem> {
         if (element.kind === ProjectTreeItemKind.project && element instanceof ProjectItem) {
             const itemId = element.project.uri.fsPath;
             const currentContext = element.treeItem.contextValue ?? '';
-            if (this.stateManager?.hasState(itemId, COPIED_STATE)) {
+            if (this.stateManager.hasState(itemId, COPIED_STATE)) {
                 if (!currentContext.includes(COPIED_STATE)) {
                     element.treeItem.contextValue = currentContext + ';' + COPIED_STATE;
                 }
@@ -154,14 +155,27 @@ export class ProjectView implements TreeDataProvider<ProjectTreeItem> {
             }
         } else if (element.kind === ProjectTreeItemKind.environment && element instanceof ProjectEnvironment) {
             const itemId = element.environment.envId.id;
-            const currentContext = element.treeItem.contextValue ?? '';
-            if (this.stateManager?.hasState(itemId, COPIED_STATE)) {
+            let currentContext = element.treeItem.contextValue ?? '';
+
+            // Handle copied state
+            if (this.stateManager.hasState(itemId, COPIED_STATE)) {
                 if (!currentContext.includes(COPIED_STATE)) {
-                    element.treeItem.contextValue = currentContext + ';' + COPIED_STATE;
+                    currentContext = currentContext + ';' + COPIED_STATE;
                 }
             } else if (currentContext.includes(COPIED_STATE)) {
-                element.treeItem.contextValue = currentContext.replace(';' + COPIED_STATE, '');
+                currentContext = currentContext.replace(';' + COPIED_STATE, '');
             }
+
+            // Handle selected state
+            if (this.stateManager.hasState(itemId, SELECTED_STATE)) {
+                if (!currentContext.includes(SELECTED_STATE)) {
+                    currentContext = currentContext + ';' + SELECTED_STATE;
+                }
+            } else if (currentContext.includes(SELECTED_STATE)) {
+                currentContext = currentContext.replace(';' + SELECTED_STATE, '');
+            }
+
+            element.treeItem.contextValue = currentContext;
         }
         return element.treeItem;
     }