Merge pull request #765 from android/refactor/notification-dot-modifier

Notification dot improvements

app/src/main/java/com/google/samples/apps/nowinandroid/ui/NiaApp.kt

@@ -15,6 +15,7 @@
  */
 
 package com.google.samples.apps.nowinandroid.ui
+
 import androidx.compose.foundation.layout.Column
 import androidx.compose.foundation.layout.ExperimentalLayoutApi
 import androidx.compose.foundation.layout.Row
@@ -46,6 +47,7 @@ import androidx.compose.runtime.saveable.rememberSaveable
 import androidx.compose.runtime.setValue
 import androidx.compose.ui.ExperimentalComposeUiApi
 import androidx.compose.ui.Modifier
+import androidx.compose.ui.composed
 import androidx.compose.ui.draw.drawWithContent
 import androidx.compose.ui.geometry.Offset
 import androidx.compose.ui.graphics.Color
@@ -129,6 +131,8 @@ fun NiaApp(
                 )
             }
 
+            val unreadDestinations by appState.topLevelDestinationsWithUnreadResources.collectAsStateWithLifecycle()
+
             Scaffold(
                 modifier = Modifier.semantics {
                     testTagsAsResourceId = true
@@ -139,7 +143,6 @@ fun NiaApp(
                 snackbarHost = { SnackbarHost(snackbarHostState) },
                 bottomBar = {
                     if (appState.shouldShowBottomBar) {
-                        val unreadDestinations by appState.topLevelDestinationsWithUnreadResources.collectAsStateWithLifecycle()
                         NiaBottomBar(
                             destinations = appState.topLevelDestinations,
                             destinationsWithUnreadResources = unreadDestinations,
@@ -164,6 +167,7 @@ fun NiaApp(
                     if (appState.shouldShowNavRail) {
                         NiaNavRail(
                             destinations = appState.topLevelDestinations,
+                            destinationsWithUnreadResources = unreadDestinations,
                             onNavigateToDestination = appState::navigateToTopLevelDestination,
                             currentDestination = appState.currentDestination,
                             modifier = Modifier
@@ -208,6 +212,7 @@ fun NiaApp(
 @Composable
 private fun NiaNavRail(
     destinations: List<TopLevelDestination>,
+    destinationsWithUnreadResources: Set<TopLevelDestination>,
     onNavigateToDestination: (TopLevelDestination) -> Unit,
     currentDestination: NavDestination?,
     modifier: Modifier = Modifier,
@@ -215,6 +220,7 @@ private fun NiaNavRail(
     NiaNavigationRail(modifier = modifier) {
         destinations.forEach { destination ->
             val selected = currentDestination.isTopLevelDestinationInHierarchy(destination)
+            val hasUnread = destinationsWithUnreadResources.contains(destination)
             NiaNavigationRailItem(
                 selected = selected,
                 onClick = { onNavigateToDestination(destination) },
@@ -237,7 +243,7 @@ private fun NiaNavRail(
                     }
                 },
                 label = { Text(stringResource(destination.iconTextId)) },
-
+                modifier = if (hasUnread) Modifier.notificationDot() else Modifier,
             )
         }
     }
@@ -279,16 +285,16 @@ private fun NiaBottomBar(
                     }
                 },
                 label = { Text(stringResource(destination.iconTextId)) },
-                modifier = if (hasUnread) notificationDot() else Modifier,
+                modifier = if (hasUnread) Modifier.notificationDot() else Modifier,
             )
         }
     }
 }
 
-@Composable
+private fun Modifier.notificationDot(): Modifier =
-private fun notificationDot(): Modifier {
+    composed {
         val tertiaryColor = MaterialTheme.colorScheme.tertiary
-    return Modifier.drawWithContent {
+        drawWithContent {
             drawContent()
             drawCircle(
                 tertiaryColor,
@@ -302,7 +308,7 @@ private fun notificationDot(): Modifier {
                 ),
             )
         }
-}
+    }
 
 private fun NavDestination?.isTopLevelDestinationInHierarchy(destination: TopLevelDestination) =
     this?.hierarchy?.any {