From 75eec52b854ebf332adfc16102f8e50346b3a731 Mon Sep 17 00:00:00 2001
From: kunfei <gekunfei@live.com>
Date: Wed, 14 Aug 2019 15:36:07 +0800
Subject: [PATCH] up

---
 app/src/main/java/io/legado/app/ui/read/ReadViewModel.kt | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/app/src/main/java/io/legado/app/ui/read/ReadViewModel.kt b/app/src/main/java/io/legado/app/ui/read/ReadViewModel.kt
index 7f45b52c2..3d0cc0b59 100644
--- a/app/src/main/java/io/legado/app/ui/read/ReadViewModel.kt
+++ b/app/src/main/java/io/legado/app/ui/read/ReadViewModel.kt
@@ -2,6 +2,7 @@ package io.legado.app.ui.read
 
 import android.app.Application
 import android.content.Intent
+import androidx.lifecycle.MediatorLiveData
 import io.legado.app.App
 import io.legado.app.base.BaseViewModel
 import io.legado.app.data.entities.Book
@@ -12,6 +13,7 @@ class ReadViewModel(application: Application) : BaseViewModel(application) {
 
     var book: Book? = null
     var bookSource: BookSource? = null
+    var chapterMaxIndex = MediatorLiveData<Int>()
 
     fun initData(intent: Intent) {
         val bookUrl = intent.getStringExtra("bookUrl")
@@ -19,7 +21,8 @@ class ReadViewModel(application: Application) : BaseViewModel(application) {
             execute {
                 book = App.db.bookDao().getBook(bookUrl)
                 book?.let { book ->
-                    if (App.db.bookChapterDao().getChapterCount(bookUrl) == 0) {
+                    val count = App.db.bookChapterDao().getChapterCount(bookUrl)
+                    if (count == 0) {
                         App.db.bookSourceDao().getBookSource(book.origin)?.let {
                             WebBook(it).getChapterList(book)
                                 .onSuccess { cList ->
@@ -30,6 +33,8 @@ class ReadViewModel(application: Application) : BaseViewModel(application) {
                         } ?: let {
 
                         }
+                    } else {
+                        chapterMaxIndex.postValue(count)
                     }
                 }