@@ -121,7 +121,6 @@ fun loginSso(
121
121
onSuccess : () -> Unit = {},
122
122
metadata : ConnectionMetadata ? = null
123
123
): AwsBearerTokenConnection ? {
124
- val source = metadata
125
124
fun createAndAuthNewConnection (profile : AuthProfile ): AwsBearerTokenConnection ? {
126
125
val authManager = ToolkitAuthManager .getInstance()
127
126
val connection = try {
@@ -130,6 +129,7 @@ fun loginSso(
130
129
project = project,
131
130
connection = transientConnection,
132
131
onPendingToken = onPendingToken,
132
+ source = metadata?.sourceId,
133
133
)
134
134
}
135
135
} catch (e: Exception ) {
@@ -233,7 +233,8 @@ fun reauthConnectionIfNeeded(
233
233
project : Project ? ,
234
234
connection : ToolkitConnection ,
235
235
onPendingToken : (InteractiveBearerTokenProvider ) -> Unit = {},
236
- isReAuth : Boolean = false
236
+ isReAuth : Boolean = false,
237
+ source : String? = null,
237
238
): BearerTokenProvider {
238
239
val tokenProvider = (connection.getConnectionSettings() as TokenConnectionSettings ).tokenProvider.delegate as BearerTokenProvider
239
240
if (tokenProvider is InteractiveBearerTokenProvider ) {
@@ -250,12 +251,14 @@ fun reauthConnectionIfNeeded(
250
251
credentialStartUrl = startUrl,
251
252
credentialSourceId = getCredentialIdForTelemetry(connection),
252
253
isReAuth = true ,
253
- result = Result .Succeeded
254
+ result = Result .Succeeded ,
255
+ source = source,
254
256
)
255
257
recordAddConnection(
256
258
credentialSourceId = getCredentialIdForTelemetry(connection),
257
259
isReAuth = true ,
258
- result = Result .Succeeded
260
+ result = Result .Succeeded ,
261
+ source = source,
259
262
)
260
263
}
261
264
} catch (e: Exception ) {
@@ -265,12 +268,14 @@ fun reauthConnectionIfNeeded(
265
268
credentialStartUrl = startUrl,
266
269
credentialSourceId = getCredentialIdForTelemetry(connection),
267
270
isReAuth = true ,
268
- result = result
271
+ result = result,
272
+ source = source,
269
273
)
270
274
recordAddConnection(
271
275
credentialSourceId = getCredentialIdForTelemetry(connection),
272
276
isReAuth = true ,
273
- result = result
277
+ result = result,
278
+ source = source,
274
279
)
275
280
}
276
281
0 commit comments