mirror of
https://github.com/geometer/FBReaderJ.git
synced 2025-10-05 10:49:24 +02:00
347 lines
10 KiB
Java
347 lines
10 KiB
Java
/*
|
|
* Copyright (C) 2007-2013 Geometer Plus <contact@geometerplus.com>
|
|
*
|
|
* This program is free software; you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation; either version 2 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program; if not, write to the Free Software
|
|
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
|
|
* 02110-1301, USA.
|
|
*/
|
|
|
|
package org.geometerplus.fbreader.library;
|
|
|
|
import java.util.*;
|
|
|
|
import org.geometerplus.zlibrary.core.filesystem.*;
|
|
import org.geometerplus.zlibrary.core.resources.ZLResource;
|
|
|
|
import org.geometerplus.fbreader.book.*;
|
|
import org.geometerplus.fbreader.tree.FBTree;
|
|
|
|
public final class Library {
|
|
public static ZLResource resource() {
|
|
return ZLResource.resource("library");
|
|
}
|
|
|
|
public static final String ROOT_FOUND = "found";
|
|
public static final String ROOT_FAVORITES = "favorites";
|
|
public static final String ROOT_RECENT = "recent";
|
|
public static final String ROOT_BY_AUTHOR = "byAuthor";
|
|
public static final String ROOT_BY_TITLE = "byTitle";
|
|
public static final String ROOT_BY_SERIES = "bySeries";
|
|
public static final String ROOT_BY_TAG = "byTag";
|
|
public static final String ROOT_FILE_TREE = "fileTree";
|
|
|
|
public static final int REMOVE_DONT_REMOVE = 0x00;
|
|
public static final int REMOVE_FROM_LIBRARY = 0x01;
|
|
public static final int REMOVE_FROM_DISK = 0x02;
|
|
public static final int REMOVE_FROM_LIBRARY_AND_DISK = REMOVE_FROM_LIBRARY | REMOVE_FROM_DISK;
|
|
|
|
private final List<ChangeListener> myListeners = Collections.synchronizedList(new LinkedList<ChangeListener>());
|
|
|
|
public interface ChangeListener {
|
|
public enum Code {
|
|
BookAdded,
|
|
BookRemoved,
|
|
StatusChanged,
|
|
Found,
|
|
NotFound
|
|
}
|
|
|
|
void onLibraryChanged(Code code);
|
|
}
|
|
|
|
public void addChangeListener(ChangeListener listener) {
|
|
myListeners.add(listener);
|
|
}
|
|
|
|
public void removeChangeListener(ChangeListener listener) {
|
|
myListeners.remove(listener);
|
|
}
|
|
|
|
protected void fireModelChangedEvent(ChangeListener.Code code) {
|
|
synchronized (myListeners) {
|
|
for (ChangeListener l : myListeners) {
|
|
l.onLibraryChanged(code);
|
|
}
|
|
}
|
|
}
|
|
|
|
public final IBookCollection Collection;
|
|
private final Map<Long,Book> myBooks = Collections.synchronizedMap(new HashMap<Long,Book>());
|
|
|
|
private final RootTree myRootTree = new RootTree();
|
|
private boolean myDoGroupTitlesByFirstLetter;
|
|
|
|
private final static int STATUS_LOADING = 1;
|
|
private final static int STATUS_SEARCHING = 2;
|
|
private volatile int myStatusMask = 0;
|
|
private final Object myStatusLock = new Object();
|
|
|
|
private void setStatus(int status) {
|
|
synchronized (myStatusLock) {
|
|
if (myStatusMask != status) {
|
|
myStatusMask = status;
|
|
fireModelChangedEvent(ChangeListener.Code.StatusChanged);
|
|
}
|
|
}
|
|
}
|
|
|
|
public Library(IBookCollection collection) {
|
|
Collection = collection;
|
|
|
|
new FavoritesTree(collection, myRootTree, ROOT_FAVORITES);
|
|
new RecentBooksTree(collection, myRootTree, ROOT_RECENT);
|
|
new FirstLevelTree(myRootTree, ROOT_BY_AUTHOR);
|
|
new FirstLevelTree(myRootTree, ROOT_BY_TITLE);
|
|
new FirstLevelTree(myRootTree, ROOT_BY_TAG);
|
|
new FileFirstLevelTree(collection, myRootTree, ROOT_FILE_TREE);
|
|
}
|
|
|
|
public void init() {
|
|
Collection.addListener(new BookCollection.Listener() {
|
|
public void onBookEvent(BookEvent event, Book book) {
|
|
switch (event) {
|
|
case Added:
|
|
addBookToLibrary(book);
|
|
synchronized (myStatusLock) {
|
|
if ((myStatusMask & STATUS_LOADING) == 0 ||
|
|
Collection.size() % 16 == 0) {
|
|
Library.this.fireModelChangedEvent(ChangeListener.Code.BookAdded);
|
|
}
|
|
}
|
|
break;
|
|
}
|
|
}
|
|
|
|
public void onBuildEvent(BuildEvent event) {
|
|
switch (event) {
|
|
case Started:
|
|
//setStatus(myStatusMask | STATUS_LOADING);
|
|
break;
|
|
case Completed:
|
|
Library.this.fireModelChangedEvent(ChangeListener.Code.BookAdded);
|
|
//setStatus(myStatusMask & ~STATUS_LOADING);
|
|
break;
|
|
}
|
|
}
|
|
});
|
|
|
|
final Thread initializer = new Thread() {
|
|
public void run() {
|
|
setStatus(myStatusMask | STATUS_LOADING);
|
|
int count = 0;
|
|
for (Book book : Collection.books()) {
|
|
addBookToLibrary(book);
|
|
if (++count % 16 == 0) {
|
|
Library.this.fireModelChangedEvent(ChangeListener.Code.BookAdded);
|
|
}
|
|
}
|
|
Library.this.fireModelChangedEvent(ChangeListener.Code.BookAdded);
|
|
setStatus(myStatusMask & ~STATUS_LOADING);
|
|
}
|
|
};
|
|
initializer.setPriority((Thread.MIN_PRIORITY + Thread.NORM_PRIORITY) / 2);
|
|
initializer.start();
|
|
}
|
|
|
|
public LibraryTree getRootTree() {
|
|
return myRootTree;
|
|
}
|
|
|
|
private FirstLevelTree getFirstLevelTree(String key) {
|
|
return (FirstLevelTree)myRootTree.getSubTree(key);
|
|
}
|
|
|
|
public LibraryTree getLibraryTree(LibraryTree.Key key) {
|
|
if (key == null) {
|
|
return null;
|
|
}
|
|
if (key.Parent == null) {
|
|
return key.Id.equals(myRootTree.getUniqueKey().Id) ? myRootTree : null;
|
|
}
|
|
final LibraryTree parentTree = getLibraryTree(key.Parent);
|
|
return parentTree != null ? (LibraryTree)parentTree.getSubTree(key.Id) : null;
|
|
}
|
|
|
|
private final List<?> myNullList = Collections.singletonList(null);
|
|
|
|
private LibraryTree getTagTree(Tag tag) {
|
|
if (tag == null || tag.Parent == null) {
|
|
return getFirstLevelTree(ROOT_BY_TAG).getTagSubTree(tag);
|
|
} else {
|
|
return getTagTree(tag.Parent).getTagSubTree(tag);
|
|
}
|
|
}
|
|
|
|
private synchronized void addBookToLibrary(Book book) {
|
|
synchronized (myBooks) {
|
|
final Book existing = myBooks.get(book.getId());
|
|
if (existing != null) {
|
|
existing.updateFrom(book);
|
|
return;
|
|
}
|
|
myBooks.put(book.getId(), book);
|
|
}
|
|
|
|
List<Author> authors = book.authors();
|
|
if (authors.isEmpty()) {
|
|
authors = (List<Author>)myNullList;
|
|
}
|
|
final SeriesInfo seriesInfo = book.getSeriesInfo();
|
|
for (Author a : authors) {
|
|
final AuthorTree authorTree = getFirstLevelTree(ROOT_BY_AUTHOR).getAuthorSubTree(a);
|
|
if (seriesInfo == null) {
|
|
authorTree.getBookSubTree(book, false);
|
|
} else {
|
|
authorTree.getSeriesSubTree(seriesInfo.Title).getBookInSeriesSubTree(book);
|
|
}
|
|
}
|
|
|
|
if (seriesInfo != null) {
|
|
FirstLevelTree seriesRoot = getFirstLevelTree(ROOT_BY_SERIES);
|
|
if (seriesRoot == null) {
|
|
seriesRoot = new FirstLevelTree(
|
|
myRootTree,
|
|
myRootTree.indexOf(getFirstLevelTree(ROOT_BY_TITLE)) + 1,
|
|
ROOT_BY_SERIES
|
|
);
|
|
}
|
|
seriesRoot.getSeriesSubTree(seriesInfo.Title).getBookInSeriesSubTree(book);
|
|
}
|
|
|
|
if (myDoGroupTitlesByFirstLetter) {
|
|
final String letter = TitleTree.firstTitleLetter(book);
|
|
if (letter != null) {
|
|
final TitleTree tree =
|
|
getFirstLevelTree(ROOT_BY_TITLE).getTitleSubTree(letter);
|
|
tree.getBookSubTree(book, true);
|
|
}
|
|
} else {
|
|
getFirstLevelTree(ROOT_BY_TITLE).getBookSubTree(book, true);
|
|
}
|
|
|
|
List<Tag> tags = book.tags();
|
|
if (tags.isEmpty()) {
|
|
tags = (List<Tag>)myNullList;
|
|
}
|
|
for (Tag t : tags) {
|
|
getTagTree(t).getBookSubTree(book, true);
|
|
}
|
|
|
|
synchronized (this) {
|
|
final SearchResultsTree found = (SearchResultsTree)getFirstLevelTree(ROOT_FOUND);
|
|
if (found != null && book.matches(found.getPattern())) {
|
|
found.getBookSubTree(book, true);
|
|
}
|
|
}
|
|
}
|
|
|
|
private void removeFromTree(String rootId, Book book) {
|
|
final FirstLevelTree tree = getFirstLevelTree(rootId);
|
|
if (tree != null) {
|
|
tree.removeBook(book, false);
|
|
}
|
|
}
|
|
|
|
public synchronized void refreshBookInfo(Book book) {
|
|
if (book == null) {
|
|
return;
|
|
}
|
|
|
|
Collection.saveBook(book, true);
|
|
myBooks.remove(book.getId());
|
|
removeFromTree(ROOT_FOUND, book);
|
|
removeFromTree(ROOT_BY_TITLE, book);
|
|
removeFromTree(ROOT_BY_SERIES, book);
|
|
removeFromTree(ROOT_BY_AUTHOR, book);
|
|
removeFromTree(ROOT_BY_TAG, book);
|
|
addBookToLibrary(book);
|
|
fireModelChangedEvent(ChangeListener.Code.BookAdded);
|
|
}
|
|
|
|
public boolean isUpToDate() {
|
|
return myStatusMask == 0;
|
|
}
|
|
|
|
public void startBookSearch(final String pattern) {
|
|
setStatus(myStatusMask | STATUS_SEARCHING);
|
|
final Thread searcher = new Thread("Library.searchBooks") {
|
|
public void run() {
|
|
try {
|
|
searchBooks(pattern);
|
|
} finally {
|
|
setStatus(myStatusMask & ~STATUS_SEARCHING);
|
|
}
|
|
}
|
|
};
|
|
searcher.setPriority((Thread.MIN_PRIORITY + Thread.NORM_PRIORITY) / 2);
|
|
searcher.start();
|
|
}
|
|
|
|
private void searchBooks(String pattern) {
|
|
if (pattern == null) {
|
|
fireModelChangedEvent(ChangeListener.Code.NotFound);
|
|
return;
|
|
}
|
|
|
|
pattern = pattern.toLowerCase();
|
|
|
|
final SearchResultsTree oldSearchResults = (SearchResultsTree)getFirstLevelTree(ROOT_FOUND);
|
|
if (oldSearchResults != null && pattern.equals(oldSearchResults.getPattern())) {
|
|
fireModelChangedEvent(ChangeListener.Code.Found);
|
|
return;
|
|
}
|
|
|
|
FirstLevelTree newSearchResults = null;
|
|
synchronized (this) {
|
|
for (Book book : Collection.books(pattern)) {
|
|
if (newSearchResults == null) {
|
|
if (oldSearchResults != null) {
|
|
oldSearchResults.removeSelf();
|
|
}
|
|
newSearchResults = new SearchResultsTree(myRootTree, ROOT_FOUND, pattern);
|
|
fireModelChangedEvent(ChangeListener.Code.Found);
|
|
}
|
|
newSearchResults.getBookSubTree(book, true);
|
|
fireModelChangedEvent(ChangeListener.Code.BookAdded);
|
|
}
|
|
}
|
|
if (newSearchResults == null) {
|
|
fireModelChangedEvent(ChangeListener.Code.NotFound);
|
|
}
|
|
}
|
|
|
|
public boolean canRemoveBookFile(Book book) {
|
|
ZLFile file = book.File;
|
|
if (file.getPhysicalFile() == null) {
|
|
return false;
|
|
}
|
|
while (file instanceof ZLArchiveEntryFile) {
|
|
file = file.getParent();
|
|
if (file.children().size() != 1) {
|
|
return false;
|
|
}
|
|
}
|
|
return true;
|
|
}
|
|
|
|
public void removeBook(Book book, int removeMode) {
|
|
if (removeMode == REMOVE_DONT_REMOVE) {
|
|
return;
|
|
}
|
|
Collection.removeBook(book, (removeMode & REMOVE_FROM_DISK) != 0);
|
|
getFirstLevelTree(ROOT_RECENT).removeBook(book, false);
|
|
getFirstLevelTree(ROOT_FAVORITES).removeBook(book, false);
|
|
myRootTree.removeBook(book, true);
|
|
}
|
|
}
|