2 * Copyright 2020-2023 Soren Stoutner <soren@stoutner.com>.
4 * This file is part of Privacy Browser Android <https://www.stoutner.com/privacy-browser-android>.
6 * Privacy Browser Android is free software: you can redistribute it and/or modify
7 * it under the terms of the GNU General Public License as published by
8 * the Free Software Foundation, either version 3 of the License, or
9 * (at your option) any later version.
11 * Privacy Browser Android is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 * GNU General Public License for more details.
16 * You should have received a copy of the GNU General Public License
17 * along with Privacy Browser Android. If not, see <http://www.gnu.org/licenses/>.
20 package com.stoutner.privacybrowser.viewmodels
22 import android.app.Application
23 import android.content.ContentResolver
24 import android.text.SpannableStringBuilder
26 import androidx.lifecycle.AndroidViewModel
28 import androidx.lifecycle.LiveData
29 import androidx.lifecycle.MutableLiveData
31 import com.stoutner.privacybrowser.backgroundtasks.GetHeadersBackgroundTask
34 import java.util.concurrent.ExecutorService
36 class HeadersViewModel(application: Application, private val urlString: String, private val userAgent: String, private val localeString: String, private val proxy: Proxy, private val contentResolver: ContentResolver,
37 private val executorService: ExecutorService): AndroidViewModel(application) {
38 // Initialize the mutable live data variables.
39 private val mutableLiveDataSourceStringArray = MutableLiveData<Array<SpannableStringBuilder>>()
40 private val mutableLiveDataErrorString = MutableLiveData<String>()
42 // Initialize the view model.
44 // Instantiate the get headers background task class.
45 val getSourceBackgroundTask = GetHeadersBackgroundTask()
48 executorService.execute { mutableLiveDataSourceStringArray.postValue(getSourceBackgroundTask.acquire(application, urlString, userAgent, localeString, proxy, contentResolver, this,
52 // The headers observer.
53 fun observeHeaders(): LiveData<Array<SpannableStringBuilder>> {
54 // Return the source to the activity.
55 return mutableLiveDataSourceStringArray
58 // The error observer.
59 fun observeErrors(): LiveData<String> {
60 // Return any errors to the activity.
61 return mutableLiveDataErrorString
64 // The interface for returning the error from the background task
65 fun returnError(errorString: String) {
66 // Update the mutable live data error string.
67 mutableLiveDataErrorString.postValue(errorString)
70 // The workhorse that gets the headers.
71 fun updateHeaders(urlString: String, ignoreSslErrors: Boolean) {
72 // Reset the mutable live data error string. This prevents the snackbar from displaying later if the activity restarts.
73 mutableLiveDataErrorString.postValue("")
75 // Instantiate the get headers background task class.
76 val getSourceBackgroundTask = GetHeadersBackgroundTask()
79 executorService.execute { mutableLiveDataSourceStringArray.postValue(getSourceBackgroundTask.acquire(getApplication(), urlString, userAgent, localeString, proxy, contentResolver, this,