Use Voyager for a few screens

2022-11-26 13:36:06 -05:00
parent bf9b2ca2ff
commit 7df12c68fd
13 changed files with 333 additions and 371 deletions
@@ -0,0 +1,108 @@
+package exh.pagepreview
+
+import androidx.compose.runtime.getValue
+import androidx.compose.runtime.mutableStateOf
+import androidx.compose.runtime.setValue
+import cafe.adriel.voyager.core.model.StateScreenModel
+import cafe.adriel.voyager.core.model.coroutineScope
+import eu.kanade.domain.chapter.model.Chapter
+import eu.kanade.domain.history.interactor.GetNextChapters
+import eu.kanade.domain.manga.interactor.GetManga
+import eu.kanade.domain.manga.interactor.GetPagePreviews
+import eu.kanade.domain.manga.model.Manga
+import eu.kanade.domain.manga.model.PagePreview
+import eu.kanade.tachiyomi.source.Source
+import eu.kanade.tachiyomi.source.SourceManager
+import eu.kanade.tachiyomi.util.lang.launchIO
+import kotlinx.coroutines.flow.MutableStateFlow
+import kotlinx.coroutines.flow.catch
+import kotlinx.coroutines.flow.collect
+import kotlinx.coroutines.flow.onEach
+import kotlinx.coroutines.flow.update
+import uy.kohesive.injekt.Injekt
+import uy.kohesive.injekt.api.get
+
+class PagePreviewScreenModel(
+    private val mangaId: Long,
+    private val getPagePreviews: GetPagePreviews = Injekt.get(),
+    private val getManga: GetManga = Injekt.get(),
+    private val getNextChapters: GetNextChapters = Injekt.get(),
+    private val sourceManager: SourceManager = Injekt.get(),
+) : StateScreenModel<PagePreviewState>(PagePreviewState.Loading) {
+
+    private val page = MutableStateFlow(1)
+
+    var pageDialogOpen by mutableStateOf(false)
+
+    init {
+        coroutineScope.launchIO {
+            val manga = getManga.await(mangaId)!!
+            val chapter = getNextChapters.await(mangaId, onlyUnread = false).lastOrNull()
+            if (chapter == null) {
+                mutableState.update {
+                    PagePreviewState.Error(Exception("No chapters found"))
+                }
+                return@launchIO
+            }
+            val source = sourceManager.getOrStub(manga.source)
+            page
+                .onEach { page ->
+                    when (
+                        val previews = getPagePreviews.await(manga, source, page)
+                    ) {
+                        is GetPagePreviews.Result.Error -> mutableState.update {
+                            PagePreviewState.Error(previews.error)
+                        }
+                        is GetPagePreviews.Result.Success -> mutableState.update {
+                            when (it) {
+                                PagePreviewState.Loading, is PagePreviewState.Error -> {
+                                    PagePreviewState.Success(
+                                        page,
+                                        previews.pagePreviews,
+                                        previews.hasNextPage,
+                                        previews.pageCount,
+                                        manga,
+                                        chapter,
+                                        source,
+                                    )
+                                }
+                                is PagePreviewState.Success -> it.copy(
+                                    page = page,
+                                    pagePreviews = previews.pagePreviews,
+                                    hasNextPage = previews.hasNextPage,
+                                    pageCount = previews.pageCount,
+                                )
+                            }
+                        }
+                        GetPagePreviews.Result.Unused -> Unit
+                    }
+                }
+                .catch { e ->
+                    mutableState.update {
+                        PagePreviewState.Error(e)
+                    }
+                }
+                .collect()
+        }
+    }
+
+    fun moveToPage(page: Int) {
+        this.page.value = page
+    }
+}
+
+sealed class PagePreviewState {
+    object Loading : PagePreviewState()
+
+    data class Success(
+        val page: Int,
+        val pagePreviews: List<PagePreview>,
+        val hasNextPage: Boolean,
+        val pageCount: Int?,
+        val manga: Manga,
+        val chapter: Chapter,
+        val source: Source,
+    ) : PagePreviewState()
+
+    data class Error(val error: Throwable) : PagePreviewState()
+}