fix: attempt at fixing CSV exports crashing app

ActivityWatch · Jan 5, 2024 · 9169e3a · 9169e3a
1 parent a082295
commit 9169e3a
Showing 1 changed file with 22 additions and 0 deletions.
diff --git a/mobile/src/main/java/net/activitywatch/android/fragments/WebUIFragment.kt b/mobile/src/main/java/net/activitywatch/android/fragments/WebUIFragment.kt
@@ -97,6 +97,7 @@ class WebUIFragment : Fragment() {
 
         myWebView.settings.javaScriptEnabled = true
         myWebView.settings.domStorageEnabled = true
+        myWebView.addJavascriptInterface(WebAppInterface(context), "Android")
         arguments?.let {
             it.getString(ARG_URL)?.let { it1 -> myWebView.loadUrl(it1) }
         }
@@ -145,3 +146,24 @@ class WebUIFragment : Fragment() {
             }
     }
 }
+
+class WebAppInterface(private val mContext: Context) {
+    @JavascriptInterface
+    fun downloadCSV(csv: String, filename: String) {
+        downloadFile(csv, filename, "text/csv")
+    }
+
+    @JavascriptInterface
+    fun downloadJSON(csv: String, filename: String) {
+        downloadFile(csv, filename, "application/json")
+    }
+
+    fun downloadFile(csv: String, filename: String, mimetype: String) {
+        val file = File(mContext.getExternalFilesDir(null), filename)
+        file.writeText(csv)
+        val intent = Intent(Intent.ACTION_VIEW)
+        intent.setDataAndType(Uri.fromFile(file), mimetype)
+        intent.flags = Intent.FLAG_ACTIVITY_NO_HISTORY
+        mContext.startActivity(intent)
+    }
+}