Feature - Emit additional Code Transform telemetry (#4063)

Author: tincheng

.changes/next-release/feature-09c9bb6d-d79b-4bbc-9241-735f0efc1098.json

@@ -0,0 +1,4 @@
+{
+  "type" : "feature",
+  "description" : "Emit additional CodeTransform telemetry"
+}

jetbrains-core/src/software/aws/toolkits/jetbrains/services/codemodernizer/ArtifactHandler.kt

@@ -27,6 +27,7 @@ import software.aws.toolkits.jetbrains.utils.notifyInfo
 import software.aws.toolkits.jetbrains.utils.notifyWarn
 import software.aws.toolkits.resources.message
 import software.aws.toolkits.telemetry.CodeTransformApiNames
+import software.aws.toolkits.telemetry.CodeTransformPatchViewerCancelSrcComponents
 import software.aws.toolkits.telemetry.CodeTransformVCSViewerSrcComponents
 import software.aws.toolkits.telemetry.CodetransformTelemetry
 import java.nio.file.Files
@@ -46,7 +47,7 @@ class ArtifactHandler(private val project: Project, private val clientAdaptor: G
         if (result.artifact == null) {
             notifyUnableToApplyPatch(result.zipPath)
         } else {
-            displayDiffUsingPatch(result.artifact.patch)
+            displayDiffUsingPatch(result.artifact.patch, job)
         }
     }
 
@@ -145,7 +146,7 @@ class ArtifactHandler(private val project: Project, private val clientAdaptor: G
     /**
      * Opens the built-in patch dialog to display the diff and allowing users to apply the changes locally.
      */
-    internal fun displayDiffUsingPatch(patchFile: VirtualFile) {
+    internal fun displayDiffUsingPatch(patchFile: VirtualFile, jobId: JobId) {
         runInEdt {
             val dialog = ApplyPatchDifferentiatedDialog(
                 project,
@@ -159,10 +160,27 @@ class ArtifactHandler(private val project: Project, private val clientAdaptor: G
                 null,
                 null,
                 null,
-                false
+                false,
             )
             dialog.isModal = true
-            dialog.showAndGet()
+
+            CodetransformTelemetry.vcsDiffViewerVisible(
+                codeTransformSessionId = CodeTransformTelemetryState.instance.getSessionId(),
+                codeTransformJobId = jobId.id
+            )
+
+            if (dialog.showAndGet()) {
+                CodetransformTelemetry.vcsViewerSubmitted(
+                    codeTransformSessionId = CodeTransformTelemetryState.instance.getSessionId(),
+                    codeTransformJobId = jobId.id
+                )
+            } else {
+                CodetransformTelemetry.vcsViewerCanceled(
+                    codeTransformPatchViewerCancelSrcComponents = CodeTransformPatchViewerCancelSrcComponents.CancelButton,
+                    codeTransformSessionId = CodeTransformTelemetryState.instance.getSessionId(),
+                    codeTransformJobId = jobId.id
+                )
+            }
         }
     }
 

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

@@ -15,6 +15,7 @@ import kotlinx.coroutines.runBlocking
 import org.junit.Before
 import org.junit.Test
 import org.mockito.Mockito
+import org.mockito.Mockito.never
 import org.mockito.Mockito.`when`
 import org.mockito.kotlin.any
 import org.mockito.kotlin.doNothing
@@ -94,6 +95,18 @@ class CodeWhispererCodeModernizerTest : CodeWhispererCodeModernizerTestBase() {
         verify(handler, times(1)).notifyUnableToApplyPatch(path)
     }
 
+    @Test
+    fun `ArtifactHandler displays patch`() = runBlocking {
+        val handler = spy(ArtifactHandler(project, clientAdaptorSpy))
+        val path = testCodeModernizerArtifact.zipPath
+        val result = DownloadArtifactResult(testCodeModernizerArtifact, path)
+        doReturn(result).whenever(handler).downloadArtifact(any())
+        doNothing().whenever(handler).displayDiffUsingPatch(any(), any())
+        handler.displayDiff(jobId)
+        verify(handler, never()).notifyUnableToApplyPatch(path)
+        verify(handler, times(1)).displayDiffUsingPatch(testCodeModernizerArtifact.patch, jobId)
+    }
+
     @Test
     fun `CodeModernizerArtifact can process a valid zip file`() {
         val artifact = CodeModernizerArtifact.create(exampleZipPath.toAbsolutePath().toString())