Use intent to pass custom flags

aws-android-sdk-cognitoauth/src/main/java/com/amazonaws/mobileconnectors/cognitoauth/Auth.java

@@ -19,6 +19,7 @@
 
 import android.content.Context;
 import android.net.Uri;
+import android.os.Bundle;
 import android.util.Patterns;
 
 import com.amazonaws.mobileconnectors.cognitoauth.exceptions.AuthInvalidParameterException;
@@ -59,6 +60,11 @@ public final class Auth {
      */
     private final String idpIdentifier;
 
+    /**
+     * Bundle containing customization flags for chrome custom tab
+     */
+    private final Bundle extraCustomBundle;
+
     /**
      * This identifies the settings for additional userPool features.
      */
@@ -143,7 +149,8 @@ private Auth(final Context context,
                  final AuthHandler userHandler,
                  final boolean advancedSecurityDataCollectionFlag,
                  final String identityProvider,
-                 final String idpIdentifier) {
+                 final String idpIdentifier,
+                 final Bundle extraCustomBundle) {
         this.context = context;
         this.appWebDomain = appWebDomain;
         this.appId = appId;
@@ -158,6 +165,7 @@ private Auth(final Context context,
         this.advancedSecurityDataCollectionFlag = advancedSecurityDataCollectionFlag;
         this.identityProvider = identityProvider;
         this.idpIdentifier = idpIdentifier;
+        this.extraCustomBundle = extraCustomBundle;
         getCurrentUser();
     }
 
@@ -222,6 +230,11 @@ public static final class Builder {
          */
         private String mIdpIdentifier;
 
+        /**
+         * Bundle of flags to customize chrome custom tab UI
+         */
+        private Bundle mExtraCustomBundle;
+
         /**
          * Flag indicating if data collection for advanced security mode is enabled.
          * By default this is enabled.
@@ -400,6 +413,20 @@ public Builder setIdpIdentifier(final String mIdpIdentifier) {
             return this;
         }
 
+        /**
+         * Sets customization bundle. This allow to customize chrome custom tab.
+         * <p>
+         *     Optional. Set a bundle to customize UI
+         * </p>
+         * @param mExtraCustomBundle Optional: Pass to chrome custom tab a bundle of customization flags
+         * @return A reference to this builder.
+         */
+        @SuppressWarnings("checkstyle:hiddenfield")
+        public Builder setExtraCustomBundle(final Bundle mExtraCustomBundle) {
+            this.mExtraCustomBundle = mExtraCustomBundle;
+            return this;
+        }
+
         /**
          * Instantiates {@link Auth} with the set options.
          * @return {@link Auth}.
@@ -417,7 +444,8 @@ public Auth build() {
                     this.mUserHandler,
                     this.mAdvancedSecurityDataCollectionFlag,
                     this.mIdentityProvider,
-                    this.mIdpIdentifier);
+                    this.mIdpIdentifier,
+                    this.mExtraCustomBundle);
         }
 
 
@@ -559,6 +587,13 @@ public String getIdpIdentifier() {
         return idpIdentifier;
     }
 
+    /**
+     * @return Extra customization bundle for this {@link Auth} instance.
+     */
+    public Bundle getExtraCustomBundle() {
+        return extraCustomBundle;
+    }
+
     /**
      * Use this method to get tokens for a user, the tokens are returned though the callback.
      * {@link AuthHandler#onSuccess(AuthUserSession)}.

aws-android-sdk-cognitoauth/src/main/java/com/amazonaws/mobileconnectors/cognitoauth/AuthClient.java

@@ -531,6 +531,8 @@ private void launchCustomTabs(final Uri uri) {
     	try {
 	        CustomTabsIntent.Builder builder = new CustomTabsIntent.Builder(mCustomTabsSession);
 	        mCustomTabsIntent = builder.build();
+	        if(pool.getExtraCustomBundle() != null)
+	            mCustomTabsIntent.intent.putExtras(pool.getExtraCustomBundle());
 	        mCustomTabsIntent.intent.setPackage(ClientConstants.CHROME_PACKAGE);
 	        mCustomTabsIntent.intent.addFlags(Intent.FLAG_ACTIVITY_NO_HISTORY);
 	        mCustomTabsIntent.intent.addFlags(Intent.FLAG_ACTIVITY_NEW_TASK);