[CodeWhisperer] CodeCoverageTracker will not be initialized with unsupportedLanguage (#3267)

Author: Will-ShaoHua

jetbrains-core/src/software/aws/toolkits/jetbrains/services/codewhisperer/editor/CodeWhispererEditorListener.kt

@@ -10,28 +10,31 @@ import com.intellij.openapi.editor.event.EditorFactoryListener
 import com.intellij.openapi.editor.impl.EditorImpl
 import com.intellij.psi.PsiDocumentManager
 import software.aws.toolkits.jetbrains.services.codewhisperer.editor.CodeWhispererEditorUtil.codeWhispererLanguage
+import software.aws.toolkits.jetbrains.services.codewhisperer.language.CodeWhispererLanguageManager
 import software.aws.toolkits.jetbrains.services.codewhisperer.service.CodeWhispererInvocationStatus
 import software.aws.toolkits.jetbrains.services.codewhisperer.telemetry.CodeWhispererCodeCoverageTracker
 
 class CodeWhispererEditorListener : EditorFactoryListener {
     override fun editorCreated(event: EditorFactoryEvent) {
         val editor = (event.editor as? EditorImpl) ?: return
-
-        editor.document.addDocumentListener(
-            object : DocumentListener {
-                override fun documentChanged(event: DocumentEvent) {
-                    CodeWhispererInvocationStatus.getInstance().documentChanged()
-                    editor.project?.let { project ->
-                        PsiDocumentManager.getInstance(project).getPsiFile(editor.document)?.codeWhispererLanguage ?. let { language ->
+        editor.project?.let { project ->
+            PsiDocumentManager.getInstance(project).getPsiFile(editor.document)?.codeWhispererLanguage ?. let { language ->
+                // If language is not supported by CodeWhisperer, no action needed
+                if (!CodeWhispererLanguageManager.getInstance().isLanguageSupported(language.toString())) return
+                // If language is supported, install document listener for CodeWhisperer service
+                editor.document.addDocumentListener(
+                    object : DocumentListener {
+                        override fun documentChanged(event: DocumentEvent) {
+                            CodeWhispererInvocationStatus.getInstance().documentChanged()
                             CodeWhispererCodeCoverageTracker.getInstance(language).apply {
                                 activateTrackerIfNotActive()
                                 documentChanged(event)
                             }
                         }
-                    }
-                }
-            },
-            editor.disposable
-        )
+                    },
+                    editor.disposable
+                )
+            }
+        }
     }
 }

jetbrains-core/src/software/aws/toolkits/jetbrains/services/codewhisperer/telemetry/CodeWhispererCodeCoverageTracker.kt

@@ -53,6 +53,8 @@ abstract class CodeWhispererCodeCoverageTracker(
     @Synchronized
     fun activateTrackerIfNotActive() {
         if (!isTelemetryEnabled() || isActive.get()) return
+
+        // tracker will only be activated if and only if IsTelemetryEnabled = true && isActive = false
         val conn = ApplicationManager.getApplication().messageBus.connect()
         conn.subscribe(
             CodeWhispererPopupManager.CODEWHISPERER_USER_ACTION_PERFORMED,

jetbrains-core/tst/software/aws/toolkits/jetbrains/services/codewhisperer/CodeWhispererTelemetryTest.kt

@@ -453,6 +453,24 @@ class CodeWhispererTelemetryTest : CodeWhispererTestBase() {
         }
     }
 
+    @Test
+    fun `test codePercentage tracker will not be initialized with unsupportedLanguage`() {
+        assertThat(CodeWhispererCodeCoverageTracker.getInstancesMap()).hasSize(0)
+        val project = projectRule.project
+        val fixture = projectRule.fixture
+        val plainTxtFile = fixture.addFileToProject("/notSupported.txt", "")
+
+        runInEdtAndWait {
+            fixture.openFileInEditor(plainTxtFile.virtualFile)
+            WriteCommandAction.runWriteCommandAction(project) {
+                fixture.editor.appendString("random txt string")
+            }
+        }
+
+        // document changes in unsupported files will not trigger initialization of tracker
+        assertThat(CodeWhispererCodeCoverageTracker.getInstancesMap()).hasSize(0)
+    }
+
     @Test
     fun `test codePercentage metric is correct - 1`() {
         val project = projectRule.project