Draff/TachiyomiSY-Plus
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Cleanup LibraryUpdateService (#8237)

Browse Source 
(cherry picked from commit 4ff5c1148e1ea89b13e5aefbd6b34d66566fb103)

# Conflicts:
#	app/src/main/java/eu/kanade/tachiyomi/data/library/LibraryUpdateService.kt
This commit is contained in:
AntsyLich 2022-10-23 01:16:44 +06:00 committed by Jobobby04
parent 1e89fb39a1
commit d853aa0c1b
2 changed files with 55 additions and 59 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
app/src/main/java/eu/kanade/tachiyomi/data/library/LibraryUpdateNotifier.kt
View File

@ -157,10 +157,6 @@ class LibraryUpdateNotifier(private val context: Context) {
* @param updates a list of manga with new updates. * @param updates a list of manga with new updates.
*/ */
fun showUpdateNotifications(updates: List<Pair<Manga, Array<Chapter>>>) { fun showUpdateNotifications(updates: List<Pair<Manga, Array<Chapter>>>) {
if (updates.isEmpty()) {
return
}
NotificationManagerCompat.from(context).apply { NotificationManagerCompat.from(context).apply {
// Parent group notification // Parent group notification
notify( notify(

78
app/src/main/java/eu/kanade/tachiyomi/data/library/LibraryUpdateService.kt
View File

@ -13,6 +13,7 @@ import eu.kanade.domain.category.model.Category
import eu.kanade.domain.chapter.interactor.GetChapterByMangaId import eu.kanade.domain.chapter.interactor.GetChapterByMangaId
import eu.kanade.domain.chapter.interactor.SyncChaptersWithSource import eu.kanade.domain.chapter.interactor.SyncChaptersWithSource
import eu.kanade.domain.chapter.interactor.SyncChaptersWithTrackServiceTwoWay import eu.kanade.domain.chapter.interactor.SyncChaptersWithTrackServiceTwoWay
import eu.kanade.domain.chapter.model.Chapter
import eu.kanade.domain.chapter.model.toDbChapter import eu.kanade.domain.chapter.model.toDbChapter
import eu.kanade.domain.download.service.DownloadPreferences import eu.kanade.domain.download.service.DownloadPreferences
import eu.kanade.domain.library.model.GroupLibraryMode import eu.kanade.domain.library.model.GroupLibraryMode
@ -35,8 +36,6 @@ import eu.kanade.domain.track.model.toDbTrack
import eu.kanade.domain.track.model.toDomainTrack import eu.kanade.domain.track.model.toDomainTrack
import eu.kanade.tachiyomi.R import eu.kanade.tachiyomi.R
import eu.kanade.tachiyomi.data.cache.CoverCache import eu.kanade.tachiyomi.data.cache.CoverCache
import eu.kanade.tachiyomi.data.database.models.Chapter
import eu.kanade.tachiyomi.data.database.models.toDomainChapter
import eu.kanade.tachiyomi.data.download.DownloadManager import eu.kanade.tachiyomi.data.download.DownloadManager
import eu.kanade.tachiyomi.data.download.DownloadService import eu.kanade.tachiyomi.data.download.DownloadService
import eu.kanade.tachiyomi.data.library.LibraryUpdateService.Companion.start import eu.kanade.tachiyomi.data.library.LibraryUpdateService.Companion.start
@ -92,7 +91,6 @@ import java.util.Date
import java.util.concurrent.CopyOnWriteArrayList import java.util.concurrent.CopyOnWriteArrayList
import java.util.concurrent.atomic.AtomicBoolean import java.util.concurrent.atomic.AtomicBoolean
import java.util.concurrent.atomic.AtomicInteger import java.util.concurrent.atomic.AtomicInteger
import eu.kanade.domain.chapter.model.Chapter as DomainChapter
/** /**
* This class will take care of updating the chapters of the manga from the library. It can be * This class will take care of updating the chapters of the manga from the library. It can be
@ -301,6 +299,9 @@ class LibraryUpdateService(
return START_REDELIVER_INTENT return START_REDELIVER_INTENT
} }
private val isUpdateJobActive: Boolean
get() = (updateJob?.isActive == true)
/** /**
* Adds list of manga to be updated. * Adds list of manga to be updated.
* *
@ -321,20 +322,21 @@ class LibraryUpdateService(
(groupLibraryUpdateType == GroupLibraryMode.ALL_BUT_UNGROUPED && group == LibraryGroup.UNGROUPED) (groupLibraryUpdateType == GroupLibraryMode.ALL_BUT_UNGROUPED && group == LibraryGroup.UNGROUPED)
) { ) {
val categoriesToUpdate = libraryPreferences.libraryUpdateCategories().get().map(String::toLong) val categoriesToUpdate = libraryPreferences.libraryUpdateCategories().get().map(String::toLong)
val listToInclude = if (categoriesToUpdate.isNotEmpty()) { val includedManga = if (categoriesToUpdate.isNotEmpty()) {
libraryManga.filter { it.category in categoriesToUpdate } libraryManga.filter { it.category in categoriesToUpdate }
} else { } else {
libraryManga libraryManga
} }
val categoriesToExclude = libraryPreferences.libraryUpdateCategoriesExclude().get().map { it.toLong() } val categoriesToExclude = libraryPreferences.libraryUpdateCategoriesExclude().get().map { it.toLong() }
val listToExclude = if (categoriesToExclude.isNotEmpty()) { val excludedMangaIds = if (categoriesToExclude.isNotEmpty()) {
libraryManga.filter { it.category in categoriesToExclude }.toSet() libraryManga.filter { it.category in categoriesToExclude }.map { it.manga.id }
} else { } else {
emptySet() emptyList()
} }
listToInclude.minus(listToExclude) includedManga
.filterNot { it.manga.id in excludedMangaIds }
} else { } else {
when (group) { when (group) {
LibraryGroup.BY_TRACK_STATUS -> { LibraryGroup.BY_TRACK_STATUS -> {
@ -395,7 +397,7 @@ class LibraryUpdateService(
val semaphore = Semaphore(5) val semaphore = Semaphore(5)
val progressCount = AtomicInteger(0) val progressCount = AtomicInteger(0)
val currentlyUpdatingManga = CopyOnWriteArrayList<Manga>() val currentlyUpdatingManga = CopyOnWriteArrayList<Manga>()
val newUpdates = CopyOnWriteArrayList<Pair<Manga, Array<DomainChapter>>>() val newUpdates = CopyOnWriteArrayList<Pair<Manga, Array<Chapter>>>()
val skippedUpdates = CopyOnWriteArrayList<Pair<Manga, String?>>() val skippedUpdates = CopyOnWriteArrayList<Pair<Manga, String?>>()
val failedUpdates = CopyOnWriteArrayList<Pair<Manga, String?>>() val failedUpdates = CopyOnWriteArrayList<Pair<Manga, String?>>()
val hasDownloads = AtomicBoolean(false) val hasDownloads = AtomicBoolean(false)
@ -412,19 +414,21 @@ class LibraryUpdateService(
semaphore.withPermit { semaphore.withPermit {
mangaInSource.forEach { libraryManga -> mangaInSource.forEach { libraryManga ->
val manga = libraryManga.manga val manga = libraryManga.manga
if (updateJob?.isActive != true) { if (!isUpdateJobActive) {
notifier.cancelProgressNotification()
return@async return@async
} }
// Don't continue to update if manga is not in library // Don't continue to update if manga is not in library
manga.id.let { getManga.await(it) } ?: return@forEach if (getManga.await(manga.id)?.favorite != true) {
return@forEach
}
withUpdateNotification( withUpdateNotification(
currentlyUpdatingManga, currentlyUpdatingManga,
progressCount, progressCount,
manga, manga,
) { ) {
try {
when { when {
MANGA_NON_COMPLETED in restrictions && manga.status.toInt() == SManga.COMPLETED -> MANGA_NON_COMPLETED in restrictions && manga.status.toInt() == SManga.COMPLETED ->
skippedUpdates.add(manga to getString(R.string.skipped_reason_completed)) skippedUpdates.add(manga to getString(R.string.skipped_reason_completed))
@ -439,26 +443,19 @@ class LibraryUpdateService(
skippedUpdates.add(manga to getString(R.string.skipped_reason_not_always_update)) skippedUpdates.add(manga to getString(R.string.skipped_reason_not_always_update))
else -> { else -> {
try {
val newChapters = updateManga(manga, loggedServices) val newChapters = updateManga(manga, loggedServices)
val newDbChapters = newChapters.map { it.toDbChapter() } .sortedByDescending { it.sourceOrder }
if (newChapters.isNotEmpty()) { if (newChapters.isNotEmpty()) {
val categoryIds = getCategories.await(manga.id).map { it.id } val categoryIds = getCategories.await(manga.id).map { it.id }
if (manga.shouldDownloadNewChapters(categoryIds, downloadPreferences)) { if (manga.shouldDownloadNewChapters(categoryIds, downloadPreferences)) {
downloadChapters(manga, newDbChapters) downloadChapters(manga, newChapters)
hasDownloads.set(true) hasDownloads.set(true)
} }
// Convert to the manga that contains new chapters // Convert to the manga that contains new chapters
newUpdates.add( newUpdates.add(manga to newChapters.toTypedArray())
manga to
newDbChapters
.map { it.toDomainChapter()!! }
.sortedByDescending { it.sourceOrder }
.toTypedArray(),
)
}
}
} }
} catch (e: Throwable) { } catch (e: Throwable) {
val errorMessage = when (e) { val errorMessage = when (e) {
@ -469,6 +466,8 @@ class LibraryUpdateService(
} }
failedUpdates.add(manga to errorMessage) failedUpdates.add(manga to errorMessage)
} }
}
}
if (libraryPreferences.autoUpdateTrackers().get()) { if (libraryPreferences.autoUpdateTrackers().get()) {
updateTrackings(manga, loggedServices) updateTrackings(manga, loggedServices)
@ -507,15 +506,16 @@ class LibraryUpdateService(
private fun downloadChapters(manga: Manga, chapters: List<Chapter>) { private fun downloadChapters(manga: Manga, chapters: List<Chapter>) {
// We don't want to start downloading while the library is updating, because websites // We don't want to start downloading while the library is updating, because websites
// may don't like it and they could ban the user. // may don't like it and they could ban the user.
val dbChapters = chapters.map { it.toDbChapter() }
// SY --> // SY -->
if (manga.source == MERGED_SOURCE_ID) { if (manga.source == MERGED_SOURCE_ID) {
val downloadingManga = runBlocking { getMergedMangaForDownloading.await(manga.id) } val downloadingManga = runBlocking { getMergedMangaForDownloading.await(manga.id) }
.associateBy { it.id } .associateBy { it.id }
chapters.groupBy { it.manga_id } dbChapters.groupBy { it.manga_id }
.forEach { .forEach {
downloadManager.downloadChapters( downloadManager.downloadChapters(
downloadingManga[it.key] ?: return@forEach, downloadingManga[it.key] ?: return@forEach,
chapters, it.value,
false, false,
) )
} }
@ -523,7 +523,7 @@ class LibraryUpdateService(
return return
} }
// SY <-- // SY <--
downloadManager.downloadChapters(manga, chapters, false) downloadManager.downloadChapters(manga, dbChapters, false)
} }
/** /**
@ -532,7 +532,7 @@ class LibraryUpdateService(
* @param manga the manga to update. * @param manga the manga to update.
* @return a pair of the inserted and removed chapters. * @return a pair of the inserted and removed chapters.
*/ */
private suspend fun updateManga(manga: Manga, loggedServices: List<TrackService>): List<DomainChapter> { private suspend fun updateManga(manga: Manga, loggedServices: List<TrackService>): List<Chapter> {
val source = sourceManager.getOrStub(manga.source) val source = sourceManager.getOrStub(manga.source)
// Update manga metadata if needed // Update manga metadata if needed
@ -562,12 +562,10 @@ class LibraryUpdateService(
val chapters = source.getChapterList(manga.toSManga()) val chapters = source.getChapterList(manga.toSManga())
// Get manga from database to account for if it was removed during the update // Get manga from database to account for if it was removed during the update and
val dbManga = getManga.await(manga.id) // to get latest data so it doesn't get overwritten later on
?: return emptyList() val dbManga = getManga.await(manga.id)?.takeIf { it.favorite } ?: return emptyList()
// [dbmanga] was used so that manga data doesn't get overwritten
// in case manga gets new chapter
return syncChaptersWithSource.await(chapters, dbManga, source) return syncChaptersWithSource.await(chapters, dbManga, source)
} }
@ -584,7 +582,8 @@ class LibraryUpdateService(
semaphore.withPermit { semaphore.withPermit {
mangaInSource.forEach { libraryManga -> mangaInSource.forEach { libraryManga ->
val manga = libraryManga.manga val manga = libraryManga.manga
if (updateJob?.isActive != true) { if (!isUpdateJobActive) {
notifier.cancelProgressNotification()
return@async return@async
} }
@ -628,7 +627,8 @@ class LibraryUpdateService(
mangaToUpdate.forEach { libraryManga -> mangaToUpdate.forEach { libraryManga ->
val manga = libraryManga.manga val manga = libraryManga.manga
if (updateJob?.isActive != true) { if (!isUpdateJobActive) {
notifier.cancelProgressNotification()
return return
} }
@ -673,7 +673,8 @@ class LibraryUpdateService(
manga: Manga, manga: Manga,
block: suspend () -> Unit, block: suspend () -> Unit,
) { ) {
if (updateJob?.isActive != true) { if (!isUpdateJobActive) {
notifier.cancelProgressNotification()
return return
} }
@ -686,7 +687,8 @@ class LibraryUpdateService(
block() block()
if (updateJob?.isActive != true) { if (!isUpdateJobActive) {
notifier.cancelProgressNotification()
return return
} }
@ -811,9 +813,7 @@ class LibraryUpdateService(
} }
return file return file
} }
} catch (e: Exception) { } catch (_: Exception) {}
// Empty
}
return File("") return File("")
} }
} }