From 909af2de2cacd66842c3084f684c296f62ab0a2e Mon Sep 17 00:00:00 2001
From: cyian-1756 <devnull64@vfemail.net>
Date: Wed, 26 Sep 2018 21:40:42 -0400
Subject: [PATCH] Port MulemaxRipper to AbstractSingleFileRipper

---
 .../rippers/{video => }/MulemaxRipper.java    | 39 ++++++++++++-------
 .../tst/ripper/rippers/MulemaxRipperTest.java | 10 ++---
 2 files changed, 30 insertions(+), 19 deletions(-)
 rename src/main/java/com/rarchives/ripme/ripper/rippers/{video => }/MulemaxRipper.java (62%)

diff --git a/src/main/java/com/rarchives/ripme/ripper/rippers/video/MulemaxRipper.java b/src/main/java/com/rarchives/ripme/ripper/rippers/MulemaxRipper.java
similarity index 62%
rename from src/main/java/com/rarchives/ripme/ripper/rippers/video/MulemaxRipper.java
rename to src/main/java/com/rarchives/ripme/ripper/rippers/MulemaxRipper.java
index 5087f15f..6f9d178d 100644
--- a/src/main/java/com/rarchives/ripme/ripper/rippers/video/MulemaxRipper.java
+++ b/src/main/java/com/rarchives/ripme/ripper/rippers/MulemaxRipper.java
@@ -1,18 +1,22 @@
-package com.rarchives.ripme.ripper.rippers.video;
+package com.rarchives.ripme.ripper.rippers;
 
 import java.io.IOException;
 import java.net.MalformedURLException;
 import java.net.URL;
+import java.util.ArrayList;
+import java.util.List;
 import java.util.regex.Matcher;
 import java.util.regex.Pattern;
 
+import com.rarchives.ripme.ripper.AbstractSingleFileRipper;
 import org.jsoup.nodes.Document;
+import org.jsoup.nodes.Element;
 import org.jsoup.select.Elements;
 
 import com.rarchives.ripme.ripper.VideoRipper;
 import com.rarchives.ripme.utils.Http;
 
-public class MulemaxRipper extends VideoRipper {
+public class MulemaxRipper extends AbstractSingleFileRipper {
 
     private static final String HOST = "mulemax";
 
@@ -22,7 +26,17 @@ public class MulemaxRipper extends VideoRipper {
 
     @Override
     public String getHost() {
-        return HOST;
+        return "mulemax";
+    }
+
+    @Override
+    public String getDomain() {
+        return "mulemax.com";
+    }
+
+    @Override
+    public Document getFirstPage() throws IOException {
+        return Http.url(url).get();
     }
 
     @Override
@@ -52,15 +66,14 @@ public class MulemaxRipper extends VideoRipper {
     }
 
     @Override
-    public void rip() throws IOException {
-        LOGGER.info("Retrieving " + this.url);
-        Document doc = Http.url(url).get();
-        Elements videos = doc.select(".video-js > source");
-        if (videos.isEmpty()) {
-            throw new IOException("Could not find Embed code at " + url);
-        }
-        String vidUrl = videos.attr("src");
-        addURLToDownload(new URL(vidUrl), HOST + "_" + getGID(this.url));
-        waitForThreads();
+    public List<String> getURLsFromPage(Document doc) {
+        List<String> result = new ArrayList<>();
+        result.add(doc.select(".video-js > source").attr("src"));
+        return result;
+    }
+
+    @Override
+    public void downloadURL(URL url, int index) {
+        addURLToDownload(url, getPrefix(index));
     }
 }
\ No newline at end of file
diff --git a/src/test/java/com/rarchives/ripme/tst/ripper/rippers/MulemaxRipperTest.java b/src/test/java/com/rarchives/ripme/tst/ripper/rippers/MulemaxRipperTest.java
index 78337cf3..0d17ab42 100644
--- a/src/test/java/com/rarchives/ripme/tst/ripper/rippers/MulemaxRipperTest.java
+++ b/src/test/java/com/rarchives/ripme/tst/ripper/rippers/MulemaxRipperTest.java
@@ -3,16 +3,14 @@ package com.rarchives.ripme.tst.ripper.rippers;
 import java.io.IOException;
 import java.net.URL;
 
-import com.rarchives.ripme.ripper.rippers.video.MulemaxRipper;
+import com.rarchives.ripme.ripper.rippers.MulemaxRipper;
 import com.rarchives.ripme.utils.Utils;
 
 public class MulemaxRipperTest extends RippersTest {
     
     public void testMulemaxVideo() throws IOException {
-        // This test fails on the CI - possibly due to checking for a file before it's written - so we're skipping it
-        if (Utils.getConfigBoolean("test.run_flaky_tests", false)) {
-            MulemaxRipper ripper = new MulemaxRipper(new URL("https://mulemax.com/video/1720/emma-and-her-older-sissy-are-home-for-a-holiday-break"));  //pick any video from the front page
-            testRipper(ripper);
-        }
+        MulemaxRipper ripper = new MulemaxRipper(new URL("https://mulemax.com/video/1720/emma-and-her-older-sissy-are-home-for-a-holiday-break"));  //pick any video from the front page
+        testRipper(ripper);
     }
+
 }
\ No newline at end of file