Add SSO_LOGIN feature IDs for SSO login flow requests

Author: muhammad-othman

sdk/src/Services/SSOOIDC/Custom/Internal/_bcl+netstandard/CoreAmazonSSOOIDC.cs

@@ -21,6 +21,8 @@
 using System.Threading.Tasks;
 using System.Xml;
 using Amazon.Runtime;
+using Amazon.Runtime.Internal;
+using Amazon.Runtime.Internal.UserAgent;
 using Amazon.Runtime.Internal.Util;
 using Amazon.Runtime.SharedInterfaces;
 using Amazon.SSOOIDC.Model;
@@ -70,6 +72,7 @@ public static GetSsoTokenResponse GetSsoToken(IAmazonSSOOIDC client, GetSsoToken
         {
             // PkceFlowOptions is a newer property that must be specified when using the PKCE flow. If it's null, we'll default to the device code flow.
             var useDeviceCodeFlow = request.PkceFlowOptions == null;
+            var featureId = useDeviceCodeFlow ? UserAgentFeatureId.SSO_LOGIN_DEVICE : UserAgentFeatureId.SSO_LOGIN_AUTH;
 
             // Identity Center supports using the start URL as the issuer URL (if one was not provided).
             var issuerUrl = request.PkceFlowOptions?.IssuerUrl ?? request.StartUrl;
@@ -113,6 +116,7 @@ public static GetSsoTokenResponse GetSsoToken(IAmazonSSOOIDC client, GetSsoToken
                     StartUrl = request.StartUrl,
                 };
                 InternalSDKUtils.ApplyValuesV2(startDeviceAuthorizationRequest, request.AdditionalProperties);
+                ((IAmazonWebServiceRequest)startDeviceAuthorizationRequest).UserAgentDetails.AddFeature(featureId);
 
                 var startDeviceAuthorizationResponse = client.StartDeviceAuthorization(startDeviceAuthorizationRequest);
 
@@ -158,6 +162,7 @@ public static GetSsoTokenResponse GetSsoToken(IAmazonSSOOIDC client, GetSsoToken
                     CodeVerifier = codeVerifier,
                 };
                 InternalSDKUtils.ApplyValuesV2(createTokenRequest, request.AdditionalProperties);
+                ((IAmazonWebServiceRequest)createTokenRequest).UserAgentDetails.AddFeature(featureId);
 
                 createTokenResponse = client.CreateToken(createTokenRequest);
             }
@@ -225,6 +230,7 @@ public static async Task<GetSsoTokenResponse> GetSsoTokenAsync(IAmazonSSOOIDC cl
         {
             // PkceFlowOptions is a newer property that must be specified when using the PKCE flow. If it's null, we'll default to the device code flow.
             var useDeviceCodeFlow = request.PkceFlowOptions == null;
+            var featureId = useDeviceCodeFlow ? UserAgentFeatureId.SSO_LOGIN_DEVICE : UserAgentFeatureId.SSO_LOGIN_AUTH;
 
             // Identity Center supports using the start URL as the issuer URL (if one was not provided).
             var issuerUrl = request.PkceFlowOptions?.IssuerUrl ?? request.StartUrl;
@@ -268,6 +274,7 @@ public static async Task<GetSsoTokenResponse> GetSsoTokenAsync(IAmazonSSOOIDC cl
                     StartUrl = request.StartUrl,
                 };
                 InternalSDKUtils.ApplyValuesV2(startDeviceAuthorizationRequest, request.AdditionalProperties);
+                ((IAmazonWebServiceRequest)startDeviceAuthorizationRequest).UserAgentDetails.AddFeature(featureId);
 
                 var startDeviceAuthorizationResponse =
                     await client.StartDeviceAuthorizationAsync(startDeviceAuthorizationRequest, cancellationToken).ConfigureAwait(false);
@@ -324,6 +331,7 @@ public static async Task<GetSsoTokenResponse> GetSsoTokenAsync(IAmazonSSOOIDC cl
                     CodeVerifier = codeVerifier,
                 };
                 InternalSDKUtils.ApplyValuesV2(createTokenRequest, request.AdditionalProperties);
+                ((IAmazonWebServiceRequest)createTokenRequest).UserAgentDetails.AddFeature(featureId);
 
                 createTokenResponse = await client.CreateTokenAsync(createTokenRequest, cancellationToken).ConfigureAwait(false);
             }