优化

2 years ago · 45059fd17a
parent 08b89b9844
commit 45059fd17a
1 changed files with 3 additions and 0 deletions
--- a/app/src/main/java/io/legado/app/ui/book/info/BookInfoViewModel.kt
+++ b/app/src/main/java/io/legado/app/ui/book/info/BookInfoViewModel.kt
@ -150,6 +150,9 @@ class BookInfoViewModel(application: Application) : BaseViewModel(application) {
                bookSource?.let { bookSource ->
                    WebBook.getBookInfo(this, bookSource, book, canReName = canReName)
                        .onSuccess(IO) {
+                            appDb.bookDao.getBook(book.name, book.author)?.let {
+                                inBookshelf = true
+                            }
                            bookData.postValue(book)
                            if (isImportBookOnLine) {
                                appDb.searchBookDao.update(book.toSearchBook())