app/src/main/java/com/stoutner/privacybrowser/asynctasks/GetUrlSize.java

   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.asynctasks;
  21
  22 import android.content.Context;
  23 import android.os.AsyncTask;
  24 import android.webkit.CookieManager;
  25 import android.widget.TextView;
  26
  27 import androidx.appcompat.app.AlertDialog;
  28
  29 import com.stoutner.privacybrowser.R;
  30 import com.stoutner.privacybrowser.helpers.ProxyHelper;
  31
  32 import java.lang.ref.WeakReference;
  33 import java.net.HttpURLConnection;
  34 import java.net.Proxy;
  35 import java.net.URL;
  36 import java.text.NumberFormat;
  37
  38 public class GetUrlSize extends AsyncTask<String, Void, String> {
  39     // Define weak references for the calling context and alert dialog.
  40     private final WeakReference<Context> contextWeakReference;
  41     private final WeakReference<AlertDialog> alertDialogWeakReference;
  42
  43     // Define the class variables.
  44     private final String userAgent;
  45     private final boolean cookiesEnabled;
  46
  47     // The public constructor.
  48     public GetUrlSize(Context context, AlertDialog alertDialog, String userAgent, boolean cookiesEnabled) {
  49         // Populate the week references for the context and alert dialog.
  50         contextWeakReference = new WeakReference<>(context);
  51         alertDialogWeakReference = new WeakReference<>(alertDialog);
  52
  53         // Store the class variables.
  54         this.userAgent = userAgent;
  55         this.cookiesEnabled = cookiesEnabled;
  56     }
  57
  58     @Override
  59     protected String doInBackground(String... urlToSave) {
  60         // Get a handle for the context and the fragment.
  61         Context context = contextWeakReference.get();
  62         AlertDialog alertDialog = alertDialogWeakReference.get();
  63
  64         // Abort if the fragment is gone.
  65         if (alertDialog == null) {
  66             return null;
  67         }
  68
  69         // Initialize the formatted file size string.
  70         String formattedFileSize = context.getString(R.string.unknown_size);
  71
  72         // Because everything relating to requesting data from a webserver can throw errors, the entire section must catch exceptions.
  73         try {
  74             // Get the URL from the calling fragment.
  75             URL url = new URL(urlToSave[0]);
  76
  77             // Instantiate the proxy helper.
  78             ProxyHelper proxyHelper = new ProxyHelper();
  79
  80             // Get the current proxy.
  81             Proxy proxy = proxyHelper.getCurrentProxy(context);
  82
  83             // Open a connection to the URL.  No data is actually sent at this point.
  84             HttpURLConnection httpUrlConnection = (HttpURLConnection) url.openConnection(proxy);
  85
  86             // Add the user agent to the header property.
  87             httpUrlConnection.setRequestProperty("User-Agent", userAgent);
  88
  89             // Add the cookies if they are enabled.
  90             if (cookiesEnabled) {
  91                 // Get the cookies for the current domain.
  92                 String cookiesString = CookieManager.getInstance().getCookie(url.toString());
  93
  94                 // Only add the cookies if they are not null.
  95                 if (cookiesString != null) {
  96                     // Add the cookies to the header property.
  97                     httpUrlConnection.setRequestProperty("Cookie", cookiesString);
  98                 }
  99             }
 100
 101             // The actual network request is in a `try` bracket so that `disconnect()` is run in the `finally` section even if an error is encountered in the main block.
 102             try {
 103                 // Exit if the task has been cancelled.
 104                 if (isCancelled()) {
 105                     // Disconnect the HTTP URL connection.
 106                     httpUrlConnection.disconnect();
 107
 108                     // Return the formatted file size string.
 109                     return formattedFileSize;
 110                 }
 111
 112                 // Get the status code.  This initiates a network connection.
 113                 int responseCode = httpUrlConnection.getResponseCode();
 114
 115                 // Exit if the task has been cancelled.
 116                 if (isCancelled()) {
 117                     // Disconnect the HTTP URL connection.
 118                     httpUrlConnection.disconnect();
 119
 120                     // Return the formatted file size string.
 121                     return formattedFileSize;
 122                 }
 123
 124                 // Check the response code.
 125                 if (responseCode >= 400) {  // The response code is an error message.
 126                     // Set the formatted file size to indicate a bad URL.
 127                     formattedFileSize = context.getString(R.string.invalid_url);
 128                 } else {  // The response code is not an error message.
 129                     // Get the content length header.
 130                     String contentLengthString = httpUrlConnection.getHeaderField("Content-Length");
 131
 132                     // Only process the content length string if it isn't null.
 133                     if (contentLengthString != null) {
 134                         // Convert the content length string to a long.
 135                         long fileSize = Long.parseLong(contentLengthString);
 136
 137                         // Format the file size.
 138                         formattedFileSize = NumberFormat.getInstance().format(fileSize) + " " + context.getString(R.string.bytes);
 139                     }
 140                 }
 141             } finally {
 142                 // Disconnect the HTTP URL connection.
 143                 httpUrlConnection.disconnect();
 144             }
 145         } catch (Exception exception) {
 146             // Set the formatted file size to indicate a bad URL.
 147             formattedFileSize = context.getString(R.string.invalid_url);
 148         }
 149
 150         // Return the formatted file size string.
 151         return formattedFileSize;
 152     }
 153
 154     // `onPostExecute()` operates on the UI thread.
 155     @Override
 156     protected void onPostExecute(String fileSize) {
 157         // Get a handle for the alert dialog.
 158         AlertDialog alertDialog = alertDialogWeakReference.get();
 159
 160         // Abort if the alert dialog is gone.
 161         if (alertDialog == null) {
 162             return;
 163         }
 164
 165         // Get a handle for the file size text view.
 166         TextView fileSizeTextView = alertDialog.findViewById(R.id.file_size_textview);
 167
 168         // Remove the incorrect warning below that the file size text view might be null.
 169         assert fileSizeTextView != null;
 170
 171         // Update the file size.
 172         fileSizeTextView.setText(fileSize);
 173     }
 174 }