+++ /dev/null
-package be.nikiroo.fanfix.supported;
-
-import java.io.File;
-import java.io.IOException;
-import java.io.InputStream;
-import java.net.URISyntaxException;
-import java.net.URL;
-import java.net.URLDecoder;
-import java.util.ArrayList;
-import java.util.zip.ZipEntry;
-import java.util.zip.ZipInputStream;
-
-import org.jsoup.nodes.Document;
-
-import be.nikiroo.fanfix.Instance;
-import be.nikiroo.fanfix.data.MetaData;
-import be.nikiroo.utils.IOUtils;
-import be.nikiroo.utils.Image;
-import be.nikiroo.utils.StringUtils;
-import be.nikiroo.utils.streams.MarkableFileInputStream;
-
-/**
- * Support class for EPUB files created with this program (as we need some
- * metadata available in those we create).
- *
- * @author niki
- */
-class Epub extends InfoText {
- private MetaData meta;
- private File tmpDir;
- private String desc;
-
- private URL fakeSource;
- private InputStream fakeIn;
-
- public File getSourceFileOriginal() {
- return super.getSourceFile();
- }
-
- @Override
- protected File getSourceFile() {
- try {
- return new File(fakeSource.toURI());
- } catch (URISyntaxException e) {
- Instance.getTraceHandler()
- .error(new IOException(
- "Cannot get the source file from the info-text URL",
- e));
- }
-
- return null;
- }
-
- @Override
- protected InputStream getInput() {
- if (fakeIn != null) {
- try {
- fakeIn.reset();
- } catch (IOException e) {
- Instance.getTraceHandler()
- .error(new IOException(
- "Cannot reset the Epub Text stream", e));
- }
-
- return fakeIn;
- }
-
- return null;
- }
-
- @Override
- protected boolean supports(URL url) {
- return url.getPath().toLowerCase().endsWith(".epub");
- }
-
- @Override
- protected MetaData getMeta() throws IOException {
- return meta;
- }
-
- @Override
- protected Document loadDocument(URL source) throws IOException {
- super.loadDocument(source); // prepares super.getSourceFile() and
- // super.getInput()
-
- InputStream in = super.getInput();
- ZipInputStream zipIn = null;
- try {
- zipIn = new ZipInputStream(in);
- tmpDir = Instance.getTempFiles().createTempDir(
- "fanfic-reader-parser");
- File tmp = new File(tmpDir, "file.txt");
- File tmpInfo = new File(tmpDir, "file.info");
-
- fakeSource = tmp.toURI().toURL();
- Image cover = null;
-
- String url;
- try {
- url = getSource().toURI().toURL().toString();
- } catch (URISyntaxException e1) {
- url = getSource().toString();
- }
- String title = null;
- String author = null;
-
- for (ZipEntry entry = zipIn.getNextEntry(); entry != null; entry = zipIn
- .getNextEntry()) {
- if (!entry.isDirectory()
- && entry.getName().startsWith(getDataPrefix())) {
- String entryLName = entry.getName().toLowerCase();
-
- boolean imageEntry = false;
- for (String ext : bsImages.getImageExt(false)) {
- if (entryLName.endsWith(ext)) {
- imageEntry = true;
- }
- }
-
- if (entry.getName().equals(getDataPrefix() + "version")) {
- // Nothing to do for now ("first"
- // version is 3.0)
- } else if (entryLName.endsWith(".info")) {
- // Info file
- IOUtils.write(zipIn, tmpInfo);
- } else if (imageEntry) {
- // Cover
- if (getCover()) {
- try {
- cover = new Image(zipIn);
- } catch (Exception e) {
- Instance.getTraceHandler().error(e);
- }
- }
- } else if (entry.getName().equals(getDataPrefix() + "URL")) {
- String[] descArray = StringUtils
- .unhtml(IOUtils.readSmallStream(zipIn)).trim()
- .split("\n");
- if (descArray.length > 0) {
- url = descArray[0].trim();
- }
- } else if (entry.getName().equals(
- getDataPrefix() + "SUMMARY")) {
- String[] descArray = StringUtils
- .unhtml(IOUtils.readSmallStream(zipIn)).trim()
- .split("\n");
- int skip = 0;
- if (descArray.length > 1) {
- title = descArray[0].trim();
- skip = 1;
- if (descArray.length > 2
- && descArray[1].startsWith("©")) {
- author = descArray[1].substring(1).trim();
- skip = 2;
- }
- }
- this.desc = "";
- for (int i = skip; i < descArray.length; i++) {
- this.desc += descArray[i].trim() + "\n";
- }
-
- this.desc = this.desc.trim();
- } else {
- // Hopefully the data file
- IOUtils.write(zipIn, tmp);
- }
- }
- }
-
- if (requireInfo() && (!tmp.exists() || !tmpInfo.exists())) {
- throw new IOException(
- "file not supported (maybe not created with this program or corrupt)");
- }
-
- if (tmp.exists()) {
- this.fakeIn = new MarkableFileInputStream(tmp);
- }
-
- if (tmpInfo.exists()) {
- meta = InfoReader.readMeta(tmpInfo, true);
- tmpInfo.delete();
- } else {
- if (title == null || title.isEmpty()) {
- title = getSourceFileOriginal().getName();
- if (title.toLowerCase().endsWith(".cbz")) {
- title = title.substring(0, title.length() - 4);
- }
- title = URLDecoder.decode(title, "UTF-8").trim();
- }
-
- meta = new MetaData();
- meta.setLang("en");
- meta.setTags(new ArrayList<String>());
- meta.setSource(getType().getSourceName());
- meta.setUuid(url);
- meta.setUrl(url);
- meta.setTitle(title);
- meta.setAuthor(author);
- meta.setImageDocument(isImagesDocumentByDefault());
- }
-
- if (meta.getCover() == null) {
- if (cover != null) {
- meta.setCover(cover);
- } else {
- meta.setCover(InfoReader
- .getCoverByName(getSourceFileOriginal().toURI()
- .toURL()));
- }
- }
- } finally {
- if (zipIn != null) {
- zipIn.close();
- }
- if (in != null) {
- in.close();
- }
- }
-
- return null;
- }
-
- @Override
- protected void close() {
- if (tmpDir != null) {
- IOUtils.deltree(tmpDir);
- }
-
- tmpDir = null;
-
- super.close();
- }
-
- protected String getDataPrefix() {
- return "DATA/";
- }
-
- protected boolean requireInfo() {
- return true;
- }
-
- protected boolean getCover() {
- return true;
- }
-
- protected boolean isImagesDocumentByDefault() {
- return false;
- }
-}
git://git.nikiroo.be / fanfix.git / blobdiff