Update nikiroo-utils, bugfixes:
[nikiroo-utils.git] / src / be / nikiroo / fanfix / output / Cbz.java
1 package be.nikiroo.fanfix.output;
2
3 import java.io.BufferedWriter;
4 import java.io.File;
5 import java.io.FileOutputStream;
6 import java.io.IOException;
7 import java.io.OutputStreamWriter;
8
9 import be.nikiroo.fanfix.data.MetaData;
10 import be.nikiroo.fanfix.data.Paragraph;
11 import be.nikiroo.fanfix.data.Story;
12 import be.nikiroo.utils.IOUtils;
13
14 class Cbz extends BasicOutput {
15 private File dir;
16
17 @Override
18 public File process(Story story, File targetDir, String targetName)
19 throws IOException {
20 String targetNameOrig = targetName;
21 targetName += getDefaultExtension(false);
22
23 File target = new File(targetDir, targetName);
24
25 dir = File.createTempFile("fanfic-reader-cbz-dir", ".wip");
26 dir.delete();
27 dir.mkdir();
28 try {
29 // will also save the images!
30 new InfoText().process(story, dir, targetNameOrig);
31
32 InfoCover.writeInfo(dir, targetNameOrig, story.getMeta());
33 if (story.getMeta() != null && !story.getMeta().isFakeCover()) {
34 InfoCover.writeCover(dir, targetNameOrig, story.getMeta());
35 }
36
37 IOUtils.writeSmallFile(dir, "version", "3.0");
38
39 try {
40 super.process(story, targetDir, targetNameOrig);
41 } finally {
42 }
43
44 IOUtils.zip(dir, target, true);
45 } finally {
46 IOUtils.deltree(dir);
47 }
48
49 return target;
50 }
51
52 @Override
53 public String getDefaultExtension(boolean readerTarget) {
54 return ".cbz";
55 }
56
57 @Override
58 protected void writeStoryHeader(Story story) throws IOException {
59 MetaData meta = story.getMeta();
60
61 StringBuilder builder = new StringBuilder();
62 if (meta != null && meta.getResume() != null) {
63 for (Paragraph para : story.getMeta().getResume()) {
64 builder.append(para.getContent());
65 builder.append("\n");
66 }
67 }
68
69 BufferedWriter writer = new BufferedWriter(new OutputStreamWriter(
70 new FileOutputStream(new File(dir, "URL")), "UTF-8"));
71 try {
72 if (meta != null) {
73 writer.write(meta.getUuid());
74 }
75 } finally {
76 writer.close();
77 }
78
79 writer = new BufferedWriter(new OutputStreamWriter(
80 new FileOutputStream(new File(dir, "SUMMARY")), "UTF-8"));
81 try {
82 String title = "";
83 if (meta != null && meta.getTitle() != null) {
84 title = meta.getTitle();
85 }
86
87 writer.write(title);
88 if (meta != null && meta.getAuthor() != null) {
89 writer.write("\n©");
90 writer.write(meta.getAuthor());
91 }
92 writer.write("\n\n");
93 writer.write(builder.toString());
94 } finally {
95 writer.close();
96 }
97 }
98 }