Reddit support
This commit is contained in:
4pr0n
parent
aa78f60ce0
commit
c8cfc57e5b
@ -0,0 +1,217 @@
|
||||
package com.rarchives.ripme.ripper.rippers;
|
||||
|
||||
import java.io.IOException;
|
||||
import java.net.MalformedURLException;
|
||||
import java.net.URL;
|
||||
import java.util.List;
|
||||
import java.util.regex.Matcher;
|
||||
import java.util.regex.Pattern;
|
||||
|
||||
import org.apache.log4j.Logger;
|
||||
import org.json.JSONArray;
|
||||
import org.json.JSONObject;
|
||||
import org.json.JSONTokener;
|
||||
import org.jsoup.Jsoup;
|
||||
import org.jsoup.nodes.Document;
|
||||
|
||||
import com.rarchives.ripme.ripper.AbstractRipper;
|
||||
import com.rarchives.ripme.utils.RipUtils;
|
||||
|
||||
public class RedditRipper extends AbstractRipper {
|
||||
|
||||
public RedditRipper(URL url) throws IOException {
|
||||
super(url);
|
||||
}
|
||||
|
||||
private static final String HOST = "reddit";
|
||||
private static final String DOMAIN = "reddit.com";
|
||||
|
||||
private static final Logger logger = Logger.getLogger(GonewildRipper.class);
|
||||
private static final int SLEEP_TIME = 2000;
|
||||
|
||||
//private static final String USER_AGENT = "ripme by /u/4_pr0n github.com/4pr0n/ripme";
|
||||
|
||||
private long lastRequestTime = 0;
|
||||
|
||||
@Override
|
||||
public boolean canRip(URL url) {
|
||||
return url.getHost().endsWith(DOMAIN);
|
||||
}
|
||||
|
||||
@Override
|
||||
public URL sanitizeURL(URL url) throws MalformedURLException {
|
||||
String u = url.toExternalForm();
|
||||
// Strip '/u/' from URL
|
||||
u = u.replaceAll("reddit\\.com/u/", "reddit.com/user/");
|
||||
return new URL(u);
|
||||
}
|
||||
|
||||
private URL getJsonURL(URL url) throws MalformedURLException {
|
||||
// Append ".json" to URL in appropriate location.
|
||||
String result = url.getProtocol() + "://" + url.getHost() + url.getPath() + ".json";
|
||||
if (url.getQuery() != null) {
|
||||
result += "?" + url.getQuery();
|
||||
}
|
||||
return new URL(result);
|
||||
}
|
||||
|
||||
@Override
|
||||
public void rip() throws IOException {
|
||||
URL jsonURL = getJsonURL(this.url);
|
||||
while (true) {
|
||||
jsonURL = getAndParseAndReturnNext(jsonURL);
|
||||
if (jsonURL == null) {
|
||||
break;
|
||||
}
|
||||
}
|
||||
waitForThreads();
|
||||
}
|
||||
|
||||
private URL getAndParseAndReturnNext(URL url) throws IOException {
|
||||
JSONArray jsonArray = getJsonArrayFromURL(url), children;
|
||||
JSONObject json, data;
|
||||
URL nextURL = null;
|
||||
for (int i = 0; i < jsonArray.length(); i++) {
|
||||
json = jsonArray.getJSONObject(i);
|
||||
if (!json.has("data")) {
|
||||
continue;
|
||||
}
|
||||
data = json.getJSONObject("data");
|
||||
if (!data.has("children")) {
|
||||
continue;
|
||||
}
|
||||
children = data.getJSONArray("children");
|
||||
for (int j = 0; j < children.length(); j++) {
|
||||
parseJsonChild(children.getJSONObject(j));
|
||||
}
|
||||
if (data.has("after") && !data.isNull("after")) {
|
||||
String nextURLString = url.toExternalForm();
|
||||
if (nextURLString.contains("?")) {
|
||||
nextURLString = nextURLString.concat("&after=" + data.getString("after"));
|
||||
}
|
||||
else {
|
||||
nextURLString = nextURLString.concat("?after=" + data.getString("after"));
|
||||
}
|
||||
nextURL = new URL(nextURLString);
|
||||
}
|
||||
}
|
||||
return nextURL;
|
||||
}
|
||||
|
||||
private JSONArray getJsonArrayFromURL(URL url) throws IOException {
|
||||
// Wait 2 seconds before the next request
|
||||
long timeDiff = System.currentTimeMillis() - lastRequestTime;
|
||||
if (timeDiff < SLEEP_TIME) {
|
||||
try {
|
||||
Thread.sleep(timeDiff);
|
||||
} catch (InterruptedException e) {
|
||||
logger.warn("[!] Interrupted while waiting to load next page", e);
|
||||
return new JSONArray();
|
||||
}
|
||||
}
|
||||
lastRequestTime = System.currentTimeMillis();
|
||||
|
||||
logger.info(" Retrieving " + url);
|
||||
Document doc= Jsoup.connect(url.toExternalForm())
|
||||
.ignoreContentType(true)
|
||||
.userAgent(USER_AGENT)
|
||||
.get();
|
||||
String jsonString = doc.body().html().replaceAll(""", "\"");
|
||||
|
||||
Object jsonObj = new JSONTokener(jsonString).nextValue();
|
||||
JSONArray jsonArray = new JSONArray();
|
||||
if (jsonObj instanceof JSONObject) {
|
||||
jsonArray.put( (JSONObject) jsonObj);
|
||||
} else if (jsonObj instanceof JSONArray){
|
||||
jsonArray = (JSONArray) jsonObj;
|
||||
} else {
|
||||
logger.warn("[!] Unable to parse child: " + jsonString);
|
||||
}
|
||||
return jsonArray;
|
||||
}
|
||||
|
||||
private void parseJsonChild(JSONObject child) {
|
||||
String kind = child.getString("kind");
|
||||
JSONObject data = child.getJSONObject("data");
|
||||
if (kind.equals("t1")) {
|
||||
// Comment
|
||||
handleBody(data.getString("body"), data.getString("id"));
|
||||
}
|
||||
else if (kind.equals("t3")) {
|
||||
// post
|
||||
if (data.getBoolean("is_self")) {
|
||||
// TODO Parse self text
|
||||
handleBody(data.getString("selftext"), data.getString("id"));
|
||||
} else {
|
||||
// Get link
|
||||
handleURL(data.getString("url"), data.getString("id"));
|
||||
}
|
||||
if (data.has("replies") && data.get("replies") instanceof JSONObject) {
|
||||
JSONArray replies = data.getJSONObject("replies")
|
||||
.getJSONObject("data")
|
||||
.getJSONArray("children");
|
||||
for (int i = 0; i < replies.length(); i++) {
|
||||
parseJsonChild(replies.getJSONObject(i));
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
public void handleBody(String body, String id) {
|
||||
Pattern p = RipUtils.getURLRegex();
|
||||
Matcher m = p.matcher(body);
|
||||
while (m.find()) {
|
||||
handleURL(m.group(1), id);
|
||||
}
|
||||
}
|
||||
|
||||
public void handleURL(String theUrl, String id) {
|
||||
URL originalURL;
|
||||
try {
|
||||
originalURL = new URL(theUrl);
|
||||
} catch (MalformedURLException e) {
|
||||
return;
|
||||
}
|
||||
|
||||
List<URL> urls = RipUtils.getFilesFromURL(originalURL);
|
||||
if (urls.size() == 1) {
|
||||
addURLToDownload(urls.get(0), id + "-");
|
||||
} else if (urls.size() > 1) {
|
||||
for (int i = 0; i < urls.size(); i++) {
|
||||
addURLToDownload(urls.get(i), id + String.format("-%03d-", i + 1));
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
@Override
|
||||
public String getHost() {
|
||||
return HOST;
|
||||
}
|
||||
|
||||
@Override
|
||||
public String getGID(URL url) throws MalformedURLException {
|
||||
// User
|
||||
Pattern p = Pattern.compile("^https?://[a-zA-Z0-9\\.]{0,4}reddit\\.com/(user|u)/([a-zA-Z0-9_\\-]{3,}).*$");
|
||||
Matcher m = p.matcher(url.toExternalForm());
|
||||
if (m.matches()) {
|
||||
return "user_" + m.group(m.groupCount());
|
||||
}
|
||||
|
||||
// Post
|
||||
p = Pattern.compile("^https?://[a-zA-Z0-9\\.]{0,4}reddit\\.com/.*comments/([a-zA-Z0-9]{1,8}).*$");
|
||||
m = p.matcher(url.toExternalForm());
|
||||
if (m.matches()) {
|
||||
return "post_" + m.group(m.groupCount());
|
||||
}
|
||||
|
||||
// Subreddit
|
||||
p = Pattern.compile("^https?://[a-zA-Z0-9\\.]{0,4}reddit\\.com/r/([a-zA-Z0-9_]{1,}).*$");
|
||||
m = p.matcher(url.toExternalForm());
|
||||
if (m.matches()) {
|
||||
return "sub_" + m.group(m.groupCount());
|
||||
}
|
||||
|
||||
throw new MalformedURLException("Only accepts user pages, subreddits, or post, can't understand " + url);
|
||||
}
|
||||
|
||||
}
|
||||
50
src/main/java/com/rarchives/ripme/utils/RipUtils.java
Normal file
50
src/main/java/com/rarchives/ripme/utils/RipUtils.java
Normal file
@ -0,0 +1,50 @@
|
||||
package com.rarchives.ripme.utils;
|
||||
|
||||
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 org.apache.log4j.Logger;
|
||||
|
||||
import com.rarchives.ripme.ripper.rippers.ImgurRipper;
|
||||
|
||||
public class RipUtils {
|
||||
private static final Logger logger = Logger.getLogger(RipUtils.class);
|
||||
|
||||
public static List<URL> getFilesFromURL(URL url) {
|
||||
List<URL> result = new ArrayList<URL>();
|
||||
|
||||
// Imgur album
|
||||
if (url.getHost().equals("imgur.com") && url.toExternalForm().contains("imgur.com/a/")) {
|
||||
try {
|
||||
return ImgurRipper.getURLsFromAlbum(url);
|
||||
} catch (IOException e) {
|
||||
logger.error("[!] Exception while loading album " + url, e);
|
||||
}
|
||||
}
|
||||
|
||||
// Direct link to image
|
||||
Pattern p = Pattern.compile("(https?://[a-zA-Z0-9\\-\\.]+\\.[a-zA-Z]{2,3}(/\\S*)\\.(jpg|jpeg|gif|png|mp4))");
|
||||
Matcher m = p.matcher(url.toExternalForm());
|
||||
if (m.matches()) {
|
||||
try {
|
||||
URL singleURL = new URL(m.group(1));
|
||||
result.add(singleURL);
|
||||
return result;
|
||||
} catch (MalformedURLException e) {
|
||||
logger.error("[!] Not a valid URL: '" + url + "'", e);
|
||||
}
|
||||
}
|
||||
|
||||
logger.error("[!] Unable to rip URL: " + url);
|
||||
return result;
|
||||
}
|
||||
|
||||
public static Pattern getURLRegex() {
|
||||
return Pattern.compile("(https?://[a-zA-Z0-9\\-\\.]+\\.[a-zA-Z]{2,3}(/\\S*))");
|
||||
}
|
||||
}
|
||||
@ -0,0 +1,34 @@
|
||||
package com.rarchives.ripme.tst.ripper.rippers;
|
||||
|
||||
import java.io.IOException;
|
||||
import java.net.URL;
|
||||
import java.util.ArrayList;
|
||||
import java.util.List;
|
||||
|
||||
import com.rarchives.ripme.ripper.rippers.RedditRipper;
|
||||
|
||||
public class RedditRipperTest extends RippersTest {
|
||||
|
||||
public void testRedditAlbums() throws IOException {
|
||||
if (false && !DOWNLOAD_CONTENT) {
|
||||
return;
|
||||
}
|
||||
List<URL> contentURLs = new ArrayList<URL>();
|
||||
//contentURLs.add(new URL("http://www.reddit.com/r/nsfw_oc"));
|
||||
//contentURLs.add(new URL("http://www.reddit.com/r/nsfw_oc/top?t=all"));
|
||||
//contentURLs.add(new URL("http://www.reddit.com/u/gingerpuss"));
|
||||
contentURLs.add(new URL("http://www.reddit.com/r/UnrealGirls/comments/1ziuhl/in_class_veronique_popa/"));
|
||||
for (URL url : contentURLs) {
|
||||
try {
|
||||
RedditRipper ripper = new RedditRipper(url);
|
||||
ripper.rip();
|
||||
assert(ripper.getWorkingDir().listFiles().length > 1);
|
||||
deleteDir(ripper.getWorkingDir());
|
||||
} catch (Exception e) {
|
||||
e.printStackTrace();
|
||||
fail("Error while ripping URL " + url + ": " + e.getMessage());
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
}
|
||||
Loading…
Reference in New Issue
Block a user