Blacklist “EVOLVE” and “KILLERS.”
[rhynodge.git] / src / main / java / net / pterodactylus / rhynodge / filters / comics / GirlGeniusComicFilter.java
index 6111ef4..1bda719 100644 (file)
 
 package net.pterodactylus.rhynodge.filters.comics;
 
-import java.util.Arrays;
 import java.util.Collections;
 import java.util.List;
 
 import net.pterodactylus.rhynodge.filters.ComicSiteFilter;
 
+import com.google.common.base.Function;
 import com.google.common.base.Optional;
+import com.google.common.collect.FluentIterable;
 import org.jsoup.nodes.Document;
+import org.jsoup.nodes.Element;
 import org.jsoup.select.Elements;
 
 /**
@@ -41,8 +43,14 @@ public class GirlGeniusComicFilter extends ComicSiteFilter {
 
        @Override
        protected List<String> extractImageUrls(Document document) {
-               Elements imageElements = document.select("#MainTable img[alt='Comic']");
-               return imageElements.hasAttr("src") ? Arrays.asList(imageElements.attr("src")) : Collections.<String>emptyList();
+               Elements imageElements = document.select("#MainTable img[alt=Comic]");
+               return imageElements.hasAttr("src") ? FluentIterable.from(imageElements).transform(new Function<Element, String>() {
+
+                       @Override
+                       public String apply(Element imageElement) {
+                               return imageElement.attr("src");
+                       }
+               }).toList() : Collections.<String>emptyList();
        }
 
        @Override