1 files changed, 280 insertions, 0 deletions
diff --git a/fbreader/src/library/Book.cpp b/fbreader/src/library/Book.cpp
new file mode 100644
index 0000000..ca3afd0
--- /dev/null
+++ b/fbreader/src/library/Book.cpp
@@ -0,0 +1,280 @@
+/*
+ * Copyright (C) 2009-2012 Geometer Plus <[email protected]>
+ *
+ * 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.
+ */
+
+#include <algorithm>
+#include <set>
+
+#include <ZLStringUtil.h>
+#include <ZLFile.h>
+#include <ZLLanguageList.h>
+
+#include "Book.h"
+#include "Author.h"
+#include "Tag.h"
+
+#include "../formats/FormatPlugin.h"
+#include "../migration/BookInfo.h"
+
+const std::string Book::AutoEncoding = "auto";
+
+Book::Book(const ZLFile &file, int id) : myBookId(id), myFile(file) {
+}
+
+Book::~Book() {
+}
+
+shared_ptr<Book> Book::createBook(
+	const ZLFile &file,
+	int id,
+	const std::string &encoding,
+	const std::string &language,
+	const std::string &title
+) {
+	Book *book = new Book(file, id);
+	book->setEncoding(encoding);
+	book->setLanguage(language);
+	book->setTitle(title);
+	return book;
+}
+
+shared_ptr<Book> Book::loadFromFile(const ZLFile &file) {
+	shared_ptr<FormatPlugin> plugin = PluginCollection::Instance().plugin(file, false);
+	if (plugin.isNull()) {
+		return 0;
+	}
+
+	shared_ptr<Book> book = new Book(file, 0);
+	if (!plugin->readMetaInfo(*book)) {
+		return 0;
+	}
+	plugin->readLanguageAndEncoding(*book);
+
+	if (book->title().empty()) {
+		book->setTitle(ZLFile::fileNameToUtf8(file.name(true)));
+	}
+
+	if (book->encoding().empty()) {
+		book->setEncoding(AutoEncoding);
+	}
+
+	if (book->language().empty()) {
+		book->setLanguage(PluginCollection::Instance().DefaultLanguageOption.value());
+	}
+
+	return book;
+}
+
+bool Book::addTag(shared_ptr<Tag> tag) {
+	if (tag.isNull()) {
+		return false;
+	}
+	TagList::const_iterator it = std::find(myTags.begin(), myTags.end(), tag);
+	if (it == myTags.end()) {
+		myTags.push_back(tag);
+		return true;
+	}
+	return false;
+}
+
+bool Book::addTag(const std::string &fullName) {
+	return addTag(Tag::getTagByFullName(fullName));
+}
+
+bool Book::removeTag(shared_ptr<Tag> tag, bool includeSubTags) {
+	bool changed = false;
+	for (TagList::iterator it = myTags.begin(); it != myTags.end();) {
+		if (tag == *it || (includeSubTags && tag->isAncestorOf(*it))) {
+			it = myTags.erase(it);
+			changed = true;
+		} else {
+			++it;
+		}
+	}
+	return changed;
+}
+
+bool Book::renameTag(shared_ptr<Tag> from, shared_ptr<Tag> to, bool includeSubTags) {
+	if (includeSubTags) {
+		std::set<shared_ptr<Tag> > tagSet;
+		bool changed = false;
+		for (TagList::const_iterator it = myTags.begin(); it != myTags.end(); ++it) {
+			if (*it == from) {
+				tagSet.insert(to);
+				changed = true;
+			} else {
+				shared_ptr<Tag> newtag = Tag::cloneSubTag(*it, from, to);
+				if (newtag.isNull()) {
+					tagSet.insert(*it);
+				} else {
+					tagSet.insert(newtag);
+					changed = true;
+				}
+			}
+		}
+		if (changed) {
+			myTags.clear();
+			myTags.insert(myTags.end(), tagSet.begin(), tagSet.end());
+			return true;
+		}
+	} else {
+		TagList::iterator it = std::find(myTags.begin(), myTags.end(), from);
+		if (it != myTags.end()) {
+			TagList::const_iterator jt = std::find(myTags.begin(), myTags.end(), to);
+			if (jt == myTags.end()) {
+				*it = to;
+			} else {
+				myTags.erase(it);
+			}
+			return true;
+		}
+	}
+	return false;
+}
+
+bool Book::cloneTag(shared_ptr<Tag> from, shared_ptr<Tag> to, bool includeSubTags) {
+	if (includeSubTags) {
+		std::set<shared_ptr<Tag> > tagSet;
+		for (TagList::const_iterator it = myTags.begin(); it != myTags.end(); ++it) {
+			if (*it == from) {
+				tagSet.insert(to);
+			} else {
+				shared_ptr<Tag> newtag = Tag::cloneSubTag(*it, from, to);
+				if (!newtag.isNull()) {
+					tagSet.insert(newtag);
+				}
+			}
+		}
+		if (!tagSet.empty()) {
+			tagSet.insert(myTags.begin(), myTags.end());
+			myTags.clear();
+			myTags.insert(myTags.end(), tagSet.begin(), tagSet.end());
+			return true;
+		}
+	} else {
+		TagList::const_iterator it = std::find(myTags.begin(), myTags.end(), from);
+		if (it != myTags.end()) {
+			TagList::const_iterator jt = std::find(myTags.begin(), myTags.end(), to);
+			if (jt == myTags.end()) {
+				myTags.push_back(to);
+				return true;
+			}
+		}
+	}
+	return false;
+}
+
+shared_ptr<Book> Book::loadFromBookInfo(const ZLFile &file) {
+	BookInfo info(file.path());
+
+	shared_ptr<Book> book = createBook(
+		file, 0,
+		info.EncodingOption.value(),
+		info.LanguageOption.value(),
+		info.TitleOption.value()
+	);
+
+	book->setSeries(
+		info.SeriesTitleOption.value(),
+		Number(info.IndexInSeriesOption.value())
+	);
+
+	if (book->language().empty()) {
+		book->setLanguage(PluginCollection::Instance().DefaultLanguageOption.value());
+	}
+
+	const std::string &tagList = info.TagsOption.value();
+	if (!tagList.empty()) {
+		std::size_t index = 0;
+		do {
+			std::size_t newIndex = tagList.find(',', index);
+			book->addTag(Tag::getTagByFullName(tagList.substr(index, newIndex - index)));
+			index = newIndex + 1;
+		} while (index != 0);
+	}
+
+	const std::string &authorList = info.AuthorDisplayNameOption.value();
+	if (!authorList.empty()) {
+		std::size_t index = 0;
+		do {
+			std::size_t newIndex = authorList.find(',', index);
+			book->addAuthor(authorList.substr(index, newIndex - index));
+			index = newIndex + 1;
+		} while (index != 0);
+	}
+
+	return book;
+}
+
+bool Book::replaceAuthor(shared_ptr<Author> from, shared_ptr<Author> to) {
+	AuthorList::iterator it = std::find(myAuthors.begin(), myAuthors.end(), from);
+	if (it == myAuthors.end()) {
+		return false;
+	}
+	if (to.isNull()) {
+		myAuthors.erase(it);
+	} else {
+		*it = to;
+	}
+	return true;
+}
+
+void Book::setTitle(const std::string &title) {
+	myTitle = title;
+}
+
+void Book::setLanguage(const std::string &language) {
+	if (!myLanguage.empty()) {
+		const std::vector<std::string> &codes = ZLLanguageList::languageCodes();
+		std::vector<std::string>::const_iterator it =
+			std::find(codes.begin(), codes.end(), myLanguage);
+		std::vector<std::string>::const_iterator jt =
+			std::find(codes.begin(), codes.end(), language);
+		if (it != codes.end() && jt == codes.end()) {
+			return;
+		}
+	}
+	myLanguage = language;
+}
+
+void Book::setEncoding(const std::string &encoding) {
+	myEncoding = encoding;
+}
+
+void Book::setSeries(const std::string &title, const Number &index) {
+	mySeriesTitle = title;
+	myIndexInSeries = index;
+}
+
+void Book::removeAllTags() {
+	myTags.clear();
+}
+
+void Book::addAuthor(const std::string &displayName, const std::string &sortKey) {
+	addAuthor(Author::getAuthor(displayName, sortKey));
+}
+
+void Book::addAuthor(shared_ptr<Author> author) {
+	if (!author.isNull()) {
+		myAuthors.push_back(author);
+	}
+}
+
+void Book::removeAllAuthors() {
+	myAuthors.clear();
+}