X-Git-Url: http://git.nikiroo.be/?a=blobdiff_plain;f=src%2Fbe%2Fnikiroo%2Ffanfix%2Flibrary%2FLocalLibrary.java;h=ffcd8af99d0a482b27e939435fbbd6548f32550d;hb=13fdb89adc017452a7a72f552b933f8e7b869413;hp=e9c67f3866d41bca6487b36b2e5db20be163dc97;hpb=3a0605e6a05e6d9dee6b596d5f9b193408470164;p=fanfix.git

diff --git a/src/be/nikiroo/fanfix/library/LocalLibrary.java b/src/be/nikiroo/fanfix/library/LocalLibrary.java
index e9c67f3..ffcd8af 100644
--- a/src/be/nikiroo/fanfix/library/LocalLibrary.java
+++ b/src/be/nikiroo/fanfix/library/LocalLibrary.java
@@ -22,6 +22,7 @@ import be.nikiroo.fanfix.supported.InfoReader;
 import be.nikiroo.utils.IOUtils;
 import be.nikiroo.utils.Image;
 import be.nikiroo.utils.Progress;
+import be.nikiroo.utils.StringUtils;
 
 /**
  * This {@link BasicLibrary} will store the stories locally on disk.
@@ -32,6 +33,7 @@ public class LocalLibrary extends BasicLibrary {
 	private int lastId;
 	private Map<MetaData, File[]> stories; // Files: [ infoFile, TargetFile ]
 	private Map<String, Image> sourceCovers;
+	private Map<String, Image> authorCovers;
 
 	private File baseDir;
 	private OutputType text;
@@ -45,8 +47,8 @@ public class LocalLibrary extends BasicLibrary {
 	 */
 	public LocalLibrary(File baseDir) {
 		this(baseDir, Instance.getConfig().getString(
-				Config.NON_IMAGES_DOCUMENT_TYPE), Instance.getConfig()
-				.getString(Config.IMAGES_DOCUMENT_TYPE), false);
+				Config.FILE_FORMAT_NON_IMAGES_DOCUMENT_TYPE), Instance.getConfig()
+				.getString(Config.FILE_FORMAT_IMAGES_DOCUMENT_TYPE), false);
 	}
 
 	/**
@@ -98,7 +100,7 @@ public class LocalLibrary extends BasicLibrary {
 	}
 
 	@Override
-	public File getFile(String luid, Progress pg) {
+	public File getFile(String luid, Progress pg) throws IOException {
 		Instance.getTraceHandler().trace(
 				this.getClass().getSimpleName() + ": get file for " + luid);
 
@@ -120,7 +122,7 @@ public class LocalLibrary extends BasicLibrary {
 	}
 
 	@Override
-	public Image getCover(String luid) {
+	public Image getCover(String luid) throws IOException {
 		MetaData meta = getInfo(luid);
 		if (meta != null) {
 			if (meta.getCover() != null) {
@@ -252,12 +254,51 @@ public class LocalLibrary extends BasicLibrary {
 	}
 
 	@Override
-	public void setSourceCover(String source, String luid) {
+	public synchronized Image getCustomAuthorCover(String author) {
+		if (authorCovers == null) {
+			authorCovers = new HashMap<String, Image>();
+		}
+
+		Image img = authorCovers.get(author);
+		if (img != null) {
+			return img;
+		}
+
+		File cover = getAuthorCoverFile(author);
+		if (cover.exists()) {
+			InputStream in;
+			try {
+				in = new FileInputStream(cover);
+				try {
+					authorCovers.put(author, new Image(in));
+				} finally {
+					in.close();
+				}
+			} catch (FileNotFoundException e) {
+				e.printStackTrace();
+			} catch (IOException e) {
+				Instance.getTraceHandler().error(
+						new IOException(
+								"Cannot load the existing custom author cover: "
+										+ cover, e));
+			}
+		}
+
+		return authorCovers.get(author);
+	}
+
+	@Override
+	public void setSourceCover(String source, String luid) throws IOException {
 		setSourceCover(source, getCover(luid));
 	}
 
+	@Override
+	public void setAuthorCover(String author, String luid) throws IOException {
+		setAuthorCover(author, getCover(luid));
+	}
+
 	/**
-	 * Fix the source cover to the given story cover.
+	 * Set the source cover to the given story cover.
 	 * 
 	 * @param source
 	 *            the source to change
@@ -278,6 +319,27 @@ public class LocalLibrary extends BasicLibrary {
 		}
 	}
 
+	/**
+	 * Set the author cover to the given story cover.
+	 * 
+	 * @param author
+	 *            the author to change
+	 * @param coverImage
+	 *            the cover image
+	 */
+	synchronized void setAuthorCover(String author, Image coverImage) {
+		File cover = getAuthorCoverFile(author);
+		cover.getParentFile().mkdirs();
+		try {
+			Instance.getCache().saveAsImage(coverImage, cover, true);
+			if (authorCovers != null) {
+				authorCovers.put(author, coverImage);
+			}
+		} catch (IOException e) {
+			Instance.getTraceHandler().error(e);
+		}
+	}
+
 	@Override
 	public void imprt(BasicLibrary other, String luid, Progress pg)
 			throws IOException {
@@ -352,6 +414,22 @@ public class LocalLibrary extends BasicLibrary {
 		return text;
 	}
 
+	/**
+	 * Return the default {@link OutputType} for this kind of {@link Story}.
+	 * 
+	 * @param imageDocument
+	 *            TRUE for images document, FALSE for text documents
+	 * 
+	 * @return the type
+	 */
+	public String getOutputType(boolean imageDocument) {
+		if (imageDocument) {
+			return image.toString();
+		}
+
+		return text.toString();
+	}
+
 	/**
 	 * Get the target {@link File} related to the given <tt>.info</tt>
 	 * {@link File} and {@link MetaData}.
@@ -387,6 +465,9 @@ public class LocalLibrary extends BasicLibrary {
 			title = "";
 		}
 		title = title.replaceAll("[^a-zA-Z0-9._+-]", "_");
+		if (title.length() > 40) {
+			title = title.substring(0, 40);
+		}
 		return new File(getExpectedDir(key.getSource()), key.getLuid() + "_"
 				+ title);
 	}
@@ -403,7 +484,8 @@ public class LocalLibrary extends BasicLibrary {
 	private File getExpectedDir(String source) {
 		String sanitizedSource = source.replaceAll("[^a-zA-Z0-9._+/-]", "_");
 
-		while (sanitizedSource.startsWith("/")) {
+		while (sanitizedSource.startsWith("/")
+				|| sanitizedSource.startsWith("_")) {
 			if (sanitizedSource.length() > 1) {
 				sanitizedSource = sanitizedSource.substring(1);
 			} else {
@@ -414,12 +496,30 @@ public class LocalLibrary extends BasicLibrary {
 		sanitizedSource = sanitizedSource.replace("/", File.separator);
 
 		if (sanitizedSource.isEmpty()) {
-			sanitizedSource = "EMPTY";
+			sanitizedSource = "_EMPTY";
 		}
 
 		return new File(baseDir, sanitizedSource);
 	}
 
+	/**
+	 * Return the full path to the file to use for the custom cover of this
+	 * author.
+	 * <p>
+	 * One or more of the parent directories <b>MAY</b> not exist.
+	 * 
+	 * @param author
+	 *            the author
+	 * 
+	 * @return the custom cover file
+	 */
+	private File getAuthorCoverFile(String author) {
+		File aDir = new File(baseDir, "_AUTHORS");
+		String hash = StringUtils.getMd5Hash(author);
+		String ext = Instance.getConfig().getString(Config.FILE_FORMAT_IMAGE_FORMAT_COVER);
+		return new File(aDir, hash + "." + ext.toLowerCase());
+	}
+
 	/**
 	 * Return the list of files/directories on disk for this {@link Story}.
 	 * <p>
@@ -462,7 +562,7 @@ public class LocalLibrary extends BasicLibrary {
 		}
 
 		String coverExt = "."
-				+ Instance.getConfig().getString(Config.IMAGE_FORMAT_COVER)
+				+ Instance.getConfig().getString(Config.FILE_FORMAT_IMAGE_FORMAT_COVER)
 						.toLowerCase();
 		File coverFile = new File(path + coverExt);
 		if (!coverFile.exists()) {
@@ -539,7 +639,8 @@ public class LocalLibrary extends BasicLibrary {
 				boolean info = file != null && file.isFile()
 						&& file.getPath().toLowerCase().endsWith(".info");
 				boolean dir = file != null && file.isDirectory();
-				return info || dir;
+				boolean isExpandedHtml = new File(file, "index.html").isFile();
+				return info || (dir && !isExpandedHtml);
 			}
 		});