1
0
mirror of https://github.com/RipMeApp/ripme.git synced 2025-08-26 07:14:38 +02:00

Refactoring of DeviantartRipper.

It is now a HTML Ripper.

NOT using Deviantart API like the old JSON ripper because it is SLOW and somehow annoying to use.

Things to consider: Using the API might be less work/maintenance later because APIs do not change as frequently as HTML source code...?
This commit is contained in:
MrPlaygon
2019-02-16 18:42:46 +01:00
parent 6c9e4200e2
commit 7b57d3cbfd
2 changed files with 484 additions and 352 deletions

View File

@@ -1,409 +1,540 @@
package com.rarchives.ripme.ripper.rippers; package com.rarchives.ripme.ripper.rippers;
import com.rarchives.ripme.ripper.AbstractJSONRipper; import com.rarchives.ripme.ripper.AbstractHTMLRipper;
import com.rarchives.ripme.utils.Base64; import com.rarchives.ripme.ripper.DownloadThreadPool;
import com.rarchives.ripme.ui.RipStatusMessage.STATUS;
import com.rarchives.ripme.utils.Http; import com.rarchives.ripme.utils.Http;
import com.rarchives.ripme.utils.RipUtils;
import com.rarchives.ripme.utils.Utils; import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException; import java.io.IOException;
import java.net.HttpURLConnection; import java.io.ObjectInputStream;
import java.io.ObjectOutputStream;
import java.net.MalformedURLException; import java.net.MalformedURLException;
import java.net.URL; import java.net.URL;
import java.util.ArrayList; import java.util.ArrayList;
import java.util.Arrays;
import java.util.HashMap; import java.util.HashMap;
import java.util.HashSet;
import java.util.List; import java.util.List;
import java.util.Map; import java.util.Map;
import java.util.Set;
import java.util.regex.Matcher; import java.util.regex.Matcher;
import java.util.regex.Pattern; import java.util.regex.Pattern;
import org.json.JSONArray; import org.jsoup.Connection.Method;
import org.json.JSONObject;
import org.jsoup.Connection.Response; import org.jsoup.Connection.Response;
import org.jsoup.Jsoup;
import org.jsoup.nodes.Document; import org.jsoup.nodes.Document;
import org.jsoup.nodes.Element; import org.jsoup.nodes.Element;
import org.jsoup.select.Elements; import org.jsoup.select.Elements;
/**
*
* @author MrPlaygon
*
* NOT using Deviantart API like the old JSON ripper because it is SLOW
* and somehow annoying to use.
*
*
* Things to consider: Using the API might be less work/maintenance later because APIs do not change as frequently as HTML source code...?
*
*
*
* Tested for:
*
* SFW:
*
* https://www.deviantart.com/apofiss/gallery/41388863/sceneries
* https://www.deviantart.com/kageuri/gallery/
* https://www.deviantart.com/kageuri/gallery/?catpath=/
* https://www.deviantart.com/apofiss/favourites/39881418/gifts-and
* https://www.deviantart.com/kageuri/favourites/
* https://www.deviantart.com/kageuri/favourites/?catpath=/
*
* NSFW:
*
* https://www.deviantart.com/revpeng/gallery/67734353/Siren-Lee-Agent-of-S-I-R-E-N-S
*
*
*
* Login Data (PLEASE DONT ACTUALLY USE!!!):
*
* email: 5g5_8l4dii5lbbpc@byom.de
*
* username: 5g58l4dii5lbbpc
*
* password: 5g5_8l4dii5lbbpc
*
*
*
*/
public class DeviantartRipper extends AbstractHTMLRipper {
public class DeviantartRipper extends AbstractJSONRipper { private final String username = "5g58l4dii5lbbpc";
String requestID; private final String password = "5g5_8l4dii5lbbpc";
String galleryID; private int offset = 0;
String username; private boolean usingCatPath = false;
String baseApiUrl = "https://www.deviantart.com/dapi/v1/gallery/"; private int downloadCount = 0;
String csrf; private Map<String, String> cookies;
Map<String, String> pageCookies = new HashMap<>(); private DownloadThreadPool deviantartThreadPool = new DownloadThreadPool("deviantart");
private ArrayList<String> names = new ArrayList<String>();
private static final int PAGE_SLEEP_TIME = 3000, @Override
IMAGE_SLEEP_TIME = 2000; public DownloadThreadPool getThreadPool() {
return deviantartThreadPool;
}
private Map<String,String> cookies = new HashMap<>(); public DeviantartRipper(URL url) throws IOException {
private Set<String> triedURLs = new HashSet<>(); super(url);
}
public DeviantartRipper(URL url) throws IOException { @Override
super(url); protected String getDomain() {
} return "deviantart.com";
}
String loginCookies = "auth=__0f9158aaec09f417b235%3B%221ff79836392a515d154216d919eae573%22;" + @Override
"auth_secure=__41d14dd0da101f411bb0%3B%2281cf2cf9477776162a1172543aae85ce%22;" + public String getHost() {
"userinfo=__bf84ac233bfa8ae642e8%3B%7B%22username%22%3A%22grabpy%22%2C%22uniqueid%22%3A%22a0a876aa37dbd4b30e1c80406ee9c280%22%2C%22vd%22%3A%22BbHUXZ%2CBbHUXZ%2CA%2CU%2CA%2C%2CB%2CA%2CB%2CBbHUXZ%2CBbHUdj%2CL%2CL%2CA%2CBbHUdj%2C13%2CA%2CB%2CA%2C%2CA%2CA%2CB%2CA%2CA%2C%2CA%22%2C%22attr%22%3A56%7D"; return "deviantart";
}
@Override @Override
public String getHost() { protected Document getFirstPage() throws IOException {
return "deviantart"; login();
} return Http.url(urlWithParams(this.offset)).cookies(getDACookie()).referrer("https://www.deviantart.com/")
.userAgent("Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:65.0) Gecko/20100101 Firefox/65.0").get();
}
@Override /**
public String getDomain() { * Stores logged in Cookies. Needed for art pieces only visible to logged in
return "deviantart.com"; * users.
} *
*
* @throws IOException when failed to load webpage or failed to read/write
* cookies in file (used when running multiple instances of
* RipMe)
*/
private void login() throws IOException {
@Override File f = new File("DACookie.toDelete");
public URL sanitizeURL(URL url) throws MalformedURLException { if (!f.exists()) {
String u = url.toExternalForm(); f.createNewFile();
if (u.contains("/gallery/")) { f.deleteOnExit();
return url;
} else if (u.contains("/favourites")) {
return url;
} else if (u.contains("/favorites")) {
return url;
}
if (!u.endsWith("/gallery/") && !u.endsWith("/gallery")) { // Load login page
if (!u.endsWith("/")) { Response res = Http.url("https://www.deviantart.com/users/login").connection().method(Method.GET)
u += "/gallery/"; .referrer("https://www.deviantart.com/")
} else { .userAgent("Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:65.0) Gecko/20100101 Firefox/65.0")
u += "gallery/"; .execute();
}
}
// Find tokens
Document doc = res.parse();
Element form = doc.getElementById("login");
String token = form.select("input[name=\"validate_token\"]").first().attr("value");
String key = form.select("input[name=\"validate_key\"]").first().attr("value");
System.out.println(
"------------------------------" + token + " & " + key + "------------------------------");
Pattern p = Pattern.compile("^https?://www\\.deviantart\\.com/([a-zA-Z0-9\\-]+)/favou?rites/([0-9]+)/*?$"); // Build Login Data
Matcher m = p.matcher(url.toExternalForm()); HashMap<String, String> loginData = new HashMap<String, String>();
if (!m.matches()) { loginData.put("challenge", "");
String subdir = "/"; loginData.put("username", username);
if (u.contains("catpath=scraps")) { loginData.put("password", password);
subdir = "scraps"; loginData.put("remember_me", "1");
} loginData.put("validate_token", token);
u = u.replaceAll("\\?.*", "?catpath=" + subdir); loginData.put("validate_key", key);
} Map<String, String> cookies = res.cookies();
return new URL(u);
}
@Override // Log in using data. Handle redirect
public String getGID(URL url) throws MalformedURLException { res = Http.url("https://www.deviantart.com/users/login").connection()
Pattern p = Pattern.compile("^https?://www\\.deviantart\\.com/([a-zA-Z0-9\\-]+)(/gallery)?/?(\\?.*)?$"); .referrer("https://www.deviantart.com/")
Matcher m = p.matcher(url.toExternalForm()); .userAgent("Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:65.0) Gecko/20100101 Firefox/65.0")
if (m.matches()) { .method(Method.POST).data(loginData).cookies(cookies).followRedirects(false).execute();
// Root gallery this.cookies = res.cookies();
if (url.toExternalForm().contains("catpath=scraps")) {
return m.group(1) + "_scraps";
}
else {
return m.group(1);
}
}
p = Pattern.compile("^https?://www\\.deviantart\\.com/([a-zA-Z0-9\\-]+)/gallery/([0-9]+).*$");
m = p.matcher(url.toExternalForm());
if (m.matches()) {
// Subgallery
return m.group(1) + "_" + m.group(2);
}
p = Pattern.compile("^https?://www\\.deviantart\\.com/([a-zA-Z0-9\\-]+)/favou?rites/([0-9]+)/.*?$");
m = p.matcher(url.toExternalForm());
if (m.matches()) {
return m.group(1) + "_faves_" + m.group(2);
}
p = Pattern.compile("^https?://www\\.deviantart\\.com/([a-zA-Z0-9\\-]+)/favou?rites/?$");
m = p.matcher(url.toExternalForm());
if (m.matches()) {
// Subgallery
return m.group(1) + "_faves";
}
throw new MalformedURLException("Expected URL format: http://www.deviantart.com/username[/gallery/#####], got: " + url);
}
private String getUsernameFromURL(String u) { res = Http.url(res.header("location")).connection().referrer("https://www.deviantart.com/")
Pattern p = Pattern.compile("^https?://www\\.deviantart\\.com/([a-zA-Z0-9\\-]+)/gallery/?(\\S+)?"); .userAgent("Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:65.0) Gecko/20100101 Firefox/65.0")
Matcher m = p.matcher(url.toExternalForm()); .method(Method.GET).cookies(cookies).followRedirects(false).execute();
if (m.matches()) {
return m.group(1);
}
return null;
} // Store cookies
updateCookie(res.cookies());
private String getFullsizedNSFWImage(String pageURL) { // Apply agegate
try { this.cookies.put("agegate_state", "1");
Document doc = Http.url(pageURL).cookies(cookies).get();
String imageToReturn = "";
String[] d = doc.select("img").attr("srcset").split(",");
String s = d[d.length -1].split(" ")[0]; // Write Cookie to file for other RipMe Instances
LOGGER.info("2:" + s); try {
FileOutputStream fileOut = new FileOutputStream(f);
ObjectOutputStream out = new ObjectOutputStream(fileOut);
out.writeObject(this.cookies);
out.close();
fileOut.close();
} catch (IOException i) {
i.printStackTrace();
}
if (s == null || s.equals("")) { } else {
LOGGER.error("Could not find full sized image at " + pageURL);
}
return s;
} catch (IOException e) {
LOGGER.error("Could not find full sized image at " + pageURL);
return null;
}
}
/** // When cookie file already exists (from another RipMe instance)
* Gets first page. while (this.cookies == null) {
* Will determine if login is supplied, try {
* if there is a login, then login and add that login cookies. Thread.sleep(2000);
* Otherwise, just bypass the age gate with an anonymous flag. FileInputStream fileIn = new FileInputStream(f);
* @return ObjectInputStream in = new ObjectInputStream(fileIn);
* @throws IOException this.cookies = (Map<String, String>) in.readObject();
*/ in.close();
@Override fileIn.close();
public JSONObject getFirstPage() throws IOException { } catch (IOException | ClassNotFoundException | InterruptedException i) {
i.printStackTrace();
}
}
}
// Base64 da login System.out.println("------------------------------" + this.cookies + "------------------------------");
// username: Z3JhYnB5 }
// password: ZmFrZXJz
/**
* Returns next page Document using offset.
*/
@Override
public Document getNextPage(Document doc) throws IOException {
this.offset += 24;
Response re = Http.url(urlWithParams(this.offset)).cookies(getDACookie())
.referrer("https://www.deviantart.com/")
.userAgent("Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:65.0) Gecko/20100101 Firefox/65.0").response();
updateCookie(re.cookies());
Document docu = re.parse();
Elements messages = docu.getElementsByClass("message");
System.out.println("------------------------------Current Offset: " + this.offset
+ " - More Pages?------------------------------");
cookies = getDACookies(); if (messages.size() > 0) {
if (cookies.isEmpty()) {
LOGGER.warn("Failed to get login cookies");
cookies.put("agegate_state","1"); // Bypasses the age gate
}
cookies.put("agegate_state", "1");
Response res = Http.url(this.url) // if message exists -> last page
.cookies(cookies) System.out.println("------------------------------Messages amount: " + messages.size()
.response(); + " - Next Page does not exists------------------------------");
Document page = res.parse(); throw new IOException("No more pages");
}
JSONObject firstPageJSON = getFirstPageJSON(page); return Http.url(urlWithParams(this.offset)).referrer("https://www.deviantart.com/")
requestID = firstPageJSON.getJSONObject("dapx").getString("requestid"); .userAgent("Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:65.0) Gecko/20100101 Firefox/65.0")
galleryID = getGalleryID(page); .cookies(getDACookie()).get();
username = getUsernameFromURL(url.toExternalForm());
csrf = firstPageJSON.getString("csrf");
pageCookies = res.cookies();
return requestPage(0, galleryID, username, requestID, csrf, pageCookies); }
}
private JSONObject requestPage(int offset, String galleryID, String username, String requestID, String csfr, Map<String, String> c) { /**
LOGGER.debug("offset: " + Integer.toString(offset)); * Returns list of Links to the Image pages. NOT links to fullsize image!!! e.g.
LOGGER.debug("galleryID: " + galleryID); * https://www.deviantart.com/kageuri/art/RUBY-568396655
LOGGER.debug("username: " + username); */
LOGGER.debug("requestID: " + requestID); @Override
String url = baseApiUrl + galleryID + "?iid=" + requestID; protected List<String> getURLsFromPage(Document page) {
try {
Document doc = Http.url(url).cookies(c).data("username", username).data("offset", Integer.toString(offset))
.data("limit", "24").data("_csrf", csfr).data("id", requestID)
.ignoreContentType().post();
return new JSONObject(doc.body().text());
} catch (IOException e) {
LOGGER.error("Got error trying to get page: " + e.getMessage());
e.printStackTrace();
return null;
}
List<String> result = new ArrayList<String>();
} Element div;
if (usingCatPath) {
div = page.getElementById("gmi-");
private JSONObject getFirstPageJSON(Document doc) { } else {
for (Element js : doc.select("script")) { div = page.getElementsByClass("folderview-art").first().child(0);
if (js.html().contains("requestid")) {
String json = js.html().replaceAll("window.__initial_body_data=", "").replaceAll("\\);", "")
.replaceAll(";__wake\\(.+", "");
JSONObject j = new JSONObject(json);
return j;
}
}
return null;
}
public String getGalleryID(Document doc) { }
// If the url contains catpath we return 0 as the DA api will provide all galery images if you sent the Elements links = div.select("a.torpedo-thumb-link");
// gallery id to 0
if (url.toExternalForm().contains("catpath=")) {
return "0";
}
Pattern p = Pattern.compile("^https?://www\\.deviantart\\.com/[a-zA-Z0-9\\-]+/gallery/([0-9]+)/?\\S+");
Matcher m = p.matcher(url.toExternalForm());
if (m.matches()) {
return m.group(1);
}
for (Element el : doc.select("input[name=set]")) {
try {
String galleryID = el.attr("value");
return galleryID;
} catch (NullPointerException e) {
continue;
}
}
LOGGER.error("Could not find gallery ID");
return null;
}
public String getUsername(Document doc) { for (Element el : links) {
return doc.select("meta[property=og:title]").attr("content") result.add(el.attr("href"));
.replaceAll("'s DeviantArt gallery", "").replaceAll("'s DeviantArt Gallery", "");
}
}
@Override System.out.println("------------------------------Amount of Images on Page: " + result.size()
public List<String> getURLsFromJSON(JSONObject json) { + "------------------------------");
List<String> imageURLs = new ArrayList<>(); System.out.println("------------------------------" + page.location() + "------------------------------");
JSONArray results = json.getJSONObject("content").getJSONArray("results");
for (int i = 0; i < results.length(); i++) {
Document doc = Jsoup.parseBodyFragment(results.getJSONObject(i).getString("html"));
if (doc.html().contains("ismature")) {
LOGGER.info("Downloading nsfw image");
String nsfwImage = getFullsizedNSFWImage(doc.select("span").attr("href"));
if (nsfwImage != null && nsfwImage.startsWith("http")) {
imageURLs.add(nsfwImage);
}
}
try {
String imageURL = doc.select("span").first().attr("data-super-full-img");
if (!imageURL.isEmpty() && imageURL.startsWith("http")) {
imageURLs.add(imageURL);
}
} catch (NullPointerException e) {
LOGGER.info(i + " does not contain any images");
}
} return result;
return imageURLs; }
}
/**
* Starts new Thread to find download link + filename + filetype
*/
@Override
protected void downloadURL(URL url, int index) {
this.downloadCount += 1;
System.out.println("------------------------------Download URL Number " + this.downloadCount
+ "------------------------------");
System.out.println(
"------------------------------DAURL: " + url.toExternalForm() + "------------------------------");
try {
Response re = Http.url(urlWithParams(this.offset)).cookies(getDACookie())
.referrer("https://www.deviantart.com/")
.userAgent("Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:65.0) Gecko/20100101 Firefox/65.0")
.response();
updateCookie(re.cookies());
} catch (IOException e) {
e.printStackTrace();
}
@Override // Start Thread and add to pool.
public JSONObject getNextPage(JSONObject page) throws IOException { DeviantartImageThread t = new DeviantartImageThread(url);
boolean hasMore = page.getJSONObject("content").getBoolean("has_more"); deviantartThreadPool.addThread(t);
if (hasMore) {
return requestPage(page.getJSONObject("content").getInt("next_offset"), galleryID, username, requestID, csrf, pageCookies);
}
throw new IOException("No more pages"); }
}
@Override @Override
public boolean keepSortOrder() { public String normalizeUrl(String url) {
// Don't keep sort order (do not add prefixes). return (urlWithParams(this.offset).toExternalForm());
// Causes file duplication, as outlined in https://github.com/4pr0n/ripme/issues/113 }
return false;
}
@Override /**
public void downloadURL(URL url, int index) { * Returns name of album. Album name consists of 3 words: - Artist (owner of
addURLToDownload(url, getPrefix(index), "", this.url.toExternalForm(), cookies); * gallery) - Type (gallery or favorites folder) - Name of the folder
sleep(IMAGE_SLEEP_TIME); *
} * Returns artist_type_name
*/
@Override
public String getGID(URL url) throws MalformedURLException {
/** String s = url.toExternalForm();
* Tries to get full size image from thumbnail URL String artist = "unknown";
* @param thumb Thumbnail URL String what = "unknown";
* @param throwException Whether or not to throw exception when full size image isn't found String albumname = "unknown";
* @return Full-size image URL
* @throws Exception If it can't find the full-size URL
*/
private static String thumbToFull(String thumb, boolean throwException) throws Exception {
thumb = thumb.replace("http://th", "http://fc");
List<String> fields = new ArrayList<>(Arrays.asList(thumb.split("/")));
fields.remove(4);
if (!fields.get(4).equals("f") && throwException) {
// Not a full-size image
throw new Exception("Can't get full size image from " + thumb);
}
StringBuilder result = new StringBuilder();
for (int i = 0; i < fields.size(); i++) {
if (i > 0) {
result.append("/");
}
result.append(fields.get(i));
}
return result.toString();
}
if (url.toExternalForm().contains("catpath=/")) {
this.usingCatPath = true;
}
Pattern p = Pattern.compile("^https?://www.deviantart\\.com/([a-zA-Z0-9]+).*$");
Matcher m = p.matcher(s);
/** // Artist
* If largest resolution for image at 'thumb' is found, starts downloading if (m.matches()) {
* and returns null. artist = m.group(1);
* If it finds a larger resolution on another page, returns the image URL. } else {
* @param thumb Thumbnail URL throw new MalformedURLException("Expected deviantart.com URL format: "
* @param page Page the thumbnail is retrieved from + "www.deviantart.com/<ARTIST>/gallery/<NUMBERS>/<NAME>\nOR\nwww.deviantart.com/<ARTIST>/favourites/<NUMBERS>/<NAME> - got "
* @return Highest-resolution version of the image based on thumbnail URL and the page. + url + " instead");
*/ }
private String smallToFull(String thumb, String page) {
try {
// Fetch the image page
Response resp = Http.url(page)
.referrer(this.url)
.cookies(cookies)
.response();
cookies.putAll(resp.cookies());
Document doc = resp.parse();
Elements els = doc.select("img.dev-content-full");
String fsimage = null;
// Get the largest resolution image on the page
if (!els.isEmpty()) {
// Large image
fsimage = els.get(0).attr("src");
LOGGER.info("Found large-scale: " + fsimage);
if (fsimage.contains("//orig")) {
return fsimage;
}
}
// Try to find the download button
els = doc.select("a.dev-page-download");
if (!els.isEmpty()) {
// Full-size image
String downloadLink = els.get(0).attr("href");
LOGGER.info("Found download button link: " + downloadLink);
HttpURLConnection con = (HttpURLConnection) new URL(downloadLink).openConnection();
con.setRequestProperty("Referer",this.url.toString());
String cookieString = "";
for (Map.Entry<String, String> entry : cookies.entrySet()) {
cookieString = cookieString + entry.getKey() + "=" + entry.getValue() + "; ";
}
cookieString = cookieString.substring(0,cookieString.length() - 1);
con.setRequestProperty("Cookie",cookieString);
con.setRequestProperty("User-Agent", USER_AGENT);
con.setInstanceFollowRedirects(true);
con.connect();
int code = con.getResponseCode();
String location = con.getURL().toString();
con.disconnect();
if (location.contains("//orig")) {
fsimage = location;
LOGGER.info("Found image download: " + location);
}
}
if (fsimage != null) {
return fsimage;
}
throw new IOException("No download page found");
} catch (IOException ioe) {
try {
LOGGER.info("Failed to get full size download image at " + page + " : '" + ioe.getMessage() + "'");
String lessThanFull = thumbToFull(thumb, false);
LOGGER.info("Falling back to less-than-full-size image " + lessThanFull);
return lessThanFull;
} catch (Exception e) {
return null;
}
}
}
/** // What is it
* Returns DA cookies. if (s.contains("/gallery/")) {
* @return Map of cookies containing session data. what = "gallery";
*/ } else if (s.contains("/favourites/")) {
private Map<String, String> getDACookies() { what = "favourites";
return RipUtils.getCookiesFromString(Utils.getConfigString("deviantart.cookies", loginCookies)); } else {
} throw new MalformedURLException("Expected deviantart.com URL format: "
+ "www.deviantart.com/<ARTIST>/gallery/<NUMBERS>/<NAME>\nOR\nwww.deviantart.com/<ARTIST>/favourites/<NUMBERS>/<NAME> - got "
+ url + " instead");
}
// Album Name
Pattern artistP = Pattern
.compile("^https?://www.deviantart\\.com/[a-zA-Z0-9]+/[a-zA-Z]+/[0-9]+/([a-zA-Z0-9-]+).*$");
Matcher artistM = artistP.matcher(s);
if (s.endsWith("?catpath=/")) {
albumname = "all";
} else if (s.endsWith("/favourites/") || s.endsWith("/gallery/")) {
albumname = "featured";
} else if (artistM.matches()) {
albumname = artistM.group(1);
}
System.out.println("------------------------------Album Name: " + artist + "_" + what + "_" + albumname
+ "------------------------------");
return artist + "_" + what + "_" + albumname;
}
/**
*
* @return Clean URL as String
*/
private String cleanURL() {
return (this.url.toExternalForm().split("\\?"))[0];
}
/**
* Return correct url with params (catpath) and current offset
*
* @return URL to page with offset
*/
private URL urlWithParams(int offset) {
try {
String url = cleanURL();
if (this.usingCatPath) {
return (new URL(url + "?catpath=/&offset=" + offset));
} else {
return (new URL(url + "?offset=" + offset));
}
} catch (MalformedURLException e) {
e.printStackTrace();
}
return null;
}
/**
* Returns Hashmap usable as Cookie for NSFW Artworks Not really needed but
* maybe useful later.
*
* @return Cookie Hashmap
*/
private Map<String, String> getDACookie() {
return this.cookies;
}
private void updateCookie(Map<String, String> m) {
System.out.println("------------------------------Updating Cookies------------------------------");
System.out.println(
"------------------------------Old Cookies: " + this.cookies + " ------------------------------");
System.out.println("------------------------------New Cookies: " + m + " ------------------------------");
this.cookies.putAll(m);
this.cookies.put("agegate_state", "1");
System.out.println(
"------------------------------Merged Cookies: " + this.cookies + " ------------------------------");
}
/**
* Analyzes an image page like
* https://www.deviantart.com/kageuri/art/RUBY-568396655 .
*
* Looks for download button, follows the authentications and redirects and adds
* the Image URL to the download queue. If no download button is present it will
* use the largest version of the image.
*
* Should work with all filetypes on Deviantart. Tested with .JPG .PNG and .PDF
*
* @author MrPlaygon
*
*/
private class DeviantartImageThread extends Thread {
private URL url;
public DeviantartImageThread(URL url) {
this.url = url;
}
@Override
public void run() {
getFullSizeURL();
}
/**
* Get URL to Artwork and return fullsize URL with file ending.
*
* @param page Like
* https://www.deviantart.com/apofiss/art/warmest-of-the-days-455668450
* @return URL like
* https://images-wixmp-ed30a86b8c4ca887773594c2.wixmp.com/intermediary/f/07f7a6bb-2d35-4630-93fc-be249af22b3e/d7jak0y-d20e5932-df72-4d13-b002-5e122037b373.jpg
*
*
*/
private void getFullSizeURL() {
System.out.println("------------------------------------------------------------");
System.out.println("------------------------------Searching max. Resolution for " + url
+ "------------------------------");
sendUpdate(STATUS.LOADING_RESOURCE, "Searching max. resolution for " + url);
try {
Response re = Http.url(url).connection().referrer("https://www.deviantart.com/")
.userAgent("Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:65.0) Gecko/20100101 Firefox/65.0")
.cookies(getDACookie()).execute();
Document doc = re.parse();
// Artwork Title
String title = doc.select("a.title").first().html();
title = title.replaceAll("[^a-zA-Z0-9\\.\\-]", "_").toLowerCase();
int counter = 1;
if (names.contains(title)) {
while (names.contains(title + "_" + counter)) {
counter++;
}
title = title + "_" + counter;
}
names.add(title);
// Check for download button
Element downloadButton = null;
downloadButton = doc.select("a.dev-page-download").first();
// Download Button
if (downloadButton != null) {
System.out.println("------------------------------Download Button found: "
+ downloadButton.attr("href") + "------------------------------");
Response download = Http.url(downloadButton.attr("href")).connection().cookies(getDACookie())
.method(Method.GET).referrer("https://www.deviantart.com/")
.userAgent("Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:65.0) Gecko/20100101 Firefox/65.0")
.ignoreContentType(true).followRedirects(true).execute();
URL location = download.url();
String[] filetypePart = download.header("Content-Disposition").split("\\.");
System.out.println("------------------------------Found Image URL------------------------------");
System.out.println("------------------------------" + url + "------------------------------");
System.out.println("------------------------------" + location + "------------------------------");
addURLToDownload(location, "", "", "", new HashMap<String, String>(),
title + "." + filetypePart[filetypePart.length - 1]);
return;
}
// No Download Button
Element div = doc.select("div.dev-view-deviation").first();
Element image = div.getElementsByTag("img").first();
String source = "";
if (image == null) {
System.out.println(
"------------------------------!!!ERROR on " + url + " !!!------------------------------");
System.out.println("------------------------------!!!Cookies: " + getDACookie()
+ " ------------------------------");
System.out.println(div);
sendUpdate(STATUS.DOWNLOAD_ERRORED, "!!!ERROR!!!\n" + url);
return;
}
// When it is text art (e.g. story) the only image is the avator (profile
// picture)
if (image.hasClass("avatar")) {
System.out.println(
"------------------------------No Image found, probably text art------------------------------");
System.out.println(url);
return;
}
source = image.attr("src");
String[] parts = source.split("/v1/");
// Image page uses scaled down version. Split at /v1/ to receive max size.
if (parts.length > 2) {
System.out.println(
"------------------------------Unexpected URL Format------------------------------");
sendUpdate(STATUS.DOWNLOAD_WARN, "Unexpected URL Format - Risky Try");
return;
}
String[] tmpParts = parts[0].split("\\.");
System.out.println("------------------------------Found Image URL------------------------------");
System.out.println("------------------------------" + url + "------------------------------");
System.out.println("------------------------------" + parts[0] + "------------------------------");
addURLToDownload(new URL(parts[0]), "", "", "", new HashMap<String, String>(),
title + "." + tmpParts[tmpParts.length - 1]);
} catch (IOException e) {
e.printStackTrace();
}
System.out.println(
"------------------------------No Full Size URL for: " + url + "------------------------------");
sendUpdate(STATUS.DOWNLOAD_ERRORED, "No image found for " + url);
return;
}
}
} }

View File

@@ -9,6 +9,7 @@ import com.rarchives.ripme.ripper.rippers.DeviantartRipper;
import com.rarchives.ripme.utils.Http; import com.rarchives.ripme.utils.Http;
import org.jsoup.nodes.Document; import org.jsoup.nodes.Document;
//TODO build some tests
public class DeviantartRipperTest extends RippersTest { public class DeviantartRipperTest extends RippersTest {
public void testDeviantartAlbum() throws IOException { public void testDeviantartAlbum() throws IOException {
DeviantartRipper ripper = new DeviantartRipper(new URL("https://www.deviantart.com/airgee/gallery/")); DeviantartRipper ripper = new DeviantartRipper(new URL("https://www.deviantart.com/airgee/gallery/"));
@@ -31,8 +32,8 @@ public class DeviantartRipperTest extends RippersTest {
URL url = new URL("https://www.deviantart.com/airgee/gallery/"); URL url = new URL("https://www.deviantart.com/airgee/gallery/");
DeviantartRipper ripper = new DeviantartRipper(url); DeviantartRipper ripper = new DeviantartRipper(url);
Document doc = Http.url(url).get(); Document doc = Http.url(url).get();
assertEquals("airgee", ripper.getUsername(doc)); //assertEquals("airgee", ripper.getUsername(doc));
assertEquals("714589", ripper.getGalleryID(doc)); //assertEquals("714589", ripper.getGalleryID(doc));
} }
public void testSanitizeURL() throws IOException { public void testSanitizeURL() throws IOException {