优化

2 years ago · 3ea742dae0
parent e9711815fe
commit 3ea742dae0
1 changed files with 2 additions and 0 deletions
--- a/app/src/main/java/io/legado/app/ui/main/MainViewModel.kt
+++ b/app/src/main/java/io/legado/app/ui/main/MainViewModel.kt
@ -114,6 +114,8 @@ class MainViewModel(application: Application) : BaseViewModel(application) {
        }
        val source = appDb.bookSourceDao.getBookSource(book.origin)
        if (source == null) {
+            book.addType(BookType.updateError)
+            appDb.bookDao.update(book)
            waitUpTocBooks.remove(book.bookUrl)
            return
        }