private boolean saveDocs() {
try {
// save split documents
for (Map.Entry<String, Document> entry : docList.entrySet()) {
if (!EpubUtils.saveDocument(
entry.getValue(),
new File(EpubUtils.EPUB_FOLDER + File.separator + entry.getKey()),
taskOutput)) {
return false;
}
}
return true;
} catch (Exception e) {
EpubUtils.outputMessage(
taskOutput, "Could not save documents. An Exception occurred: " + e.getMessage());
}
return false;
}
@Override
public Void doInBackground() {
if (!EpubUtils.start(taskOutput)) return null;
if (!EpubUtils.unzip(taskOutput)) return null;
if (!EpubUtils.canSplit(taskOutput)) return null;
if (!EpubUtils.backup(taskOutput)) return null;
EpubUtils.outputProcess("PREPARING AND PARSING", true, taskOutput);
// create package handler instance
packageHandler = new PackageHandler();
if (!EpubUtils.parseFile(
new File(EpubUtils.EPUB_FOLDER + File.separator + EpubUtils.PACKAGE_FILENAME),
packageHandler,
taskOutput)) return null;
docNumber = 0;
docList = new TreeMap<String, Document>();
ids = new HashMap<String, String>();
// get concat xhtml file from extracted zip file
listOfFiles = EpubUtils.getFiles(true, false);
// create split handler instance
splitHandler = new SplitHandler();
// prepare source file
if (!EpubUtils.prepareFile(listOfFiles[0], taskOutput)) return null;
// parse source file
if (!EpubUtils.parseFile(listOfFiles[0], splitHandler, taskOutput)) return null;
if (splitHandler.getMetaNodes().containsKey("dc:identifier")) {
idPrefix = splitHandler.getMetaNodes().get("dc:identifier");
}
EpubUtils.outputProcess("BUILDING SPLIT DOCUMENTS", true, taskOutput);
if (!splitConcatDocument(listOfFiles[0])) return null;
if (!mapRefs()) return null;
if (!saveDocs()) return null;
EpubUtils.outputProcess("MODIFYING PACKAGE DOCUMENT", true, taskOutput);
listOfFiles = EpubUtils.getFiles(false, true);
if (listOfFiles.length == 0) {
EpubUtils.outputMessage(taskOutput, "No split files found");
return null;
} else if (listOfFiles.length == 1) {
EpubUtils.outputMessage(
taskOutput, "Only one file found - concat file have not been splitted correctly");
return null;
}
Document packageDoc =
EpubUtils.createDocument(
new File(EpubUtils.EPUB_FOLDER + File.separator + EpubUtils.PACKAGE_FILENAME),
taskOutput);
if (packageDoc == null) {
return null;
}
for (int i = 0; i < listOfFiles.length; i++) {
// add split document to opf document
if (!EpubUtils.addOpfItem(packageDoc, listOfFiles[i].getName(), i + 1, taskOutput))
return null;
}
// remove concat document from opf document
if (!EpubUtils.removeOpfItem(packageDoc, EpubUtils.CONCAT_FILENAME, taskOutput)) return null;
// remove fallback from non xhtml spine elements
if (!EpubUtils.removeFallbackFromOpf(packageDoc, taskOutput)) return null;
// save opf document
if (!EpubUtils.saveDocument(
packageDoc,
new File(EpubUtils.EPUB_FOLDER + File.separator + EpubUtils.PACKAGE_FILENAME),
taskOutput)) return null;
EpubUtils.outputProcess("MODIFYING EPUB", true, taskOutput);
// obtain the global configuration
TConfig config = TConfig.get();
config.setArchiveDetector(
new TArchiveDetector("epub", new JarDriver(IOPoolLocator.SINGLETON)));
// get epub file destination
String epubPath = EpubUtils.EPUB.getPath();
String epubFolder =
EpubUtils.EPUB_FOLDER
.substring(EpubUtils.EPUB_FOLDER.lastIndexOf(File.separator))
.replace(File.separator, "");
TFile destination = new TFile(epubPath + File.separator + epubFolder);
// modify epub file destination
for (int i = 0; i < listOfFiles.length; i++) {
if (!EpubUtils.addFileToEpub(new TFile(listOfFiles[i]), destination, taskOutput))
return null;
}
if (!EpubUtils.addFileToEpub(
new TFile(EpubUtils.EPUB_FOLDER + File.separator + EpubUtils.PACKAGE_FILENAME),
destination,
taskOutput)) return null;
if (!EpubUtils.removeFileFromEpub(
new TFile(destination, EpubUtils.CONCAT_FILENAME), taskOutput)) return null;
// commit changes to epub file destination
if (!EpubUtils.commitChanges(taskOutput)) return null;
if (!EpubUtils.finish(taskOutput)) return null;
return null;
}