Migrate to AndroidX ViewModel from Voyager ScreenModel (#3594)

2026-07-15 21:19:11 +06:00
parent ef1c529672
commit c3b99aea07
91 changed files with 1503 additions and 1091 deletions
@@ -0,0 +1,81 @@
+package mihon.feature.upcoming
+
+import androidx.compose.ui.util.fastMap
+import androidx.compose.ui.util.fastMapIndexedNotNull
+import androidx.lifecycle.viewModelScope
+import eu.kanade.core.util.insertSeparatorsReversed
+import eu.kanade.tachiyomi.util.lang.toLocalDate
+import kotlinx.coroutines.flow.collectLatest
+import kotlinx.coroutines.flow.update
+import kotlinx.coroutines.launch
+import mihon.core.viewmodel.StateViewModel
+import mihon.domain.upcoming.interactor.GetUpcomingManga
+import tachiyomi.domain.manga.model.Manga
+import uy.kohesive.injekt.Injekt
+import uy.kohesive.injekt.api.get
+import java.time.LocalDate
+import java.time.YearMonth
+
+class UpcomingViewModel(
+    private val getUpcomingManga: GetUpcomingManga = Injekt.get(),
+) : StateViewModel<UpcomingViewModel.State>(State()) {
+
+    init {
+        viewModelScope.launch {
+            getUpcomingManga.subscribe().collectLatest {
+                mutableState.update { state ->
+                    val upcomingItems = it.toUpcomingUIModels()
+                    state.copy(
+                        items = upcomingItems,
+                        events = upcomingItems.toEvents(),
+                        headerIndexes = upcomingItems.getHeaderIndexes(),
+                    )
+                }
+            }
+        }
+    }
+
+    private fun List<Manga>.toUpcomingUIModels(): List<UpcomingUIModel> {
+        var mangaCount = 0
+        return fastMap { UpcomingUIModel.Item(it) }
+            .insertSeparatorsReversed { before, after ->
+                if (after != null) mangaCount++
+
+                val beforeDate = before?.manga?.expectedNextUpdate?.toLocalDate()
+                val afterDate = after?.manga?.expectedNextUpdate?.toLocalDate()
+
+                if (beforeDate != afterDate && afterDate != null) {
+                    UpcomingUIModel.Header(afterDate, mangaCount).also { mangaCount = 0 }
+                } else {
+                    null
+                }
+            }
+    }
+
+    private fun List<UpcomingUIModel>.toEvents(): Map<LocalDate, Int> {
+        return filterIsInstance<UpcomingUIModel.Header>()
+            .associate { it.date to it.mangaCount }
+    }
+
+    private fun List<UpcomingUIModel>.getHeaderIndexes(): Map<LocalDate, Int> {
+        return fastMapIndexedNotNull { index, upcomingUIModel ->
+            if (upcomingUIModel is UpcomingUIModel.Header) {
+                upcomingUIModel.date to index
+            } else {
+                null
+            }
+        }
+            .toMap()
+    }
+
+    fun setSelectedYearMonth(yearMonth: YearMonth) {
+        mutableState.update { it.copy(selectedYearMonth = yearMonth) }
+    }
+
+    data class State(
+        val selectedYearMonth: YearMonth = YearMonth.now(),
+        val items: List<UpcomingUIModel> = listOf(),
+        val events: Map<LocalDate, Int> = mapOf(),
+        val headerIndexes: Map<LocalDate, Int> = mapOf(),
+    )
+}