added ripper for e621.net
refactored paheal.net and xbooru.com rippers: - used absUrl() instead of plain attr() - album names (and file names for paheal) now passed through filesystemSafe()
This commit is contained in:
parent
bbc78ef67f
commit
5951d51c7e
142
src/main/java/com/rarchives/ripme/ripper/rippers/E621Ripper.java
Normal file
142
src/main/java/com/rarchives/ripme/ripper/rippers/E621Ripper.java
Normal file
@ -0,0 +1,142 @@
|
|||||||
|
|
||||||
|
package com.rarchives.ripme.ripper.rippers;
|
||||||
|
|
||||||
|
import com.rarchives.ripme.ripper.AbstractHTMLRipper;
|
||||||
|
import com.rarchives.ripme.ripper.DownloadThreadPool;
|
||||||
|
import com.rarchives.ripme.utils.Http;
|
||||||
|
import com.rarchives.ripme.utils.Utils;
|
||||||
|
import java.io.IOException;
|
||||||
|
import java.net.MalformedURLException;
|
||||||
|
import java.net.URI;
|
||||||
|
import java.net.URISyntaxException;
|
||||||
|
import java.net.URL;
|
||||||
|
import java.util.ArrayList;
|
||||||
|
import java.util.List;
|
||||||
|
import java.util.logging.Level;
|
||||||
|
import java.util.logging.Logger;
|
||||||
|
import java.util.regex.Matcher;
|
||||||
|
import java.util.regex.Pattern;
|
||||||
|
import org.jsoup.nodes.Document;
|
||||||
|
import org.jsoup.nodes.Element;
|
||||||
|
import org.jsoup.select.Elements;
|
||||||
|
|
||||||
|
/**
|
||||||
|
*
|
||||||
|
* @author
|
||||||
|
*/
|
||||||
|
public class E621Ripper extends AbstractHTMLRipper{
|
||||||
|
private static Pattern gidPattern=null;
|
||||||
|
private static Pattern gidPattern2=null;
|
||||||
|
private static Pattern gidPatternPool=null;
|
||||||
|
|
||||||
|
private DownloadThreadPool e621ThreadPool=new DownloadThreadPool("e621");
|
||||||
|
|
||||||
|
public E621Ripper(URL url) throws IOException {
|
||||||
|
super(url);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public DownloadThreadPool getThreadPool() {
|
||||||
|
return e621ThreadPool;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public String getDomain() {
|
||||||
|
return "e621.net";
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public String getHost() {
|
||||||
|
return "e621";
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public Document getFirstPage() throws IOException {
|
||||||
|
if(url.getPath().startsWith("/pool/show/"))
|
||||||
|
return Http.url("https://e621.net/pool/show/"+getTerm(url)).get();
|
||||||
|
else
|
||||||
|
return Http.url("https://e621.net/post/index/1/"+getTerm(url)).get();
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public List<String> getURLsFromPage(Document page) {
|
||||||
|
Elements elements=page.select("#post-list .thumb a,#pool-show .thumb a");
|
||||||
|
List<String> res=new ArrayList<String>(elements.size());
|
||||||
|
|
||||||
|
for(Element e:elements){
|
||||||
|
res.add(e.absUrl("href")+"#"+e.child(0).attr("id").substring(1));
|
||||||
|
}
|
||||||
|
|
||||||
|
return res;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public Document getNextPage(Document page) throws IOException {
|
||||||
|
for(Element e:page.select("#paginator a")){
|
||||||
|
if(e.attr("rel").equals("next"))
|
||||||
|
return Http.url(e.absUrl("href")).get();
|
||||||
|
}
|
||||||
|
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void downloadURL(final URL url, int index) {
|
||||||
|
e621ThreadPool.addThread(new Thread(new Runnable() {
|
||||||
|
public void run() {
|
||||||
|
try {
|
||||||
|
Document page=Http.url(url).get();
|
||||||
|
|
||||||
|
addURLToDownload(new URL(page.getElementById("image").absUrl("src")),Utils.getConfigBoolean("download.save_order",true)?url.getRef()+"-":"");
|
||||||
|
} catch (IOException ex) {
|
||||||
|
Logger.getLogger(E621Ripper.class.getName()).log(Level.SEVERE, null, ex);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}));
|
||||||
|
}
|
||||||
|
|
||||||
|
private String getTerm(URL url) throws MalformedURLException{
|
||||||
|
if(gidPattern==null)
|
||||||
|
gidPattern=Pattern.compile("^https?://(www\\.)?e621\\.net/post/index/[^/]+/([a-zA-Z0-9$_.+!*'(),%-]+)(/.*)?(#.*)?$");
|
||||||
|
if(gidPatternPool==null)
|
||||||
|
gidPatternPool=Pattern.compile("^https?://(www\\.)?e621\\.net/pool/show/([a-zA-Z0-9$_.+!*'(),%-]+)(\\?.*)?(/.*)?(#.*)?$");
|
||||||
|
|
||||||
|
Matcher m = gidPattern.matcher(url.toExternalForm());
|
||||||
|
if(m.matches())
|
||||||
|
return m.group(2);
|
||||||
|
|
||||||
|
m = gidPatternPool.matcher(url.toExternalForm());
|
||||||
|
if(m.matches())
|
||||||
|
return m.group(2);
|
||||||
|
|
||||||
|
throw new MalformedURLException("Expected e621.net URL format: e621.net/post/index/1/searchterm - got "+url+" instead");
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public String getGID(URL url) throws MalformedURLException {
|
||||||
|
try {
|
||||||
|
String prefix="";
|
||||||
|
if(url.getPath().startsWith("/pool/show/"))
|
||||||
|
prefix="pool_";
|
||||||
|
|
||||||
|
return Utils.filesystemSafe(prefix+new URI(getTerm(url)).getPath());
|
||||||
|
} catch (URISyntaxException ex) {
|
||||||
|
Logger.getLogger(PahealRipper.class.getName()).log(Level.SEVERE, null, ex);
|
||||||
|
}
|
||||||
|
|
||||||
|
throw new MalformedURLException("Expected e621.net URL format: e621.net/post/index/1/searchterm - got "+url+" instead");
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public URL sanitizeURL(URL url) throws MalformedURLException {
|
||||||
|
if(gidPattern2==null)
|
||||||
|
gidPattern2=Pattern.compile("^https?://(www\\.)?e621\\.net/post/search\\?tags=([a-zA-Z0-9$_.+!*'(),%-]+)(/.*)?(#.*)?$");
|
||||||
|
|
||||||
|
Matcher m = gidPattern2.matcher(url.toExternalForm());
|
||||||
|
if(m.matches())
|
||||||
|
return new URL("https://e621.net/post/index/1/"+m.group(2).replace("+","%20"));
|
||||||
|
|
||||||
|
return url;
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
@ -2,8 +2,12 @@ package com.rarchives.ripme.ripper.rippers;
|
|||||||
|
|
||||||
import com.rarchives.ripme.ripper.AbstractHTMLRipper;
|
import com.rarchives.ripme.ripper.AbstractHTMLRipper;
|
||||||
import com.rarchives.ripme.utils.Http;
|
import com.rarchives.ripme.utils.Http;
|
||||||
|
import com.rarchives.ripme.utils.Utils;
|
||||||
|
import java.io.File;
|
||||||
import java.io.IOException;
|
import java.io.IOException;
|
||||||
import java.net.MalformedURLException;
|
import java.net.MalformedURLException;
|
||||||
|
import java.net.URI;
|
||||||
|
import java.net.URISyntaxException;
|
||||||
import java.net.URL;
|
import java.net.URL;
|
||||||
import java.util.ArrayList;
|
import java.util.ArrayList;
|
||||||
import java.util.HashMap;
|
import java.util.HashMap;
|
||||||
@ -25,7 +29,7 @@ public class PahealRipper extends AbstractHTMLRipper{
|
|||||||
private static Map<String,String> cookies=null;
|
private static Map<String,String> cookies=null;
|
||||||
private static Pattern gidPattern=null;
|
private static Pattern gidPattern=null;
|
||||||
|
|
||||||
public static Map<String, String> getCookies() {
|
private static Map<String, String> getCookies() {
|
||||||
if(cookies==null){
|
if(cookies==null){
|
||||||
cookies=new HashMap<String, String>(1);
|
cookies=new HashMap<String, String>(1);
|
||||||
cookies.put("ui-tnc-agreed","true");
|
cookies.put("ui-tnc-agreed","true");
|
||||||
@ -49,7 +53,7 @@ public class PahealRipper extends AbstractHTMLRipper{
|
|||||||
|
|
||||||
@Override
|
@Override
|
||||||
public Document getFirstPage() throws IOException {
|
public Document getFirstPage() throws IOException {
|
||||||
return Http.url("http://rule34.paheal.net/post/list/"+getGID(url)+"/1").cookies(getCookies()).get();
|
return Http.url("http://rule34.paheal.net/post/list/"+getTerm(url)+"/1").cookies(getCookies()).get();
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
@ -68,31 +72,51 @@ public class PahealRipper extends AbstractHTMLRipper{
|
|||||||
List<String> res=new ArrayList<String>(elements.size());
|
List<String> res=new ArrayList<String>(elements.size());
|
||||||
|
|
||||||
for(Element e:elements)
|
for(Element e:elements)
|
||||||
res.add(e.attr("href"));
|
res.add(e.absUrl("href"));
|
||||||
|
|
||||||
return res;
|
return res;
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public void downloadURL(URL url, int index) {
|
public void downloadURL(URL url, int index) {
|
||||||
String file=url.getFile();
|
|
||||||
try {
|
try {
|
||||||
addURLToDownload(new URL(url.getProtocol(),url.getHost(),url.getPort(),file.substring(0, Math.min(128,file.lastIndexOf('.')))+file.substring(file.lastIndexOf('.'))));
|
String name=url.getPath();
|
||||||
} catch (MalformedURLException ex) {
|
String ext=".png";
|
||||||
|
|
||||||
|
name=name.substring(name.lastIndexOf('/')+1);
|
||||||
|
if(name.indexOf('.')>=0){
|
||||||
|
ext=name.substring(name.lastIndexOf('.'));
|
||||||
|
name=name.substring(0,name.length()-ext.length());
|
||||||
|
}
|
||||||
|
|
||||||
|
addURLToDownload(url,new File(workingDir.getCanonicalPath()+File.separator+Utils.filesystemSafe(new URI(name).getPath())+ext));
|
||||||
|
} catch (IOException ex) {
|
||||||
|
Logger.getLogger(PahealRipper.class.getName()).log(Level.SEVERE, null, ex);
|
||||||
|
} catch (URISyntaxException ex) {
|
||||||
Logger.getLogger(PahealRipper.class.getName()).log(Level.SEVERE, null, ex);
|
Logger.getLogger(PahealRipper.class.getName()).log(Level.SEVERE, null, ex);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
private String getTerm(URL url) throws MalformedURLException{
|
||||||
public String getGID(URL url) throws MalformedURLException {
|
|
||||||
if(gidPattern==null)
|
if(gidPattern==null)
|
||||||
gidPattern=Pattern.compile("^https?://(www\\.)?rule34\\.paheal\\.net/post/list/([a-zA-Z0-9$_.+!*'(),%-]+)(/.*)?(#.*)?$");
|
gidPattern=Pattern.compile("^https?://(www\\.)?rule34\\.paheal\\.net/post/list/([a-zA-Z0-9$_.+!*'(),%-]+)(/.*)?(#.*)?$");
|
||||||
|
|
||||||
Matcher m = gidPattern.matcher(url.toExternalForm());
|
Matcher m = gidPattern.matcher(url.toExternalForm());
|
||||||
if(m.matches())
|
if(m.matches())
|
||||||
return m.group(2);
|
return m.group(2);
|
||||||
|
|
||||||
throw new MalformedURLException("Expected paheal.net URL format: rule34.paheal.net/post/list/searchterm - got "+url+" instead");
|
throw new MalformedURLException("Expected paheal.net URL format: rule34.paheal.net/post/list/searchterm - got "+url+" instead");
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public String getGID(URL url) throws MalformedURLException {
|
||||||
|
try {
|
||||||
|
return Utils.filesystemSafe(new URI(getTerm(url)).getPath());
|
||||||
|
} catch (URISyntaxException ex) {
|
||||||
|
Logger.getLogger(PahealRipper.class.getName()).log(Level.SEVERE, null, ex);
|
||||||
|
}
|
||||||
|
|
||||||
|
throw new MalformedURLException("Expected paheal.net URL format: rule34.paheal.net/post/list/searchterm - got "+url+" instead");
|
||||||
|
}
|
||||||
|
|
||||||
}
|
}
|
||||||
|
@ -6,9 +6,13 @@ import com.rarchives.ripme.utils.Http;
|
|||||||
import com.rarchives.ripme.utils.Utils;
|
import com.rarchives.ripme.utils.Utils;
|
||||||
import java.io.IOException;
|
import java.io.IOException;
|
||||||
import java.net.MalformedURLException;
|
import java.net.MalformedURLException;
|
||||||
|
import java.net.URI;
|
||||||
|
import java.net.URISyntaxException;
|
||||||
import java.net.URL;
|
import java.net.URL;
|
||||||
import java.util.ArrayList;
|
import java.util.ArrayList;
|
||||||
import java.util.List;
|
import java.util.List;
|
||||||
|
import java.util.logging.Level;
|
||||||
|
import java.util.logging.Logger;
|
||||||
import java.util.regex.Matcher;
|
import java.util.regex.Matcher;
|
||||||
import java.util.regex.Pattern;
|
import java.util.regex.Pattern;
|
||||||
import org.jsoup.nodes.Document;
|
import org.jsoup.nodes.Document;
|
||||||
@ -19,7 +23,7 @@ import org.jsoup.nodes.Element;
|
|||||||
* @author
|
* @author
|
||||||
*/
|
*/
|
||||||
public class XbooruRipper extends AbstractHTMLRipper{
|
public class XbooruRipper extends AbstractHTMLRipper{
|
||||||
private Pattern gidPattern=null;
|
private static Pattern gidPattern=null;
|
||||||
|
|
||||||
public XbooruRipper(URL url) throws IOException {
|
public XbooruRipper(URL url) throws IOException {
|
||||||
super(url);
|
super(url);
|
||||||
@ -36,7 +40,7 @@ public class XbooruRipper extends AbstractHTMLRipper{
|
|||||||
}
|
}
|
||||||
|
|
||||||
private String getPage(int num) throws MalformedURLException{
|
private String getPage(int num) throws MalformedURLException{
|
||||||
return "http://xbooru.com/index.php?page=dapi&s=post&q=index&pid="+num+"&tags="+getGID(url);
|
return "http://xbooru.com/index.php?page=dapi&s=post&q=index&pid="+num+"&tags="+getTerm(url);
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
@ -59,7 +63,7 @@ public class XbooruRipper extends AbstractHTMLRipper{
|
|||||||
public List<String> getURLsFromPage(Document page) {
|
public List<String> getURLsFromPage(Document page) {
|
||||||
List<String> res=new ArrayList<String>(100);
|
List<String> res=new ArrayList<String>(100);
|
||||||
for(Element e:page.getElementsByTag("post"))
|
for(Element e:page.getElementsByTag("post"))
|
||||||
res.add(e.attr("file_url")+"#"+e.attr("id"));
|
res.add(e.absUrl("file_url")+"#"+e.attr("id"));
|
||||||
return res;
|
return res;
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -67,17 +71,27 @@ public class XbooruRipper extends AbstractHTMLRipper{
|
|||||||
public void downloadURL(URL url, int index) {
|
public void downloadURL(URL url, int index) {
|
||||||
addURLToDownload(url,Utils.getConfigBoolean("download.save_order",true)?url.getRef()+"-":"");
|
addURLToDownload(url,Utils.getConfigBoolean("download.save_order",true)?url.getRef()+"-":"");
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
private String getTerm(URL url) throws MalformedURLException{
|
||||||
public String getGID(URL url) throws MalformedURLException {
|
|
||||||
if(gidPattern==null)
|
if(gidPattern==null)
|
||||||
gidPattern=Pattern.compile("^https?://(www\\.)?xbooru\\.com/(index.php)?.*([?&]tags=([a-zA-Z0-9$_.+!*'(),%-]+))(\\&|(#.*)?$)");
|
gidPattern=Pattern.compile("^https?://(www\\.)?xbooru\\.com/(index.php)?.*([?&]tags=([a-zA-Z0-9$_.+!*'(),%-]+))(\\&|(#.*)?$)");
|
||||||
|
|
||||||
Matcher m = gidPattern.matcher(url.toExternalForm());
|
Matcher m = gidPattern.matcher(url.toExternalForm());
|
||||||
if(m.matches())
|
if(m.matches())
|
||||||
return m.group(4);
|
return m.group(4);
|
||||||
|
|
||||||
throw new MalformedURLException("Expected xbooru.com URL format: xbooru.com - got "+url+" instead");
|
throw new MalformedURLException("Expected xbooru.com URL format: xbooru.com/index.php?tags=searchterm - got "+url+" instead");
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public String getGID(URL url) throws MalformedURLException {
|
||||||
|
try {
|
||||||
|
return Utils.filesystemSafe(new URI(getTerm(url)).getPath());
|
||||||
|
} catch (URISyntaxException ex) {
|
||||||
|
Logger.getLogger(PahealRipper.class.getName()).log(Level.SEVERE, null, ex);
|
||||||
|
}
|
||||||
|
|
||||||
|
throw new MalformedURLException("Expected xbooru.com URL format: xbooru.com/index.php?tags=searchterm - got "+url+" instead");
|
||||||
}
|
}
|
||||||
|
|
||||||
}
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user