From fb4b6664b4867502b6d1997faedbc19b93f2afb6 Mon Sep 17 00:00:00 2001
From: Horis <821938089@qq.com>
Date: Mon, 5 Feb 2024 12:13:24 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BC=98=E5=8C=96?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 app/src/main/java/io/legado/app/model/CacheBook.kt | 13 +++++++++++--
 1 file changed, 11 insertions(+), 2 deletions(-)

diff --git a/app/src/main/java/io/legado/app/model/CacheBook.kt b/app/src/main/java/io/legado/app/model/CacheBook.kt
index 02a4309f1..7904fa4ff 100644
--- a/app/src/main/java/io/legado/app/model/CacheBook.kt
+++ b/app/src/main/java/io/legado/app/model/CacheBook.kt
@@ -36,7 +36,7 @@ object CacheBook {
         var cacheBook = cacheBookMap[bookUrl]
         if (cacheBook != null) {
             //存在时更新,书源可能会变化,必须更新
-            cacheBook.bookSource = bookSource
+            updateBookSource(bookSource)
             cacheBook.book = book
             return cacheBook
         }
@@ -50,7 +50,7 @@ object CacheBook {
         var cacheBook = cacheBookMap[book.bookUrl]
         if (cacheBook != null) {
             //存在时更新,书源可能会变化,必须更新
-            cacheBook.bookSource = bookSource
+            updateBookSource(bookSource)
             cacheBook.book = book
             return cacheBook
         }
@@ -59,6 +59,15 @@ object CacheBook {
         return cacheBook
     }
 
+    private fun updateBookSource(newBookSource: BookSource) {
+        cacheBookMap.forEach {
+            val model = it.value
+            if (model.bookSource.bookSourceUrl == newBookSource.bookSourceUrl) {
+                model.bookSource = newBookSource
+            }
+        }
+    }
+
     fun start(context: Context, book: Book, start: Int, end: Int) {
         if (!book.isLocal) {
             context.startService<CacheBookService> {