fix loading logic

Author: dosukoi-android

ui/view/src/main/java/jp/dosukoi/ui/view/top/MainActivity.kt

@@ -28,12 +28,8 @@ class MainActivity : AppCompatActivity() {
     private val myPageViewModel: MyPageViewModel by produceViewModels {
         myPageViewModelFactory.create(viewModel)
     }
-
-    @Inject
-    lateinit var searchViewModelFactory: SearchViewModel.Factory
-    private val searchViewModel: SearchViewModel by produceViewModels {
-        searchViewModelFactory.create(viewModel)
-    }
+    
+    private val searchViewModel: SearchViewModel by viewModels()
 
     @Inject
     lateinit var compositionLocalProvider: CompositionLocalProvider

ui/viewModel/src/main/java/jp/dosukoi/ui/viewmodel/search/SearchViewModel.kt

@@ -81,7 +81,13 @@ class SearchViewModel @Inject constructor(
             }
         }
         _searchUiState.update {
-            it.copy(searchState = LoadState.Loading)
+            when (val loadState = it.searchState) {
+                is LoadState.Loaded -> when (loadState.data) {
+                    SearchState.Initialized, SearchState.Empty -> it.copy(searchState = LoadState.Loading)
+                    else -> it
+                }
+                else -> it.copy(searchState = LoadState.Loading)
+            }
         }
         refresh(true)
     }