From 65bdc5541cd2686445ed5d4d0ec9fab6d1ff3325 Mon Sep 17 00:00:00 2001
From: kunfei <gekunfei@live.com>
Date: Tue, 7 Jan 2020 08:57:41 +0800
Subject: [PATCH] up

---
 .../java/io/legado/app/help/storage/Backup.kt | 10 +++++++---
 .../io/legado/app/ui/main/MainActivity.kt     | 20 +++++++++++++++++--
 2 files changed, 25 insertions(+), 5 deletions(-)

diff --git a/app/src/main/java/io/legado/app/help/storage/Backup.kt b/app/src/main/java/io/legado/app/help/storage/Backup.kt
index 7f5b726ae..90bf3b9e2 100644
--- a/app/src/main/java/io/legado/app/help/storage/Backup.kt
+++ b/app/src/main/java/io/legado/app/help/storage/Backup.kt
@@ -120,9 +120,13 @@ object Backup {
     }
 
     private fun copyBackup() {
-        for (fileName in backupFileNames) {
-            FileHelp.getFile(backupPath + File.separator + "bookshelf.json")
-                .copyTo(FileHelp.getFile(legadoPath + File.separator + "bookshelf.json"), true)
+        try {
+            for (fileName in backupFileNames) {
+                FileHelp.getFile(backupPath + File.separator + "bookshelf.json")
+                    .copyTo(FileHelp.getFile(legadoPath + File.separator + "bookshelf.json"), true)
+            }
+        } catch (e: Exception) {
+            e.printStackTrace()
         }
     }
 }
\ No newline at end of file
diff --git a/app/src/main/java/io/legado/app/ui/main/MainActivity.kt b/app/src/main/java/io/legado/app/ui/main/MainActivity.kt
index 39a5ad05f..571b32139 100644
--- a/app/src/main/java/io/legado/app/ui/main/MainActivity.kt
+++ b/app/src/main/java/io/legado/app/ui/main/MainActivity.kt
@@ -1,8 +1,10 @@
 package io.legado.app.ui.main
 
+import android.net.Uri
 import android.os.Bundle
 import android.view.KeyEvent
 import android.view.MenuItem
+import androidx.documentfile.provider.DocumentFile
 import androidx.fragment.app.Fragment
 import androidx.fragment.app.FragmentManager
 import androidx.fragment.app.FragmentStatePagerAdapter
@@ -13,6 +15,8 @@ import io.legado.app.BuildConfig
 import io.legado.app.R
 import io.legado.app.base.VMBaseActivity
 import io.legado.app.constant.Bus
+import io.legado.app.constant.PreferKey
+import io.legado.app.help.storage.Backup
 import io.legado.app.lib.theme.ATH
 import io.legado.app.service.BaseReadAloudService
 import io.legado.app.service.help.ReadAloud
@@ -28,8 +32,6 @@ import kotlinx.coroutines.launch
 class MainActivity : VMBaseActivity<MainViewModel>(R.layout.activity_main),
     BottomNavigationView.OnNavigationItemSelectedListener,
     ViewPager.OnPageChangeListener by ViewPager.SimpleOnPageChangeListener() {
-    private val backupSelectRequestCode = 11
-    private val restoreSelectRequestCode = 22
     override val viewModel: MainViewModel
         get() = getViewModel(MainViewModel::class.java)
 
@@ -115,6 +117,7 @@ class MainActivity : VMBaseActivity<MainViewModel>(R.layout.activity_main),
     override fun finish() {
         if (!BuildConfig.DEBUG) {
             launch {
+                backup()
                 super.finish()
             }
         } else {
@@ -122,6 +125,19 @@ class MainActivity : VMBaseActivity<MainViewModel>(R.layout.activity_main),
         }
     }
 
+    private suspend fun backup() {
+        val backupPath = getPrefString(PreferKey.backupPath)
+        if (backupPath?.isNotEmpty() == true) {
+            val uri = Uri.parse(backupPath)
+            val doc = DocumentFile.fromTreeUri(this, uri)
+            if (doc?.canWrite() == true) {
+                Backup.backup(this@MainActivity, uri)
+            }
+        } else {
+            Backup.backup(this@MainActivity, null)
+        }
+    }
+
     override fun onDestroy() {
         super.onDestroy()
         ReadAloud.stop(this)