X-Git-Url: http://git.nikiroo.be/?a=blobdiff_plain;f=src%2Fbe%2Fnikiroo%2Ffanfix%2FLibrary.java;h=69b497fd84f1730feda48aae6fab96b02a76d74c;hb=4f661b2bb0f64ed54c3e849c0c2c5b6bcfe844df;hp=88228497d8e1b6722e42ca35490006d331fadc9b;hpb=3b2b638f7e1395702f843b5b19d7959327f604b2;p=fanfix.git diff --git a/src/be/nikiroo/fanfix/Library.java b/src/be/nikiroo/fanfix/Library.java index 8822849..69b497f 100644 --- a/src/be/nikiroo/fanfix/Library.java +++ b/src/be/nikiroo/fanfix/Library.java @@ -1,14 +1,17 @@ package be.nikiroo.fanfix; import java.io.File; +import java.io.FileFilter; import java.io.IOException; import java.net.URL; import java.util.ArrayList; +import java.util.Collections; import java.util.HashMap; import java.util.List; import java.util.Map; import java.util.Map.Entry; +import be.nikiroo.fanfix.bundles.Config; import be.nikiroo.fanfix.data.MetaData; import be.nikiroo.fanfix.data.Story; import be.nikiroo.fanfix.output.BasicOutput; @@ -16,6 +19,7 @@ import be.nikiroo.fanfix.output.BasicOutput.OutputType; import be.nikiroo.fanfix.supported.BasicSupport; import be.nikiroo.fanfix.supported.BasicSupport.SupportType; import be.nikiroo.fanfix.supported.InfoReader; +import be.nikiroo.utils.IOUtils; import be.nikiroo.utils.Progress; /** @@ -53,20 +57,71 @@ public class Library { dir.mkdirs(); } + /** + * Refresh the {@link Library}, that is, make sure all stories are loaded. + * + * @param pg + * the optional progress reporter + */ + public void refresh(Progress pg) { + getStories(pg); + } + /** * List all the known types of stories. * * @return the types */ - public List getTypes() { + public synchronized List getTypes() { List list = new ArrayList(); - for (Entry entry : getStories().entrySet()) { - String storyType = entry.getValue().getParentFile().getName(); + for (Entry entry : getStories(null).entrySet()) { + String storyType = entry.getKey().getSource(); if (!list.contains(storyType)) { list.add(storyType); } } + Collections.sort(list); + return list; + } + + /** + * List all the known authors of stories. + * + * @return the authors + */ + public synchronized List getAuthors() { + List list = new ArrayList(); + for (Entry entry : getStories(null).entrySet()) { + String storyAuthor = entry.getKey().getAuthor(); + if (!list.contains(storyAuthor)) { + list.add(storyAuthor); + } + } + + Collections.sort(list); + return list; + } + + /** + * List all the stories of the given author in the {@link Library}, or all + * the stories if NULL is passed as an author. + * + * @param author + * the author of the stories to retrieve, or NULL for all + * + * @return the stories + */ + public synchronized List getListByAuthor(String author) { + List list = new ArrayList(); + for (Entry entry : getStories(null).entrySet()) { + String storyAuthor = entry.getKey().getAuthor(); + if (author == null || author.equalsIgnoreCase(storyAuthor)) { + list.add(entry.getKey()); + } + } + + Collections.sort(list); return list; } @@ -79,15 +134,16 @@ public class Library { * * @return the stories */ - public List getList(String type) { + public synchronized List getListByType(String type) { List list = new ArrayList(); - for (Entry entry : getStories().entrySet()) { + for (Entry entry : getStories(null).entrySet()) { String storyType = entry.getValue().getParentFile().getName(); if (type == null || type.equalsIgnoreCase(storyType)) { list.add(entry.getKey()); } } + Collections.sort(list); return list; } @@ -99,9 +155,9 @@ public class Library { * * @return the corresponding {@link Story} */ - public MetaData getInfo(String luid) { + public synchronized MetaData getInfo(String luid) { if (luid != null) { - for (Entry entry : getStories().entrySet()) { + for (Entry entry : getStories(null).entrySet()) { if (luid.equals(entry.getKey().getLuid())) { return entry.getKey(); } @@ -119,9 +175,9 @@ public class Library { * * @return the corresponding {@link Story} */ - public File getFile(String luid) { + public synchronized File getFile(String luid) { if (luid != null) { - for (Entry entry : getStories().entrySet()) { + for (Entry entry : getStories(null).entrySet()) { if (luid.equals(entry.getKey().getLuid())) { return entry.getValue(); } @@ -141,9 +197,9 @@ public class Library { * * @return the corresponding {@link Story} or NULL if not found */ - public Story getStory(String luid, Progress pg) { + public synchronized Story getStory(String luid, Progress pg) { if (luid != null) { - for (Entry entry : getStories().entrySet()) { + for (Entry entry : getStories(null).entrySet()) { if (luid.equals(entry.getKey().getLuid())) { try { SupportType type = SupportType.valueOfAllOkUC(entry @@ -271,13 +327,14 @@ public class Library { * @throws IOException * in case of I/O error */ - public Story save(Story story, String luid, Progress pg) throws IOException { + public synchronized Story save(Story story, String luid, Progress pg) + throws IOException { // Do not change the original metadata, but change the original story MetaData key = story.getMeta().clone(); story.setMeta(key); if (luid == null || luid.isEmpty()) { - getStories(); // refresh lastId if needed + getStories(null); // refresh lastId if needed key.setLuid(String.format("%03d", (++lastId))); } else { key.setLuid(luid); @@ -296,12 +353,71 @@ public class Library { } BasicOutput it = BasicOutput.getOutput(out, true); - File file = it.process(story, getFile(key).getPath(), pg); - getStories().put(story.getMeta(), file); + it.process(story, getFile(key).getPath(), pg); + + // empty cache + stories.clear(); return story; } + /** + * Delete the given {@link Story} from this {@link Library}. + * + * @param luid + * the LUID of the target {@link Story} + * + * @return TRUE if it was deleted + */ + public synchronized boolean delete(String luid) { + boolean ok = false; + + MetaData meta = getInfo(luid); + File file = getStories(null).get(meta); + + if (file != null) { + if (file.delete()) { + String readerExt = getOutputType(meta) + .getDefaultExtension(true); + String fileExt = getOutputType(meta).getDefaultExtension(false); + + String path = file.getAbsolutePath(); + if (readerExt != null && !readerExt.equals(fileExt)) { + path = path + .substring(0, path.length() - readerExt.length()) + + fileExt; + file = new File(path); + IOUtils.deltree(file); + } + + File infoFile = new File(path + ".info"); + if (!infoFile.exists()) { + infoFile = new File(path.substring(0, path.length() + - fileExt.length()) + + ".info"); + } + infoFile.delete(); + + 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())); + } + coverFile.delete(); + + ok = true; + } + + // clear cache + stories.clear(); + } + + return ok; + } + /** * The directory (full path) where the {@link Story} related to this * {@link MetaData} should be located on disk. @@ -326,63 +442,104 @@ public class Library { * @return the target */ private File getFile(MetaData key) { - String title = key.getTitle().replaceAll("[^a-zA-Z0-9._+-]", "_"); + String title = key.getTitle(); + if (title == null) { + title = ""; + } + title = title.replaceAll("[^a-zA-Z0-9._+-]", "_"); return new File(getDir(key), key.getLuid() + "_" + title); } /** * Return all the known stories in this {@link Library} object. * + * @param pg + * the optional progress reporter + * * @return the stories */ - private Map getStories() { + private synchronized Map getStories(Progress pg) { + if (pg == null) { + pg = new Progress(); + } else { + pg.setMinMax(0, 100); + } + if (stories.isEmpty()) { lastId = 0; - String ext = ".info"; - for (File dir : baseDir.listFiles()) { - if (dir.isDirectory()) { - for (File file : dir.listFiles()) { + File[] dirs = baseDir.listFiles(new FileFilter() { + public boolean accept(File file) { + return file != null && file.isDirectory(); + } + }); + + Progress pgDirs = new Progress(0, 100 * dirs.length); + pg.addProgress(pgDirs, 100); + + final String ext = ".info"; + for (File dir : dirs) { + File[] files = dir.listFiles(new FileFilter() { + public boolean accept(File file) { + return file != null + && file.getPath().toLowerCase().endsWith(ext); + } + }); + + Progress pgFiles = new Progress(0, files.length); + pgDirs.addProgress(pgFiles, 100); + pgDirs.setName("Loading from: " + dir.getName()); + + for (File file : files) { + try { + pgFiles.setName(file.getName()); + MetaData meta = InfoReader.readMeta(file); try { - if (file.getPath().toLowerCase().endsWith(ext)) { - MetaData meta = InfoReader.readMeta(file); - try { - int id = Integer.parseInt(meta.getLuid()); - if (id > lastId) { - lastId = id; - } - - // Replace .info with whatever is needed: - String path = file.getPath(); - path = path.substring(0, path.length() - - ext.length()); - - String newExt = getOutputType(meta) - .getDefaultExtension(); - - file = new File(path + newExt); - // - - stories.put(meta, file); - - } catch (Exception e) { - // not normal!! - Instance.syserr(new IOException( - "Cannot understand the LUID of " - + file.getPath() + ": " - + meta.getLuid(), e)); - } + int id = Integer.parseInt(meta.getLuid()); + if (id > lastId) { + lastId = id; } - } catch (IOException e) { - // We should not have not-supported files in the - // library + + // Replace .info with whatever is needed: + String path = file.getPath(); + path = path.substring(0, + path.length() - ext.length()); + + String newExt = getOutputType(meta) + .getDefaultExtension(true); + + file = new File(path + newExt); + // + + stories.put(meta, file); + + } catch (Exception e) { + // not normal!! Instance.syserr(new IOException( - "Cannot load file from library: " - + file.getPath(), e)); + "Cannot understand the LUID of " + + file.getPath() + ": " + + meta.getLuid(), e)); } + } catch (IOException e) { + // We should not have not-supported files in the + // library + Instance.syserr(new IOException( + "Cannot load file from library: " + + file.getPath(), e)); + } finally { + pgFiles.setProgress(pgFiles.getProgress() + 1); + + System.out.println("files: " + pgFiles.getProgress() + + "/" + pgFiles.getMax()); + System.out.println("dirs : " + pgDirs.getProgress() + + "/" + pgDirs.getMax()); } } + + pgFiles.setName(null); } + + pgDirs.setName("Loading directories"); } return stories;