app/src/main/java/com/stoutner/privacybrowser/dialogs/DownloadFileDialog.java

   1 /*
   2  * Copyright © 2016-2017 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.dialogs;
  21
  22 import android.annotation.SuppressLint;
  23 import android.app.AlertDialog;
  24 import android.app.Dialog;
  25 import android.content.Context;
  26 import android.content.DialogInterface;
  27 import android.net.Uri;
  28 import android.os.Bundle;
  29 import android.support.annotation.NonNull;
  30 // We have to use `AppCompatDialogFragment` instead of `DialogFragment` or an error is produced on API <=22.
  31 import android.support.v7.app.AppCompatDialogFragment;
  32 import android.view.KeyEvent;
  33 import android.view.LayoutInflater;
  34 import android.view.View;
  35 import android.view.WindowManager;
  36 import android.widget.EditText;
  37 import android.widget.TextView;
  38
  39 import com.stoutner.privacybrowser.R;
  40 import com.stoutner.privacybrowser.activities.MainWebViewActivity;
  41
  42 import java.util.Locale;
  43
  44 public class DownloadFileDialog extends AppCompatDialogFragment {
  45
  46     private String downloadUrl;
  47     private String downloadFileName;
  48     private String fileSize;
  49
  50     public static DownloadFileDialog fromUrl(String urlString, String contentDisposition, long contentLength) {
  51         // Create `argumentsBundle`.
  52         Bundle argumentsBundle = new Bundle();
  53
  54         String fileNameString;
  55
  56         // Parse `filename` from `contentDisposition`.
  57         if (contentDisposition.contains("filename=\"")) {  // The file name is contained in a string surrounded by `""`.
  58             fileNameString = contentDisposition.substring(contentDisposition.indexOf("filename=\"") + 10, contentDisposition.indexOf("\"", contentDisposition.indexOf("filename=\"") + 10));
  59         } else if (contentDisposition.contains("filename=") && ((contentDisposition.indexOf(";", contentDisposition.indexOf("filename=") + 9)) > 0 )) {  // The file name is contained in a string beginning with `filename=` and ending with `;`.
  60             fileNameString = contentDisposition.substring(contentDisposition.indexOf("filename=") + 9, contentDisposition.indexOf(";", contentDisposition.indexOf("filename=") + 9));
  61         } else if (contentDisposition.contains("filename=")) {  // The file name is contained in a string beginning with `filename=` and proceeding to the end of `contentDisposition`.
  62             fileNameString = contentDisposition.substring(contentDisposition.indexOf("filename=") + 9, contentDisposition.length());
  63         } else {  // `contentDisposition` does not contain the filename, so use the last path segment of the URL.
  64             Uri downloadUri = Uri.parse(urlString);
  65             fileNameString = downloadUri.getLastPathSegment();
  66         }
  67
  68         // Store the variables in the `Bundle`.
  69         argumentsBundle.putString("URL", urlString);
  70         argumentsBundle.putString("File_Name", fileNameString);
  71         argumentsBundle.putLong("File_Size", contentLength);
  72
  73         // Add `argumentsBundle` to this instance of `DownloadFileDialog`.
  74         DownloadFileDialog thisDownloadFileDialog = new DownloadFileDialog();
  75         thisDownloadFileDialog.setArguments(argumentsBundle);
  76         return thisDownloadFileDialog;
  77     }
  78
  79     @Override
  80     public void onCreate(Bundle savedInstanceState) {
  81         super.onCreate(savedInstanceState);
  82
  83         // Store the strings in the local class variables.
  84         downloadUrl = getArguments().getString("URL");
  85         downloadFileName = getArguments().getString("File_Name");
  86
  87         // Get the `File_Size`.
  88         long fileSizeLong = getArguments().getLong("File_Size");
  89
  90         // Convert `fileSizeLong` to a String.
  91         if (fileSizeLong == -1) {  // We don't know the file size.
  92             fileSize = getString(R.string.unknown_size);
  93         } else {  // Convert `fileSize` to MB and store it in `fileSizeString`.  `%.3g` displays the three most significant digits.
  94             fileSize = String.format(Locale.getDefault(), "%.3g", (float) fileSizeLong / 1048576) + " MB";
  95         }
  96     }
  97
  98     // The public interface is used to send information back to the parent activity.
  99     public interface DownloadFileListener {
 100         void onDownloadFile(AppCompatDialogFragment dialogFragment, String downloadUrl);
 101     }
 102
 103     // `downloadFileListener` is used in `onAttach()` and `onCreateDialog()`.
 104     private DownloadFileListener downloadFileListener;
 105
 106     @Override
 107     public void onAttach(Context context) {
 108         super.onAttach(context);
 109
 110         // Check to make sure the parent activity implements the listener.
 111         try {
 112             downloadFileListener = (DownloadFileListener) context;
 113         } catch (ClassCastException exception) {
 114             throw new ClassCastException(context.toString() + " must implement DownloadFileListener.");
 115         }
 116     }
 117
 118     @Override
 119     @NonNull
 120     // `@SuppressLing("InflateParams")` removes the warning about using `null` as the parent view group when inflating the `AlertDialog`.
 121     @SuppressLint("InflateParams")
 122     public Dialog onCreateDialog(Bundle savedInstanceState) {
 123         // Get the activity's layout inflater.
 124         LayoutInflater layoutInflater = getActivity().getLayoutInflater();
 125
 126         // Use `AlertDialog.Builder` to create the `AlertDialog`.
 127         AlertDialog.Builder dialogBuilder;
 128
 129         // Set the style according to the theme.
 130         if (MainWebViewActivity.darkTheme) {
 131             dialogBuilder = new AlertDialog.Builder(getActivity(), R.style.PrivacyBrowserAlertDialogDark);
 132         } else {
 133             dialogBuilder = new AlertDialog.Builder(getActivity(), R.style.PrivacyBrowserAlertDialogLight);
 134         }
 135
 136         // Set the title.
 137         dialogBuilder.setTitle(R.string.save_as);
 138
 139         // Set the view.  The parent view is `null` because it will be assigned by `AlertDialog`.
 140         dialogBuilder.setView(layoutInflater.inflate(R.layout.download_file_dialog, null));
 141
 142         // Set an `onClick()` listener on the negative button.
 143         dialogBuilder.setNegativeButton(R.string.cancel, new DialogInterface.OnClickListener() {
 144             @Override
 145             public void onClick(DialogInterface dialog, int which) {
 146                 // Do nothing if `Cancel` is clicked.  The `Dialog` will automatically close.
 147             }
 148         });
 149
 150         // Set an `onClick()` listener on the positive button
 151         dialogBuilder.setPositiveButton(R.string.download, new DialogInterface.OnClickListener() {
 152             @Override
 153             public void onClick(DialogInterface dialog, int which) {
 154                 // trigger `onDownloadFile()` and return the `DialogFragment` and the download URL to the parent activity.
 155                 downloadFileListener.onDownloadFile(DownloadFileDialog.this, downloadUrl);
 156             }
 157         });
 158
 159         // Create an `AlertDialog` from the `AlertDialog.Builder`.
 160         final AlertDialog alertDialog = dialogBuilder.create();
 161
 162         // Remove the warning below that `setSoftInputMode` might produce `java.lang.NullPointerException`.
 163         assert alertDialog.getWindow() != null;
 164
 165         // Show the keyboard when `alertDialog` is displayed on the screen.
 166         alertDialog.getWindow().setSoftInputMode(WindowManager.LayoutParams.SOFT_INPUT_STATE_ALWAYS_VISIBLE);
 167
 168         // We need to show `alertDialog` before we can modify the contents.
 169         alertDialog.show();
 170
 171         // Set the text for `downloadFileSizeTextView`.
 172         TextView downloadFileSizeTextView = (TextView) alertDialog.findViewById(R.id.download_file_size);
 173         downloadFileSizeTextView.setText(fileSize);
 174
 175         // Set the text for `downloadFileNameTextView`.
 176         EditText downloadFileNameTextView = (EditText) alertDialog.findViewById(R.id.download_file_name);
 177         downloadFileNameTextView.setText(downloadFileName);
 178
 179         // Allow the `enter` key on the keyboard to save the file from `downloadFileNameTextView`.
 180         downloadFileNameTextView.setOnKeyListener(new View.OnKeyListener() {
 181             @Override
 182             public boolean onKey (View v, int keyCode, KeyEvent event) {
 183                 // If the event is an `ACTION_DOWN` on the `enter` key, initiate the download.
 184                 if ((event.getAction() == KeyEvent.ACTION_DOWN) && (keyCode == KeyEvent.KEYCODE_ENTER)) {
 185                     // trigger `onDownloadFile()` and return the `DialogFragment` and the URL to the parent activity.
 186                     downloadFileListener.onDownloadFile(DownloadFileDialog.this, downloadUrl);
 187                     // Manually dismiss `alertDialog`.
 188                     alertDialog.dismiss();
 189                     // Consume the event.
 190                     return true;
 191                 } else {  // If any other key was pressed, do not consume the event.
 192                     return false;
 193                 }
 194             }
 195         });
 196
 197         // `onCreateDialog` requires the return of an `AlertDialog`.
 198         return alertDialog;
 199     }
 200 }