2 files changed, 93 insertions, 0 deletions
diff --git a/src/main/java/de/thedevstack/conversationsplus/utils/ConversationUtil.java b/src/main/java/de/thedevstack/conversationsplus/utils/ConversationUtil.java
new file mode 100644
index 00000000..958e9de8
--- /dev/null
+++ b/src/main/java/de/thedevstack/conversationsplus/utils/ConversationUtil.java
@@ -0,0 +1,77 @@
+package de.thedevstack.conversationsplus.utils;
+
+import android.net.Uri;
+
+import de.thedevstack.conversationsplus.ConversationsPlusApplication;
+import de.thedevstack.conversationsplus.exceptions.FileCopyException;
+
+import eu.siacs.conversations.crypto.PgpEngine;
+import eu.siacs.conversations.entities.Conversation;
+import eu.siacs.conversations.entities.Message;
+import eu.siacs.conversations.persistance.FileBackend;
+import eu.siacs.conversations.ui.UiCallback;
+import eu.siacs.conversations.utils.FileUtils;
+
+/**
+ * Utility class to work with conversations.
+ */
+public class ConversationUtil {
+
+    public static void attachLocationToConversation(final Conversation conversation,
+                                             final Uri uri,
+                                             final UiCallback<Message> callback) {
+        int encryption = conversation.getNextEncryption();
+        if (encryption == Message.ENCRYPTION_PGP) {
+            encryption = Message.ENCRYPTION_DECRYPTED;
+        }
+        Message message = new Message(conversation, uri.toString(), encryption);
+        if (conversation.getNextCounterpart() != null) {
+            message.setCounterpart(conversation.getNextCounterpart());
+        }
+        if (encryption == Message.ENCRYPTION_DECRYPTED) {
+            PgpEngine.getInstance().encrypt(message, callback);
+        } else {
+            callback.success(message);
+        }
+    }
+
+    public static void attachFileToConversation(final Conversation conversation,
+                                         final Uri uri,
+                                         final UiCallback<Message> callback) {
+        final Message message;
+        if (conversation.getNextEncryption() == Message.ENCRYPTION_PGP) {
+            message = new Message(conversation, "", Message.ENCRYPTION_DECRYPTED);
+        } else {
+            message = new Message(conversation, "", conversation.getNextEncryption());
+        }
+        message.setCounterpart(conversation.getNextCounterpart());
+        message.setType(Message.TYPE_FILE);
+        String path = FileUtils.getPath(uri);
+        if (path != null) {
+            message.setRelativeFilePath(path);
+            MessageUtil.updateFileParams(message);
+            if (message.getEncryption() == Message.ENCRYPTION_DECRYPTED) {
+                PgpEngine.getInstance().encrypt(message, callback);
+            } else {
+                callback.success(message);
+            }
+        } else {
+            ConversationsPlusApplication.executeFileAdding(new Runnable() {
+                @Override
+                public void run() {
+                    try {
+                        FileBackend.copyFileToPrivateStorage(message, uri);
+                        MessageUtil.updateFileParams(message);
+                        if (message.getEncryption() == Message.ENCRYPTION_DECRYPTED) {
+                            PgpEngine.getInstance().encrypt(message, callback);
+                        } else {
+                            callback.success(message);
+                        }
+                    } catch (FileCopyException e) {
+                        callback.error(e.getResId(), message);
+                    }
+                }
+            });
+        }
+    }
+}
diff --git a/src/main/java/de/thedevstack/conversationsplus/utils/MessageUtil.java b/src/main/java/de/thedevstack/conversationsplus/utils/MessageUtil.java
index afb8387d..ca24bd1d 100644
--- a/src/main/java/de/thedevstack/conversationsplus/utils/MessageUtil.java
+++ b/src/main/java/de/thedevstack/conversationsplus/utils/MessageUtil.java
@@ -8,6 +8,7 @@ import java.util.regex.Pattern;
 
 import de.thedevstack.conversationsplus.ConversationsPlusApplication;
 
+import eu.siacs.conversations.entities.Conversation;
 import eu.siacs.conversations.entities.DownloadableFile;
 import eu.siacs.conversations.entities.Message;
 import eu.siacs.conversations.persistance.DatabaseBackend;
@@ -18,6 +19,21 @@ import eu.siacs.conversations.persistance.FileBackend;
  */
 public final class MessageUtil {
 
+
+    public static boolean markMessage(Conversation conversation, String uuid, int status) {
+        if (uuid == null) {
+            return false;
+        } else {
+            Message message = conversation.findSentMessageWithUuid(uuid);
+            if (message != null) {
+                markMessage(message, status);
+                return true;
+            } else {
+                return false;
+            }
+        }
+    }
+
     public static void markMessage(Message message, int status) {
         if (status == Message.STATUS_SEND_FAILED
                 && (message.getStatus() == Message.STATUS_SEND_RECEIVED || message