1 files changed, 35 insertions, 0 deletions
diff --git a/src/main/java/de/thedevstack/conversationsplus/services/filetransfer/httpupload/HttpFileTransferEntity.java b/src/main/java/de/thedevstack/conversationsplus/services/filetransfer/httpupload/HttpFileTransferEntity.java
index 8efd498b..a8e5734f 100644
--- a/src/main/java/de/thedevstack/conversationsplus/services/filetransfer/httpupload/HttpFileTransferEntity.java
+++ b/src/main/java/de/thedevstack/conversationsplus/services/filetransfer/httpupload/HttpFileTransferEntity.java
@@ -1,9 +1,16 @@
 package de.thedevstack.conversationsplus.services.filetransfer.httpupload;
 
+import java.net.MalformedURLException;
+import java.net.URL;
+
+import de.thedevstack.android.logcat.Logging;
 import de.thedevstack.conversationsplus.Config;
 import de.thedevstack.conversationsplus.ConversationsPlusApplication;
 import de.thedevstack.conversationsplus.entities.Message;
 import de.thedevstack.conversationsplus.services.filetransfer.FileTransferEntity;
+import de.thedevstack.conversationsplus.services.filetransfer.FileTransferFailureReason;
+import de.thedevstack.conversationsplus.utils.CryptoHelper;
+import de.thedevstack.conversationsplus.utils.MessageUtil;
 import de.thedevstack.conversationsplus.xmpp.httpupload.HttpUploadSlot;
 
 /**
@@ -49,4 +56,32 @@ public class HttpFileTransferEntity extends FileTransferEntity {
     public boolean isDelayed() {
         return this.delayed;
     }
+
+    @Override
+    public void fail(FileTransferFailureReason failureReason) {
+        this.getMessage().setHttpUploaded(false);
+        super.fail(failureReason);
+    }
+
+    @Override
+    public void cancel() {
+        this.getMessage().setHttpUploaded(false);
+        super.cancel();
+    }
+
+    @Override
+    public void transferred() {
+        try {
+            URL getUrl = new URL(this.getGetUrl());
+            if (this.getKey() != null) {
+                getUrl = new URL(getUrl.toString() + "#" + CryptoHelper.bytesToHex(this.getKey()));
+            }
+
+            MessageUtil.updateFileParams(this.getMessage(), getUrl);
+        } catch (MalformedURLException e) {
+            Logging.e("httpupload", "Not a valid get url");
+        }
+
+        super.transferred();
+    }
 }