From 4a7375494268f9070a45b7bd4d82ccca14bff909 Mon Sep 17 00:00:00 2001
From: kunfei <gekunfei@live.com>
Date: Tue, 8 Feb 2022 11:52:09 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=9D=E5=AD=98=E4=B9=A6=E7=B1=8D=E8=A6=81?=
 =?UTF-8?q?=E5=88=86=E4=B8=A4=E7=A7=8D=E6=83=85=E5=86=B5=E5=88=A4=E6=96=AD?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../app/api/controller/BookController.kt      | 59 +++++++++++++------
 1 file changed, 40 insertions(+), 19 deletions(-)

diff --git a/app/src/main/java/io/legado/app/api/controller/BookController.kt b/app/src/main/java/io/legado/app/api/controller/BookController.kt
index c2b480aae..17be8f1c3 100644
--- a/app/src/main/java/io/legado/app/api/controller/BookController.kt
+++ b/app/src/main/java/io/legado/app/api/controller/BookController.kt
@@ -1,8 +1,9 @@
 package io.legado.app.api.controller
 
+import android.net.Uri
 import android.util.Base64
 import androidx.core.graphics.drawable.toBitmap
-import io.legado.app.R
+import androidx.documentfile.provider.DocumentFile
 import io.legado.app.api.ReturnData
 import io.legado.app.constant.PreferKey
 import io.legado.app.data.appDb
@@ -22,7 +23,8 @@ import io.legado.app.model.webBook.WebBook
 import io.legado.app.utils.*
 import kotlinx.coroutines.runBlocking
 import splitties.init.appCtx
-import timber.log.Timber
+import java.io.File
+import java.io.FileOutputStream
 
 object BookController {
 
@@ -202,36 +204,55 @@ object BookController {
      */
     fun addLocalBook(parameters: Map<String, List<String>>): ReturnData {
         val returnData = ReturnData()
-        try {
-            val fileName = parameters["fileName"]?.firstOrNull()
-                ?: return returnData.setErrorMsg("fileName 不能为空")
-            val fileData = parameters["fileData"]?.firstOrNull()
-                ?: return returnData.setErrorMsg("fileData 不能为空")
-            if (AppConfig.defaultBookTreeUri == null) return returnData.setErrorMsg("没有设置书籍保存位置!")
-            val bookFolder = FileUtils.createFolderIfNotExist(AppConfig.defaultBookTreeUri!!)
-            val file = FileUtils.createFileIfNotExist(bookFolder, fileName)
-            val fileBytes = Base64.decode(fileData.substringAfter("base64,"), Base64.DEFAULT)
-            file.writeBytes(fileBytes)
+        val fileName = parameters["fileName"]?.firstOrNull()
+            ?: return returnData.setErrorMsg("fileName 不能为空")
+        val fileData = parameters["fileData"]?.firstOrNull()
+            ?: return returnData.setErrorMsg("fileData 不能为空")
+        kotlin.runCatching {
+            val defaultBookTreeUri = AppConfig.defaultBookTreeUri
+            if (defaultBookTreeUri.isNullOrBlank()) return returnData.setErrorMsg("没有设置书籍保存位置!")
+            val treeUri = Uri.parse(defaultBookTreeUri)
+            val fileBytes =
+                Base64.decode(fileData.substringAfter("base64,"), Base64.DEFAULT)
+            val uri = if (treeUri.isContentScheme()) {
+                val treeDoc = DocumentFile.fromTreeUri(appCtx, treeUri)
+                var doc = treeDoc!!.findFile(fileName)
+                if (doc == null) {
+                    doc = treeDoc.createFile(FileUtils.getMimeType(fileName), fileName)
+                        ?: throw SecurityException("Permission Denial")
+                }
+                appCtx.contentResolver.openOutputStream(doc.uri)!!.use { oStream ->
+                    oStream.write(fileBytes)
+                }
+                doc.uri
+            } else {
+                val treeFile = File(treeUri.path!!)
+                val file = treeFile.getFile(fileName)
+                FileOutputStream(file).use { oStream ->
+                    oStream.write(fileBytes)
+                }
+                Uri.fromFile(file)
+            }
             val nameAuthor = LocalBook.analyzeNameAuthor(fileName)
             val book = Book(
-                bookUrl = file.absolutePath,
+                bookUrl = uri.toString(),
                 name = nameAuthor.first,
                 author = nameAuthor.second,
                 originName = fileName,
                 coverUrl = FileUtils.getPath(
                     appCtx.externalFiles,
                     "covers",
-                    "${MD5Utils.md5Encode16(file.absolutePath)}.jpg"
+                    "${MD5Utils.md5Encode16(uri.toString())}.jpg"
                 )
             )
             if (book.isEpub()) EpubFile.upBookInfo(book)
             if (book.isUmd()) UmdFile.upBookInfo(book)
             appDb.bookDao.insert(book)
-        } catch (e: Exception) {
-            Timber.e(e)
-            return returnData.setErrorMsg(
-                e.localizedMessage ?: appCtx.getString(R.string.unknown_error)
-            )
+        }.onFailure {
+            return when (it) {
+                is SecurityException -> returnData.setErrorMsg("需重新设置书籍保存位置!")
+                else -> returnData.setErrorMsg("保存书籍错误\n${it.localizedMessage}")
+            }
         }
         return returnData.setData(true)
     }