Add test for “Girl Genius,” fix parsing

[rhynodge.git] / src / main / java / net / pterodactylus / rhynodge / filters / comics / GirlGeniusComicFilter.java

diff --git a/src/main/java/net/pterodactylus/rhynodge/filters/comics/GirlGeniusComicFilter.java b/src/main/java/net/pterodactylus/rhynodge/filters/comics/GirlGeniusComicFilter.java
index 6111ef4..d2ed5aa 100644 (file)
--- a/src/main/java/net/pterodactylus/rhynodge/filters/comics/GirlGeniusComicFilter.java
+++ b/src/main/java/net/pterodactylus/rhynodge/filters/comics/GirlGeniusComicFilter.java
@@ -17,9 +17,12 @@
 
 package net.pterodactylus.rhynodge.filters.comics;
 
-import java.util.Arrays;
+import static com.google.common.base.Optional.absent;
+import static com.google.common.base.Optional.of;
+
 import java.util.Collections;
 import java.util.List;
+import java.util.stream.Collectors;
 
 import net.pterodactylus.rhynodge.filters.ComicSiteFilter;
 
@@ -36,17 +39,18 @@ public class GirlGeniusComicFilter extends ComicSiteFilter {
 
        @Override
        protected Optional<String> extractTitle(Document document) {
-               return Optional.of("");
+               return extractImageUrls(document).isEmpty() ? absent() : of("");
        }
 
        @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("#comicbody img[alt=Comic]");
+               return imageElements.stream().filter(e -> e.hasAttr("src")).map(e -> e.attr("src")).collect(Collectors.toList());
        }
 
        @Override
        protected List<String> extractImageComments(Document document) {
                return Collections.emptyList();
        }
+
 }