Fix snapping behavior (#24)

darkokoa · Sep 12, 2024 · 2d5fc9c · 2d5fc9c
1 parent 89a15f3
commit 2d5fc9c
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/...e-wheel-picker/src/commonMain/kotlin/dev/darkokoa/datetimewheelpicker/core/WheelPicker.kt b/...e-wheel-picker/src/commonMain/kotlin/dev/darkokoa/datetimewheelpicker/core/WheelPicker.kt
@@ -92,12 +92,16 @@ internal fun WheelPicker(
 }
 
 private fun calculateSnappedItemIndex(lazyListState: LazyListState): Int {
-  var currentItemIndex = lazyListState.firstVisibleItemIndex
+  val currentItemIndex = lazyListState.firstVisibleItemIndex
+  val itemCount = lazyListState.layoutInfo.totalItemsCount
+  val offset = lazyListState.firstVisibleItemScrollOffset
+  val itemHeight = lazyListState.layoutInfo.visibleItemsInfo.firstOrNull()?.size ?: return currentItemIndex
 
-  if (lazyListState.firstVisibleItemScrollOffset != 0) {
-    currentItemIndex++
+  return if (offset > itemHeight / 2 && currentItemIndex < itemCount - 1) {
+    currentItemIndex + 1
+  } else {
+    currentItemIndex
   }
-  return currentItemIndex
 }
 
 @Composable