cover: allow custom author covers

[fanfix.git] / src / be / nikiroo / fanfix / library / LocalLibrary.java

diff --git a/src/be/nikiroo/fanfix/library/LocalLibrary.java b/src/be/nikiroo/fanfix/library/LocalLibrary.java
index e9c67f3866d41bca6487b36b2e5db20be163dc97..1752dc7f3e253e57586fdc0e675d380700bc31ee 100644 (file)
--- 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;
@@ -251,13 +253,43 @@ public class LocalLibrary extends BasicLibrary {
                return sourceCovers.get(source);
        }
 
+       @Override
+       public synchronized Image getCustomAuthorCover(String author) {
+               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) {
                setSourceCover(source, getCover(luid));
        }
 
+       @Override
+       public void setAuthorCover(String author, String luid) {
+               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 +310,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 {
@@ -387,6 +440,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 +459,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 +471,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.IMAGE_FORMAT_COVER);
+               return new File(aDir, hash + "." + ext.toLowerCase());
+       }
+
        /**
         * Return the list of files/directories on disk for this {@link Story}.
         * <p>