From f9af0b1c1a4a4cd1bf003c63db335903aefc7ff2 Mon Sep 17 00:00:00 2001
From: M66B <M66B@users.noreply.github.com>
Date: Sun, 2 Feb 2020 14:05:49 +0100
Subject: [PATCH] Added blocking sender domain

---
 .../eu/faircode/email/AdapterMessage.java     | 16 +++++-
 .../eu/faircode/email/FragmentMessages.java   | 56 ++++++++++---------
 app/src/main/res/layout/dialog_junk.xml       | 15 ++++-
 app/src/main/res/values/strings.xml           |  1 +
 4 files changed, 58 insertions(+), 30 deletions(-)

diff --git a/app/src/main/java/eu/faircode/email/AdapterMessage.java b/app/src/main/java/eu/faircode/email/AdapterMessage.java
index 7305459fcd..aa5cc2d1bb 100644
--- a/app/src/main/java/eu/faircode/email/AdapterMessage.java
+++ b/app/src/main/java/eu/faircode/email/AdapterMessage.java
@@ -4951,17 +4951,27 @@ public class AdapterMessage extends RecyclerView.Adapter<AdapterMessage.ViewHold
 
             View view = LayoutInflater.from(getContext()).inflate(R.layout.dialog_junk, null);
             final TextView tvMessage = view.findViewById(R.id.tvMessage);
-            final CheckBox cbBlock = view.findViewById(R.id.cbBlock);
+            final CheckBox cbBlockSender = view.findViewById(R.id.cbBlockSender);
+            final CheckBox cbBlockDomain = view.findViewById(R.id.cbBlockDomain);
 
             tvMessage.setText(getString(R.string.title_ask_spam_who, from));
-            cbBlock.setEnabled(ActivityBilling.isPro(getContext()));
+            cbBlockSender.setEnabled(ActivityBilling.isPro(getContext()));
+            cbBlockDomain.setEnabled(false);
+
+            cbBlockSender.setOnCheckedChangeListener(new CompoundButton.OnCheckedChangeListener() {
+                @Override
+                public void onCheckedChanged(CompoundButton buttonView, boolean isChecked) {
+                    cbBlockDomain.setEnabled(isChecked);
+                }
+            });
 
             return new AlertDialog.Builder(getContext())
                     .setView(view)
                     .setPositiveButton(android.R.string.ok, new DialogInterface.OnClickListener() {
                         @Override
                         public void onClick(DialogInterface dialog, int which) {
-                            getArguments().putBoolean("block", cbBlock.isChecked());
+                            getArguments().putBoolean("block_sender", cbBlockSender.isChecked());
+                            getArguments().putBoolean("block_domain", cbBlockDomain.isChecked());
                             sendResult(RESULT_OK);
                         }
                     })
diff --git a/app/src/main/java/eu/faircode/email/FragmentMessages.java b/app/src/main/java/eu/faircode/email/FragmentMessages.java
index 1ecdd8a647..4f3771df58 100644
--- a/app/src/main/java/eu/faircode/email/FragmentMessages.java
+++ b/app/src/main/java/eu/faircode/email/FragmentMessages.java
@@ -5353,7 +5353,8 @@ public class FragmentMessages extends FragmentBase implements SharedPreferences.
             @Override
             protected Void onExecute(Context context, Bundle args) throws JSONException {
                 long id = args.getLong("id");
-                boolean block = args.getBoolean("block");
+                boolean block_sender = args.getBoolean("block_sender");
+                boolean block_domain = args.getBoolean("block_domain");
 
                 DB db = DB.getInstance(context);
                 try {
@@ -5369,33 +5370,38 @@ public class FragmentMessages extends FragmentBase implements SharedPreferences.
 
                     EntityOperation.queue(context, message, EntityOperation.MOVE, junk.id);
 
-                    if (block && message.from != null)
-                        for (Address from : message.from) {
-                            String sender = ((InternetAddress) from).getAddress();
-                            String name = MessageHelper.formatAddresses(new Address[]{from});
+                    if ((block_sender || block_domain) &&
+                            (message.from != null && message.from.length > 0)) {
+                        String sender = ((InternetAddress) message.from[0]).getAddress();
+                        String name = MessageHelper.formatAddresses(new Address[]{message.from[0]});
 
-                            JSONObject jsender = new JSONObject();
-                            jsender.put("value", sender);
-                            jsender.put("regex", false);
-
-                            JSONObject jcondition = new JSONObject();
-                            jcondition.put("sender", jsender);
-
-                            JSONObject jaction = new JSONObject();
-                            jaction.put("type", EntityRule.TYPE_MOVE);
-                            jaction.put("target", junk.id);
-
-                            EntityRule rule = new EntityRule();
-                            rule.folder = message.folder;
-                            rule.name = context.getString(R.string.title_block, name);
-                            rule.order = 1000;
-                            rule.enabled = true;
-                            rule.stop = true;
-                            rule.condition = jcondition.toString();
-                            rule.action = jaction.toString();
-                            rule.id = db.rule().insertRule(rule);
+                        if (block_domain) {
+                            int at = sender.indexOf('@');
+                            if (at > 0)
+                                sender = sender.substring(at);
                         }
 
+                        JSONObject jsender = new JSONObject();
+                        jsender.put("value", sender);
+                        jsender.put("regex", false);
+
+                        JSONObject jcondition = new JSONObject();
+                        jcondition.put("sender", jsender);
+
+                        JSONObject jaction = new JSONObject();
+                        jaction.put("type", EntityRule.TYPE_MOVE);
+                        jaction.put("target", junk.id);
+
+                        EntityRule rule = new EntityRule();
+                        rule.folder = message.folder;
+                        rule.name = context.getString(R.string.title_block, name);
+                        rule.order = 1000;
+                        rule.enabled = true;
+                        rule.stop = true;
+                        rule.condition = jcondition.toString();
+                        rule.action = jaction.toString();
+                        rule.id = db.rule().insertRule(rule);
+                    }
 
                     db.setTransactionSuccessful();
                 } finally {
diff --git a/app/src/main/res/layout/dialog_junk.xml b/app/src/main/res/layout/dialog_junk.xml
index bc2f73e39d..5636d6070f 100644
--- a/app/src/main/res/layout/dialog_junk.xml
+++ b/app/src/main/res/layout/dialog_junk.xml
@@ -21,7 +21,7 @@
             app:layout_constraintTop_toTopOf="parent" />
 
         <CheckBox
-            android:id="@+id/cbBlock"
+            android:id="@+id/cbBlockSender"
             android:layout_width="wrap_content"
             android:layout_height="wrap_content"
             android:layout_marginTop="12dp"
@@ -30,14 +30,25 @@
             app:layout_constraintStart_toStartOf="parent"
             app:layout_constraintTop_toBottomOf="@id/tvMessage" />
 
+        <CheckBox
+            android:id="@+id/cbBlockDomain"
+            android:layout_width="wrap_content"
+            android:layout_height="wrap_content"
+            android:layout_marginTop="12dp"
+            android:text="@string/title_block_domain"
+            android:textAppearance="@style/TextAppearance.AppCompat.Small"
+            app:layout_constraintStart_toStartOf="parent"
+            app:layout_constraintTop_toBottomOf="@id/cbBlockSender" />
+
         <TextView
             android:id="@+id/tvBlockHint"
             android:layout_width="wrap_content"
             android:layout_height="wrap_content"
+            android:layout_marginTop="12dp"
             android:text="@string/title_block_sender_hint"
             android:textAppearance="@style/TextAppearance.AppCompat.Small"
             android:textColor="?android:attr/textColorPrimary"
             app:layout_constraintStart_toStartOf="parent"
-            app:layout_constraintTop_toBottomOf="@+id/cbBlock" />
+            app:layout_constraintTop_toBottomOf="@+id/cbBlockDomain" />
     </androidx.constraintlayout.widget.ConstraintLayout>
 </ScrollView>
\ No newline at end of file
diff --git a/app/src/main/res/values/strings.xml b/app/src/main/res/values/strings.xml
index 057d1a1067..247c00de43 100644
--- a/app/src/main/res/values/strings.xml
+++ b/app/src/main/res/values/strings.xml
@@ -87,6 +87,7 @@
 
     <string name="title_block">Block %1$s</string>
     <string name="title_block_sender">Block sender</string>
+    <string name="title_block_domain">Block sender domain</string>
     <string name="title_block_sender_hint">
         Blocking a sender will create a rule to automatically move future messages to the spam folder.
         Creating and using rules is a pro feature.