Backup & Restore for custom attributes

customattrs/src/main/java/de/mm20/launcher2/customattrs/CustomAttributesRepository.kt

@@ -1,17 +1,24 @@
 package de.mm20.launcher2.customattrs
 
+import de.mm20.launcher2.crashreporter.CrashReporter
 import de.mm20.launcher2.database.AppDatabase
+import de.mm20.launcher2.database.entities.CustomAttributeEntity
+import de.mm20.launcher2.database.entities.WebsearchEntity
+import de.mm20.launcher2.ktx.jsonObjectOf
 import de.mm20.launcher2.search.data.Searchable
-import kotlinx.coroutines.CoroutineScope
-import kotlinx.coroutines.Dispatchers
-import kotlinx.coroutines.Job
+import kotlinx.coroutines.*
 import kotlinx.coroutines.flow.Flow
 import kotlinx.coroutines.flow.map
-import kotlinx.coroutines.launch
+import org.json.JSONArray
+import org.json.JSONException
+import java.io.File
 
 interface CustomAttributesRepository {
     fun getCustomIcon(searchable: Searchable): Flow<CustomIcon?>
     fun setCustomIcon(searchable: Searchable, icon: CustomIcon?)
+
+    suspend fun export(toDir: File)
+    suspend fun import(fromDir: File)
 }
 
 internal class CustomAttributesRepositoryImpl(
@@ -36,4 +43,59 @@ internal class CustomAttributesRepositoryImpl(
             }
         }
     }
+
+    override suspend fun export(toDir: File) = withContext(Dispatchers.IO) {
+        val dao = appDatabase.backupDao()
+        var page = 0
+        do {
+            val customAttrs = dao.exportCustomAttributes(limit = 100, offset = page * 100)
+            val jsonArray = JSONArray()
+            for (customAttr in customAttrs) {
+                jsonArray.put(
+                    jsonObjectOf(
+                        "key" to customAttr.key,
+                        "value" to customAttr.value,
+                        "type" to customAttr.type,
+                    )
+                )
+            }
+
+            val file = File(toDir, "customizations.${page.toString().padStart(4, '0')}")
+            file.bufferedWriter().use {
+                it.write(jsonArray.toString())
+            }
+            page++
+        } while (customAttrs.size == 100)
+    }
+
+    override suspend fun import(fromDir: File) = withContext(Dispatchers.IO) {
+        val dao = appDatabase.backupDao()
+        dao.wipeCustomAttributes()
+
+        val files = fromDir.listFiles { _, name -> name.startsWith("customizations.") } ?: return@withContext
+
+        for (file in files) {
+            val customAttrs = mutableListOf<CustomAttributeEntity>()
+            try {
+                val jsonArray = JSONArray(file.inputStream().reader().readText())
+
+                for (i in 0 until jsonArray.length()) {
+                    val json = jsonArray.getJSONObject(i)
+
+                    val entity = CustomAttributeEntity(
+                        id = null,
+                        type = json.getString("type"),
+                        value = json.optString("value"),
+                        key = json.optString("key"),
+                    )
+                    customAttrs.add(entity)
+                }
+
+                dao.importCustomAttributes(customAttrs)
+
+            } catch (e: JSONException) {
+                CrashReporter.logException(e)
+            }
+        }
+    }
 }