fix(EditMedia): empty custom lists on new entry (#126)

axiel7 · May 24, 2024 · a650bee · a650bee
1 parent 4e7162e
commit a650bee
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 2 deletions.
diff --git a/app/src/main/java/com/axiel7/anihyou/ui/screens/mediadetails/edit/EditMediaSheet.kt b/app/src/main/java/com/axiel7/anihyou/ui/screens/mediadetails/edit/EditMediaSheet.kt
@@ -93,6 +93,9 @@ fun EditMediaSheet(
 
     LaunchedEffect(listEntry) {
         viewModel.setListEntry(listEntry)
+        if (listEntry == null) {
+            viewModel.fillCustomLists(mediaDetails.type)
+        }
     }
 
     EditMediaSheetContent(

diff --git a/app/src/main/java/com/axiel7/anihyou/ui/screens/mediadetails/edit/EditMediaViewModel.kt b/app/src/main/java/com/axiel7/anihyou/ui/screens/mediadetails/edit/EditMediaViewModel.kt
@@ -9,6 +9,7 @@ import com.axiel7.anihyou.data.repository.MediaListRepository
 import com.axiel7.anihyou.fragment.BasicMediaDetails
 import com.axiel7.anihyou.fragment.BasicMediaListEntry
 import com.axiel7.anihyou.type.MediaListStatus
+import com.axiel7.anihyou.type.MediaType
 import com.axiel7.anihyou.ui.common.viewmodel.UiStateViewModel
 import com.axiel7.anihyou.utils.DateUtils.millisToLocalDate
 import com.axiel7.anihyou.utils.DateUtils.toFuzzyDate
@@ -28,7 +29,7 @@ import javax.inject.Inject
 @HiltViewModel
 class EditMediaViewModel @Inject constructor(
     private val mediaListRepository: MediaListRepository,
-    defaultPreferencesRepository: DefaultPreferencesRepository,
+    private val defaultPreferencesRepository: DefaultPreferencesRepository,
 ) : UiStateViewModel<EditMediaUiState>(), EditMediaEvent {
 
     override val initialState = EditMediaUiState()
@@ -54,10 +55,22 @@ class EditMediaViewModel @Inject constructor(
             isPrivate = value?.private,
             isHiddenFromStatusLists = value?.hiddenFromStatusLists,
             notes = value?.notes,
-            customLists = null,
         )
     }
 
+    fun fillCustomLists(mediaType: MediaType?) {
+        viewModelScope.launch {
+            val customLists = linkedMapOf<String, Boolean>()
+            val savedCustomsLists = if (mediaType == MediaType.ANIME) {
+                defaultPreferencesRepository.animeCustomLists.first()
+            } else {
+                defaultPreferencesRepository.mangaCustomLists.first()
+            }
+            savedCustomsLists?.forEach { customLists[it] = false }
+            mutableUiState.update { it.copy(customLists = customLists) }
+        }
+    }
+
     override fun onChangeStatus(value: MediaListStatus) {
         if (mutableUiState.value.isNewEntry && value == MediaListStatus.CURRENT) {
             mutableUiState.update {