X-Git-Url: https://gitweb.stoutner.com/?p=PrivacyBrowserAndroid.git;a=blobdiff_plain;f=app%2Fsrc%2Fmain%2Fjava%2Fcom%2Fstoutner%2Fprivacybrowser%2Fviewmodels%2FHeadersViewModel.kt;fp=app%2Fsrc%2Fmain%2Fjava%2Fcom%2Fstoutner%2Fprivacybrowser%2Fviewmodels%2FHeadersViewModel.kt;h=ccfe79ab2ca2baf4ee7964731ba0aad1d857eb54;hp=0000000000000000000000000000000000000000;hb=a54a66c7d169d2edf55ba560ec2d951e709188e6;hpb=4395165965c4a211f7c03583ebca6edaa5bf19fe

diff --git a/app/src/main/java/com/stoutner/privacybrowser/viewmodels/HeadersViewModel.kt b/app/src/main/java/com/stoutner/privacybrowser/viewmodels/HeadersViewModel.kt
new file mode 100644
index 00000000..ccfe79ab
--- /dev/null
+++ b/app/src/main/java/com/stoutner/privacybrowser/viewmodels/HeadersViewModel.kt
@@ -0,0 +1,80 @@
+/*
+ * Copyright 2020-2023 Soren Stoutner <soren@stoutner.com>.
+ *
+ * This file is part of Privacy Browser Android <https://www.stoutner.com/privacy-browser-android>.
+ *
+ * Privacy Browser Android is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * Privacy Browser Android is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with Privacy Browser Android.  If not, see <http://www.gnu.org/licenses/>.
+ */
+
+package com.stoutner.privacybrowser.viewmodels
+
+import android.content.ContentResolver
+import android.text.SpannableStringBuilder
+
+import androidx.lifecycle.LiveData
+import androidx.lifecycle.MutableLiveData
+import androidx.lifecycle.ViewModel
+
+import com.stoutner.privacybrowser.backgroundtasks.GetHeadersBackgroundTask
+
+import java.net.Proxy
+import java.util.concurrent.ExecutorService
+
+class HeadersViewModel(private val urlString: String, private val userAgent: String, private val localeString: String, private val proxy: Proxy, private val contentResolver: ContentResolver,
+                       private val executorService: ExecutorService): ViewModel() {
+    // Initialize the mutable live data variables.
+    private val mutableLiveDataSourceStringArray = MutableLiveData<Array<SpannableStringBuilder>>()
+    private val mutableLiveDataErrorString = MutableLiveData<String>()
+
+    // Initialize the view model.
+    init {
+        // Instantiate the get headers background task class.
+        val getSourceBackgroundTask = GetHeadersBackgroundTask()
+
+        // Get the headers.
+        executorService.execute { mutableLiveDataSourceStringArray.postValue(getSourceBackgroundTask.acquire(urlString, userAgent, localeString, proxy, contentResolver, this,
+            false)) }
+    }
+
+    // The headers observer.
+    fun observeHeaders(): LiveData<Array<SpannableStringBuilder>> {
+        // Return the source to the activity.
+        return mutableLiveDataSourceStringArray
+    }
+
+    // The error observer.
+    fun observeErrors(): LiveData<String> {
+        // Return any errors to the activity.
+        return mutableLiveDataErrorString
+    }
+
+    // The interface for returning the error from the background task
+    fun returnError(errorString: String) {
+        // Update the mutable live data error string.
+        mutableLiveDataErrorString.postValue(errorString)
+    }
+
+    // The workhorse that gets the headers.
+    fun updateHeaders(urlString: String, ignoreSslErrors: Boolean) {
+        // Reset the mutable live data error string.  This prevents the snackbar from displaying later if the activity restarts.
+        mutableLiveDataErrorString.postValue("")
+
+        // Instantiate the get headers background task class.
+        val getSourceBackgroundTask = GetHeadersBackgroundTask()
+
+        // Get the headers.
+        executorService.execute { mutableLiveDataSourceStringArray.postValue(getSourceBackgroundTask.acquire(urlString, userAgent, localeString, proxy, contentResolver, this,
+            ignoreSslErrors)) }
+    }
+}