03ab37464b21782137fad85e0f533c51b4a6c5bf
[PrivacyBrowser.git] / app / src / main / java / com / stoutner / privacybrowser / asynctasks / SaveUrl.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.app.Activity;
23 import android.content.Context;
24 import android.net.Uri;
25 import android.os.AsyncTask;
26 import android.util.Base64;
27 import android.webkit.CookieManager;
28
29 import com.google.android.material.snackbar.Snackbar;
30 import com.stoutner.privacybrowser.R;
31 import com.stoutner.privacybrowser.helpers.ProxyHelper;
32 import com.stoutner.privacybrowser.views.NoSwipeViewPager;
33
34 import java.io.BufferedInputStream;
35 import java.io.InputStream;
36 import java.io.OutputStream;
37 import java.lang.ref.WeakReference;
38 import java.net.HttpURLConnection;
39 import java.net.Proxy;
40 import java.net.URL;
41 import java.text.NumberFormat;
42
43 public class SaveUrl extends AsyncTask<String, Long, String> {
44     // Define a weak references.
45     private final WeakReference<Context> contextWeakReference;
46     private final WeakReference<Activity> activityWeakReference;
47
48     // Define a success string constant.
49     private final String SUCCESS = "Success";
50
51     // Define the class variables.
52     private final String filePathString;
53     private final String userAgent;
54     private final boolean cookiesEnabled;
55     private Snackbar savingFileSnackbar;
56     private long fileSize;
57     private String formattedFileSize;
58     private String urlString = "";
59
60     // The public constructor.
61     public SaveUrl(Context context, Activity activity, String filePathString, String userAgent, boolean cookiesEnabled) {
62         // Populate weak references to the calling context and activity.
63         contextWeakReference = new WeakReference<>(context);
64         activityWeakReference = new WeakReference<>(activity);
65
66         // Store the class variables.
67         this.filePathString = filePathString;
68         this.userAgent = userAgent;
69         this.cookiesEnabled = cookiesEnabled;
70     }
71
72     // `onPreExecute()` operates on the UI thread.
73     @Override
74     protected void onPreExecute() {
75         // Get a handle for the activity.
76         Activity activity = activityWeakReference.get();
77
78         // Abort if the activity is gone.
79         if ((activity==null) || activity.isFinishing()) {
80             return;
81         }
82
83         // Get a handle for the no swipe view pager.
84         NoSwipeViewPager noSwipeViewPager = activity.findViewById(R.id.webviewpager);
85
86         // Create a saving file snackbar.
87         savingFileSnackbar = Snackbar.make(noSwipeViewPager, activity.getString(R.string.saving_file) + "  0%  -  " + urlString, Snackbar.LENGTH_INDEFINITE);
88
89         // Display the saving file snackbar.
90         savingFileSnackbar.show();
91     }
92
93     @Override
94     protected String doInBackground(String... urlToSave) {
95         // Get handles for the context and activity.
96         Context context = contextWeakReference.get();
97         Activity activity = activityWeakReference.get();
98
99         // Abort if the activity is gone.
100         if ((activity == null) || activity.isFinishing()) {
101             return null;
102         }
103
104         // Define a save disposition string.
105         String saveDisposition = SUCCESS;
106
107         // Get the URL string.
108         urlString = urlToSave[0];
109
110         try {
111             // Open an output stream.
112             OutputStream outputStream = activity.getContentResolver().openOutputStream(Uri.parse(filePathString));
113
114             // Save the URL.
115             if (urlString.startsWith("data:")) {  // The URL contains the entire data of an image.
116                 // Get the Base64 data, which begins after a `,`.
117                 String base64DataString = urlString.substring(urlString.indexOf(",") + 1);
118
119                 // Decode the Base64 string to a byte array.
120                 byte[] base64DecodedDataByteArray = Base64.decode(base64DataString, Base64.DEFAULT);
121
122                 // Write the Base64 byte array to the output stream.
123                 outputStream.write(base64DecodedDataByteArray);
124             } else {  // The URL points to the data location on the internet.
125                 // Get the URL from the calling activity.
126                 URL url = new URL(urlString);
127
128                 // Instantiate the proxy helper.
129                 ProxyHelper proxyHelper = new ProxyHelper();
130
131                 // Get the current proxy.
132                 Proxy proxy = proxyHelper.getCurrentProxy(context);
133
134                 // Open a connection to the URL.  No data is actually sent at this point.
135                 HttpURLConnection httpUrlConnection = (HttpURLConnection) url.openConnection(proxy);
136
137                 // Add the user agent to the header property.
138                 httpUrlConnection.setRequestProperty("User-Agent", userAgent);
139
140                 // Add the cookies if they are enabled.
141                 if (cookiesEnabled) {
142                     // Get the cookies for the current domain.
143                     String cookiesString = CookieManager.getInstance().getCookie(url.toString());
144
145                     // Only add the cookies if they are not null.
146                     if (cookiesString != null) {
147                         // Add the cookies to the header property.
148                         httpUrlConnection.setRequestProperty("Cookie", cookiesString);
149                     }
150                 }
151
152                 // 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.
153                 try {
154                     // Get the content length header, which causes the connection to the server to be made.
155                     String contentLengthString = httpUrlConnection.getHeaderField("Content-Length");
156
157                     // Make sure the content length isn't null.
158                     if (contentLengthString != null) {  // The content length isn't null.
159                         // Convert the content length to an long.
160                         fileSize = Long.parseLong(contentLengthString);
161
162                         // Format the file size for display.
163                         formattedFileSize = NumberFormat.getInstance().format(fileSize);
164                     } else {  // The content length is null.
165                         // Set the file size to be `-1`.
166                         fileSize = -1;
167                     }
168
169                     // Get the response body stream.
170                     InputStream inputStream = new BufferedInputStream(httpUrlConnection.getInputStream());
171
172                     // Initialize the conversion buffer byte array.
173                     byte[] conversionBufferByteArray = new byte[1024];
174
175                     // Initialize the downloaded kilobytes counter.
176                     long downloadedKilobytesCounter = 0;
177
178                     // Define the buffer length variable.
179                     int bufferLength;
180
181                     // Attempt to read data from the input stream and store it in the output stream.  Also store the amount of data read in the buffer length variable.
182                     while ((bufferLength = inputStream.read(conversionBufferByteArray)) > 0) {  // Proceed while the amount of data stored in the buffer in > 0.
183                         // Write the contents of the conversion buffer to the file output stream.
184                         outputStream.write(conversionBufferByteArray, 0, bufferLength);
185
186                         // Update the downloaded kilobytes counter.
187                         downloadedKilobytesCounter = downloadedKilobytesCounter + bufferLength;
188
189                         // Update the file download progress snackbar.
190                         publishProgress(downloadedKilobytesCounter);
191                     }
192
193                     // Close the input stream.
194                     inputStream.close();
195                 } finally {
196                     // Disconnect the HTTP URL connection.
197                     httpUrlConnection.disconnect();
198                 }
199             }
200
201             // Close the output stream.
202             outputStream.close();
203         } catch (Exception exception) {
204             // Store the error in the save disposition string.
205             saveDisposition = exception.toString();
206         }
207
208         // Return the save disposition string.
209         return saveDisposition;
210     }
211
212     // `onProgressUpdate()` operates on the UI thread.
213     @Override
214     protected void onProgressUpdate(Long... numberOfBytesDownloaded) {
215         // Get a handle for the activity.
216         Activity activity = activityWeakReference.get();
217
218         // Abort if the activity is gone.
219         if ((activity == null) || activity.isFinishing()) {
220             return;
221         }
222
223         // Format the number of bytes downloaded.
224         String formattedNumberOfBytesDownloaded = NumberFormat.getInstance().format(numberOfBytesDownloaded[0]);
225
226         // Check to see if the file size is known.
227         if (fileSize == -1) {  // The size of the download file is not known.
228             // Update the snackbar.
229             savingFileSnackbar.setText(activity.getString(R.string.saving_file) + "  " + formattedNumberOfBytesDownloaded + " " + activity.getString(R.string.bytes) + "  -  " + urlString);
230         } else {  // The size of the download file is known.
231             // Calculate the download percentage.
232             long downloadPercentage = (numberOfBytesDownloaded[0] * 100) / fileSize;
233
234             // Update the snackbar.
235             savingFileSnackbar.setText(activity.getString(R.string.saving_file) + "  " + downloadPercentage + "%  -  " + formattedNumberOfBytesDownloaded + " " + activity.getString(R.string.bytes) + " / " + formattedFileSize + " " +
236                     activity.getString(R.string.bytes) + "  -  " + urlString);
237         }
238     }
239
240     // `onPostExecute()` operates on the UI thread.
241     @Override
242     protected void onPostExecute(String saveDisposition) {
243         // Get handles for the context and activity.
244         Activity activity = activityWeakReference.get();
245
246         // Abort if the activity is gone.
247         if ((activity == null) || activity.isFinishing()) {
248             return;
249         }
250
251         // Get a handle for the no swipe view pager.
252         NoSwipeViewPager noSwipeViewPager = activity.findViewById(R.id.webviewpager);
253
254         // Dismiss the saving file snackbar.
255         savingFileSnackbar.dismiss();
256
257         // Display a save disposition snackbar.
258         if (saveDisposition.equals(SUCCESS)) {
259             // Display the file saved snackbar.
260             Snackbar.make(noSwipeViewPager, activity.getString(R.string.file_saved) + "  " + urlString, Snackbar.LENGTH_LONG).show();
261         } else {
262             // Display the file saving error.
263             Snackbar.make(noSwipeViewPager, activity.getString(R.string.error_saving_file) + "  " + saveDisposition, Snackbar.LENGTH_INDEFINITE).show();
264         }
265     }
266 }