Merge branch 'subtree'
[nikiroo-utils.git] / src / be / nikiroo / fanfix / library / CacheLibrary.java
1 package be.nikiroo.fanfix.library;
2
3 import java.io.File;
4 import java.io.IOException;
5 import java.net.URL;
6 import java.util.ArrayList;
7 import java.util.List;
8 import java.util.TreeSet;
9
10 import be.nikiroo.fanfix.Instance;
11 import be.nikiroo.fanfix.bundles.UiConfig;
12 import be.nikiroo.fanfix.bundles.UiConfigBundle;
13 import be.nikiroo.fanfix.data.MetaData;
14 import be.nikiroo.fanfix.data.Story;
15 import be.nikiroo.fanfix.output.BasicOutput.OutputType;
16 import be.nikiroo.utils.Image;
17 import be.nikiroo.utils.Progress;
18
19 /**
20 * This library will cache another pre-existing {@link BasicLibrary}.
21 *
22 * @author niki
23 */
24 public class CacheLibrary extends BasicLibrary {
25 private List<MetaData> metasReal;
26 private List<MetaData> metasMixed;
27 private Object metasLock = new Object();
28
29 private BasicLibrary lib;
30 private LocalLibrary cacheLib;
31
32 /**
33 * Create a cache library around the given one.
34 * <p>
35 * It will return the same result, but those will be saved to disk at the
36 * same time to be fetched quicker the next time.
37 *
38 * @param cacheDir
39 * the cache directory where to save the files to disk
40 * @param lib
41 * the original library to wrap
42 * @param config
43 * the configuration used to know which kind of default
44 * {@link OutputType} to use for images and non-images stories
45 */
46 public CacheLibrary(File cacheDir, BasicLibrary lib,
47 UiConfigBundle config) {
48 this.cacheLib = new LocalLibrary(cacheDir, //
49 config.getString(UiConfig.GUI_NON_IMAGES_DOCUMENT_TYPE),
50 config.getString(UiConfig.GUI_IMAGES_DOCUMENT_TYPE), true);
51 this.lib = lib;
52 }
53
54 @Override
55 public String getLibraryName() {
56 return lib.getLibraryName();
57 }
58
59 @Override
60 public Status getStatus() {
61 return lib.getStatus();
62 }
63
64 @Override
65 protected List<MetaData> getMetas(Progress pg) throws IOException {
66 if (pg == null) {
67 pg = new Progress();
68 }
69
70 synchronized (metasLock) {
71 // We make sure that cached metas have precedence
72 if (metasMixed == null) {
73 if (metasReal == null) {
74 metasReal = lib.getMetas(pg);
75 }
76
77 metasMixed = new ArrayList<MetaData>();
78 TreeSet<String> cachedLuids = new TreeSet<String>();
79 for (MetaData cachedMeta : cacheLib.getMetas(null)) {
80 metasMixed.add(cachedMeta);
81 cachedLuids.add(cachedMeta.getLuid());
82 }
83 for (MetaData realMeta : metasReal) {
84 if (!cachedLuids.contains(realMeta.getLuid())) {
85 metasMixed.add(realMeta);
86 }
87 }
88 }
89 }
90
91 pg.done();
92 return new ArrayList<MetaData>(metasMixed);
93 }
94
95 @Override
96 public synchronized Story getStory(String luid, MetaData meta, Progress pg)
97 throws IOException {
98 if (pg == null) {
99 pg = new Progress();
100 }
101
102 Progress pgImport = new Progress();
103 Progress pgGet = new Progress();
104
105 pg.setMinMax(0, 4);
106 pg.addProgress(pgImport, 3);
107 pg.addProgress(pgGet, 1);
108
109 if (!isCached(luid)) {
110 try {
111 cacheLib.imprt(lib, luid, pgImport);
112 updateMetaCache(metasMixed, cacheLib.getInfo(luid));
113 pgImport.done();
114 } catch (IOException e) {
115 Instance.getInstance().getTraceHandler().error(e);
116 }
117
118 pgImport.done();
119 pgGet.done();
120 }
121
122 String type = cacheLib.getOutputType(meta.isImageDocument());
123 MetaData cachedMeta = meta.clone();
124 cachedMeta.setType(type);
125
126 return cacheLib.getStory(luid, cachedMeta, pg);
127 }
128
129 @Override
130 public synchronized File getFile(final String luid, Progress pg)
131 throws IOException {
132 if (pg == null) {
133 pg = new Progress();
134 }
135
136 Progress pgGet = new Progress();
137 Progress pgRecall = new Progress();
138
139 pg.setMinMax(0, 5);
140 pg.addProgress(pgGet, 4);
141 pg.addProgress(pgRecall, 1);
142
143 if (!isCached(luid)) {
144 getStory(luid, pgGet);
145 pgGet.done();
146 }
147
148 File file = cacheLib.getFile(luid, pgRecall);
149 pgRecall.done();
150
151 pg.done();
152 return file;
153 }
154
155 @Override
156 public Image getCover(final String luid) throws IOException {
157 if (isCached(luid)) {
158 return cacheLib.getCover(luid);
159 }
160
161 // We could update the cache here, but it's not easy
162 return lib.getCover(luid);
163 }
164
165 @Override
166 public Image getSourceCover(String source) throws IOException {
167 Image custom = getCustomSourceCover(source);
168 if (custom != null) {
169 return custom;
170 }
171
172 Image cached = cacheLib.getSourceCover(source);
173 if (cached != null) {
174 return cached;
175 }
176
177 return lib.getSourceCover(source);
178 }
179
180 @Override
181 public Image getAuthorCover(String author) throws IOException {
182 Image custom = getCustomAuthorCover(author);
183 if (custom != null) {
184 return custom;
185 }
186
187 Image cached = cacheLib.getAuthorCover(author);
188 if (cached != null) {
189 return cached;
190 }
191
192 return lib.getAuthorCover(author);
193 }
194
195 @Override
196 public Image getCustomSourceCover(String source) throws IOException {
197 Image custom = cacheLib.getCustomSourceCover(source);
198 if (custom == null) {
199 custom = lib.getCustomSourceCover(source);
200 if (custom != null) {
201 cacheLib.setSourceCover(source, custom);
202 }
203 }
204
205 return custom;
206 }
207
208 @Override
209 public Image getCustomAuthorCover(String author) throws IOException {
210 Image custom = cacheLib.getCustomAuthorCover(author);
211 if (custom == null) {
212 custom = lib.getCustomAuthorCover(author);
213 if (custom != null) {
214 cacheLib.setAuthorCover(author, custom);
215 }
216 }
217
218 return custom;
219 }
220
221 @Override
222 public void setSourceCover(String source, String luid) throws IOException {
223 lib.setSourceCover(source, luid);
224 cacheLib.setSourceCover(source, getCover(luid));
225 }
226
227 @Override
228 public void setAuthorCover(String author, String luid) throws IOException {
229 lib.setAuthorCover(author, luid);
230 cacheLib.setAuthorCover(author, getCover(luid));
231 }
232
233 /**
234 * Invalidate the {@link Story} cache (when the content has changed, but we
235 * already have it) with the new given meta.
236 * <p>
237 * <b>Make sure to always use {@link MetaData} from the cached library in
238 * priority, here.</b>
239 *
240 * @param meta
241 * the {@link Story} to clear from the cache
242 *
243 * @throws IOException
244 * in case of IOException
245 */
246 @Override
247 @Deprecated
248 protected void updateInfo(MetaData meta) throws IOException {
249 throw new IOException(
250 "This method is not supported in a CacheLibrary, please use updateMetaCache");
251 }
252
253 // relplace the meta in Metas by Meta, add it if needed
254 // return TRUE = added
255 private boolean updateMetaCache(List<MetaData> metas, MetaData meta) {
256 if (meta != null && metas != null) {
257 synchronized (metasLock) {
258 boolean changed = false;
259 for (int i = 0; i < metas.size(); i++) {
260 if (metas.get(i).getLuid().equals(meta.getLuid())) {
261 metas.set(i, meta);
262 changed = true;
263 }
264 }
265
266 if (!changed) {
267 metas.add(meta);
268 return true;
269 }
270 }
271 }
272
273 return false;
274 }
275
276 @Override
277 protected void invalidateInfo(String luid) {
278 if (luid == null) {
279 synchronized (metasLock) {
280 metasReal = null;
281 metasMixed = null;
282 }
283 } else {
284 invalidateInfo(metasReal, luid);
285 invalidateInfo(metasMixed, luid);
286 }
287
288 cacheLib.invalidateInfo(luid);
289 lib.invalidateInfo(luid);
290 }
291
292 // luid cannot be null
293 private void invalidateInfo(List<MetaData> metas, String luid) {
294 if (metas != null) {
295 synchronized (metasLock) {
296 for (int i = 0; i < metas.size(); i++) {
297 if (metas.get(i).getLuid().equals(luid)) {
298 metas.remove(i--);
299 }
300 }
301 }
302 }
303 }
304
305 @Override
306 public synchronized Story save(Story story, String luid, Progress pg)
307 throws IOException {
308 Progress pgLib = new Progress();
309 Progress pgCacheLib = new Progress();
310
311 if (pg == null) {
312 pg = new Progress();
313 }
314
315 pg.setMinMax(0, 2);
316 pg.addProgress(pgLib, 1);
317 pg.addProgress(pgCacheLib, 1);
318
319 story = lib.save(story, luid, pgLib);
320 updateMetaCache(metasReal, story.getMeta());
321
322 story = cacheLib.save(story, story.getMeta().getLuid(), pgCacheLib);
323 updateMetaCache(metasMixed, story.getMeta());
324
325 return story;
326 }
327
328 @Override
329 public void delete(String luid) throws IOException {
330 if (isCached(luid)) {
331 cacheLib.delete(luid);
332 }
333 lib.delete(luid);
334
335 invalidateInfo(luid);
336 }
337
338 @Override
339 protected synchronized void changeSTA(String luid, String newSource,
340 String newTitle, String newAuthor, Progress pg) throws IOException {
341 if (pg == null) {
342 pg = new Progress();
343 }
344
345 Progress pgCache = new Progress();
346 Progress pgOrig = new Progress();
347 pg.setMinMax(0, 2);
348 pg.addProgress(pgCache, 1);
349 pg.addProgress(pgOrig, 1);
350
351 MetaData meta = getInfo(luid);
352 if (meta == null) {
353 throw new IOException("Story not found: " + luid);
354 }
355
356 if (isCached(luid)) {
357 cacheLib.changeSTA(luid, newSource, newTitle, newAuthor, pgCache);
358 }
359 pgCache.done();
360
361 lib.changeSTA(luid, newSource, newTitle, newAuthor, pgOrig);
362 pgOrig.done();
363
364 meta.setSource(newSource);
365 meta.setTitle(newTitle);
366 meta.setAuthor(newAuthor);
367 pg.done();
368
369 if (isCached(luid)) {
370 updateMetaCache(metasMixed, meta);
371 updateMetaCache(metasReal, lib.getInfo(luid));
372 } else {
373 updateMetaCache(metasReal, meta);
374 }
375 }
376
377 @Override
378 public boolean isCached(String luid) {
379 try {
380 return cacheLib.getInfo(luid) != null;
381 } catch (IOException e) {
382 return false;
383 }
384 }
385
386 @Override
387 public void clearFromCache(String luid) throws IOException {
388 if (isCached(luid)) {
389 cacheLib.delete(luid);
390 }
391 }
392
393 @Override
394 public MetaData imprt(URL url, Progress pg) throws IOException {
395 if (pg == null) {
396 pg = new Progress();
397 }
398
399 Progress pgImprt = new Progress();
400 Progress pgCache = new Progress();
401 pg.setMinMax(0, 10);
402 pg.addProgress(pgImprt, 7);
403 pg.addProgress(pgCache, 3);
404
405 MetaData meta = lib.imprt(url, pgImprt);
406 updateMetaCache(metasReal, meta);
407 synchronized (metasLock) {
408 metasMixed = null;
409 }
410
411 clearFromCache(meta.getLuid());
412
413 pg.done();
414 return meta;
415 }
416
417 // All the following methods are only used by Save and Delete in
418 // BasicLibrary:
419
420 @Override
421 protected int getNextId() {
422 throw new java.lang.InternalError("Should not have been called");
423 }
424
425 @Override
426 protected void doDelete(String luid) throws IOException {
427 throw new java.lang.InternalError("Should not have been called");
428 }
429
430 @Override
431 protected Story doSave(Story story, Progress pg) throws IOException {
432 throw new java.lang.InternalError("Should not have been called");
433 }
434 }