[Summit prep] Add a search feature

feature/src/main/java/com/example/feature/countrylist/CountryListAdapter.kt

@@ -29,7 +29,11 @@ fun CountryListAdapter(
             viewModel.onCountrySelect(country)
         },
         onRefreshTap = { viewModel.onRefreshTap() },
-        onFailOtherTap = { viewModel.onFailOtherTap() }
+        onFailOtherTap = { viewModel.onFailOtherTap() },
+        onSearchQueryChange = { query ->
+            viewModel.updateSearchQuery(query)
+        },
+        filteredContinents = viewModel.filteredContinents
     )
 
     FloatingAlertNotifier(

feature/src/main/java/com/example/feature/countrylist/CountryListPage.kt

@@ -5,6 +5,7 @@ import androidx.compose.foundation.layout.*
 import androidx.compose.foundation.shape.RoundedCornerShape
 import androidx.compose.material.Button
 import androidx.compose.material.Text
+import androidx.compose.material.TextField
 import androidx.compose.runtime.Composable
 import androidx.compose.ui.Alignment
 import androidx.compose.ui.Modifier
@@ -15,8 +16,8 @@ import androidx.compose.ui.tooling.preview.Preview
 import androidx.compose.ui.unit.dp
 import com.example.domainmodels.Continent
 import com.example.domainmodels.Country
-import com.example.feature.countrylist.componenets.CountryListButton
-import com.example.feature.countrylist.componenets.CountryListList
+import com.example.feature.countrylist.components.CountryListButton
+import com.example.feature.countrylist.components.CountryListList
 import com.example.features.R
 import com.example.uicomponents.library.ProgressIndicator
 import com.example.viewmodels.CountryListViewModel
@@ -26,14 +27,30 @@ fun CountryListPage(
     listUiState: CountryListViewModel.UiState,
     onCountrySelect: ((Country) -> Unit)? = null,
     onRefreshTap: (() -> Unit)? = null,
-    onFailOtherTap: (() -> Unit)? = null
+    onFailOtherTap: (() -> Unit)? = null,
+    onSearchQueryChange: ((String) -> Unit)? = null,
+    filteredContinents: List<Continent> = emptyList()
 ) {
     Box {
         ProgressIndicator(isLoading = listUiState.isLoading)
         Column(modifier = Modifier.fillMaxHeight()) {
+            Row(
+                modifier = Modifier
+                    .fillMaxWidth()
+                    .padding(8.dp),
+                verticalAlignment = Alignment.CenterVertically
+            ) {
+                TextField(
+                    value = listUiState.searchQuery,
+                    onValueChange = { onSearchQueryChange?.invoke(it) },
+                    label = { Text("Search countries") },
+                    modifier = Modifier.fillMaxWidth(),
+                    singleLine = true
+                )
+            }
             Column(modifier = Modifier.weight(1f)) {
                 CountryListList(
-                    list = listUiState.continents,
+                    list = filteredContinents,
                     onClick = onCountrySelect
                 )
             }

feature/src/main/java/com/example/feature/countrylist/componenets/CountrySelectingButton.kt → feature/src/main/java/com/example/feature/countrylist/components/CountrySelectingButton.kt

@@ -1,4 +1,4 @@
-package com.example.feature.countrylist.componenets
+package com.example.feature.countrylist.components
 
 import androidx.compose.foundation.layout.fillMaxWidth
 import androidx.compose.foundation.layout.padding

feature/src/main/java/com/example/feature/countrylist/componenets/CountrySelectingList.kt → feature/src/main/java/com/example/feature/countrylist/components/CountrySelectingList.kt

@@ -1,4 +1,4 @@
-package com.example.feature.countrylist.componenets
+package com.example.feature.countrylist.components
 
 import androidx.compose.foundation.ExperimentalFoundationApi
 import androidx.compose.foundation.background

viewmodels/src/main/java/com/example/viewmodels/CountryListViewModel.kt

@@ -20,8 +20,28 @@ class CountryListViewModel(
         val error: CountryListError? = null,
         val serverStatus: ServerStatus? = null,
         val navigationTarget: Country? = null,
+        var searchQuery: String = "",
     )
 
+    val filteredContinents: List<Continent>
+        get() {
+            val query = _state.value?.searchQuery.orEmpty()
+            return _state.value?.continents
+                ?.map { continent ->
+                    val filteredCountries = continent.countries.filter {
+                        it.countryName.contains(query, ignoreCase = true) ||
+                                it.regionCode.contains(query, ignoreCase = true)
+                    }
+                    continent.copy(countries = filteredCountries)
+                }
+                ?.filter { it.countries.isNotEmpty() }
+                ?: emptyList()
+        }
+
+    fun updateSearchQuery(query: String) {
+        _state.onNext(_state.value!!.copy(searchQuery = query))
+    }
+
     private val _state: BehaviorSubject<UiState> = BehaviorSubject.createDefault(UiState())
     val state: Observable<UiState> = _state
 

viewmodels/src/test/kotlin/com/example/viewmodels/CountryListViewModelTests.kt

@@ -80,4 +80,44 @@ class CountryListViewModelTests: KoinTest {
             }
         }
     }
+
+    @Nested
+    @DisplayName("filteredContinents")
+    inner class FilteredContinents {
+        @Test
+        fun `returns only continents with matching countries`() {
+            // Arrange: set up continents with countries
+            val continents = listOf(
+                com.example.domainmodels.Continent(
+                    name = "Europe",
+                    countries = listOf(
+                        com.example.domainmodels.Country(regionCode = "FR"),
+                        com.example.domainmodels.Country(regionCode = "DE")
+                    )
+                ),
+                com.example.domainmodels.Continent(
+                    name = "Asia",
+                    countries = listOf(
+                        com.example.domainmodels.Country(regionCode = "JP"),
+                        com.example.domainmodels.Country(regionCode = "CN")
+                    )
+                )
+            )
+            // Set continents in state
+            val stateField = CountryListViewModel::class.java.getDeclaredField("_state")
+            stateField.isAccessible = true
+            val subject = stateField.get(viewModel) as io.reactivex.rxjava3.subjects.BehaviorSubject<CountryListViewModel.UiState>
+            subject.onNext(CountryListViewModel.UiState(continents = continents))
+
+            // Act: update search query
+            viewModel.updateSearchQuery("fr")
+
+            // Assert: only Europe with France should match
+            val filtered = viewModel.filteredContinents
+            filtered.size.shouldBeEqualTo(1)
+            filtered[0].name.shouldBeEqualTo("Europe")
+            filtered[0].countries.size.shouldBeEqualTo(1)
+            filtered[0].countries[0].regionCode.shouldBeEqualTo("FR")
+        }
+    }
 }