From dd730d6e01b6649dbbdec0099f155747eebfa499 Mon Sep 17 00:00:00 2001
From: Kshitij Pramod Ghongadi <kshitij.pramod@navi.com>
Date: Thu, 4 Jul 2024 23:36:37 +0530
Subject: [PATCH] TP-72289 | Wheel picker | Fix scroll issue (#11716)

---
 .../purchase/journey/composables/reusable/WheelPicker.kt  | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/android/navi-insurance/src/main/java/com/navi/insurance/pre/purchase/journey/composables/reusable/WheelPicker.kt b/android/navi-insurance/src/main/java/com/navi/insurance/pre/purchase/journey/composables/reusable/WheelPicker.kt
index b8b09d0cf4..acba7e962b 100644
--- a/android/navi-insurance/src/main/java/com/navi/insurance/pre/purchase/journey/composables/reusable/WheelPicker.kt
+++ b/android/navi-insurance/src/main/java/com/navi/insurance/pre/purchase/journey/composables/reusable/WheelPicker.kt
@@ -156,8 +156,9 @@ fun OpenAgePicker(
     onDismiss: (value: PickerItemData) -> Unit
 ) {
     var selectedOption by remember { mutableStateOf(selectedOptions) }
+    val pickerState = rememberPickerState()
     Dialog(
-        onDismissRequest = { selectedOption?.let { onDismiss(it) } },
+        onDismissRequest = { selectedOption?.let { onDismiss(pickerState.selectedItem) } },
         properties = DialogProperties()
     ) {
         Box(modifier = Modifier.fillMaxWidth().wrapContentHeight()) {
@@ -189,10 +190,11 @@ fun OpenAgePicker(
                     InfiniteItemsPicker(
                         items = dataList,
                         firstIndex = dataList.indexOf(selectedOption) - 1,
-                        onItemSelected = { selectedOption = it }
+                        onItemSelected = { selectedOption = it },
+                        state = pickerState
                     )
                     Button(
-                        onClick = { selectedOption?.let { onDismiss(it) } },
+                        onClick = { selectedOption?.let { onDismiss(pickerState.selectedItem) } },
                         modifier =
                             Modifier.fillMaxWidth()
                                 .padding(