Update TabLayout demos in MDC catalog to include badging.

PiperOrigin-RevId: 250356916

Author: wcshi

catalog/java/io/material/catalog/tabs/TabsControllableDemoFragment.java

@@ -27,6 +27,8 @@
 import androidx.annotation.StringRes;
 import com.google.android.material.tabs.TabLayout;
 import com.google.android.material.tabs.TabLayout.LabelVisibility;
+import com.google.android.material.tabs.TabLayout.OnTabSelectedListener;
+import com.google.android.material.tabs.TabLayout.Tab;
 import androidx.appcompat.widget.SwitchCompat;
 import android.view.LayoutInflater;
 import android.view.View;
@@ -74,6 +76,7 @@ public View onCreateDemoView(
     setupViewPager();
     setAllTabLayoutIcons(ICON_DRAWABLE_RES);
     setAllTabLayoutText(LABEL_STRING_RES);
+    setAllTabLayoutBadges();
 
     SwitchCompat iconsToggle = view.findViewById(R.id.toggle_icons_switch);
     iconsToggle.setOnCheckedChangeListener(
@@ -211,6 +214,33 @@ private void setTabLayoutText(TabLayout tabLayout, @StringRes int stringResId) {
     }
   }
 
+  private void setAllTabLayoutBadges() {
+    for (TabLayout tabLayout : tabLayouts) {
+      setupBadging(tabLayout);
+      tabLayout.addOnTabSelectedListener(
+          new OnTabSelectedListener() {
+            @Override
+            public void onTabSelected(Tab tab) {
+              tab.removeBadge();
+            }
+
+            @Override
+            public void onTabUnselected(Tab tab) {}
+
+            @Override
+            public void onTabReselected(Tab tab) {
+              tab.removeBadge();
+            }
+          });
+    }
+  }
+
+  private void setupBadging(TabLayout tabLayout) {
+    tabLayout.getTabAt(0).showBadge().setNumber(1);
+    tabLayout.getTabAt(1).showBadge().setNumber(88);
+    tabLayout.getTabAt(2).showBadge().setNumber(999);
+  }
+
   private void setLabelVisibility(TabLayout tabLayout, @LabelVisibility int mode) {
      for (int i = 0; i < tabLayout.getTabCount(); i++) {
       tabLayout.getTabAt(i).setTabLabelVisibility(mode);

catalog/java/io/material/catalog/tabs/TabsMainDemoFragment.java

@@ -21,23 +21,92 @@
 import android.os.Bundle;
 import androidx.annotation.LayoutRes;
 import androidx.annotation.Nullable;
+import com.google.android.material.badge.BadgeDrawable;
+import com.google.android.material.button.MaterialButton;
+import com.google.android.material.tabs.TabLayout;
+import com.google.android.material.tabs.TabLayout.OnTabSelectedListener;
+import com.google.android.material.tabs.TabLayout.Tab;
 import android.view.LayoutInflater;
 import android.view.View;
+import android.view.View.OnClickListener;
 import android.view.ViewGroup;
 import io.material.catalog.feature.DemoFragment;
+import io.material.catalog.feature.DemoUtils;
+import java.util.List;
 
 /** A fragment that displays the main tabs demo for the Catalog app. */
 public class TabsMainDemoFragment extends DemoFragment {
 
+  private List<TabLayout> tabLayouts;
+
   @Nullable
   @Override
   public View onCreateDemoView(
       LayoutInflater layoutInflater, @Nullable ViewGroup viewGroup, @Nullable Bundle bundle) {
-    return layoutInflater.inflate(getTabsContent(), viewGroup, false /* attachToRoot */);
+    View view = layoutInflater.inflate(getTabsContent(), viewGroup, /* attachToRoot= */ false);
+
+    tabLayouts = DemoUtils.findViewsWithType(view, TabLayout.class);
+    MaterialButton incrementBadgeNumberButton =
+        view.findViewById(R.id.increment_badge_number_button);
+    incrementBadgeNumberButton.setOnClickListener(
+        new OnClickListener() {
+          @Override
+          public void onClick(View v) {
+            incrementBadgeNumber();
+          }
+        });
+
+    setupBadging();
+    return view;
   }
 
   @LayoutRes
   protected int getTabsContent() {
     return R.layout.cat_tabs_main_content;
   }
+
+  private void setupBadging() {
+    for (TabLayout tabLayout : tabLayouts) {
+      // An icon only badge will be displayed.
+      tabLayout.getTabAt(0).showBadge();
+
+      // A badge with the text "99" will be displayed.
+      tabLayout.getTabAt(1).showBadge().setNumber(99);
+
+      // A badge with the text "999+" will be displayed.
+      tabLayout.getTabAt(2).showBadge().setNumber(9999);
+
+      tabLayout.addOnTabSelectedListener(
+          new OnTabSelectedListener() {
+            @Override
+            public void onTabSelected(Tab tab) {
+              clearAndHideBadge(tab.getPosition());
+            }
+
+            @Override
+            public void onTabUnselected(Tab tab) {}
+
+            @Override
+            public void onTabReselected(Tab tab) {
+              clearAndHideBadge(tab.getPosition());
+            }
+          });
+    }
+  }
+
+  private void incrementBadgeNumber() {
+    for (TabLayout tabLayout : tabLayouts) {
+      // Increase the badge number on the first tab position.
+      // In case the first tab has been selected and the badge was hidden, call #showBadge() instead
+      // of #getBadge() to ensure the badge is visible.
+      BadgeDrawable badgeDrawable = tabLayout.getTabAt(0).showBadge();
+      badgeDrawable.setNumber(badgeDrawable.getNumber() + 1);
+    }
+  }
+
+  private void clearAndHideBadge(int tabPosition) {
+    for (TabLayout tabLayout : tabLayouts) {
+      tabLayout.getTabAt(tabPosition).removeBadge();
+    }
+  }
 }

catalog/java/io/material/catalog/tabs/res/layout/cat_tabs_main_content.xml

@@ -96,4 +96,11 @@
 
   </com.google.android.material.tabs.TabLayout>
 
+  <Button
+      android:id="@+id/increment_badge_number_button"
+      android:layout_width="wrap_content"
+      android:layout_height="wrap_content"
+      android:layout_marginVertical="8dp"
+      android:layout_gravity="center_horizontal"
+      android:text="@string/cat_tabs_increment_badge_number"/>
 </LinearLayout>

catalog/java/io/material/catalog/tabs/res/values/strings.xml

@@ -67,4 +67,7 @@
   <string name="cat_tabs_label_plugins">Plugins</string>
   <string name="cat_tabs_label_more">More</string>
 
+  <!-- Badge increment button label -->
+  <string name="cat_tabs_increment_badge_number" description="The title of a button to increment the notification number. [CHAR_LIMIT=20]">+Badge number</string>
+
 </resources>