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