mirror of https://github.com/kiwix/libkiwix.git
134 lines
4.0 KiB
C++
134 lines
4.0 KiB
C++
/*
|
|
* Copyright 2011 Emmanuel Engelhart <kelson@kiwix.org>
|
|
*
|
|
* 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 3 of the License, or
|
|
* 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 "manager.h"
|
|
#include <pugixml.hpp>
|
|
|
|
namespace kiwix {
|
|
|
|
/* Constructor */
|
|
Manager::Manager() {
|
|
}
|
|
|
|
/* Destructor */
|
|
Manager::~Manager() {
|
|
}
|
|
|
|
bool Manager::readFile(const string path) {
|
|
pugi::xml_document doc;
|
|
pugi::xml_parse_result result = doc.load_file(path.c_str());
|
|
|
|
if (result) {
|
|
pugi::xml_node libraryNode = doc.child("library");
|
|
library.current = libraryNode.attribute("current").value();
|
|
|
|
for (pugi::xml_node bookNode = libraryNode.child("book"); bookNode; bookNode = bookNode.next_sibling("book")) {
|
|
kiwix::Book book;
|
|
book.id = bookNode.attribute("id").value();
|
|
book.path = bookNode.attribute("path").value();
|
|
book.last = bookNode.attribute("last").value();
|
|
book.indexPath = bookNode.attribute("indexPath").value();
|
|
book.indexType = bookNode.attribute("indexType").value() == "xapian" ? XAPIAN: CLUCENE;
|
|
book.title = bookNode.attribute("title").value();
|
|
book.description = bookNode.attribute("description").value();
|
|
book.language = bookNode.attribute("language").value();
|
|
book.date = bookNode.attribute("date").value();
|
|
book.creator = bookNode.attribute("creator").value();
|
|
library.addBook(book);
|
|
}
|
|
|
|
}
|
|
|
|
return result;
|
|
}
|
|
|
|
bool Manager::writeFile(const string path) {
|
|
pugi::xml_document doc;
|
|
|
|
/* Add the library node */
|
|
pugi::xml_node libraryNode = doc.append_child("library");
|
|
libraryNode.append_attribute("current") = library.current.c_str();
|
|
|
|
/* Add each book */
|
|
std::vector<kiwix::Book>::iterator itr;
|
|
for ( itr = library.books.begin(); itr != library.books.end(); ++itr ) {
|
|
pugi::xml_node bookNode = libraryNode.append_child("book");
|
|
bookNode.append_attribute("id") = itr->id.c_str();
|
|
bookNode.append_attribute("path") = itr->path.c_str();
|
|
|
|
if (itr->last != "")
|
|
bookNode.append_attribute("last") = itr->last.c_str();
|
|
|
|
if (itr->indexPath != "") {
|
|
bookNode.append_attribute("indexPath") = itr->indexPath.c_str();
|
|
if (itr->indexType == XAPIAN)
|
|
bookNode.append_attribute("indexType") = "xapian";
|
|
else
|
|
bookNode.append_attribute("indexType") = "clucene";
|
|
}
|
|
|
|
if (itr->title != "")
|
|
bookNode.append_attribute("title") = itr->title.c_str();
|
|
|
|
if (itr->description != "")
|
|
bookNode.append_attribute("description") = itr->description.c_str();
|
|
|
|
if (itr->language != "")
|
|
bookNode.append_attribute("language") = itr->language.c_str();
|
|
|
|
if (itr->date != "")
|
|
bookNode.append_attribute("date") = itr->date.c_str();
|
|
|
|
if (itr->creator != "")
|
|
bookNode.append_attribute("creator") = itr->creator.c_str();
|
|
|
|
}
|
|
|
|
/* saving file */
|
|
doc.save_file(path.c_str());
|
|
|
|
return true;
|
|
}
|
|
|
|
bool Manager::addBookFromPath(const string path) {
|
|
kiwix::Book book;
|
|
|
|
/* Open the ZIM file */
|
|
kiwix::Reader reader = kiwix::Reader(path);
|
|
book.path = path;
|
|
book.id = reader.getId();
|
|
book.title = reader.getTitle();
|
|
book.description = reader.getDescription();
|
|
book.language = reader.getLanguage();
|
|
book.date = reader.getDate();
|
|
book.creator = reader.getCreator();
|
|
library.addBook(book);
|
|
return true;
|
|
}
|
|
|
|
bool Manager::removeBookByIndex(const unsigned int bookIndex) {
|
|
return this->library.removeBookByIndex(bookIndex);
|
|
}
|
|
|
|
kiwix::Library Manager::cloneLibrary() {
|
|
return this->library;
|
|
}
|
|
|
|
}
|