package ani.dantotsu.notifications.subscription import ani.dantotsu.R import ani.dantotsu.currContext import ani.dantotsu.media.Media import ani.dantotsu.media.MediaNameAdapter import ani.dantotsu.media.Selected import ani.dantotsu.parsers.AnimeParser import ani.dantotsu.parsers.AnimeSources import ani.dantotsu.parsers.Episode import ani.dantotsu.parsers.MangaChapter import ani.dantotsu.parsers.MangaParser import ani.dantotsu.parsers.MangaSources import ani.dantotsu.settings.saving.PrefManager import ani.dantotsu.settings.saving.PrefName import ani.dantotsu.tryWithSuspend import ani.dantotsu.util.Logger import kotlinx.coroutines.withTimeoutOrNull class SubscriptionHelper { companion object { private fun loadSelected( mediaId: Int ): Selected { val data = PrefManager.getNullableCustomVal("Selected-${mediaId}", null, Selected::class.java) ?: Selected().let { it.sourceIndex = 0 it.preferDub = PrefManager.getVal(PrefName.SettingsPreferDub) it } return data } private fun saveSelected(mediaId: Int, data: Selected) { PrefManager.setCustomVal("Selected-${mediaId}", data) } fun getAnimeParser(id: Int): AnimeParser { val sources = AnimeSources Logger.log("getAnimeParser size: ${sources.list.size}") val selected = loadSelected(id) if (selected.sourceIndex >= sources.list.size) { selected.sourceIndex = 0 } val parser = sources[selected.sourceIndex] parser.selectDub = selected.preferDub return parser } suspend fun getEpisode( parser: AnimeParser, id: Int ): Episode? { val selected = loadSelected(id) val ep = withTimeoutOrNull(10 * 1000) { tryWithSuspend { val show = parser.loadSavedShowResponse(id) ?: throw Exception( currContext()?.getString( R.string.failed_to_load_data, id ) ) show.sAnime?.let { parser.getLatestEpisode( show.link, show.extra, it, selected.latest ) } } } return ep?.apply { selected.latest = number.toFloat() saveSelected(id, selected) } } fun getMangaParser(id: Int): MangaParser { val sources = MangaSources Logger.log("getMangaParser size: ${sources.list.size}") val selected = loadSelected(id) if (selected.sourceIndex >= sources.list.size) { selected.sourceIndex = 0 } return sources[selected.sourceIndex] } suspend fun getChapter( parser: MangaParser, id: Int ): MangaChapter? { val selected = loadSelected(id) val chp = withTimeoutOrNull(10 * 1000) { tryWithSuspend { val show = parser.loadSavedShowResponse(id) ?: throw Exception( currContext()?.getString( R.string.failed_to_load_data, id ) ) show.sManga?.let { parser.getLatestChapter( show.link, show.extra, it, selected.latest ) } } } return chp?.apply { selected.latest = MediaNameAdapter.findChapterNumber(number) ?: 0f saveSelected(id, selected) } } data class SubscribeMedia( val isAnime: Boolean, val isAdult: Boolean, val id: Int, val name: String, val image: String? ) : java.io.Serializable private const val SUBSCRIPTIONS = "subscriptions" @Suppress("UNCHECKED_CAST") fun getSubscriptions(): Map = (PrefManager.getNullableCustomVal( SUBSCRIPTIONS, null, Map::class.java ) as? Map) ?: mapOf().also { PrefManager.setCustomVal(SUBSCRIPTIONS, it) } @Suppress("UNCHECKED_CAST") fun saveSubscription(media: Media, subscribed: Boolean) { val data = PrefManager.getNullableCustomVal( SUBSCRIPTIONS, null, Map::class.java ) as? MutableMap ?: mutableMapOf() if (subscribed) { if (!data.containsKey(media.id)) { val new = SubscribeMedia( media.anime != null, media.isAdult, media.id, media.userPreferredName, media.cover ) data[media.id] = new } } else { data.remove(media.id) } PrefManager.setCustomVal(SUBSCRIPTIONS, data) } } }