refactor TemporaryStateManager with interface

Author: karthiknadig

src/features/views/envManagersView.ts

@@ -11,7 +11,7 @@ import {
     InternalEnvironmentManager,
     InternalPackageManager,
 } from '../../internal.api';
-import { TemporaryStateManager } from './temporaryStateManager';
+import { ITemporaryStateManager } from './temporaryStateManager';
 import {
     EnvInfoTreeItem,
     EnvManagerTreeItem,
@@ -37,7 +37,7 @@ export class EnvManagerView implements TreeDataProvider<EnvTreeItem>, Disposable
     private selected: Map<string, string> = new Map();
     private disposables: Disposable[] = [];
 
-    public constructor(public providers: EnvironmentManagers, private stateManager: TemporaryStateManager) {
+    public constructor(public providers: EnvironmentManagers, private stateManager: ITemporaryStateManager) {
         this.treeView = window.createTreeView<EnvTreeItem>('env-managers', {
             treeDataProvider: this,
         });

src/features/views/projectView.ts

@@ -14,7 +14,7 @@ import { ProjectViews } from '../../common/localize';
 import { createSimpleDebounce } from '../../common/utils/debounce';
 import { onDidChangeConfiguration } from '../../common/workspace.apis';
 import { EnvironmentManagers, PythonProjectManager } from '../../internal.api';
-import { TemporaryStateManager } from './temporaryStateManager';
+import { ITemporaryStateManager } from './temporaryStateManager';
 import {
     GlobalProjectItem,
     NoProjectEnvironment,
@@ -43,7 +43,7 @@ export class ProjectView implements TreeDataProvider<ProjectTreeItem> {
     public constructor(
         private envManagers: EnvironmentManagers,
         private projectManager: PythonProjectManager,
-        private stateManager: TemporaryStateManager,
+        private stateManager: ITemporaryStateManager,
     ) {
         this.treeView = window.createTreeView<ProjectTreeItem>('python-projects', {
             treeDataProvider: this,

src/features/views/temporaryStateManager.ts

@@ -2,12 +2,23 @@ import { Disposable, Event, EventEmitter } from 'vscode';
 
 const DEFAULT_TIMEOUT_MS = 2000;
 
+/**
+ * Interface for managing temporary state on tree items.
+ */
+export interface ITemporaryStateManager {
+    readonly onDidChangeState: Event<{ itemId: string; stateKey: string }>;
+    setState(itemId: string, stateKey: string): void;
+    clearState(itemId: string, stateKey: string): void;
+    hasState(itemId: string, stateKey: string): boolean;
+    updateContextValue(itemId: string, currentContext: string, stateKeys: string[], separator?: string): string;
+}
+
 /**
  * Manages temporary state for tree items that auto-clears after a timeout.
  * Useful for visual feedback like showing a checkmark after copying,
  * or highlighting a recently selected environment.
  */
-export class TemporaryStateManager implements Disposable {
+export class TemporaryStateManager implements ITemporaryStateManager, Disposable {
     private activeItems: Map<string, Set<string>> = new Map();
     private timeouts: Map<string, NodeJS.Timeout> = new Map();
     private readonly _onDidChangeState = new EventEmitter<{ itemId: string; stateKey: string }>();