From 197e366144e7d3c6fdb0ce788c8f1f7084ed238e Mon Sep 17 00:00:00 2001
From: M66B <M66B@users.noreply.github.com>
Date: Sun, 5 Sep 2021 22:01:28 +0200
Subject: [PATCH] Added POP3 raw save

---
 .../eu/faircode/email/AdapterMessage.java     | 12 +++++-----
 app/src/main/java/eu/faircode/email/Core.java | 24 +++++++++++++++++++
 .../email/FragmentDialogForwardRaw.java       | 16 +++++++++++++
 .../eu/faircode/email/FragmentMessages.java   |  2 +-
 4 files changed, 47 insertions(+), 7 deletions(-)

diff --git a/app/src/main/java/eu/faircode/email/AdapterMessage.java b/app/src/main/java/eu/faircode/email/AdapterMessage.java
index 79e77ff899..5c2782b4ed 100644
--- a/app/src/main/java/eu/faircode/email/AdapterMessage.java
+++ b/app/src/main/java/eu/faircode/email/AdapterMessage.java
@@ -4624,12 +4624,12 @@ public class AdapterMessage extends RecyclerView.Adapter<AdapterMessage.ViewHold
 
             popupMenu.getMenu().findItem(R.id.menu_share_as_html).setVisible(message.content && BuildConfig.DEBUG);
 
-            popupMenu.getMenu().findItem(R.id.menu_raw_save).setEnabled(message.uid != null);
-            popupMenu.getMenu().findItem(R.id.menu_raw_send_message).setEnabled(message.uid != null);
-            popupMenu.getMenu().findItem(R.id.menu_raw_send_thread).setEnabled(message.uid != null);
-
-            popupMenu.getMenu().findItem(R.id.menu_raw_save).setVisible(message.accountProtocol == EntityAccount.TYPE_IMAP);
-            popupMenu.getMenu().findItem(R.id.menu_raw_send).setVisible(message.accountProtocol == EntityAccount.TYPE_IMAP);
+            boolean canRaw = (message.uid != null ||
+                    (EntityFolder.INBOX.equals(message.folderType) &&
+                            message.accountProtocol == EntityAccount.TYPE_POP));
+            popupMenu.getMenu().findItem(R.id.menu_raw_save).setEnabled(canRaw);
+            popupMenu.getMenu().findItem(R.id.menu_raw_send_message).setEnabled(canRaw);
+            popupMenu.getMenu().findItem(R.id.menu_raw_send_thread).setEnabled(canRaw);
 
             popupMenu.getMenu().findItem(R.id.menu_resync)
                     .setEnabled(message.uid != null)
diff --git a/app/src/main/java/eu/faircode/email/Core.java b/app/src/main/java/eu/faircode/email/Core.java
index 5abe4f06e2..a9bbcbf015 100644
--- a/app/src/main/java/eu/faircode/email/Core.java
+++ b/app/src/main/java/eu/faircode/email/Core.java
@@ -350,6 +350,10 @@ class Core {
                                     onDelete(context, jargs, account, folder, message, (POP3Folder) ifolder, (POP3Store) istore, state);
                                     break;
 
+                                case EntityOperation.RAW:
+                                    onRaw(context, jargs, folder, message, (POP3Store) istore, (POP3Folder) ifolder);
+                                    break;
+
                                 case EntityOperation.SYNC:
                                     Helper.gc();
                                     onSynchronizeMessages(context, jargs, account, folder, (POP3Folder) ifolder, (POP3Store) istore, state);
@@ -1759,6 +1763,26 @@ class Core {
         }
     }
 
+    private static void onRaw(Context context, JSONArray jargs, EntityFolder folder, EntityMessage message, POP3Store istore, POP3Folder ifolder) throws MessagingException, IOException, JSONException {
+        // Download raw message
+        if (!EntityFolder.INBOX.equals(folder.type))
+            throw new IllegalArgumentException("Unexpected folder=" + folder.type);
+
+        if (message.raw == null || !message.raw) {
+            Message imessage = findMessage(context, folder, message, istore, ifolder);
+            if (imessage == null)
+                throw new IllegalArgumentException("Message not found msgid=" + message.msgid);
+
+            File file = message.getRawFile(context);
+            try (OutputStream os = new BufferedOutputStream(new FileOutputStream(file))) {
+                imessage.writeTo(os);
+            }
+
+            DB db = DB.getInstance(context);
+            db.message().setMessageRaw(message.id, true);
+        }
+    }
+
     private static void onBody(Context context, JSONArray jargs, EntityFolder folder, EntityMessage message, IMAPFolder ifolder) throws MessagingException, IOException {
         // Download message body
         DB db = DB.getInstance(context);
diff --git a/app/src/main/java/eu/faircode/email/FragmentDialogForwardRaw.java b/app/src/main/java/eu/faircode/email/FragmentDialogForwardRaw.java
index 07d81a5f0a..9e45dfb775 100644
--- a/app/src/main/java/eu/faircode/email/FragmentDialogForwardRaw.java
+++ b/app/src/main/java/eu/faircode/email/FragmentDialogForwardRaw.java
@@ -86,6 +86,22 @@ public class FragmentDialogForwardRaw extends FragmentDialogBase {
                         if (message == null)
                             continue;
 
+                        EntityFolder folder = db.folder().getFolder(message.folder);
+                        if (folder == null)
+                            continue;
+
+                        EntityAccount account = db.account().getAccount(message.account);
+                        if (account == null)
+                            continue;
+
+                        if (account.protocol == EntityAccount.TYPE_IMAP) {
+                            if (message.uid == null)
+                                continue;
+                        } else if (account.protocol == EntityAccount.TYPE_POP) {
+                            if (!EntityFolder.INBOX.equals(folder.type))
+                                continue;
+                        }
+
                         List<EntityMessage> messages = db.message().getMessagesByThread(
                                 message.account, message.thread, threads ? null : id, null);
                         if (messages == null)
diff --git a/app/src/main/java/eu/faircode/email/FragmentMessages.java b/app/src/main/java/eu/faircode/email/FragmentMessages.java
index a64b5078b9..b7d4ca0e74 100644
--- a/app/src/main/java/eu/faircode/email/FragmentMessages.java
+++ b/app/src/main/java/eu/faircode/email/FragmentMessages.java
@@ -3068,7 +3068,7 @@ public class FragmentMessages extends FragmentBase implements SharedPreferences.
                         .setIcon(R.drawable.baseline_arrow_downward_24)
                         .setEnabled(!EntityMessage.PRIORITIY_LOW.equals(result.importance));
 
-                if (result.accounts.size() > 0 /* IMAP */ && ids.length < MAX_SEND_RAW)
+                if (ids.length < MAX_SEND_RAW)
                     popupMenu.getMenu().add(Menu.NONE, R.string.title_raw_send, order++, R.string.title_raw_send)
                             .setIcon(R.drawable.twotone_attachment_24);