app/src/main/java/com/stoutner/privacybrowser/viewmodels/WebViewSource.kt

   1 /*
   2  * Copyright © 2020 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 doNotTrack: Boolean, 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, doNotTrack, 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, doNotTrack, localeString, proxy, this)) }
  76     }
  77 }