Add metadata to authAddConnection metrics

plugins/amazonq/chat/jetbrains-community/src/software/aws/toolkits/jetbrains/services/amazonq/auth/AuthController.kt

@@ -88,7 +88,7 @@
             AuthFollowUpType.FullAuth,
             -> runInEdt {
                 UiTelemetry.click(project, "amazonq_chatAuthenticate")
-                requestCredentialsForQ(project, connectionInitiatedFromQChatPanel = true)
+                requestCredentialsForQ(project, connectionInitiatedFromQChatPanel = true, isReauth = false)
             }
 
             AuthFollowUpType.ReAuth,

plugins/amazonq/chat/jetbrains-community/src/software/aws/toolkits/jetbrains/services/amazonq/explorerActions/SignInToQAction.kt

@@ -24,7 +24,7 @@
         UiTelemetry.click(project, "auth_start_Q")
 
         if (!isQWebviewsAvailable()) {
-            requestCredentialsForQ(project)
+            requestCredentialsForQ(project, isReauth = false)
         } else {
             ToolWindowManager.getInstance(project).getToolWindow(AmazonQToolWindowFactory.WINDOW_ID)?.show()
         }
@@ -42,7 +42,7 @@
             reauthConnectionIfNeeded(project, it, isReAuth = true)
         } ?: run {
             runInEdt {
-                if (requestCredentialsForQ(project)) {
+                if (requestCredentialsForQ(project, isReauth = false)) {
                     if (!openMeetQPage(project)) {
                         return@runInEdt
                     }

plugins/core/jetbrains-community/resources/telemetryOverride.json

@@ -810,50 +810,6 @@
                     "type": "authType"
                 }
             ]
-        },
-        {
-            "name": "auth_addConnection",
-            "description": "Captures the result of adding a new connection in the 'Add New Connection' workflow",
-            "metadata": [
-                {
-                    "type": "attempts",
-                    "required": false
-                },
-                {
-                    "type": "credentialSourceId"
-                },
-                {
-                    "type": "featureId",
-                    "required": false
-                },
-                {
-                    "type": "invalidInputFields",
-                    "required": false
-                },
-                {
-                    "type": "isAggregated",
-                    "required": false
-                },
-                {
-                    "type": "reason",
-                    "required": false
-                },
-                {
-                    "type": "result"
-                },
-                {
-                    "type": "source",
-                    "required": false
-                },
-                {
-                    "type": "credentialStartUrl",
-                    "required": false
-                },
-                {
-                    "type": "isReAuth",
-                    "required": false
-                }
-            ]
         }
     ]
 }

plugins/core/jetbrains-community/src/software/aws/toolkits/jetbrains/core/gettingstarted/GettingStartedAuthUtils.kt

@@ -36,6 +36,7 @@
     ),
     isFirstInstance: Boolean = false,
     connectionInitiatedFromExplorer: Boolean = false,
+    isReauth: Boolean = false,
 ): Boolean {
     val authenticationDialog = SetupAuthenticationDialog(
         project,
@@ -87,7 +88,8 @@
             credentialSourceId = authenticationDialog.authType,
             isAggregated = true,
             attempts = authenticationDialog.attempts + 1,
-            result = Result.Succeeded
+            result = Result.Succeeded,
+            isReAuth = isReauth
         )
         AuthTelemetry.addedConnections(
             project,
@@ -108,6 +110,7 @@
             isAggregated = false,
             attempts = authenticationDialog.attempts + 1,
             result = Result.Cancelled,
+            isReAuth = isReauth
         )
     }
     return isAuthenticationSuccessful
@@ -123,6 +126,7 @@
     isFirstInstance: Boolean = false,
     connectionInitiatedFromExplorer: Boolean = false,
     connectionInitiatedFromQChatPanel: Boolean = false,
+    isReauth: Boolean,
 ): Boolean {
     // try to scope upgrade if we have a codewhisperer connection
     val codeWhispererConnection = ToolkitConnectionManager.getInstance(project).activeConnectionForFeature(CodeWhispererConnection.getInstance())
@@ -194,7 +198,8 @@
             credentialSourceId = authenticationDialog.authType,
             isAggregated = true,
             attempts = authenticationDialog.attempts + 1,
-            result = Result.Succeeded
+            result = Result.Succeeded,
+            isReAuth = isReauth
         )
         AuthTelemetry.addedConnections(
             project,
@@ -215,6 +220,7 @@
             isAggregated = false,
             attempts = authenticationDialog.attempts + 1,
             result = Result.Cancelled,
+            isReAuth = isReauth
         )
     }
     return isAuthenticationSuccessful

plugins/core/jetbrains-community/src/software/aws/toolkits/jetbrains/core/gettingstarted/SetupAuthenticationDialog.kt

@@ -287,7 +287,8 @@
                         isAggregated = false,
                         attempts = ++attempts,
                         result = Result.Failed,
-                        reason = "ConnectionUnsuccessful"
+                        reason = "ConnectionUnsuccessful",
+                        isReAuth = false
                     )
                 } ?: return
 
@@ -339,7 +340,8 @@
                         isAggregated = false,
                         attempts = ++attempts,
                         result = Result.Failed,
-                        reason = "DuplicateProfileName"
+                        reason = "DuplicateProfileName",
+                        isReAuth = false
                     )
                     return
                 }
@@ -365,7 +367,8 @@
                         isAggregated = false,
                         attempts = ++attempts,
                         result = Result.Failed,
-                        reason = "InvalidCredentials"
+                        reason = "InvalidCredentials",
+                        isReAuth = false
                     )
                     return
                 }
@@ -438,7 +441,8 @@
             isAggregated = false,
             attempts = ++attempts,
             result = Result.Failed,
-            reason = errorType
+            reason = errorType,
+            isReAuth = false
         )
 
         LOG.error(e) { errorMessage }

plugins/core/jetbrains-community/src/software/aws/toolkits/jetbrains/core/gettingstarted/editor/GettingStartedPanelUtils.kt

@@ -181,6 +181,7 @@ enum class SourceOfEntry {
     FIRST_STARTUP,
     Q,
     AMAZONQ_CHAT_PANEL,
+    LOGIN_BROWSER,
     UNKNOWN,
     ;
     override fun toString(): String {

plugins/core/jetbrains-community/src/software/aws/toolkits/jetbrains/core/webview/LoginBrowser.kt

@@ -39,6 +39,7 @@
 import software.aws.toolkits.jetbrains.core.credentials.sso.PendingAuthorization
 import software.aws.toolkits.jetbrains.core.credentials.sso.bearer.InteractiveBearerTokenProvider
 import software.aws.toolkits.jetbrains.core.credentials.ssoErrorMessageFromException
+import software.aws.toolkits.jetbrains.core.gettingstarted.editor.SourceOfEntry
 import software.aws.toolkits.jetbrains.utils.pluginAwareExecuteOnPooledThread
 import software.aws.toolkits.jetbrains.utils.pollFor
 import software.aws.toolkits.resources.AwsCoreBundle
@@ -87,7 +88,7 @@
 
     private var browserOpenTimer: Timer? = null
 
-    private fun startBrowserOpenTimer(startUrl: String, ssoRegion: String) {
+    private fun startBrowserOpenTimer(startUrl: String, ssoRegion: String, scopes: List<String>) {
         browserOpenTimer = Timer()
         browserOpenTimer?.schedule(
             object : TimerTask() {
@@ -103,7 +104,11 @@
                     AuthTelemetry.addConnection(
                         result = Result.Failed,
                         reason = "Browser authentication idle for more than 15min",
-                        credentialSourceId = if (startUrl == SONO_URL) CredentialSourceId.AwsId else CredentialSourceId.IamIdentityCenter
+                        credentialSourceId = if (startUrl == SONO_URL) CredentialSourceId.AwsId else CredentialSourceId.IamIdentityCenter,
+                        isAggregated = false,
+                        source = SourceOfEntry.LOGIN_BROWSER.toString(),
+                        featureId = getFeatureId(scopes),
+                        isReAuth = isReAuth(scopes, startUrl)
                     )
                     stopAndClearBrowserOpenTimer()
                 }
@@ -121,7 +126,7 @@
     }
 
     protected val onPendingToken: (InteractiveBearerTokenProvider) -> Unit = { provider ->
-        startBrowserOpenTimer(provider.startUrl, provider.region)
+        startBrowserOpenTimer(provider.startUrl, provider.region, provider.scopes)
         projectCoroutineScope(project).launch {
             val authorization = pollForAuthorization(provider)
             if (authorization != null) {
@@ -180,6 +185,7 @@
 
     open fun loginBuilderId(scopes: List<String>) {
         val isReauth = isReAuth(scopes, SONO_URL)
+        val featureId = getFeatureId(scopes)
         val onError: (Exception) -> Unit = { e ->
             stopAndClearBrowserOpenTimer()
             isUserCancellation(e)
@@ -196,7 +202,10 @@
                 result = Result.Failed,
                 credentialSourceId = CredentialSourceId.AwsId,
                 reason = e.message,
-                isReAuth = isReauth
+                isReAuth = isReauth,
+                featureId = featureId,
+                isAggregated = false,
+                source = SourceOfEntry.LOGIN_BROWSER.toString()
             )
         }
         val onSuccess: () -> Unit = {
@@ -212,7 +221,10 @@
             AuthTelemetry.addConnection(
                 result = Result.Succeeded,
                 credentialSourceId = CredentialSourceId.AwsId,
-                isReAuth = isReauth
+                isReAuth = isReauth,
+                featureId = featureId,
+                isAggregated = true,
+                source = SourceOfEntry.LOGIN_BROWSER.toString()
             )
         }
 
@@ -239,7 +251,7 @@
         region: AwsRegion,
     ): Pair<(Exception) -> Unit, () -> Unit> {
         val isReAuth = isReAuth(scopes, url)
-
+        val featureId = getFeatureId(scopes)
         val onError: (Exception) -> Unit = { e ->
             stopAndClearBrowserOpenTimer()
             val message = ssoErrorMessageFromException(e)
@@ -269,6 +281,9 @@
                 credentialSourceId = CredentialSourceId.IamIdentityCenter,
                 reason = message,
                 isReAuth = isReAuth,
+                featureId = featureId,
+                isAggregated = false,
+                source = SourceOfEntry.LOGIN_BROWSER.toString()
             )
         }
         val onSuccess: () -> Unit = {
@@ -286,7 +301,10 @@
                 project = null,
                 result = Result.Succeeded,
                 isReAuth = isReAuth,
-                credentialSourceId = CredentialSourceId.IamIdentityCenter
+                credentialSourceId = CredentialSourceId.IamIdentityCenter,
+                featureId = featureId,
+                isAggregated = true,
+                source = SourceOfEntry.LOGIN_BROWSER.toString()
             )
         }
         return Pair(onError, onSuccess)

plugins/core/jetbrains-community/src/software/aws/toolkits/jetbrains/core/webview/WebviewTelemetryUtils.kt

@@ -4,7 +4,10 @@
 package software.aws.toolkits.jetbrains.core.webview
 
 import com.intellij.openapi.util.registry.Registry
+import software.aws.toolkits.jetbrains.core.credentials.sono.CODECATALYST_SCOPES
+import software.aws.toolkits.jetbrains.core.credentials.sono.Q_SCOPES
 import software.aws.toolkits.telemetry.AuthType
+import software.aws.toolkits.telemetry.FeatureId
 
 fun getAuthType(region: String = "us-east-1"): AuthType {
     val isCommercialRegion = !region.startsWith("us-gov") && !region.startsWith("us-iso") && !region.startsWith("cn")
@@ -14,3 +17,12 @@
         return AuthType.DeviceCode
     }
 }
+
+fun getFeatureId(scopes: List<String>): FeatureId =
+    if (scopes.intersect(Q_SCOPES.toSet()).isNotEmpty()) {
+        FeatureId.Q
+    } else if (scopes.intersect(CODECATALYST_SCOPES.toSet()).isNotEmpty()) {
+        FeatureId.Codecatalyst
+    } else {
+        FeatureId.AwsExplorer
+    }

plugins/toolkit/jetbrains-core/src/software/aws/toolkits/jetbrains/core/gettingstarted/GettingStartedOnStartup.kt

@@ -43,7 +43,8 @@
                     featureId = FeatureId.Unknown,
                     credentialSourceId = CredentialSourceId.Unknown,
                     isAggregated = true,
-                    result = Result.Succeeded
+                    result = Result.Succeeded,
+                    isReAuth = false
                 )
                 AuthTelemetry.addedConnections(
                     project,
@@ -66,7 +67,8 @@
                 credentialSourceId = CredentialSourceId.Unknown,
                 isAggregated = false,
                 result = Result.Failed,
-                reason = "Error opening getting started panel"
+                reason = "Error opening getting started panel",
+                isReAuth = false
             )
             AuthTelemetry.addedConnections(
                 project,

plugins/toolkit/jetbrains-core/src/software/aws/toolkits/jetbrains/core/gettingstarted/ToolkitGettingStartedAuthUtils.kt

@@ -89,7 +89,8 @@
             credentialSourceId = authenticationDialog.authType,
             isAggregated = true,
             attempts = authenticationDialog.attempts + 1,
-            result = Result.Succeeded
+            result = Result.Succeeded,
+            isReAuth = false
         )
         AuthTelemetry.addedConnections(
             project,
@@ -110,6 +111,7 @@
             isAggregated = false,
             attempts = authenticationDialog.attempts + 1,
             result = Result.Cancelled,
+            isReAuth = false
         )
     }
     return isAuthenticationSuccessful
@@ -157,7 +159,8 @@
             credentialSourceId = authenticationDialog.authType,
             isAggregated = true,
             attempts = authenticationDialog.attempts + 1,
-            result = Result.Succeeded
+            result = Result.Succeeded,
+            isReAuth = false
         )
         AuthTelemetry.addedConnections(
             project,
@@ -178,6 +181,7 @@
             isAggregated = false,
             attempts = authenticationDialog.attempts + 1,
             result = Result.Cancelled,
+            isReAuth = false
         )
     }
     return isAuthSuccessful

plugins/toolkit/jetbrains-core/src/software/aws/toolkits/jetbrains/core/gettingstarted/editor/GettingStartedPanel.kt

@@ -957,7 +957,8 @@
                                             oldConnectionCount,
                                             initialEnabledConnection,
                                             isFirstInstance,
-                                            connectionInitiatedFromExplorer
+                                            connectionInitiatedFromExplorer,
+                                            isReauth = true
                                         ),
                                         panelReauthenticationRequired
                                     )