From b800200304eb1b588a620b9fb466f2869cef39dc Mon Sep 17 00:00:00 2001
From: 4pr0n <admin@rarchives.com>
Date: Thu, 22 May 2014 19:41:13 -0700
Subject: [PATCH] Get proper imgur file extension when ripping reddit albums

---
 .../com/rarchives/ripme/utils/RipUtils.java   | 22 +++++++++++++++----
 1 file changed, 18 insertions(+), 4 deletions(-)

diff --git a/src/main/java/com/rarchives/ripme/utils/RipUtils.java b/src/main/java/com/rarchives/ripme/utils/RipUtils.java
index b77f13f2..3053614b 100644
--- a/src/main/java/com/rarchives/ripme/utils/RipUtils.java
+++ b/src/main/java/com/rarchives/ripme/utils/RipUtils.java
@@ -9,7 +9,12 @@ import java.util.regex.Matcher;
 import java.util.regex.Pattern;
 
 import org.apache.log4j.Logger;
+import org.jsoup.Jsoup;
+import org.jsoup.nodes.Document;
+import org.jsoup.nodes.Element;
+import org.jsoup.select.Elements;
 
+import com.rarchives.ripme.ripper.AbstractRipper;
 import com.rarchives.ripme.ripper.rippers.ImgurRipper;
 import com.rarchives.ripme.ripper.rippers.ImgurRipper.ImgurAlbum;
 import com.rarchives.ripme.ripper.rippers.ImgurRipper.ImgurImage;
@@ -31,6 +36,7 @@ public class RipUtils {
                 return result;
             } catch (IOException e) {
                 logger.error("[!] Exception while loading album " + url, e);
+                return result;
             }
            
         }
@@ -51,10 +57,18 @@ public class RipUtils {
         if(url.getHost().equals("imgur.com") || 
                 url.getHost().equals("m.imgur.com")){
             try {
-                result.add(new URL(url.toExternalForm() + ".png"));
-                return result;
-            } catch (MalformedURLException ex) {
-                logger.error("[!] Exception while loading album " + url, ex);
+                // Fetch the page
+                Document doc = Jsoup.connect(url.toExternalForm())
+                                    .userAgent(AbstractRipper.USER_AGENT)
+                                    .get();
+                for (Element el : doc.select("meta")) {
+                    if (el.attr("property").equals("og:image")) {
+                        result.add(new URL(el.attr("content")));
+                        return result;
+                    }
+                }
+            } catch (IOException ex) {
+                logger.error("[!] Error", ex);
             }
             
         }