X-Git-Url: http://git.nikiroo.be/?p=fanfix.git;a=blobdiff_plain;f=src%2Fbe%2Fnikiroo%2Ffanfix%2Fsupported%2FE621.java;h=665bdd5100cb0e7ef2b10ebc972a0c0729dca607;hp=bc6ba5b1501e83a1c1afbbdf38e6804d7147fdf7;hb=d98a29006897a1ae31ff3a039afe9643a48e0704;hpb=68686a37a591a767f6d1af428ea0d5f3d3a1ddc1

diff --git a/src/be/nikiroo/fanfix/supported/E621.java b/src/be/nikiroo/fanfix/supported/E621.java
index bc6ba5b..665bdd5 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;
@@ -39,6 +40,7 @@ class E621 extends BasicSupport {
 		meta.setDate("");
 		meta.setTags(new ArrayList<String>()); // TODDO ???
 		meta.setSource(getSourceName());
+		meta.setUrl(source.toString());
 		meta.setPublisher(getSourceName());
 		meta.setUuid(source.toString());
 		meta.setLuid("");
@@ -46,7 +48,7 @@ class E621 extends BasicSupport {
 		meta.setSubject("");
 		meta.setType(getType().toString());
 		meta.setImageDocument(true);
-		meta.setCover(null);
+		meta.setCover(getCover(source));
 
 		return meta;
 	}
@@ -83,6 +85,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) {
@@ -95,24 +112,28 @@ class E621 extends BasicSupport {
 					author = author.substring(0, pos - 1);
 					String page = source.getProtocol() + "://"
 							+ source.getHost() + author;
-					InputStream pageIn = Instance.getCache().open(
-							new URL(page), this, false);
 					try {
-						key = "class=\"tag-type-artist\"";
-						author = getLine(pageIn, key, 0);
-						if (author != null) {
-							pos = author.indexOf("<a href=\"");
-							if (pos >= 0) {
-								author = author.substring(pos);
-								pos = author.indexOf("</a>");
+						InputStream pageIn = Instance.getCache().open(
+								new URL(page), this, false);
+						try {
+							key = "class=\"tag-type-artist\"";
+							author = getLine(pageIn, key, 0);
+							if (author != null) {
+								pos = author.indexOf("<a href=\"");
 								if (pos >= 0) {
-									author = author.substring(0, pos);
-									return StringUtils.unhtml(author);
+									author = author.substring(pos);
+									pos = author.indexOf("</a>");
+									if (pos >= 0) {
+										author = author.substring(0, pos);
+										return StringUtils.unhtml(author);
+									}
 								}
 							}
+						} finally {
+							pageIn.close();
 						}
-					} finally {
-						pageIn.close();
+					} catch (Exception e) {
+						// No author found
 					}
 				}
 			}
@@ -232,7 +253,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);