diff --git a/app/src/main/java/dev/msfjarvis/lobsters/MainActivity.kt b/app/src/main/java/dev/msfjarvis/lobsters/MainActivity.kt index 9195be14..a2dee873 100644 --- a/app/src/main/java/dev/msfjarvis/lobsters/MainActivity.kt +++ b/app/src/main/java/dev/msfjarvis/lobsters/MainActivity.kt @@ -62,14 +62,7 @@ fun LobstersApp() { val currentRoute = navBackStackEntry?.arguments?.getString(KEY_ROUTE) destinations.forEach { screen -> BottomNavigationItem( - icon = { - IconResource( - resourceId = when (screen) { - Destination.Hottest -> R.drawable.ic_whatshot_24px - Destination.Saved -> R.drawable.ic_favorite_24px - } - ) - }, + icon = { IconResource(resourceId = screen.badgeRes) }, label = { Text(stringResource(id = screen.labelRes)) }, selected = currentRoute == screen.route, onClick = { diff --git a/app/src/main/java/dev/msfjarvis/lobsters/ui/Destination.kt b/app/src/main/java/dev/msfjarvis/lobsters/ui/Destination.kt index 0c6ebc45..6f55ef51 100644 --- a/app/src/main/java/dev/msfjarvis/lobsters/ui/Destination.kt +++ b/app/src/main/java/dev/msfjarvis/lobsters/ui/Destination.kt @@ -1,5 +1,6 @@ package dev.msfjarvis.lobsters.ui +import androidx.annotation.DrawableRes import androidx.annotation.StringRes import dev.msfjarvis.lobsters.R @@ -9,7 +10,8 @@ import dev.msfjarvis.lobsters.R sealed class Destination( val route: String, @StringRes val labelRes: Int, + @DrawableRes val badgeRes: Int, ) { - object Hottest : Destination("hottest", R.string.hottest_posts) - object Saved : Destination("saved", R.string.saved_posts) + object Hottest : Destination("hottest", R.string.hottest_posts, R.drawable.ic_whatshot_24px) + object Saved : Destination("saved", R.string.saved_posts, R.drawable.ic_favorite_24px) }