]> gitweb.stoutner.com Git - PrivacyBrowserAndroid.git/blob - HeadersViewModel.kt
ccfe79ab2ca2baf4ee7964731ba0aad1d857eb54
[PrivacyBrowserAndroid.git] / HeadersViewModel.kt
1 /*
2  * Copyright 2020-2023 Soren Stoutner <soren@stoutner.com>.
3  *
4  * This file is part of Privacy Browser Android <https://www.stoutner.com/privacy-browser-android>.
5  *
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.
10  *
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.
15  *
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/>.
18  */
19
20 package com.stoutner.privacybrowser.viewmodels
21
22 import android.content.ContentResolver
23 import android.text.SpannableStringBuilder
24
25 import androidx.lifecycle.LiveData
26 import androidx.lifecycle.MutableLiveData
27 import androidx.lifecycle.ViewModel
28
29 import com.stoutner.privacybrowser.backgroundtasks.GetHeadersBackgroundTask
30
31 import java.net.Proxy
32 import java.util.concurrent.ExecutorService
33
34 class HeadersViewModel(private val urlString: String, private val userAgent: String, private val localeString: String, private val proxy: Proxy, private val contentResolver: ContentResolver,
35                        private val executorService: ExecutorService): ViewModel() {
36     // Initialize the mutable live data variables.
37     private val mutableLiveDataSourceStringArray = MutableLiveData<Array<SpannableStringBuilder>>()
38     private val mutableLiveDataErrorString = MutableLiveData<String>()
39
40     // Initialize the view model.
41     init {
42         // Instantiate the get headers background task class.
43         val getSourceBackgroundTask = GetHeadersBackgroundTask()
44
45         // Get the headers.
46         executorService.execute { mutableLiveDataSourceStringArray.postValue(getSourceBackgroundTask.acquire(urlString, userAgent, localeString, proxy, contentResolver, this,
47             false)) }
48     }
49
50     // The headers observer.
51     fun observeHeaders(): LiveData<Array<SpannableStringBuilder>> {
52         // Return the source to the activity.
53         return mutableLiveDataSourceStringArray
54     }
55
56     // The error observer.
57     fun observeErrors(): LiveData<String> {
58         // Return any errors to the activity.
59         return mutableLiveDataErrorString
60     }
61
62     // The interface for returning the error from the background task
63     fun returnError(errorString: String) {
64         // Update the mutable live data error string.
65         mutableLiveDataErrorString.postValue(errorString)
66     }
67
68     // The workhorse that gets the headers.
69     fun updateHeaders(urlString: String, ignoreSslErrors: Boolean) {
70         // Reset the mutable live data error string.  This prevents the snackbar from displaying later if the activity restarts.
71         mutableLiveDataErrorString.postValue("")
72
73         // Instantiate the get headers background task class.
74         val getSourceBackgroundTask = GetHeadersBackgroundTask()
75
76         // Get the headers.
77         executorService.execute { mutableLiveDataSourceStringArray.postValue(getSourceBackgroundTask.acquire(urlString, userAgent, localeString, proxy, contentResolver, this,
78             ignoreSslErrors)) }
79     }
80 }