ask if user wants to delete database before importing

greenart7c3 · Jul 26, 2024 · 428a88d · 428a88d
1 parent 790ef30
commit 428a88d
Show file tree

Hide file tree

Showing 2 changed files with 87 additions and 23 deletions.
diff --git a/app/src/main/java/com/greenart7c3/citrine/MainActivity.kt b/app/src/main/java/com/greenart7c3/citrine/MainActivity.kt
@@ -21,11 +21,13 @@ import androidx.compose.foundation.layout.fillMaxWidth
 import androidx.compose.foundation.layout.padding
 import androidx.compose.foundation.rememberScrollState
 import androidx.compose.foundation.verticalScroll
+import androidx.compose.material3.AlertDialog
 import androidx.compose.material3.CircularProgressIndicator
 import androidx.compose.material3.ElevatedButton
 import androidx.compose.material3.MaterialTheme
 import androidx.compose.material3.Surface
 import androidx.compose.material3.Text
+import androidx.compose.material3.TextButton
 import androidx.compose.runtime.getValue
 import androidx.compose.runtime.mutableStateOf
 import androidx.compose.runtime.remember
@@ -117,25 +119,6 @@ class MainActivity : ComponentActivity() {
 
     override fun onCreate(savedInstanceState: Bundle?) {
         val progress = mutableStateOf("")
-
-        storageHelper.onFolderSelected = { _, folder ->
-            exportDatabase(
-                folder = folder,
-                onProgress = {
-                    progress.value = it
-                },
-            )
-        }
-
-        storageHelper.onFileSelected = { _, files ->
-            importDatabase(
-                files = files,
-                onProgress = {
-                    progress.value = it
-                },
-            )
-        }
-
         super.onCreate(savedInstanceState)
 
         requestPermissionLauncher.launch("android.permission.POST_NOTIFICATIONS")
@@ -170,6 +153,78 @@ class MainActivity : ComponentActivity() {
             )
 
             CitrineTheme {
+                var showDialog by remember { mutableStateOf(false) }
+                val selectedFiles = remember {
+                    mutableListOf<DocumentFile>()
+                }
+
+                storageHelper.onFolderSelected = { _, folder ->
+                    exportDatabase(
+                        folder = folder,
+                        onProgress = {
+                            progress.value = it
+                        },
+                    )
+                }
+
+                storageHelper.onFileSelected = { _, files ->
+                    selectedFiles.clear()
+                    selectedFiles.addAll(files)
+                    showDialog = true
+                }
+
+                if (showDialog) {
+                    AlertDialog(
+                        onDismissRequest = {
+                            showDialog = false
+                        },
+                        title = {
+                            Text(getString(R.string.import_events))
+                        },
+                        text = {
+                            Text(getString(R.string.import_events_warning))
+                        },
+                        confirmButton = {
+                            TextButton(
+                                onClick = {
+                                    showDialog = false
+                                    importDatabase(
+                                        files = selectedFiles,
+                                        shouldDelete = true,
+                                        onProgress = {
+                                            progress.value = it
+                                        },
+                                        onFinished = {
+                                            selectedFiles.clear()
+                                        },
+                                    )
+                                },
+                            ) {
+                                Text(getString(R.string.yes))
+                            }
+                        },
+                        dismissButton = {
+                            TextButton(
+                                onClick = {
+                                    showDialog = false
+                                    importDatabase(
+                                        files = selectedFiles,
+                                        shouldDelete = false,
+                                        onProgress = {
+                                            progress.value = it
+                                        },
+                                        onFinished = {
+                                            selectedFiles.clear()
+                                        },
+                                    )
+                                },
+                            ) {
+                                Text(getString(R.string.no))
+                            }
+                        },
+                    )
+                }
+
                 Surface(
                     modifier = Modifier
                         .fillMaxSize()
@@ -309,7 +364,7 @@ class MainActivity : ComponentActivity() {
     }
 
     @OptIn(DelicateCoroutinesApi::class)
-    private fun importDatabase(files: List<DocumentFile>, onProgress: (String) -> Unit) {
+    private fun importDatabase(files: List<DocumentFile>, shouldDelete: Boolean, onProgress: (String) -> Unit, onFinished: () -> Unit) {
         GlobalScope.launch(Dispatchers.IO) {
             val file = files.first()
             if (file.extension != "jsonl") {
@@ -330,7 +385,7 @@ class MainActivity : ComponentActivity() {
                 val input = file.openInputStream(this@MainActivity) ?: return@launch
                 input.use { ip ->
                     ip.bufferedReader().use {
-                        var line: String? = ""
+                        var line: String?
                         while (it.readLine().also { readLine -> line = readLine } != null) {
                             if (line?.isNotBlank() == true) {
                                 totalLines++
@@ -343,8 +398,10 @@ class MainActivity : ComponentActivity() {
                     ip.bufferedReader().use {
                         var linesRead = 0
 
-                        onProgress("deleting all events")
-                        database.eventDao().deleteAll()
+                        if (shouldDelete) {
+                            onProgress("deleting all events")
+                            database.eventDao().deleteAll()
+                        }
 
                         it.useLines { lines ->
                             lines.forEach { line ->
@@ -363,6 +420,7 @@ class MainActivity : ComponentActivity() {
 
                 onProgress("")
                 isLoading.value = false
+                onFinished()
                 GlobalScope.launch(Dispatchers.Main) {
                     Toast.makeText(
                         this@MainActivity,
@@ -382,6 +440,7 @@ class MainActivity : ComponentActivity() {
                 }
                 onProgress("")
                 isLoading.value = false
+                onFinished()
             }
         }
     }

diff --git a/app/src/main/res/values/strings.xml b/app/src/main/res/values/strings.xml
@@ -17,4 +17,9 @@
     <string name="import_failed">Import failed</string>
     <string name="invalid_file_extension">File not supported</string>
     <string name="imported_events_successfully">Imported %1$d events successfully</string>
+    <string name="import_events">Import events</string>
+    <string name="import_events_warning">Delete existing events? This action cannot be undone.</string>
+    <string name="yes">Yes</string>
+    <string name="no">No</string>
+    <string name="cancel">Cancel</string>
 </resources>