X-Git-Url: https://gitweb.stoutner.com/?a=blobdiff_plain;ds=sidebyside;f=app%2Fsrc%2Fmain%2Fjava%2Fcom%2Fstoutner%2Fprivacybrowser%2Fdialogs%2FDownloadImageDialog.java;h=9c1afe67e91bb7f96f3cb69b56b66cbfd0c21c58;hb=4d51aa9acb8daaec1326f14e5025fde6d1f0dcd8;hp=06d765c22276cc529ec442fb6ff56e9b1dd4156a;hpb=f0393ca22075be3e5fe9199c7db87381256236fa;p=PrivacyBrowserAndroid.git

diff --git a/app/src/main/java/com/stoutner/privacybrowser/dialogs/DownloadImageDialog.java b/app/src/main/java/com/stoutner/privacybrowser/dialogs/DownloadImageDialog.java
index 06d765c2..9c1afe67 100644
--- a/app/src/main/java/com/stoutner/privacybrowser/dialogs/DownloadImageDialog.java
+++ b/app/src/main/java/com/stoutner/privacybrowser/dialogs/DownloadImageDialog.java
@@ -29,7 +29,6 @@ import android.net.Uri;
 import android.os.Bundle;
 import android.preference.PreferenceManager;
 import android.view.KeyEvent;
-import android.view.LayoutInflater;
 import android.view.View;
 import android.view.WindowManager;
 import android.widget.EditText;
@@ -50,7 +49,7 @@ public class DownloadImageDialog extends DialogFragment {
 
     // Check to make sure tha the parent activity implements the listener.
     @Override
-    public void onAttach(Context context) {
+    public void onAttach(@NonNull Context context) {
         // Run the default commands.
         super.onAttach(context);
 
@@ -94,12 +93,6 @@ public class DownloadImageDialog extends DialogFragment {
         // Remove the warning below that `.getActivity()` might be null.
         assert getActivity() != null;
 
-        // Get the activity's layout inflater.
-        LayoutInflater layoutInflater = getActivity().getLayoutInflater();
-
-        // Use and alert dialog builder to create the alert dialog.
-        AlertDialog.Builder dialogBuilder;
-
         // Get a handle for the shared preferences.
         SharedPreferences sharedPreferences = PreferenceManager.getDefaultSharedPreferences(getContext());
 
@@ -107,6 +100,9 @@ public class DownloadImageDialog extends DialogFragment {
         boolean allowScreenshots = sharedPreferences.getBoolean("allow_screenshots", false);
         boolean darkTheme = sharedPreferences.getBoolean("dark_theme", false);
 
+        // Use and alert dialog builder to create the alert dialog.
+        AlertDialog.Builder dialogBuilder;
+
         // Set the style according to the theme.
         if (darkTheme) {
             dialogBuilder = new AlertDialog.Builder(getActivity(), R.style.PrivacyBrowserAlertDialogDark);
@@ -117,8 +113,18 @@ public class DownloadImageDialog extends DialogFragment {
         // Set the title.
         dialogBuilder.setTitle(R.string.save_image_as);
 
+        // Set the icon according to the theme.
+        if (darkTheme) {
+            dialogBuilder.setIcon(R.drawable.images_enabled_dark);
+        } else {
+            dialogBuilder.setIcon(R.drawable.images_enabled_light);
+        }
+
+        // Remove the incorrect lint warning below that `getActivity() might be null.
+        assert getActivity() != null;
+
         // Set the view.  The parent view is `null` because it will be assigned by `AlertDialog`.
-        dialogBuilder.setView(layoutInflater.inflate(R.layout.download_image_dialog, null));
+        dialogBuilder.setView(getActivity().getLayoutInflater().inflate(R.layout.download_image_dialog, null));
 
         // Set an listener on the negative button.
         dialogBuilder.setNegativeButton(R.string.cancel, (DialogInterface dialog, int which) -> {