93 lines
3.1 KiB
Java
93 lines
3.1 KiB
Java
package com.rarchives.ripme.ripper.rippers;
|
|
|
|
import java.io.IOException;
|
|
import java.net.MalformedURLException;
|
|
import java.net.URL;
|
|
import java.util.regex.Matcher;
|
|
import java.util.regex.Pattern;
|
|
|
|
import org.jsoup.Jsoup;
|
|
import org.jsoup.nodes.Document;
|
|
import org.jsoup.nodes.Element;
|
|
|
|
import com.rarchives.ripme.ripper.AlbumRipper;
|
|
import com.rarchives.ripme.utils.Utils;
|
|
|
|
public class ImagearnRipper extends AlbumRipper {
|
|
|
|
private static final String DOMAIN = "imagearn.com",
|
|
HOST = "imagearn";
|
|
|
|
public ImagearnRipper(URL url) throws IOException {
|
|
super(url);
|
|
}
|
|
|
|
public boolean canRip(URL url) {
|
|
return url.getHost().endsWith(DOMAIN);
|
|
}
|
|
|
|
public URL sanitizeURL(URL url) throws MalformedURLException {
|
|
Pattern p = Pattern.compile("^.*imagearn.com/{1,}image.php\\?id=[0-9]{1,}.*$");
|
|
Matcher m = p.matcher(url.toExternalForm());
|
|
if (m.matches()) {
|
|
// URL points to imagearn *image*, not gallery
|
|
try {
|
|
url = getGalleryFromImage(url);
|
|
} catch (Exception e) {
|
|
logger.error("[!] " + e.getMessage(), e);
|
|
}
|
|
}
|
|
return url;
|
|
}
|
|
|
|
private URL getGalleryFromImage(URL url) throws IOException {
|
|
Document doc = Jsoup.connect(url.toExternalForm()).get();
|
|
for (Element link : doc.select("a[href~=^gallery\\.php.*$]")) {
|
|
logger.info("LINK: " + link.toString());
|
|
if (link.hasAttr("href")
|
|
&& link.attr("href").contains("gallery.php")) {
|
|
url = new URL("http://imagearn.com/" + link.attr("href"));
|
|
logger.info("[!] Found gallery from given link: " + url);
|
|
return url;
|
|
}
|
|
}
|
|
throw new IOException("Failed to find gallery at URL " + url);
|
|
}
|
|
|
|
@Override
|
|
public void rip() throws IOException {
|
|
int index = 0;
|
|
logger.info("[ ] Retrieving " + this.url.toExternalForm());
|
|
Document doc = Jsoup.connect(url.toExternalForm()).get();
|
|
for (Element thumb : doc.select("img.border")) {
|
|
String image = thumb.attr("src");
|
|
image = image.replaceAll("thumbs[0-9]*\\.imagearn\\.com/", "img.imagearn.com/imags/");
|
|
index += 1;
|
|
String prefix = "";
|
|
if (Utils.getConfigBoolean("download.save_order", true)) {
|
|
prefix = String.format("%03d_", index);
|
|
}
|
|
addURLToDownload(new URL(image), prefix);
|
|
}
|
|
waitForThreads();
|
|
}
|
|
|
|
@Override
|
|
public String getHost() {
|
|
return HOST;
|
|
}
|
|
|
|
@Override
|
|
public String getGID(URL url) throws MalformedURLException {
|
|
Pattern p = Pattern.compile("^.*imagearn.com/{1,}gallery.php\\?id=([0-9]{1,}).*$");
|
|
Matcher m = p.matcher(url.toExternalForm());
|
|
if (m.matches()) {
|
|
return m.group(1);
|
|
}
|
|
throw new MalformedURLException(
|
|
"Expected imagearn.com gallery formats: "
|
|
+ "imagearn.com/gallery.php?id=####..."
|
|
+ " Got: " + url);
|
|
}
|
|
}
|