From dd0867abd12a6e16e5b874df9da746e1940f4a7a Mon Sep 17 00:00:00 2001
From: AndyBernie <zhenghaotian2016@gmail.com>
Date: Thu, 3 Sep 2020 13:22:50 +0800
Subject: [PATCH] images can be exported

---
 .../app/ui/book/download/DownloadViewModel.kt | 42 +++++++++++++++++--
 1 file changed, 38 insertions(+), 4 deletions(-)

diff --git a/app/src/main/java/io/legado/app/ui/book/download/DownloadViewModel.kt b/app/src/main/java/io/legado/app/ui/book/download/DownloadViewModel.kt
index a2a2f00bb..7402dd18b 100644
--- a/app/src/main/java/io/legado/app/ui/book/download/DownloadViewModel.kt
+++ b/app/src/main/java/io/legado/app/ui/book/download/DownloadViewModel.kt
@@ -5,13 +5,11 @@ import android.net.Uri
 import androidx.documentfile.provider.DocumentFile
 import io.legado.app.App
 import io.legado.app.R
+import io.legado.app.constant.AppPattern
 import io.legado.app.base.BaseViewModel
 import io.legado.app.data.entities.Book
 import io.legado.app.help.BookHelp
-import io.legado.app.utils.DocumentUtils
-import io.legado.app.utils.FileUtils
-import io.legado.app.utils.isContentPath
-import io.legado.app.utils.writeText
+import io.legado.app.utils.*
 import java.io.File
 
 
@@ -38,11 +36,47 @@ class DownloadViewModel(application: Application) : BaseViewModel(application) {
     private fun export(doc: DocumentFile, book: Book) {
         DocumentUtils.createFileIfNotExist(doc, "${book.name} 作者:${book.author}.txt")
             ?.writeText(context, getAllContents(book))
+        App.db.bookChapterDao().getChapterList(book.bookUrl).forEach { chapter ->
+            BookHelp.getContent(book, chapter).let { content ->
+                content?.split("\n")?.forEachIndexed { index, text ->
+                    val matcher = AppPattern.imgPattern.matcher(text)
+                    if (matcher.find()) {
+                        var src = matcher.group(1)
+                        src = NetworkUtils.getAbsoluteURL(chapter.url, src)
+                        src?.let {
+                            val vfile = BookHelp.getImage(book, src)
+                            if(vfile.exists()) {
+                                DocumentUtils.createFileIfNotExist(doc, "${index}-${MD5Utils.md5Encode16(src)}.jpg", subDirs = arrayOf("${book.name}_${book.author}", "images", chapter.title))
+                                    ?.writeBytes(context, vfile.readBytes())
+                            }
+                        }
+                    }
+                }
+            }
+        }
     }
 
     private fun export(file: File, book: Book) {
         FileUtils.createFileIfNotExist(file, "${book.name} 作者:${book.author}.txt")
             .writeText(getAllContents(book))
+        App.db.bookChapterDao().getChapterList(book.bookUrl).forEach { chapter ->
+            BookHelp.getContent(book, chapter).let { content ->
+                content?.split("\n")?.forEachIndexed { index, text ->
+                    val matcher = AppPattern.imgPattern.matcher(text)
+                    if (matcher.find()) {
+                        var src = matcher.group(1)
+                        src = NetworkUtils.getAbsoluteURL(chapter.url, src)
+                        src?.let {
+                            val vfile = BookHelp.getImage(book, src)
+                            if(vfile.exists()) {
+                                FileUtils.createFileIfNotExist(file, "${book.name}_${book.author}", "images", chapter.title, "${index}-${MD5Utils.md5Encode16(src)}.jpg")
+                                    .writeBytes(vfile.readBytes())
+                            }
+                        }
+                    }
+                }
+            }
+        }
     }
 
     private fun getAllContents(book: Book): String {