diff --git a/src/main/java/com/rarchives/ripme/ripper/rippers/FuraffinityRipper.java b/src/main/java/com/rarchives/ripme/ripper/rippers/FuraffinityRipper.java index aa0e07a7..9a44e162 100644 --- a/src/main/java/com/rarchives/ripme/ripper/rippers/FuraffinityRipper.java +++ b/src/main/java/com/rarchives/ripme/ripper/rippers/FuraffinityRipper.java @@ -4,16 +4,12 @@ import java.io.IOException; import java.net.MalformedURLException; import java.net.URL; import java.util.ArrayList; -import java.util.Arrays; import java.util.HashMap; import java.util.List; import java.util.Map; import java.util.regex.Matcher; import java.util.regex.Pattern; -import javax.swing.JOptionPane; -import javax.swing.JPasswordField; - import org.jsoup.Connection.Method; import org.jsoup.Connection.Response; import org.jsoup.nodes.Document; @@ -22,146 +18,143 @@ import org.jsoup.select.Elements; import com.rarchives.ripme.ripper.AbstractHTMLRipper; import com.rarchives.ripme.ripper.DownloadThreadPool; +import com.rarchives.ripme.utils.Base64; import com.rarchives.ripme.utils.Http; public class FuraffinityRipper extends AbstractHTMLRipper { - static Map cookies=null; - static final String urlBase = "http://www.furaffinity.net"; + static Map cookies=null; + static final String urlBase = "https://www.furaffinity.net"; - // Thread pool for finding direct image links from "image" pages (html) - private DownloadThreadPool furaffinityThreadPool = new DownloadThreadPool( - "furaffinity"); + // Thread pool for finding direct image links from "image" pages (html) + private DownloadThreadPool furaffinityThreadPool + = new DownloadThreadPool( "furaffinity"); - @Override - public DownloadThreadPool getThreadPool() { - return furaffinityThreadPool; - } + @Override + public DownloadThreadPool getThreadPool() { + return furaffinityThreadPool; + } - public FuraffinityRipper(URL url) throws IOException { - super(url); - } + public FuraffinityRipper(URL url) throws IOException { + super(url); + } - @Override - public String getDomain() { - return "furaffinity.net"; - } + @Override + public String getDomain() { + return "furaffinity.net"; + } - @Override - public String getHost() { - return "furaffinity"; - } + @Override + public String getHost() { + return "furaffinity"; + } - @Override - public Document getFirstPage() throws IOException { - if (cookies == null || cookies.size() == 0) { - JPasswordField passwordField=new JPasswordField(); - String user = JOptionPane.showInputDialog("Username"); - JOptionPane.showMessageDialog(null,passwordField,"Password",JOptionPane.QUESTION_MESSAGE|JOptionPane.OK_OPTION); - String pass = Arrays.toString(passwordField.getPassword()); + @Override + public Document getFirstPage() throws IOException { + if (cookies == null || cookies.size() == 0) { + login(); + } - Response loginPage=Http.url(urlBase+"/login/") - .referrer(urlBase) - .response(); - cookies=loginPage.cookies(); - System.out.println("Cookies: "+cookies); - - Map formData=new HashMap(); - formData.put("action", "login"); - formData.put("retard_protection", "1"); - formData.put("name", user); - formData.put("pass", pass); - formData.put("login", "Login to FurAffinity"); - - Response doLogin=Http.url(urlBase+"/login/?ref=http://www.furaffinity.net/") - .referrer(urlBase+"/login/") - .cookies(cookies) - .data(formData) - .method(Method.POST) - .response(); - cookies.putAll(doLogin.cookies()); - System.out.println("Cookies: "+cookies); - } + return Http.url(url).cookies(cookies).get(); + } + + private void login() throws IOException { + String user = new String(Base64.decode("cmlwbWU=")); + String pass = new String(Base64.decode("cmlwbWVwYXNzd29yZA==")); - return Http.url(url).cookies(cookies).get(); - } + Response loginPage = Http.url(urlBase + "/login/") + .referrer(urlBase) + .response(); + cookies = loginPage.cookies(); - @Override - public Document getNextPage(Document doc) throws IOException { - // Find next page - Elements nextPageUrl = doc.select("td[align=right] form"); - String nextUrl = urlBase+nextPageUrl.first().attr("action"); - if (nextPageUrl.size() == 0) { - throw new IOException("No more pages"); - } - sleep(500); - Document nextPage = Http.url(nextUrl).cookies(cookies).get(); + Map formData = new HashMap(); + formData.put("action", "login"); + formData.put("retard_protection", "1"); + formData.put("name", user); + formData.put("pass", pass); + formData.put("login", "Login to FurAffinity"); - Elements hrefs = nextPage.select("div#no-images"); - if (hrefs.size() != 0) { - throw new IOException("No more pages"); - } - return nextPage; - } + Response doLogin = Http.url(urlBase + "/login/?ref=" + url) + .referrer(urlBase + "/login/") + .cookies(cookies) + .data(formData) + .method(Method.POST) + .response(); + cookies.putAll(doLogin.cookies()); + } - @Override - public List getURLsFromPage(Document page) { - List urls = new ArrayList(); - Elements urlElements = page.select("b[id^=sid_]"); - for (Element e : urlElements) { - urls.add(urlBase + e.select("a").first().attr("href")); - } - return urls; - } + @Override + public Document getNextPage(Document doc) throws IOException { + // Find next page + Elements nextPageUrl = doc.select("td[align=right] form"); + String nextUrl = urlBase+nextPageUrl.first().attr("action"); + if (nextPageUrl.size() == 0) { + throw new IOException("No more pages"); + } + sleep(500); + Document nextPage = Http.url(nextUrl).cookies(cookies).get(); - @Override - public void downloadURL(URL url, int index) { - furaffinityThreadPool.addThread(new FuraffinityDocumentThread(url, - index)); - sleep(250); - } + Elements hrefs = nextPage.select("div#no-images"); + if (hrefs.size() != 0) { + throw new IOException("No more pages"); + } + return nextPage; + } - @Override - public String getGID(URL url) throws MalformedURLException { - Pattern p = Pattern - .compile("^https?://www\\.furaffinity\\.net/gallery/([-_.0-9a-zA-Z]+).*$"); - Matcher m = p.matcher(url.toExternalForm()); - if (m.matches()) { - return m.group(1); - } - throw new MalformedURLException("Expected furaffinity.net URL format: " - + "www.furaffinity.net/gallery/username - got " + url - + " instead"); - } + @Override + public List getURLsFromPage(Document page) { + List urls = new ArrayList(); + Elements urlElements = page.select("b[id^=sid_]"); + for (Element e : urlElements) { + urls.add(urlBase + e.select("a").first().attr("href")); + } + return urls; + } - private class FuraffinityDocumentThread extends Thread { - private URL url; - private int index; + @Override + public void downloadURL(URL url, int index) { + furaffinityThreadPool.addThread(new FuraffinityDocumentThread(url)); + sleep(250); + } - public FuraffinityDocumentThread(URL url, int index) { - super(); - this.url = url; - this.index = index; - } + @Override + public String getGID(URL url) throws MalformedURLException { + Pattern p = Pattern + .compile("^https?://www\\.furaffinity\\.net/gallery/([-_.0-9a-zA-Z]+).*$"); + Matcher m = p.matcher(url.toExternalForm()); + if (m.matches()) { + return m.group(1); + } + throw new MalformedURLException("Expected furaffinity.net URL format: " + + "www.furaffinity.net/gallery/username - got " + url + + " instead"); + } - @Override - public void run() { - try { - Document doc = Http.url(url).cookies(cookies).get(); - // Find image - Elements donwloadLink = doc.select("div.alt1 b a[href^=//d.facdn.net/]"); - if (donwloadLink.size() == 0) { - logger.warn("Could not download " + this.url); - return; - } - String link = "http:" + donwloadLink.first().attr("href"); - logger.info("Found URL " + link); - addURLToDownload(new URL(link),"","",url.toExternalForm(),cookies); - } catch (IOException e) { - logger.error("[!] Exception while loading/parsing " + this.url, - e); - } - } - } + private class FuraffinityDocumentThread extends Thread { + private URL url; + + public FuraffinityDocumentThread(URL url) { + super(); + this.url = url; + } + + @Override + public void run() { + try { + Document doc = Http.url(url).cookies(cookies).get(); + // Find image + Elements donwloadLink = doc.select("div.alt1 b a[href^=//d.facdn.net/]"); + if (donwloadLink.size() == 0) { + logger.warn("Could not download " + this.url); + return; + } + String link = "http:" + donwloadLink.first().attr("href"); + logger.info("Found URL " + link); + addURLToDownload(new URL(link),"","",url.toExternalForm(),cookies); + } catch (IOException e) { + logger.error("[!] Exception while loading/parsing " + this.url, e); + } + } + } }