From 5098a82e4a8c498ab3e6fe3275f619432b855420 Mon Sep 17 00:00:00 2001
From: kunfei <gekunfei@live.com>
Date: Sun, 3 Apr 2022 23:34:00 +0800
Subject: [PATCH] =?UTF-8?q?=E6=B7=BB=E5=8A=A0=E6=89=B9=E9=87=8F=E6=8D=A2?=
 =?UTF-8?q?=E6=BA=90?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../io/legado/app/ui/book/arrange/ArrangeBookViewModel.kt | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/app/src/main/java/io/legado/app/ui/book/arrange/ArrangeBookViewModel.kt b/app/src/main/java/io/legado/app/ui/book/arrange/ArrangeBookViewModel.kt
index 399fcb9e0..e5bd06731 100644
--- a/app/src/main/java/io/legado/app/ui/book/arrange/ArrangeBookViewModel.kt
+++ b/app/src/main/java/io/legado/app/ui/book/arrange/ArrangeBookViewModel.kt
@@ -47,7 +47,13 @@ class ArrangeBookViewModel(application: Application) : BaseViewModel(application
                 if (book.isLocalBook()) return@forEachIndexed
                 if (book.origin == source.bookSourceUrl) return@forEachIndexed
                 WebBook.preciseSearchAwait(this, book.name, book.author, source)?.let {
-                    book.changeTo(it.second)
+                    val newBook = it.second
+                    book.changeTo(newBook)
+                    if (newBook.tocUrl.isEmpty()) {
+                        WebBook.getBookInfoAwait(this, source, newBook)
+                    }
+                    val toc = WebBook.getChapterListAwait(this, source, newBook)
+                    appDb.bookChapterDao.insert(*toc.toTypedArray())
                 }
             }
         }.onFinally {