86ad8189eb20eda7f9d9c71c4564d7820f88f0cc
[meviz.git] / src / main / java / eu / svjatoslav / meviz / htmlindexer / FilesystemIndexer.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;
11
12 import eu.svjatoslav.commons.file.IOHelper;
13 import eu.svjatoslav.meviz.htmlindexer.metadata.Dimension;
14 import eu.svjatoslav.meviz.htmlindexer.metadata.DirectoryMetadata;
15 import eu.svjatoslav.meviz.htmlindexer.metadata.MetadadaHelper;
16 import eu.svjatoslav.meviz.htmlindexer.metadata.fileTypes.AbstractFile;
17 import eu.svjatoslav.meviz.htmlindexer.metadata.fileTypes.Picture;
18 import org.apache.log4j.Logger;
19
20 import java.io.File;
21 import java.io.IOException;
22 import java.util.ArrayList;
23 import java.util.HashSet;
24 import java.util.List;
25
26 /**
27  * Main class of HTML indexer.
28  */
29
30 public class FilesystemIndexer extends AbstractIndexer {
31
32     public static final Logger logger = Logger
33             .getLogger(FilesystemIndexer.class);
34
35     private final HashSet<String> validMetadataFiles = new HashSet<String>();
36
37     private final File directoryToIndex;
38     private final List<String> pathElements;
39     private final CommandlineHandler commandlineHandler;
40     private File metadataFile;
41     private DirectoryMetadata metadata;
42     private HashSet<Layout> layouts;
43
44     public FilesystemIndexer(final File directoryToIndex,
45                              final CommandlineHandler commandlineHandler,
46                              final List<String> pathElements) throws Exception {
47
48         this.directoryToIndex = directoryToIndex;
49         this.commandlineHandler = commandlineHandler;
50         this.pathElements = pathElements;
51
52         if (!directoryToIndex.canRead())
53             return;
54
55         validMetadataFiles.add(Constants.METADATA_FILE_NAME);
56
57         initializeThumbnailsDirectory();
58         loadOrCreateMetadata();
59
60         initializeLayouts();
61
62         updateMetadata();
63
64         metadata.removeUnusedMetadataEntries();
65
66         generateHtmlFromMetadata(layouts);
67
68         removeUnusedThumbnailFiles();
69
70         // save directory metadata
71         if (metadata.changed)
72             MetadadaHelper.saveDirectoryMetadata(metadataFile, metadata);
73
74     }
75
76     public boolean canWriteIndexFile(final File indexFile)
77             throws IOException {
78
79         return !indexFile.exists() || Utils.isMevizGeneratedIndexFile(indexFile);
80
81     }
82
83     public void generateHtmlFromMetadata(final HashSet<Layout> layouts) {
84         // Generate HTML from metadata
85         for (final Layout layout : layouts)
86             try {
87                 final File indexFile = Utils.getLayoutIndexFile(layout,
88                         directoryToIndex);
89
90                 if (canWriteIndexFile(indexFile)) {
91
92                     compileHtml(layout, metadata);
93
94                     IOHelper.overwriteFileIfContentDiffers(indexFile, layout
95                             .getHtml(true, true, metadata).getBytes());
96                 }
97             } catch (final Exception e) {
98                 logger.error("Error writing index file. ", e);
99             }
100     }
101
102     public List<String> getChildPath(final File file) {
103
104         final List<String> result = new ArrayList<String>();
105         result.addAll(pathElements);
106         result.add(file.getName());
107
108         return result;
109     }
110
111     @Override
112     public String getDirectoryUrl(final AbstractFile directory,
113                                   final IndexingContext context) {
114         return Utils.urlEncode(directory.fileName) + "/index.html";
115     }
116
117     @Override
118     public String getParentDirectoryUrl(final IndexingContext context) {
119         return "../index.html";
120     }
121
122     @Override
123     public String getThumbnailUrl(final Picture picture,
124                                   final Dimension desiredDimension, final IndexingContext context) {
125
126         // in case thumbnail size was equal to original, then return original
127         // file path
128         if (picture.getDimensions().equals(desiredDimension))
129             return picture.fileName;
130
131         final String thumbnailFileName = picture
132                 .getRelativeThumbnailFileName(desiredDimension);
133
134         validMetadataFiles.add(thumbnailFileName);
135         final File thumbnailFile = new File(
136                 Utils.getThumbnailsDirectoryPath(directoryToIndex)
137                         + thumbnailFileName);
138
139         if (!thumbnailFile.exists()) {
140
141             final File originalFile = new File(
142                     directoryToIndex.getAbsolutePath() + "/" + picture.fileName);
143
144             // generate new thumbnail
145             Picture.makeThumbnail(originalFile, thumbnailFile,
146                     desiredDimension.getAwtDimension());
147         }
148         return Constants.THUMBNAILS_DIRECTORY_NAME + "/"
149                 + Utils.urlEncode(thumbnailFileName);
150     }
151
152     public void initializeLayouts() {
153
154         layouts = Utils.getLayouts();
155
156         final String galleryTitle = commandlineHandler.getGalleryTitle();
157
158         for (final Layout layout : layouts)
159             layout.init(galleryTitle, pathElements, this, new IndexingContext(
160                     "", "."));
161     }
162
163     public void initializeThumbnailsDirectory() {
164         final File thumbnailsDirectory = Utils
165                 .getThumbnailsDirectory(directoryToIndex);
166
167         if (!thumbnailsDirectory.exists())
168             thumbnailsDirectory.mkdirs();
169     }
170
171     public void loadOrCreateMetadata() {
172
173         metadataFile = new File(
174                 Utils.getThumbnailsDirectoryPath(directoryToIndex)
175                         + Constants.METADATA_FILE_NAME);
176
177         metadata = MetadadaHelper.initDirectoryMetadata(metadataFile);
178     }
179
180     private void removeUnusedThumbnailFiles() {
181
182         for (final File file : Utils.getThumbnailsDirectory(directoryToIndex)
183                 .listFiles())
184             if (!validMetadataFiles.contains(file.getName()))
185                 file.delete();
186
187     }
188
189     public void updateMetadata() throws Exception {
190
191         for (final File file : directoryToIndex.listFiles())
192             if (shallFileBeIndexed(file)) {
193
194                 metadata.ensureFileMetainfoIsUpToDate(directoryToIndex, file);
195
196                 if (file.isDirectory())
197                     new FilesystemIndexer(file, commandlineHandler,
198                             getChildPath(file));
199             }
200     }
201 }