src/main/java/eu/svjatoslav/meviz/htmlindexer/metadata/DirectoryMetadata.java

   1 /*
   2  * Meviz - Various tools collection to work with multimedia.
   3  * Copyright (C) 2012, Svjatoslav Agejenko, svjatoslav@svjatoslav.eu
   4  *
   5  * This program is free software; you can redistribute it and/or
   6  * modify it under the terms of version 2 of the GNU General Public License
   7  * as published by the Free Software Foundation.
   8  */
   9
  10 package eu.svjatoslav.meviz.htmlindexer.metadata;
  11
  12 import eu.svjatoslav.commons.file.FilePathParser;
  13 import eu.svjatoslav.meviz.htmlindexer.AbstractIndexer;
  14 import eu.svjatoslav.meviz.htmlindexer.FileSortComparator;
  15 import eu.svjatoslav.meviz.htmlindexer.ImageFormatError;
  16 import eu.svjatoslav.meviz.htmlindexer.metadata.fileTypes.*;
  17
  18 import java.io.File;
  19 import java.io.IOException;
  20 import java.io.Serializable;
  21 import java.util.*;
  22
  23 /**
  24  * Corresponds to single filesystem directory.
  25  */
  26 public class DirectoryMetadata implements Serializable {
  27
  28     private static final long serialVersionUID = 3526489102725366334L;
  29
  30     private final HashMap<String, AbstractFile> fileNameToFileMap = new HashMap<String, AbstractFile>();
  31
  32     public transient boolean changed;
  33
  34     /**
  35      * Performs cleanup of directory model. In particular removes metadata for
  36      * images that no longer exist in the filesystem.
  37      */
  38     public void cleanup() {
  39         final Iterator<AbstractFile> fileIterator = fileNameToFileMap.values()
  40                 .iterator();
  41
  42         while (fileIterator.hasNext()) {
  43             final AbstractFile file = fileIterator.next();
  44
  45             if (!file.isMetaInfoVerified())
  46                 fileIterator.remove();
  47         }
  48     }
  49
  50     public void ensureFileMetainfoIsUpToDate(final File parentDirectory,
  51                                              final File file) throws Exception {
  52
  53         final String fileName = file.getName();
  54
  55         if (fileNameToFileMap.containsKey(fileName)) {
  56             final boolean metainfoUpdated = fileNameToFileMap.get(fileName)
  57                     .ensureFileMetainfoIsUpToDate(parentDirectory);
  58
  59             if (metainfoUpdated)
  60                 changed = true;
  61         } else
  62             registerFile(parentDirectory, file);
  63
  64     }
  65
  66     private AbstractFile getAbstractFile(final File parentDirectory,
  67                                          final File file, final String fileName) throws Exception {
  68
  69         if (file.isDirectory())
  70             return new DirectoryFile(parentDirectory, fileName);
  71
  72         final String fileExtension = FilePathParser.getFileExtension(fileName)
  73                 .toLowerCase();
  74
  75         if (AbstractIndexer.isImage(fileExtension))
  76             try {
  77                 return new Picture(parentDirectory, fileName);
  78             } catch (final Exception exception) {
  79                 // in case image decoding failed, handle image as general file
  80                 return new GeneralFile(parentDirectory, fileName);
  81             }
  82
  83         if (fileName.endsWith(".meviz.org"))
  84             return new MevizSpecialFile(parentDirectory, fileName);
  85
  86         if (fileName.endsWith(".meviz.html"))
  87             return new MevizSpecialFile(parentDirectory, fileName);
  88
  89         return new GeneralFile(parentDirectory, fileName);
  90     }
  91
  92     public AbstractFile getFile(final String fileName) {
  93         return fileNameToFileMap.get(fileName);
  94     }
  95
  96     public Collection<AbstractFile> getFiles() {
  97
  98         final ArrayList<AbstractFile> files = new ArrayList<AbstractFile>(
  99                 fileNameToFileMap.values());
 100
 101         Collections.sort(files, new FileSortComparator());
 102
 103         return files;
 104     }
 105
 106     public String getHeaderMevizMetadataHtml() {
 107         final AbstractFile headerFile = getFile("header.meviz.html");
 108         if (headerFile == null)
 109             return "";
 110
 111         final MevizSpecialFile mevizSpecialFile = (MevizSpecialFile) headerFile;
 112         return mevizSpecialFile.getFileContent();
 113     }
 114
 115     public Dimension getImageDimensions(final String fileName)
 116             throws IOException, ImageFormatError {
 117
 118         return ((Picture) getFile(fileName)).getDimensions();
 119     }
 120
 121     public AbstractFile registerFile(final File parentDirectory, final File file)
 122             throws Exception {
 123
 124         final String fileName = file.getName();
 125
 126         final AbstractFile abstractFile = getAbstractFile(parentDirectory,
 127                 file, fileName);
 128
 129         fileNameToFileMap.put(fileName, abstractFile);
 130
 131         changed = true;
 132
 133         return abstractFile;
 134     }
 135
 136     public void removeUnusedMetadataEntries() {
 137
 138         for (final String key : new HashSet<String>(fileNameToFileMap.keySet()))
 139             if (!fileNameToFileMap.get(key).isMetaInfoVerified()) {
 140                 fileNameToFileMap.remove(key);
 141                 changed = true;
 142             }
 143
 144     }
 145 }