From b17b882a3b79999b82732adc6923499741fa602b Mon Sep 17 00:00:00 2001
From: arkon <eugcheung94@gmail.com>
Date: Mon, 11 Oct 2021 16:08:32 -0400
Subject: [PATCH] Adjust update/download warning threshold

---
 .../java/eu/kanade/tachiyomi/data/download/Downloader.kt    | 5 +++--
 .../kanade/tachiyomi/data/library/LibraryUpdateService.kt   | 6 ++++--
 2 files changed, 7 insertions(+), 4 deletions(-)

diff --git a/app/src/main/java/eu/kanade/tachiyomi/data/download/Downloader.kt b/app/src/main/java/eu/kanade/tachiyomi/data/download/Downloader.kt
index e1bf847722..8cd520cbf7 100644
--- a/app/src/main/java/eu/kanade/tachiyomi/data/download/Downloader.kt
+++ b/app/src/main/java/eu/kanade/tachiyomi/data/download/Downloader.kt
@@ -11,7 +11,7 @@ import eu.kanade.tachiyomi.data.database.models.Chapter
 import eu.kanade.tachiyomi.data.database.models.Manga
 import eu.kanade.tachiyomi.data.download.model.Download
 import eu.kanade.tachiyomi.data.download.model.DownloadQueue
-import eu.kanade.tachiyomi.data.library.QUEUE_SIZE_WARNING_THRESHOLD
+import eu.kanade.tachiyomi.data.library.PER_SOURCE_QUEUE_WARNING_THRESHOLD
 import eu.kanade.tachiyomi.source.SourceManager
 import eu.kanade.tachiyomi.source.model.Page
 import eu.kanade.tachiyomi.source.online.HttpSource
@@ -264,7 +264,8 @@ class Downloader(
 
             // Start downloader if needed
             if (autoStart && wasEmpty) {
-                if (queue.size > QUEUE_SIZE_WARNING_THRESHOLD) {
+                val maxDownloadsFromSource = queue.groupBy { it.source }.maxOf { it.value.size }
+                if (maxDownloadsFromSource > PER_SOURCE_QUEUE_WARNING_THRESHOLD) {
                     notifier.onWarning(context.getString(R.string.notification_size_warning))
                 }
                 DownloadService.start(context)
diff --git a/app/src/main/java/eu/kanade/tachiyomi/data/library/LibraryUpdateService.kt b/app/src/main/java/eu/kanade/tachiyomi/data/library/LibraryUpdateService.kt
index ebdef98e52..268b70628f 100644
--- a/app/src/main/java/eu/kanade/tachiyomi/data/library/LibraryUpdateService.kt
+++ b/app/src/main/java/eu/kanade/tachiyomi/data/library/LibraryUpdateService.kt
@@ -265,7 +265,9 @@ class LibraryUpdateService(
             .distinctBy { it.id }
             .sortedWith(rankingScheme[selectedScheme])
 
-        if (mangaToUpdate.size > QUEUE_SIZE_WARNING_THRESHOLD) {
+        // Warn when excessively checking a single source
+        val maxUpdatesFromSource = mangaToUpdate.groupBy { it.source }.maxOf { it.value.size }
+        if (maxUpdatesFromSource > PER_SOURCE_QUEUE_WARNING_THRESHOLD) {
             notifier.showQueueSizeWarningNotification()
         }
     }
@@ -572,4 +574,4 @@ class LibraryUpdateService(
     }
 }
 
-const val QUEUE_SIZE_WARNING_THRESHOLD = 100
+const val PER_SOURCE_QUEUE_WARNING_THRESHOLD = 50