Add favorite entry alternative handling, allowing parennt versions to take priority for favorites sync

2023-05-12 22:50:23 -04:00
parent e9a3463455
commit 282a0c4e16
12 changed files with 333 additions and 34 deletions
@@ -0,0 +1,178 @@
+
+import eu.kanade.tachiyomi.data.backup.models.BackupSerializer
+import eu.kanade.tachiyomi.source.model.SManga
+import eu.kanade.tachiyomi.source.online.all.EHentai
+import exh.favorites.LocalFavoritesStorage
+import exh.metadata.metadata.EHentaiSearchMetadata
+import exh.source.EXH_SOURCE_ID
+import io.kotest.inspectors.shouldForAll
+import io.mockk.coEvery
+import io.mockk.mockk
+import kotlinx.coroutines.runBlocking
+import kotlinx.serialization.protobuf.ProtoBuf
+import okio.buffer
+import okio.gzip
+import okio.sink
+import okio.source
+import org.junit.jupiter.api.BeforeAll
+import org.junit.jupiter.api.Disabled
+import org.junit.jupiter.api.Test
+import tachiyomi.domain.category.interactor.GetCategories
+import tachiyomi.domain.category.model.Category
+import tachiyomi.domain.manga.interactor.GetCustomMangaInfo
+import tachiyomi.domain.manga.interactor.GetFavoriteEntries
+import tachiyomi.domain.manga.interactor.GetFavorites
+import tachiyomi.domain.manga.model.CustomMangaInfo
+import tachiyomi.domain.manga.model.FavoriteEntry
+import tachiyomi.domain.manga.model.Manga
+import tachiyomi.domain.manga.repository.CustomMangaRepository
+import uy.kohesive.injekt.Injekt
+import uy.kohesive.injekt.api.addSingletonFactory
+import java.io.File
+
+class Tester {
+
+    @Disabled
+    @Test
+    fun stripBackup() {
+        val bytes = File("D:\\Downloads\\pacthiyomi_2023-05-08_13-30.proto (1).gz")
+            .inputStream().source().buffer()
+            .gzip().buffer()
+            .readByteArray()
+        val backup = ProtoBuf.decodeFromByteArray(BackupSerializer, bytes)
+        val newBytes = ProtoBuf.encodeToByteArray(
+            BackupSerializer,
+            backup.copy(
+                backupManga = backup.backupManga.filter { it.favorite },
+            ),
+        )
+        File("D:\\Downloads\\pacthiyomi_2023-05-08_13-30 (2).proto.gz").outputStream().sink().gzip().buffer().use {
+            it.write(newBytes)
+        }
+    }
+
+    @Test
+    fun localFavoritesStorageTester(): Unit = runBlocking {
+        val favorites = listOf(
+            Manga.create().copy(
+                id = 1,
+                favorite = true,
+                source = EXH_SOURCE_ID,
+                url = "/g/gid/token",
+            ),
+            // an alias for gid2/token2
+            Manga.create().copy(
+                id = 3,
+                favorite = true,
+                source = EXH_SOURCE_ID,
+                url = "/g/gid3/token3",
+            ),
+            // add this one to library
+            Manga.create().copy(
+                id = 3,
+                favorite = true,
+                source = EXH_SOURCE_ID,
+                url = "/g/gid4/token4",
+            ),
+        )
+        val categories = listOf(
+            Category(
+                id = 1,
+                name = "a",
+                order = 1,
+                flags = 0,
+            ),
+        )
+        val favoriteEntries = listOf(
+            FavoriteEntry(
+                gid = "gid",
+                token = "token",
+                title = "a",
+                category = 0,
+            ),
+            FavoriteEntry(
+                gid = "gid2",
+                token = "token2",
+                title = "a",
+                category = 0,
+            ),
+            // the alias for gid2/token2
+            FavoriteEntry(
+                gid = "gid2",
+                token = "token2",
+                title = "a",
+                category = 0,
+                otherGid = "gid3",
+                otherToken = "token3",
+            ),
+            // removed on remote and local
+            FavoriteEntry(
+                gid = "gid6",
+                token = "token6",
+                title = "a",
+                category = 0,
+            ),
+        )
+
+        val getFavorites = mockk<GetFavorites>()
+        coEvery { getFavorites.await() } returns favorites
+
+        val getCategories = mockk<GetCategories>()
+        coEvery { getCategories.await() } returns categories
+        coEvery { getCategories.await(any()) } returns categories
+
+        val getFavoriteEntries = mockk<GetFavoriteEntries>()
+        coEvery { getFavoriteEntries.await() } returns favoriteEntries
+
+        val storage = LocalFavoritesStorage(
+            getFavorites = getFavorites,
+            getCategories = getCategories,
+            deleteFavoriteEntries = mockk(),
+            getFavoriteEntries = getFavoriteEntries,
+            insertFavoriteEntries = mockk(),
+        )
+
+        val (added, removed) = storage.getChangedDbEntries()
+        added.shouldForAll { it.gid == "gid4" && it.token == "token4" }
+        removed.shouldForAll { it.gid == "gid6" && it.token == "token6" }
+
+        val (remoteAdded, remoteRemoved) = storage.getChangedRemoteEntries(
+            listOf(
+                EHentai.ParsedManga(
+                    0,
+                    SManga("/g/gid/token", "a"),
+                    EHentaiSearchMetadata(),
+                ),
+                EHentai.ParsedManga(
+                    0,
+                    SManga("/g/gid2/token2", "a"),
+                    EHentaiSearchMetadata(),
+                ),
+                // added on remote
+                EHentai.ParsedManga(
+                    0,
+                    SManga("/g/gid5/token5", "a"),
+                    EHentaiSearchMetadata(),
+                ),
+            ),
+        )
+
+        remoteAdded.shouldForAll { it.gid == "gid5" && it.token == "token5" }
+        remoteRemoved.shouldForAll { it.gid == "gid6" && it.token == "token6" }
+    }
+
+    companion object {
+        @JvmStatic
+        @BeforeAll
+        fun before() {
+            Injekt.addSingletonFactory {
+                GetCustomMangaInfo(
+                    object : CustomMangaRepository {
+                        override fun get(mangaId: Long) = null
+                        override fun set(mangaInfo: CustomMangaInfo) = Unit
+                    },
+                )
+            }
+        }
+    }
+}