feat: 优化代码

5 years ago · 619c9eb887
parent c8179fa59a
commit 619c9eb887
1 changed files with 5 additions and 2 deletions
--- a/app/src/main/java/io/legado/app/ui/download/DownloadViewModel.kt
+++ b/app/src/main/java/io/legado/app/ui/download/DownloadViewModel.kt
@ -50,8 +50,11 @@ class DownloadViewModel(application: Application) : BaseViewModel(application) {
            .append("\n")
            .append(context.getString(R.string.author_show, book.author))
        App.db.bookChapterDao().getChapterList(book.bookUrl).forEach { chapter ->
-            BookHelp.getContent(book, chapter)?.let {
-                stringBuilder.append("\n").append(it)
+            BookHelp.getContent(book, chapter).let {
+                stringBuilder.append("\n\n")
+                    .append(chapter.title)
+                    .append("\n")
+                    .append(it)
            }
        }
        return stringBuilder.toString()