X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fnet%2Fpterodactylus%2Frhynodge%2Ffilters%2Fcomics%2FGeneralProtectionFaultComicFilter.java;h=0a8f16aa7d99deec9f665b22563f655a7fc9b9ce;hb=7eaf9f52b3ef3eead4695ec9ef03611b71618078;hp=660867a07a80e3271c73529e12c3e089996a03b2;hpb=829e0558e83db219a2f2a6be8630fe4af31d016c;p=rhynodge.git

diff --git a/src/main/java/net/pterodactylus/rhynodge/filters/comics/GeneralProtectionFaultComicFilter.java b/src/main/java/net/pterodactylus/rhynodge/filters/comics/GeneralProtectionFaultComicFilter.java
index 660867a..0a8f16a 100644
--- a/src/main/java/net/pterodactylus/rhynodge/filters/comics/GeneralProtectionFaultComicFilter.java
+++ b/src/main/java/net/pterodactylus/rhynodge/filters/comics/GeneralProtectionFaultComicFilter.java
@@ -17,14 +17,19 @@
 
 package net.pterodactylus.rhynodge.filters.comics;
 
-import java.util.Arrays;
+import static com.google.common.collect.FluentIterable.from;
+import static java.util.Arrays.asList;
+import static java.util.Collections.emptyList;
+
 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 org.jsoup.nodes.Document;
+import org.jsoup.nodes.Element;
 import org.jsoup.select.Elements;
 
 /**
@@ -43,12 +48,17 @@ public class GeneralProtectionFaultComicFilter extends ComicSiteFilter {
 	@Override
 	protected List<String> extractImageUrls(Document document) {
 		Elements imageElements = document.select(".content img[alt~=.Comic.for]");
-		return imageElements.hasAttr("src") ? Arrays.asList(imageElements.attr("src")) : Collections.<String>emptyList();
+		return from(imageElements).transformAndConcat(new Function<Element, Iterable<String>>() {
+			@Override
+			public Iterable<String> apply(Element element) {
+				return ((element != null) && element.hasAttr("src")) ? asList(element.attr("src")) : Collections.<String>emptyList();
+			}
+		}).toList();
 	}
 
 	@Override
 	protected List<String> extractImageComments(Document document) {
-		return Collections.emptyList();
+		return emptyList();
 	}
 
 }