X-Git-Url: http://git.nikiroo.be/?p=fanfix.git;a=blobdiff_plain;f=src%2Fbe%2Fnikiroo%2Ffanfix%2Foutput%2FEpub.java;h=fc2dc8c982127210c308d6a9dc70a20b667162eb;hp=869a0bc2647f3e7642ee458faf389684ed48186b;hb=HEAD;hpb=fa4dcafe32a95f725eb2573e42c7c2990cbeacd1 diff --git a/src/be/nikiroo/fanfix/output/Epub.java b/src/be/nikiroo/fanfix/output/Epub.java index 869a0bc..fc2dc8c 100644 --- a/src/be/nikiroo/fanfix/output/Epub.java +++ b/src/be/nikiroo/fanfix/output/Epub.java @@ -2,13 +2,14 @@ package be.nikiroo.fanfix.output; import java.io.BufferedWriter; import java.io.File; +import java.io.FileInputStream; import java.io.FileOutputStream; import java.io.IOException; import java.io.InputStream; +import java.io.OutputStream; import java.io.OutputStreamWriter; -import java.net.URL; - -import javax.imageio.ImageIO; +import java.util.zip.ZipEntry; +import java.util.zip.ZipOutputStream; import be.nikiroo.fanfix.Instance; import be.nikiroo.fanfix.bundles.Config; @@ -27,6 +28,7 @@ class Epub extends BasicOutput { private boolean inDialogue = false; private boolean inNormal = false; private File images; + private boolean nextParaIsCover = true; @Override public File process(Story story, File targetDir, String targetName) @@ -34,7 +36,7 @@ class Epub extends BasicOutput { String targetNameOrig = targetName; targetName += getDefaultExtension(false); - tmpDir = File.createTempFile("fanfic-reader-epub_", ".wip"); + tmpDir = Instance.getInstance().getTempFiles().createTempDir("fanfic-reader-epub"); tmpDir.delete(); if (!tmpDir.mkdir()) { @@ -42,20 +44,55 @@ class Epub extends BasicOutput { "Cannot create a temporary directory: no space left on device?"); } - // "Originals" - File data = new File(tmpDir, "DATA"); - data.mkdir(); - new InfoText().process(story, data, targetNameOrig); - InfoCover.writeInfo(data, targetNameOrig, story.getMeta()); - IOUtils.writeSmallFile(data, "version", "3.0"); - super.process(story, targetDir, targetNameOrig); - // zip/epub - File epub = new File(targetDir, targetName); - IOUtils.zip(tmpDir, epub, true); - IOUtils.deltree(tmpDir); - tmpDir = null; + File epub = null; + try { + // "Originals" + File data = new File(tmpDir, "DATA"); + data.mkdir(); + BasicOutput.getOutput(OutputType.TEXT, isWriteInfo(), + isWriteCover()).process(story, data, targetNameOrig); + InfoCover.writeInfo(data, targetNameOrig, story.getMeta()); + IOUtils.writeSmallFile(data, "version", "3.0"); + + // zip/epub + epub = new File(targetDir, targetName); + IOUtils.zip(tmpDir, epub, true); + + OutputStream out = new FileOutputStream(epub); + try { + ZipOutputStream zip = new ZipOutputStream(out); + try { + // "mimetype" MUST be the first element and not compressed + zip.setLevel(ZipOutputStream.STORED); + File mimetype = new File(tmpDir, "mimetype"); + IOUtils.writeSmallFile(tmpDir, "mimetype", + "application/epub+zip"); + ZipEntry entry = new ZipEntry("mimetype"); + entry.setExtra(new byte[] {}); + zip.putNextEntry(entry); + FileInputStream in = new FileInputStream(mimetype); + try { + IOUtils.write(in, zip); + } finally { + in.close(); + } + IOUtils.deltree(mimetype); + zip.setLevel(ZipOutputStream.DEFLATED); + // + + IOUtils.zip(zip, "", tmpDir, true); + } finally { + zip.close(); + } + } finally { + out.close(); + } + } finally { + IOUtils.deltree(tmpDir); + tmpDir = null; + } return epub; } @@ -76,11 +113,8 @@ class Epub extends BasicOutput { File metaInf = new File(tmpDir, "META-INF"); metaInf.mkdirs(); - // "root" - IOUtils.writeSmallFile(tmpDir, "mimetype", "application/epub+zip"); - // META-INF - String containerContent = "\n" + String containerContent = "\n" + "\n" + "\t\n" + "\t\t\n" @@ -101,16 +135,18 @@ class Epub extends BasicOutput { // OPS/images if (story.getMeta() != null && story.getMeta().getCover() != null) { - String format = Instance.getConfig() - .getString(Config.IMAGE_FORMAT_COVER).toLowerCase(); - File file = new File(images, "cover." + format); - ImageIO.write(story.getMeta().getCover(), format, file); + File file = new File(images, "cover"); + try { + Instance.getInstance().getCache().saveAsImage(story.getMeta().getCover(), file, true); + } catch (Exception e) { + Instance.getInstance().getTraceHandler().error(e); + } } // OPS/* except chapters IOUtils.writeSmallFile(ops, "epb.ncx", generateNcx(story)); IOUtils.writeSmallFile(ops, "epb.opf", generateOpf(story)); - IOUtils.writeSmallFile(ops, "title.xml", generateTitleXml(story)); + IOUtils.writeSmallFile(ops, "title.xhtml", generateTitleXml(story)); // Resume if (story.getMeta() != null && story.getMeta().getResume() != null) { @@ -121,7 +157,7 @@ class Epub extends BasicOutput { @Override protected void writeChapterHeader(Chapter chap) throws IOException { String filename = String.format("%s%03d%s", "chapter-", - chap.getNumber(), ".xml"); + chap.getNumber(), ".xhtml"); writer = new BufferedWriter(new OutputStreamWriter( new FileOutputStream(new File(tmpDir + File.separator + "OPS", filename)), "UTF-8")); @@ -135,9 +171,9 @@ class Epub extends BasicOutput { nameOrNum = chap.getName(); } - writer.write(""); - writer.write("\n"); - writer.write("\n"); + writer.append(""); + writer.append("\n"); + writer.append("\n"); writer.write("\n"); writer.write("\n " + StringUtils.xmlEscape(title) + ""); @@ -209,11 +245,13 @@ class Epub extends BasicOutput { break; case IMAGE: File file = new File(images, getCurrentImageBestName(false)); - Instance.getCache().saveAsImage(new URL(para.getContent()), file); - writer.write(" "); break; } + + nextParaIsCover = false; } @Override @@ -267,7 +305,7 @@ class Epub extends BasicOutput { title = meta.getTitle(); } - builder.append(""); + builder.append(""); builder.append("\n"); builder.append("\n"); @@ -295,7 +333,7 @@ class Epub extends BasicOutput { builder.append("\n "); builder.append("\n Title Page"); builder.append("\n "); - builder.append("\n "); + builder.append("\n "); builder.append("\n "); int navPoint = 2; // 1 is above @@ -318,11 +356,10 @@ class Epub extends BasicOutput { private void generateNcx(Chapter chap, StringBuilder builder, int navPoint) { String name; if (chap.getName() != null && !chap.getName().isEmpty()) { - name = Instance.getTrans().getString(StringId.CHAPTER_NAMED, - chap.getNumber(), chap.getName()); + name = Instance.getInstance().getTrans().getString(StringId.CHAPTER_NAMED, chap.getNumber(), + chap.getName()); } else { - name = Instance.getTrans().getString(StringId.CHAPTER_UNNAMED, - chap.getNumber()); + name = Instance.getInstance().getTrans().getString(StringId.CHAPTER_UNNAMED, chap.getNumber()); } String nnn = String.format("%03d", (navPoint - 2)); @@ -332,7 +369,7 @@ class Epub extends BasicOutput { builder.append("\n "); builder.append("\n " + name + ""); builder.append("\n "); - builder.append("\n "); + builder.append("\n "); builder.append("\n \n"); } @@ -359,9 +396,8 @@ class Epub extends BasicOutput { lang = meta.getLang(); } - builder.append(""); - builder.append("\n"); + builder.append(""); + builder.append("\n"); builder.append("\n "); builder.append("\n " + StringUtils.xmlEscape(title) @@ -379,20 +415,20 @@ class Epub extends BasicOutput { builder.append("\n " + StringUtils.xmlEscape(source) + ""); builder.append("\n Not for commercial use."); - builder.append("\n " + builder.append("\n " + StringUtils.xmlEscape(uuid) + ""); builder.append("\n " + StringUtils.xmlEscape(lang) + ""); builder.append("\n "); builder.append("\n "); builder.append("\n "); - builder.append("\n "); + builder.append("\n "); for (int i = 0; i <= story.getChapters().size(); i++) { String name = String.format("%s%03d", "chapter-", i); builder.append("\n "); + + ".xhtml\" media-type=\"application/xhtml+xml\"/>"); } builder.append("\n "); @@ -401,8 +437,9 @@ class Epub extends BasicOutput { builder.append("\n "); if (story.getMeta() != null && story.getMeta().getCover() != null) { - String format = Instance.getConfig() - .getString(Config.IMAGE_FORMAT_COVER).toLowerCase(); + String format = Instance.getInstance().getConfig() + .getString(Config.FILE_FORMAT_IMAGE_FORMAT_COVER) + .toLowerCase(); builder.append("\n "); } @@ -447,12 +484,12 @@ class Epub extends BasicOutput { author = meta.getAuthor(); } - String format = Instance.getConfig() - .getString(Config.IMAGE_FORMAT_COVER).toLowerCase(); + String format = Instance.getInstance().getConfig() + .getString(Config.FILE_FORMAT_IMAGE_FORMAT_COVER).toLowerCase(); - builder.append(""); - builder.append("\n"); - builder.append("\n"); + builder.append(""); + builder.append("\n"); + builder.append("\n"); builder.append("\n"); builder.append("\n " + StringUtils.xmlEscape(title) + ""); builder.append("\n "); @@ -463,7 +500,8 @@ class Epub extends BasicOutput { builder.append("\n
" + StringUtils.xmlEscape(tags) + "
"); builder.append("\n
"); - builder.append("\n "); + builder.append("\n \"cover"); builder.append("\n
"); builder.append("\n
" + StringUtils.xmlEscape(author) + "
");