Fix Ghost Text and CodeLens issue in Jetbrains

sourcegraph · Dec 31, 2024 · 898b1c8 · 898b1c8
1 parent f3fe8b6
commit 898b1c8
Showing 1 changed file with 10 additions and 7 deletions.
diff --git a/jetbrains/src/main/kotlin/com/sourcegraph/cody/edit/lenses/LensesService.kt b/jetbrains/src/main/kotlin/com/sourcegraph/cody/edit/lenses/LensesService.kt
@@ -63,13 +63,16 @@ class LensesService(val project: Project) {
 
     runInEdt {
       if (project.isDisposed) return@runInEdt
-      CodyEditorUtil.getSelectedEditors(project).forEach { editor ->
-        CodeVisionInitializer.getInstance(project)
-            .getCodeVisionHost()
-            .invalidateProvider(
-                CodeVisionHost.LensInvalidateSignal(
-                    editor, EditCodeVisionProvider.allEditProviders().map { it.id }))
-      }
+      // Find the specific editor matching the file
+      CodyEditorUtil.getAllOpenEditors()
+        .find { editor -> editor.virtualFile == vf }
+        ?.let { matchingEditor ->
+          CodeVisionInitializer.getInstance(project)
+              .getCodeVisionHost()
+              .invalidateProvider(
+                  CodeVisionHost.LensInvalidateSignal(
+                      matchingEditor, EditCodeVisionProvider.allEditProviders().map { it.id }))
+        }
     }
     listeners.forEach { it.onLensesUpdate(vf, codeLens) }
   }