Improve remote, fix bugs, update nikiroo-utils

[nikiroo-utils.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 d811d2b2249ec365232c9ea31657d5b1504f5a11..11c36630adcef96ea73d17c8712d84495260299a 100644 (file)
--- a/src/be/nikiroo/fanfix/library/LocalLibrary.java
+++ b/src/be/nikiroo/fanfix/library/LocalLibrary.java
@@ -57,12 +57,19 @@ public class LocalLibrary extends BasicLibrary {
         * 
         * @param baseDir
         *            the directory where to find the {@link Story} objects
+        * @param text
+        *            the {@link OutputType} to use for non-image documents
+        * @param image
+        *            the {@link OutputType} to use for image documents
+        * @param defaultIsHtml
+        *            if the given text or image is invalid, use HTML by default (if
+        *            not, it will be INFO_TEXT/CBZ by default)
         */
        public LocalLibrary(File baseDir, String text, String image,
                        boolean defaultIsHtml) {
-               this(baseDir, OutputType.valueOfNullOkUC(text,
+               this(baseDir, OutputType.valueOfAllOkUC(text,
                                defaultIsHtml ? OutputType.HTML : OutputType.INFO_TEXT),
-                               OutputType.valueOfNullOkUC(image,
+                               OutputType.valueOfAllOkUC(image,
                                                defaultIsHtml ? OutputType.HTML : OutputType.CBZ));
        }
 
@@ -94,8 +101,8 @@ public class LocalLibrary extends BasicLibrary {
        }
 
        @Override
-       public File getFile(String luid) {
-               File[] files = getStories(null).get(getInfo(luid));
+       public File getFile(String luid, Progress pg) {
+               File[] files = getStories(pg).get(getInfo(luid));
                if (files != null) {
                        return files[1];
                }
@@ -214,17 +221,10 @@ public class LocalLibrary extends BasicLibrary {
                        pg = new Progress();
                }
 
-               LocalLibrary otherLocalLibrary = null;
-               if (other instanceof RemoteLibrary) {
-                       otherLocalLibrary = ((RemoteLibrary) other).getLocalLibrary();
-               }
-
+               // Check if we can simply copy the files instead of the whole process
                if (other instanceof LocalLibrary) {
-                       otherLocalLibrary = (LocalLibrary) other;
-               }
+                       LocalLibrary otherLocalLibrary = (LocalLibrary) other;
 
-               // Check if we can simply copy the files instead of the whole process
-               if (otherLocalLibrary != null) {
                        MetaData meta = otherLocalLibrary.getInfo(luid);
                        String expectedType = ""
                                        + (meta != null && meta.isImageDocument() ? image : text);
@@ -331,14 +331,14 @@ public class LocalLibrary extends BasicLibrary {
         * The directory (full path) where the new {@link Story} related to this
         * {@link MetaData} should be located on disk.
         * 
-        * @param type
+        * @param source
         *            the type (source)
         * 
         * @return the target directory
         */
-       private File getExpectedDir(String type) {
-               String source = type.replaceAll("[^a-zA-Z0-9._+-]", "_");
-               return new File(baseDir, source);
+       private File getExpectedDir(String source) {
+               String sanitizedSource = source.replaceAll("[^a-zA-Z0-9._+-]", "_");
+               return new File(baseDir, sanitizedSource);
        }
 
        /**
@@ -501,4 +501,23 @@ public class LocalLibrary extends BasicLibrary {
 
                return stories;
        }
+
+       /**
+        * Fix the source cover to the given story cover.
+        * 
+        * @param source
+        *            the source to change
+        * @param coverImage
+        *            the cover image
+        */
+       void setSourceCover(String source, BufferedImage coverImage) {
+               sourceCovers.put(source, coverImage);
+               File cover = new File(getExpectedDir(source), ".cover.png");
+               try {
+                       ImageIO.write(sourceCovers.get(source), "png", cover);
+               } catch (IOException e) {
+                       Instance.syserr(e);
+                       sourceCovers.remove(source);
+               }
+       }
 }