X-Git-Url: http://git.nikiroo.be/?a=blobdiff_plain;ds=sidebyside;f=src%2Fbe%2Fnikiroo%2Ffanfix%2Fsupported%2FE621.java;h=5ebd6b317661cf0ff2b1ff0cf357559df469b380;hb=92fb0719f84f5b6734b51e528332546d78e9ccec;hp=45c110f283ebaf825d5f7c65361917333f69add9;hpb=7d0d2be6b0e9bd4e03ac8a7d749f49d6d1436242;p=fanfix.git

diff --git a/src/be/nikiroo/fanfix/supported/E621.java b/src/be/nikiroo/fanfix/supported/E621.java
index 45c110f..5ebd6b3 100644
--- a/src/be/nikiroo/fanfix/supported/E621.java
+++ b/src/be/nikiroo/fanfix/supported/E621.java
@@ -1,5 +1,6 @@
 package be.nikiroo.fanfix.supported;
 
+import java.awt.image.BufferedImage;
 import java.io.IOException;
 import java.io.InputStream;
 import java.net.URL;
@@ -13,6 +14,7 @@ import be.nikiroo.fanfix.data.Chapter;
 import be.nikiroo.fanfix.data.MetaData;
 import be.nikiroo.fanfix.data.Story;
 import be.nikiroo.utils.StringUtils;
+import be.nikiroo.utils.ui.Progress;
 
 /**
  * Support class for <a href="http://e621.net/">e621.net</a> and <a
@@ -47,15 +49,15 @@ class E621 extends BasicSupport {
 		meta.setSubject("");
 		meta.setType(getType().toString());
 		meta.setImageDocument(true);
-		meta.setCover(null);
+		meta.setCover(getCover(source));
 
 		return meta;
 	}
 
 	@Override
-	public Story process(URL url) throws IOException {
+	public Story process(URL url, Progress pg) throws IOException {
 		// There is no chapters on e621, just pagination...
-		Story story = super.process(url);
+		Story story = super.process(url, pg);
 
 		Chapter only = new Chapter(1, null);
 		for (Chapter chap : story) {
@@ -84,6 +86,21 @@ class E621 extends BasicSupport {
 		return true;
 	}
 
+	private BufferedImage getCover(URL source) throws IOException {
+		InputStream in = Instance.getCache().open(source, this, true);
+		String images = getChapterContent(new URL(source.toString() + "?page="
+				+ 1), in, 1);
+		if (!images.isEmpty()) {
+			int pos = images.indexOf('\n');
+			if (pos >= 0) {
+				images = images.substring(1, pos - 1);
+				return getImage(this, null, images);
+			}
+		}
+
+		return null;
+	}
+
 	private String getAuthor(URL source, InputStream in) throws IOException {
 		String author = getLine(in, "href=\"/post/show/", 0);
 		if (author != null) {
@@ -237,7 +254,7 @@ class E621 extends BasicSupport {
 		scan.useDelimiter("\\n");
 		while (scan.hasNext()) {
 			String line = scan.next();
-			if (line.contains("class=\"preview\"")) {
+			if (line.contains("class=\"preview")) {
 				for (int pos = line.indexOf(key); pos >= 0; pos = line.indexOf(
 						key, pos + key.length())) {
 					int endPos = line.indexOf("\"", pos);