Implement functionality for node selection

Signed-off-by: Elly Kitoto <[email protected]>

android/engine/src/main/java/org/smartregister/fhircore/engine/datastore/PreferenceDataStore.kt

@@ -54,6 +54,7 @@ class PreferenceDataStore @Inject constructor(@ApplicationContext val context: C
   companion object Keys {
     val APP_ID by lazy { stringPreferencesKey("appId") }
     val LANG by lazy { stringPreferencesKey("lang") }
+    val SYNC_LOCATION_IDS by lazy { stringPreferencesKey("syncLocationIds") }
     val MIGRATION_VERSION by lazy { intPreferencesKey("migrationVersion") }
   }
 }

android/engine/src/main/java/org/smartregister/fhircore/engine/datastore/serializers/PractitionerDetailsDataStoreSerializer.kt

@@ -34,8 +34,8 @@ object PractitionerDetailsDataStoreSerializer : Serializer<PractitionerDetails>
         deserializer = PractitionerDetails.serializer(),
         string = input.readBytes().decodeToString(),
       )
-    } catch (e: SerializationException) {
-      Timber.tag(SerializerConstants.PROTOSTORE_SERIALIZER_TAG).d(e)
+    } catch (serializationException: SerializationException) {
+      Timber.e(serializationException)
       defaultValue
     }
   }

android/engine/src/main/java/org/smartregister/fhircore/engine/datastore/serializers/UserInfoDataStoreSerializer.kt

@@ -34,8 +34,8 @@ object UserInfoDataStoreSerializer : Serializer<UserInfo> {
         deserializer = UserInfo.serializer(),
         string = input.readBytes().decodeToString(),
       )
-    } catch (e: SerializationException) {
-      Timber.tag(SerializerConstants.PROTOSTORE_SERIALIZER_TAG).d(e)
+    } catch (serializationException: SerializationException) {
+      Timber.e(serializationException)
       defaultValue
     }
   }

android/engine/src/main/java/org/smartregister/fhircore/engine/ui/multiselect/MultiSelectView.kt

@@ -52,8 +52,8 @@ import org.smartregister.fhircore.engine.ui.theme.AppTheme
 @Composable
 fun <T> ColumnScope.MultiSelectView(
   rootNodeId: String,
-  treeNodeMap: SnapshotStateMap<String, TreeNode<T>>,
-  selectedNodes: SnapshotStateMap<String, ToggleableState>,
+  treeNodeMap: Map<String, TreeNode<T>>,
+  selectedNodes: MutableMap<String, ToggleableState>,
   depth: Int = 0,
   content: @Composable (TreeNode<T>) -> Unit,
 ) {
@@ -91,7 +91,7 @@ fun <T> ColumnScope.MultiSelectView(
 
 @Composable
 fun <T> MultiSelectCheckbox(
-  selectedNodes: SnapshotStateMap<String, ToggleableState>,
+  selectedNodes: MutableMap<String, ToggleableState>,
   treeNodeMap: Map<String, TreeNode<T>>,
   currentTreeNode: TreeNode<T>,
   depth: Int,

android/engine/src/main/java/org/smartregister/fhircore/engine/ui/multiselect/TreeMap.kt

@@ -31,7 +31,7 @@ object TreeMap {
   fun <T> populateLookupMap(
     items: List<TreeNode<T>>,
     lookup: MutableMap<String, TreeNode<T>>,
-  ): MutableMap<String, TreeNode<T>> {
+  ): Map<String, TreeNode<T>> {
     items.forEach { item ->
       val childNode = findOrCreate(item.id, item, lookup)
       val parentNode = findOrCreate(item.parentId, item, lookup)

android/quest/src/main/java/org/smartregister/fhircore/quest/ui/multiselect/MultiSelectBottomSheetFragment.kt

@@ -56,7 +56,8 @@ class MultiSelectBottomSheetFragment() : BottomSheetDialogFragment() {
             title = bottomSheetArgs.screenTitle,
             onDismiss = { dismiss() },
             searchTextState = multiSelectViewModel.searchTextState,
-            onTextChanged = multiSelectViewModel::onTextChanged,
+            onSearchTextChanged = multiSelectViewModel::onTextChanged,
+            onSelectionDone = multiSelectViewModel::onSelectionDone,
           )
         }
       }

android/quest/src/main/java/org/smartregister/fhircore/quest/ui/multiselect/MultiSelectBottomSheetView.kt

@@ -47,7 +47,6 @@ import androidx.compose.ui.res.stringResource
 import androidx.compose.ui.state.ToggleableState
 import androidx.compose.ui.text.TextStyle
 import androidx.compose.ui.text.font.FontWeight
-import androidx.compose.ui.text.input.TextFieldValue
 import androidx.compose.ui.text.style.TextAlign
 import androidx.compose.ui.unit.dp
 import androidx.compose.ui.unit.sp
@@ -59,12 +58,13 @@ import org.smartregister.fhircore.engine.ui.theme.DividerColor
 @Composable
 fun MultiSelectBottomSheetView(
   rootNodeIds: SnapshotStateList<String>,
-  treeNodeMap: SnapshotStateMap<String, TreeNode<String>>,
+  treeNodeMap: Map<String, TreeNode<String>>,
   selectedNodes: SnapshotStateMap<String, ToggleableState>,
   title: String?,
   onDismiss: () -> Unit,
-  searchTextState: MutableState<TextFieldValue>,
-  onTextChanged: (String) -> Unit,
+  searchTextState: MutableState<String>,
+  onSearchTextChanged: (String) -> Unit,
+  onSelectionDone: (() -> Unit) -> Unit,
 ) {
   Column(modifier = Modifier.fillMaxWidth()) {
     Row(
@@ -91,20 +91,12 @@ fun MultiSelectBottomSheetView(
     ) {
       OutlinedTextField(
         value = searchTextState.value,
-        onValueChange = { value ->
-          searchTextState.value = value
-          onTextChanged(value.text)
-        },
+        onValueChange = { value -> onSearchTextChanged(value) },
         modifier = Modifier.fillMaxWidth(),
         textStyle = TextStyle(fontSize = 18.sp),
         trailingIcon = {
-          if (searchTextState.value.text.isNotEmpty()) {
-            IconButton(
-              onClick = {
-                searchTextState.value = TextFieldValue("")
-                onTextChanged(searchTextState.value.text)
-              },
-            ) {
+          if (searchTextState.value.isNotEmpty()) {
+            IconButton(onClick = { onSearchTextChanged("") }) {
               Icon(
                 Icons.Default.Close,
                 contentDescription = "",
@@ -137,7 +129,7 @@ fun MultiSelectBottomSheetView(
 
       item {
         Button(
-          onClick = { /*TODO Get selected nodes*/},
+          onClick = { onSelectionDone(onDismiss) },
           modifier = Modifier.fillMaxWidth().padding(vertical = 16.dp, horizontal = 8.dp),
         ) {
           Text(

android/quest/src/main/java/org/smartregister/fhircore/quest/ui/multiselect/MultiSelectViewModel.kt

@@ -21,14 +21,15 @@ import androidx.compose.runtime.mutableStateOf
 import androidx.compose.runtime.snapshots.SnapshotStateList
 import androidx.compose.runtime.snapshots.SnapshotStateMap
 import androidx.compose.ui.state.ToggleableState
-import androidx.compose.ui.text.input.TextFieldValue
 import androidx.lifecycle.ViewModel
 import androidx.lifecycle.viewModelScope
 import com.google.android.fhir.logicalId
 import dagger.hilt.android.lifecycle.HiltViewModel
 import javax.inject.Inject
+import kotlinx.coroutines.flow.firstOrNull
 import kotlinx.coroutines.launch
 import org.smartregister.fhircore.engine.data.local.DefaultRepository
+import org.smartregister.fhircore.engine.datastore.PreferenceDataStore
 import org.smartregister.fhircore.engine.domain.model.MultiSelectViewConfig
 import org.smartregister.fhircore.engine.ui.multiselect.TreeMap
 import org.smartregister.fhircore.engine.ui.multiselect.TreeNode
@@ -41,15 +42,29 @@ class MultiSelectViewModel
 constructor(
   val defaultRepository: DefaultRepository,
   val fhirPathDataExtractor: FhirPathDataExtractor,
+  val preferenceDataStore: PreferenceDataStore,
 ) : ViewModel() {
 
-  val searchTextState: MutableState<TextFieldValue> = mutableStateOf(TextFieldValue())
+  val searchTextState: MutableState<String> = mutableStateOf("")
   val rootNodeIds: SnapshotStateList<String> = SnapshotStateList()
-  val lookupMap: SnapshotStateMap<String, TreeNode<String>> = SnapshotStateMap()
   val selectedNodes: SnapshotStateMap<String, ToggleableState> = SnapshotStateMap()
+  val lookupMap = SnapshotStateMap<String, TreeNode<String>>()
 
   fun populateLookupMap(multiSelectViewConfig: MultiSelectViewConfig) {
+    // Mark previously selected nodes
     viewModelScope.launch {
+      val previouslySelectedNodes =
+        preferenceDataStore.read(PreferenceDataStore.SYNC_LOCATION_IDS).firstOrNull()
+      if (!previouslySelectedNodes.isNullOrEmpty()) {
+        previouslySelectedNodes
+          .split(",")
+          .asSequence()
+          .map { it.split(":") }
+          .filter { it.size == 2 }
+          .map { Pair(it.first(), it.last()) }
+          .forEach { selectedNodes[it.first] = ToggleableState.valueOf(it.second) }
+      }
+
       val repositoryResourceDataList =
         defaultRepository.searchResourcesRecursively(
           fhirResourceConfig = multiSelectViewConfig.resourceConfig,
@@ -85,6 +100,17 @@ constructor(
   }
 
   fun onTextChanged(searchTerm: String) {
-    searchTextState.value = TextFieldValue(searchTerm)
+    searchTextState.value = searchTerm
+  }
+
+  fun onSelectionDone(dismiss: () -> Unit) {
+    viewModelScope.launch {
+      // Consider using a proto-datastore here
+      preferenceDataStore.write(
+        PreferenceDataStore.SYNC_LOCATION_IDS,
+        selectedNodes.map { "${it.key}:${it.value}" }.joinToString(","),
+      )
+      dismiss()
+    }
   }
 }