Update NavigationSuiteScaffold to align NavRail's icon to the center

Author: tiwiz

AdaptiveJetStream/jetstream/src/main/java/com/google/jetstream/presentation/app/withNavigationSuiteScaffold/AppWithNavigationSuiteScaffold.kt

@@ -19,6 +19,7 @@ package com.google.jetstream.presentation.app.withNavigationSuiteScaffold
 import androidx.compose.animation.AnimatedVisibility
 import androidx.compose.animation.slideInVertically
 import androidx.compose.animation.slideOutVertically
+import androidx.compose.foundation.layout.Arrangement
 import androidx.compose.foundation.layout.fillMaxSize
 import androidx.compose.foundation.layout.padding
 import androidx.compose.material3.Scaffold
@@ -82,20 +83,21 @@ fun AppWithNavigationSuiteScaffold(
     NavigationSuiteScaffold(
         modifier = modifier.fillMaxSize(),
         state = navigationSuiteScaffoldState,
-        navigationSuiteItems = {
-            navigationSuiteItems(appState.selectedScreen, screensInGlobalNavigation) {
+        navigationItems = {
+            AdaptiveAppNavigationItems(appState.selectedScreen, screensInGlobalNavigation) {
                 if (it != appState.selectedScreen) {
                     navController.navigate(it())
                 }
             }
             if (xrSession != null) {
-                toggleFullSpaceMode(
+                ToggleFullSpaceModeItems(
                     xrSession = xrSession,
                     isSpatialUiEnabled = isSpatialUiEnabled,
                     spatialConfiguration = spatialConfiguration
                 )
             }
-        }
+        },
+        navigationItemVerticalArrangement = Arrangement.Center
     ) {
         Scaffold(
             modifier = Modifier.fillMaxSize(),

AdaptiveJetStream/jetstream/src/main/java/com/google/jetstream/presentation/app/withNavigationSuiteScaffold/navigationSuiteItems.kt

@@ -20,7 +20,8 @@ import androidx.compose.foundation.layout.size
 import androidx.compose.material3.Icon
 import androidx.compose.material3.MaterialTheme
 import androidx.compose.material3.Text
-import androidx.compose.material3.adaptive.navigationsuite.NavigationSuiteScope
+import androidx.compose.material3.adaptive.navigationsuite.NavigationSuiteItem
+import androidx.compose.runtime.Composable
 import androidx.compose.ui.Modifier
 import androidx.compose.ui.graphics.vector.ImageVector
 import androidx.compose.ui.res.stringResource
@@ -31,13 +32,14 @@ import androidx.xr.runtime.Session
 import com.google.jetstream.R
 import com.google.jetstream.presentation.screens.Screens
 
-fun NavigationSuiteScope.navigationSuiteItems(
+@Composable
+fun AdaptiveAppNavigationItems(
     currentScreen: Screens,
     screens: List<Screens>,
     onSelectScreen: (Screens) -> Unit,
 ) {
     screens.forEach { screen ->
-        item(
+        NavigationSuiteItem(
             selected = screen == currentScreen,
             onClick = {
                 onSelectScreen(screen)
@@ -55,7 +57,8 @@ fun NavigationSuiteScope.navigationSuiteItems(
     }
 }
 
-fun NavigationSuiteScope.toggleFullSpaceMode(
+@Composable
+fun ToggleFullSpaceModeItems(
     xrSession: Session?,
     spatialConfiguration: SpatialConfiguration,
     isSpatialUiEnabled: Boolean,
@@ -73,7 +76,7 @@ fun NavigationSuiteScope.toggleFullSpaceMode(
             R.string.full_space_mode
         }
 
-        item(
+        NavigationSuiteItem(
             selected = false,
             onClick = {
                 if (isSpatialUiEnabled) {