up

app/src/main/java/io/legado/app/help/storage/Restore.kt

@@ -17,8 +17,6 @@ import io.legado.app.help.ReadBookConfig
 import io.legado.app.help.coroutine.Coroutine
 import io.legado.app.help.storage.Backup.defaultPath
 import io.legado.app.utils.*
-import kotlinx.coroutines.Dispatchers.IO
-import kotlinx.coroutines.withContext
 import org.jetbrains.anko.defaultSharedPreferences
 import org.jetbrains.anko.doAsync
 import org.jetbrains.anko.toast
@@ -103,11 +101,9 @@ object Restore {
     fun importYueDuData(context: Context) {
         Coroutine.async {
             val file = File(FileUtils.getSdCardPath(), "YueDu")
-
             // 导入书架
             val shelfFile = File(file, "myBookShelf.json")
             val books = mutableListOf<Book>()
-            if (shelfFile.exists()) try {
             val items: List<Map<String, Any>> = jsonPath.parse(shelfFile.readText()).read("$")
             val existingBooks = App.db.bookDao().allBookUrls.toSet()
             for (item in items) {
@@ -145,20 +141,18 @@ object Restore {
                 books.add(book)
             }
             App.db.bookDao().insert(*books.toTypedArray())
-                withContext(IO) {
-                    context.toast("成功导入书籍${books.size}")
-                }
-            } catch (e: Exception) {
-                e.printStackTrace()
-                withContext(IO) {
-                    context.toast("导入书籍失败\n${e.localizedMessage}")
-                }
+            books
+        }.onSuccess {
+            context.toast("成功导入书籍${it?.size}")
+        }.onError {
+            context.toast("导入书籍失败\n${it.localizedMessage}")
         }
 
+        Coroutine.async {
             // Book source
+            val file = File(FileUtils.getSdCardPath(), "YueDu")
             val sourceFile = File(file, "myBookSource.json")
             val bookSources = mutableListOf<BookSource>()
-            if (sourceFile.exists()) try {
             val items: List<Map<String, Any>> = jsonPath.parse(sourceFile.readText()).read("$")
             for (item in items) {
                 val jsonItem = jsonPath.parse(item)
@@ -167,20 +161,19 @@ object Restore {
                 }
             }
             App.db.bookSourceDao().insert(*bookSources.toTypedArray())
-                withContext(IO) {
-                    context.toast("成功导入书源${bookSources.size}")
-                }
-            } catch (e: Exception) {
-                e.printStackTrace()
-                withContext(IO) {
-                    context.toast("导入源失败\n${e.localizedMessage}")
-                }
+            bookSources
+        }.onSuccess {
+            context.toast("成功导入书源${it?.size}")
+        }.onError {
+            context.toast("导入源失败\n${it.localizedMessage}")
         }
 
+        Coroutine.async {
+            // Book source
+            val file = File(FileUtils.getSdCardPath(), "YueDu")
             // Replace rules
             val ruleFile = File(file, "myBookReplaceRule.json")
             val replaceRules = mutableListOf<ReplaceRule>()
-            if (ruleFile.exists()) try {
             val items: List<Map<String, Any>> = jsonPath.parse(ruleFile.readText()).read("$")
             val existingRules = App.db.replaceRuleDao().all.map { it.pattern }.toSet()
             for ((index: Int, item: Map<String, Any>) in items.withIndex()) {
@@ -197,16 +190,12 @@ object Restore {
                 rule.order = jsonItem.readInt("$.serialNumber") ?: index
                 replaceRules.add(rule)
             }
-                withContext(IO) {
-                    context.toast("成功导入替换规则${replaceRules.size}")
-                }
-            } catch (e: Exception) {
-                e.printStackTrace()
-                withContext(IO) {
-                    context.toast("导入替换规则失败\n${e.localizedMessage}")
-                }
+            replaceRules
+        }.onSuccess {
+            context.toast("成功导入替换规则${it?.size}")
+        }.onError {
+            context.toast("导入替换规则失败\n${it.localizedMessage}")
         }
 
     }
 }
-}