nl.siegmann.epublib.bookprocessor.HtmlBookProcessor Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of epublib-tools Show documentation

A java library for reading/writing/manipulating epub files

The newest version!

package nl.siegmann.epublib.bookprocessor;

import java.io.IOException;

import nl.siegmann.epublib.Constants;
import nl.siegmann.epublib.domain.Book;
import nl.siegmann.epublib.domain.Resource;
import nl.siegmann.epublib.epub.BookProcessor;
import nl.siegmann.epublib.service.MediatypeService;
import org.apache.commons.io.FileUtils;
import org.apache.commons.io.FilenameUtils;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

/**
 * Helper class for BookProcessors that only manipulate html type resources.
 *
 * @author paul
 *
 */
public abstract class HtmlBookProcessor implements BookProcessor {

    private final static Logger log = LoggerFactory.getLogger(HtmlBookProcessor.class);
    public static final String OUTPUT_ENCODING = "UTF-8";

    public HtmlBookProcessor() {
    }

    @Override
    public Book processBook(Book book) {
        for (Resource resource : book.getResources().getAll()) {
            try {
                cleanupResource(resource, book);
            } catch (IOException e) {
                log.error(e.getMessage(), e);
            }
        }
        return book;
    }

    private void cleanupResource(Resource resource, Book book) throws IOException {
        if (resource.getMediaType() == MediatypeService.XHTML) {         
            if (FilenameUtils.getExtension(resource.getHref()).equals("html")) {
                byte[] cleanedHtml = processHtml(resource, book, Constants.CHARACTER_ENCODING);
                resource.setData(cleanedHtml);
                resource.setInputEncoding(Constants.CHARACTER_ENCODING);
                
                String filename = resource.getHref();
                filename = FilenameUtils.removeExtension(filename);
                filename += ".xhtml";
                resource.setHref(filename);
            }
        }
    }

    protected abstract byte[] processHtml(Resource resource, Book book, String encoding) throws IOException;
}