add startUrl to security scan events (#4151)

Author: leigaol

.changes/next-release/feature-9bc489b4-9702-4a97-9809-b1978175ffaa.json

@@ -0,0 +1,4 @@
+{
+  "type" : "feature",
+  "description" : "CodeWhisperer: Add startUrl to security scan telemetry"
+}

gradle/libs.versions.toml

@@ -21,7 +21,7 @@ mockito = "5.10.0"
 mockitoKotlin = "5.2.1"
 mockk = "1.13.8"
 node-gradle = "7.0.1"
-telemetryGenerator = "1.0.190"
+telemetryGenerator = "1.0.192"
 testLogger = "3.1.0"
 testRetry = "1.5.2"
 # test-only; platform provides slf4j transitively at runtime. <233, 1.7.36; >=233, 2.0.9

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

@@ -31,6 +31,7 @@ import software.aws.toolkits.jetbrains.services.codewhisperer.service.CodeWhispe
 import software.aws.toolkits.jetbrains.services.codewhisperer.service.CodeWhispererService
 import software.aws.toolkits.jetbrains.services.codewhisperer.service.CodeWhispererUserGroupSettings
 import software.aws.toolkits.jetbrains.services.codewhisperer.util.CodeWhispererConstants.TOTAL_SECONDS_IN_MINUTE
+import software.aws.toolkits.jetbrains.services.codewhisperer.util.CodeWhispererUtil.getCodeWhispererStartUrl
 import software.aws.toolkits.jetbrains.services.codewhisperer.util.runIfIdcConnectionOrTelemetryEnabled
 import software.aws.toolkits.telemetry.CodewhispererTelemetry
 import java.time.Duration
@@ -245,7 +246,8 @@ abstract class CodeWhispererCodeCoverageTracker(
                 totalTokensSize,
                 successCount = myServiceInvocationCount.get(),
                 codewhispererCustomizationArn = customizationArn,
-                codewhispererUserGroup = CodeWhispererUserGroupSettings.getInstance().getUserGroup().name
+                codewhispererUserGroup = CodeWhispererUserGroupSettings.getInstance().getUserGroup().name,
+                credentialStartUrl = getCodeWhispererStartUrl(project)
             )
         }
     }

plugins/toolkit/jetbrains-core/src/software/aws/toolkits/jetbrains/services/codewhisperer/telemetry/CodeWhispererTelemetryService.kt

@@ -32,6 +32,7 @@ import software.aws.toolkits.jetbrains.services.codewhisperer.service.CodeWhispe
 import software.aws.toolkits.jetbrains.services.codewhisperer.service.RequestContext
 import software.aws.toolkits.jetbrains.services.codewhisperer.service.ResponseContext
 import software.aws.toolkits.jetbrains.services.codewhisperer.settings.CodeWhispererSettings
+import software.aws.toolkits.jetbrains.services.codewhisperer.util.CodeWhispererUtil.getCodeWhispererStartUrl
 import software.aws.toolkits.jetbrains.services.codewhisperer.util.CodeWhispererUtil.getConnectionStartUrl
 import software.aws.toolkits.jetbrains.services.codewhisperer.util.CodeWhispererUtil.getGettingStartedTaskType
 import software.aws.toolkits.jetbrains.services.codewhisperer.util.runIfIdcConnectionOrTelemetryEnabled
@@ -330,7 +331,8 @@ class CodeWhispererTelemetryService {
             findingId = issue.findingId,
             detectorId = issue.detectorId,
             ruleId = issue.ruleId,
-            includesFix = issue.suggestedFixes.isNotEmpty()
+            includesFix = issue.suggestedFixes.isNotEmpty(),
+            credentialStartUrl = getCodeWhispererStartUrl(issue.project)
         )
     }
 
@@ -341,7 +343,8 @@ class CodeWhispererTelemetryService {
             ruleId = issue.ruleId,
             component = Component.Hover,
             result = result,
-            reason = reason
+            reason = reason,
+            credentialStartUrl = getCodeWhispererStartUrl(issue.project)
         )
     }
 

plugins/toolkit/jetbrains-core/src/software/aws/toolkits/jetbrains/services/codewhisperer/util/CodeWhispererUtil.kt

@@ -248,6 +248,13 @@ object CodeWhispererUtil {
         return connection.startUrl
     }
 
+    fun getCodeWhispererStartUrl(project: Project): String? {
+        val connection = ToolkitConnectionManager.getInstance(
+            project
+        ).activeConnectionForFeature(CodeWhispererConnection.getInstance()) as? AwsBearerTokenConnection?
+        return connection?.startUrl
+    }
+
     private fun tokenConnection(project: Project) = (
         ToolkitConnectionManager
             .getInstance(project)