app/src/main/java/com/stoutner/privacybrowser/viewmodels/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.app.Application
  23 import android.content.ContentResolver
  24 import android.text.SpannableStringBuilder
  25
  26 import androidx.lifecycle.AndroidViewModel
  27
  28 import androidx.lifecycle.LiveData
  29 import androidx.lifecycle.MutableLiveData
  30
  31 import com.stoutner.privacybrowser.backgroundtasks.GetHeadersBackgroundTask
  32
  33 import java.net.Proxy
  34 import java.util.concurrent.ExecutorService
  35
  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>()
  41
  42     // Initialize the view model.
  43     init {
  44         // Instantiate the get headers background task class.
  45         val getSourceBackgroundTask = GetHeadersBackgroundTask()
  46
  47         // Get the headers.
  48         executorService.execute { mutableLiveDataSourceStringArray.postValue(getSourceBackgroundTask.acquire(application, urlString, userAgent, localeString, proxy, contentResolver, this,
  49             false)) }
  50     }
  51
  52     // The headers observer.
  53     fun observeHeaders(): LiveData<Array<SpannableStringBuilder>> {
  54         // Return the source to the activity.
  55         return mutableLiveDataSourceStringArray
  56     }
  57
  58     // The error observer.
  59     fun observeErrors(): LiveData<String> {
  60         // Return any errors to the activity.
  61         return mutableLiveDataErrorString
  62     }
  63
  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)
  68     }
  69
  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("")
  74
  75         // Instantiate the get headers background task class.
  76         val getSourceBackgroundTask = GetHeadersBackgroundTask()
  77
  78         // Get the headers.
  79         executorService.execute { mutableLiveDataSourceStringArray.postValue(getSourceBackgroundTask.acquire(getApplication(), urlString, userAgent, localeString, proxy, contentResolver, this,
  80             ignoreSslErrors)) }
  81     }
  82 }