New one-item-per-source-type mode

[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 e4dc5e98bce53547c4c7d5ad2cee833e0e95bee0..bc41157328e7a0656b065c7b45ba2afc07872ba0 100644 (file)
--- a/src/be/nikiroo/fanfix/library/LocalLibrary.java
+++ b/src/be/nikiroo/fanfix/library/LocalLibrary.java
@@ -3,12 +3,16 @@ package be.nikiroo.fanfix.library;
 import java.awt.image.BufferedImage;
 import java.io.File;
 import java.io.FileFilter;
+import java.io.FileInputStream;
 import java.io.IOException;
+import java.io.InputStream;
 import java.util.ArrayList;
 import java.util.HashMap;
 import java.util.List;
 import java.util.Map;
 
+import javax.imageio.ImageIO;
+
 import be.nikiroo.fanfix.Instance;
 import be.nikiroo.fanfix.bundles.Config;
 import be.nikiroo.fanfix.data.MetaData;
@@ -18,6 +22,8 @@ import be.nikiroo.fanfix.output.BasicOutput.OutputType;
 import be.nikiroo.fanfix.output.InfoCover;
 import be.nikiroo.fanfix.supported.InfoReader;
 import be.nikiroo.utils.IOUtils;
+import be.nikiroo.utils.ImageUtils;
+import be.nikiroo.utils.MarkableFileInputStream;
 import be.nikiroo.utils.Progress;
 
 /**
@@ -28,6 +34,7 @@ import be.nikiroo.utils.Progress;
 public class LocalLibrary extends BasicLibrary {
        private int lastId;
        private Map<MetaData, File[]> stories; // Files: [ infoFile, TargetFile ]
+       private Map<String, BufferedImage> sourceCovers;
 
        private File baseDir;
        private OutputType text;
@@ -50,6 +57,7 @@ public class LocalLibrary extends BasicLibrary {
 
                this.lastId = 0;
                this.stories = null;
+               this.sourceCovers = new HashMap<String, BufferedImage>();
 
                baseDir.mkdirs();
        }
@@ -92,10 +100,12 @@ public class LocalLibrary extends BasicLibrary {
        @Override
        protected void clearCache() {
                stories = null;
+               sourceCovers = new HashMap<String, BufferedImage>();
        }
 
        @Override
        protected synchronized int getNextId() {
+               getStories(null); // make sure lastId is set
                return ++lastId;
        }
 
@@ -150,6 +160,27 @@ public class LocalLibrary extends BasicLibrary {
                clearCache();
        }
 
+       @Override
+       public BufferedImage getSourceCover(String source) {
+               if (!sourceCovers.containsKey(source)) {
+                       sourceCovers.put(source, super.getSourceCover(source));
+               }
+
+               return sourceCovers.get(source);
+       }
+
+       @Override
+       public void setSourceCover(String source, String luid) {
+               sourceCovers.put(source, getCover(luid));
+               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);
+               }
+       }
+
        /**
         * Return the {@link OutputType} for this {@link Story}.
         * 
@@ -161,9 +192,9 @@ public class LocalLibrary extends BasicLibrary {
        private OutputType getOutputType(MetaData meta) {
                if (meta != null && meta.isImageDocument()) {
                        return image;
-               } else {
-                       return text;
                }
+
+               return text;
        }
 
        /**
@@ -238,39 +269,39 @@ public class LocalLibrary extends BasicLibrary {
                MetaData meta = getInfo(luid);
                if (meta == null) {
                        throw new IOException("Story not found: " + luid);
-               } else {
-                       File infoFile = getStories(null).get(meta)[0];
-                       File targetFile = getStories(null).get(meta)[1];
+               }
 
-                       files.add(infoFile);
-                       files.add(targetFile);
+               File infoFile = getStories(null).get(meta)[0];
+               File targetFile = getStories(null).get(meta)[1];
 
-                       String readerExt = getOutputType(meta).getDefaultExtension(true);
-                       String fileExt = getOutputType(meta).getDefaultExtension(false);
+               files.add(infoFile);
+               files.add(targetFile);
 
-                       String path = targetFile.getAbsolutePath();
-                       if (readerExt != null && !readerExt.equals(fileExt)) {
-                               path = path.substring(0, path.length() - readerExt.length())
-                                               + fileExt;
-                               File relatedFile = new File(path);
+               String readerExt = getOutputType(meta).getDefaultExtension(true);
+               String fileExt = getOutputType(meta).getDefaultExtension(false);
 
-                               if (relatedFile.exists()) {
-                                       files.add(relatedFile);
-                               }
-                       }
+               String path = targetFile.getAbsolutePath();
+               if (readerExt != null && !readerExt.equals(fileExt)) {
+                       path = path.substring(0, path.length() - readerExt.length())
+                                       + fileExt;
+                       File relatedFile = new File(path);
 
-                       String coverExt = "."
-                                       + Instance.getConfig().getString(Config.IMAGE_FORMAT_COVER);
-                       File coverFile = new File(path + coverExt);
-                       if (!coverFile.exists()) {
-                               coverFile = new File(path.substring(0,
-                                               path.length() - fileExt.length())
-                                               + coverExt);
+                       if (relatedFile.exists()) {
+                               files.add(relatedFile);
                        }
+               }
 
-                       if (coverFile.exists()) {
-                               files.add(coverFile);
-                       }
+               String coverExt = "."
+                               + Instance.getConfig().getString(Config.IMAGE_FORMAT_COVER);
+               File coverFile = new File(path + coverExt);
+               if (!coverFile.exists()) {
+                       coverFile = new File(path.substring(0,
+                                       path.length() - fileExt.length())
+                                       + coverExt);
+               }
+
+               if (coverFile.exists()) {
+                       files.add(coverFile);
                }
 
                return files;
@@ -301,6 +332,7 @@ public class LocalLibrary extends BasicLibrary {
                        lastId = 0;
 
                        File[] dirs = baseDir.listFiles(new FileFilter() {
+                               @Override
                                public boolean accept(File file) {
                                        return file != null && file.isDirectory();
                                }
@@ -311,6 +343,7 @@ public class LocalLibrary extends BasicLibrary {
 
                        for (File dir : dirs) {
                                File[] infoFiles = dir.listFiles(new FileFilter() {
+                                       @Override
                                        public boolean accept(File file) {
                                                return file != null
                                                                && file.getPath().toLowerCase()
@@ -322,10 +355,12 @@ public class LocalLibrary extends BasicLibrary {
                                pgDirs.addProgress(pgFiles, 100);
                                pgDirs.setName("Loading from: " + dir.getName());
 
+                               String source = null;
                                for (File infoFile : infoFiles) {
                                        pgFiles.setName(infoFile.getName());
                                        try {
                                                MetaData meta = InfoReader.readMeta(infoFile, false);
+                                               source = meta.getSource();
                                                try {
                                                        int id = Integer.parseInt(meta.getLuid());
                                                        if (id > lastId) {
@@ -352,6 +387,21 @@ public class LocalLibrary extends BasicLibrary {
                                        pgFiles.add(1);
                                }
 
+                               File cover = new File(dir, ".cover.png");
+                               if (cover.exists()) {
+                                       try {
+                                               InputStream in = new MarkableFileInputStream(
+                                                               new FileInputStream(cover));
+                                               try {
+                                                       sourceCovers.put(source, ImageUtils.fromStream(in));
+                                               } finally {
+                                                       in.close();
+                                               }
+                                       } catch (IOException e) {
+                                               Instance.syserr(e);
+                                       }
+                               }
+
                                pgFiles.setName(null);
                        }