From 851b22373353f182750c53ffa648424788980bcf Mon Sep 17 00:00:00 2001
From: vanitasvitae <vanitasvitae@fsfe.org>
Date: Sun, 6 Nov 2016 22:19:22 +0100
Subject: [PATCH] Fixed faulty extensions on downloaded images

---
 .../service/SaveImageTask.java                | 74 +++++++++++++++++++
 .../web/ContextMenuWebView.java               | 23 +++---
 2 files changed, 86 insertions(+), 11 deletions(-)
 create mode 100644 app/src/main/java/com/github/dfa/diaspora_android/service/SaveImageTask.java

diff --git a/app/src/main/java/com/github/dfa/diaspora_android/service/SaveImageTask.java b/app/src/main/java/com/github/dfa/diaspora_android/service/SaveImageTask.java
new file mode 100644
index 00000000..49cf6ab0
--- /dev/null
+++ b/app/src/main/java/com/github/dfa/diaspora_android/service/SaveImageTask.java
@@ -0,0 +1,74 @@
+package com.github.dfa.diaspora_android.service;
+
+import android.app.DownloadManager;
+import android.content.Context;
+import android.net.Uri;
+import android.os.AsyncTask;
+import android.os.Environment;
+import android.os.Looper;
+import android.os.StrictMode;
+import android.widget.Toast;
+
+import com.github.dfa.diaspora_android.R;
+import com.github.dfa.diaspora_android.util.AppLog;
+
+import java.io.File;
+import java.io.IOException;
+import java.io.InputStream;
+import java.net.URL;
+
+/**
+ * Created by vanitas on 06.11.16.
+ */
+
+public class SaveImageTask extends AsyncTask<String, Void, String> {
+
+    protected Context context;
+
+    public void setContext(Context context) {
+        this.context = context;
+    }
+
+    protected String doInBackground(String... urls) {
+        String url = urls[0];
+            if (url != null) {
+                Uri source = Uri.parse(url);
+                DownloadManager.Request request = new DownloadManager.Request(source);
+                URL sourceUrl;
+                InputStream is;
+                byte[] a = new byte[8];
+                String extension = ".png";
+                try {
+                    sourceUrl = new URL(source.toString());
+                    is = sourceUrl.openStream();
+                    is.read(a);
+                    is.close();
+                    AppLog.d(this, "Array: " + new String(a));
+                    //JPG
+                    if (new String(a).startsWith(new String(new byte[]{-1, -40}))) {
+                        AppLog.d(this, "is jpg");
+                        extension = ".jpg";
+                    } else
+                        //GIF
+                        if (new String(a).startsWith("GIF")) {
+                            AppLog.d(this, "is gif");
+                            extension = ".gif";
+                        } else {
+                            AppLog.d(this, "is SPARTAAAA! (GIF)");
+                        }
+                } catch (IOException ignored) {
+                }
+                File destinationFile = new File(Environment.getExternalStorageDirectory() + "/Pictures/Diaspora/"
+                        + System.currentTimeMillis() + extension);
+                request.setDestinationUri(Uri.fromFile(destinationFile));
+                ((DownloadManager) context.getSystemService(Context.DOWNLOAD_SERVICE)).enqueue(request);
+                return destinationFile.getAbsolutePath();
+            }
+        return null;
+    }
+
+    @Override
+    protected void onPostExecute(String s) {
+        Toast.makeText(context, context.getString(R.string.share__toast_saved_image_to_location)+" "+s, Toast.LENGTH_LONG).show();
+    }
+}
\ No newline at end of file
diff --git a/app/src/main/java/com/github/dfa/diaspora_android/web/ContextMenuWebView.java b/app/src/main/java/com/github/dfa/diaspora_android/web/ContextMenuWebView.java
index f9fc22fb..351c12ea 100644
--- a/app/src/main/java/com/github/dfa/diaspora_android/web/ContextMenuWebView.java
+++ b/app/src/main/java/com/github/dfa/diaspora_android/web/ContextMenuWebView.java
@@ -30,18 +30,27 @@ import android.content.Intent;
 import android.content.pm.PackageManager;
 import android.graphics.Bitmap;
 import android.net.Uri;
+import android.os.AsyncTask;
 import android.os.Environment;
+import android.os.StrictMode;
 import android.support.v4.content.LocalBroadcastManager;
 import android.util.AttributeSet;
 import android.view.ContextMenu;
 import android.view.MenuItem;
 import android.widget.Toast;
 
+import com.github.dfa.diaspora_android.App;
 import com.github.dfa.diaspora_android.R;
 import com.github.dfa.diaspora_android.activity.MainActivity;
 import com.github.dfa.diaspora_android.service.ImageDownloadTask;
+import com.github.dfa.diaspora_android.service.SaveImageTask;
+import com.github.dfa.diaspora_android.util.AppLog;
 
 import java.io.File;
+import java.io.IOException;
+import java.io.InputStream;
+import java.net.MalformedURLException;
+import java.net.URL;
 
 /**
  * Subclass of WebView which adds a context menu for long clicks on images or links to share, save
@@ -107,17 +116,9 @@ public class ContextMenuWebView extends NestedWebView {
                             }
                         }
                         if (writeToStoragePermitted) {
-                            if (url != null) {
-                                Uri source = Uri.parse(url);
-                                DownloadManager.Request request = new DownloadManager.Request(source);
-                                File destinationFile = new File(Environment.getExternalStorageDirectory() + "/Pictures/Diaspora/"
-                                        + System.currentTimeMillis() + ".png");
-                                request.setDestinationUri(Uri.fromFile(destinationFile));
-                                ((DownloadManager) context.getSystemService(Context.DOWNLOAD_SERVICE)).enqueue(request);
-
-                                Toast.makeText(context, context.getText(R.string.share__toast_saved_image_to_location) + " " +
-                                        destinationFile.getAbsolutePath(), Toast.LENGTH_LONG).show();
-                            }
+                            SaveImageTask saveTask = new SaveImageTask();
+                            saveTask.setContext(context);
+                            saveTask.execute(url);
                         }
                     }
                     break;