Commit | Line | Data |
---|---|---|
08fe2e33 NR |
1 | package be.nikiroo.fanfix.supported; |
2 | ||
3 | import java.awt.image.BufferedImage; | |
4 | import java.io.File; | |
5 | import java.io.FileInputStream; | |
6 | import java.io.IOException; | |
7 | import java.io.InputStream; | |
8 | import java.net.URL; | |
68686a37 | 9 | import java.util.ArrayList; |
08fe2e33 NR |
10 | import java.util.List; |
11 | import java.util.Map.Entry; | |
12 | import java.util.zip.ZipEntry; | |
13 | import java.util.zip.ZipInputStream; | |
14 | ||
08fe2e33 | 15 | import be.nikiroo.fanfix.Instance; |
68686a37 | 16 | import be.nikiroo.fanfix.data.MetaData; |
08fe2e33 NR |
17 | import be.nikiroo.utils.IOUtils; |
18 | import be.nikiroo.utils.MarkableFileInputStream; | |
ed08c171 | 19 | import be.nikiroo.utils.Progress; |
08fe2e33 NR |
20 | |
21 | /** | |
22 | * Support class for EPUB files created with this program (as we need some | |
23 | * metadata available in those we create). | |
24 | * | |
25 | * @author niki | |
26 | */ | |
68686a37 | 27 | class Epub extends InfoText { |
08fe2e33 | 28 | private File tmp; |
68686a37 | 29 | protected MetaData meta; |
08fe2e33 | 30 | |
68686a37 NR |
31 | private URL fakeSource; |
32 | private InputStream fakeIn; | |
08fe2e33 NR |
33 | |
34 | @Override | |
35 | public String getSourceName() { | |
36 | return "epub"; | |
37 | } | |
38 | ||
39 | @Override | |
40 | protected boolean supports(URL url) { | |
41 | if (url.getPath().toLowerCase().endsWith(".epub")) { | |
42 | return true; | |
43 | } | |
44 | ||
45 | return false; | |
46 | } | |
47 | ||
48 | @Override | |
68686a37 NR |
49 | protected MetaData getMeta(URL source, InputStream in) throws IOException { |
50 | return meta; | |
08fe2e33 NR |
51 | } |
52 | ||
53 | @Override | |
54 | protected String getDesc(URL source, InputStream in) throws IOException { | |
68686a37 NR |
55 | if (fakeIn != null) { |
56 | fakeIn.reset(); | |
57 | return super.getDesc(fakeSource, fakeIn); | |
08fe2e33 NR |
58 | } |
59 | ||
60 | return null; | |
61 | } | |
62 | ||
63 | @Override | |
ed08c171 NR |
64 | protected List<Entry<String, URL>> getChapters(URL source, InputStream in, |
65 | Progress pg) throws IOException { | |
68686a37 NR |
66 | if (fakeIn != null) { |
67 | fakeIn.reset(); | |
ed08c171 | 68 | return super.getChapters(fakeSource, fakeIn, pg); |
08fe2e33 NR |
69 | } |
70 | ||
71 | return null; | |
72 | } | |
73 | ||
74 | @Override | |
ed08c171 NR |
75 | protected String getChapterContent(URL source, InputStream in, int number, |
76 | Progress pg) throws IOException { | |
68686a37 NR |
77 | if (fakeIn != null) { |
78 | fakeIn.reset(); | |
ed08c171 | 79 | return super.getChapterContent(fakeSource, fakeIn, number, pg); |
08fe2e33 NR |
80 | } |
81 | ||
82 | return null; | |
83 | } | |
84 | ||
85 | @Override | |
68686a37 | 86 | protected void preprocess(URL source, InputStream in) throws IOException { |
08fe2e33 NR |
87 | // Note: do NOT close this stream, as it would also close "in" |
88 | ZipInputStream zipIn = new ZipInputStream(in); | |
89 | tmp = File.createTempFile("fanfic-reader-parser_", ".tmp"); | |
68686a37 | 90 | File tmpInfo = new File(tmp + ".info"); |
08fe2e33 | 91 | fakeSource = tmp.toURI().toURL(); |
68686a37 | 92 | BufferedImage cover = null; |
08fe2e33 NR |
93 | |
94 | for (ZipEntry entry = zipIn.getNextEntry(); entry != null; entry = zipIn | |
95 | .getNextEntry()) { | |
96 | if (!entry.isDirectory() | |
97 | && entry.getName().startsWith(getDataPrefix())) { | |
98 | String entryLName = entry.getName().toLowerCase(); | |
fe999aa4 | 99 | |
08fe2e33 NR |
100 | boolean imageEntry = false; |
101 | for (String ext : getImageExt(false)) { | |
102 | if (entryLName.endsWith(ext)) { | |
103 | imageEntry = true; | |
104 | } | |
105 | } | |
106 | ||
107 | if (entry.getName().equals(getDataPrefix() + "version")) { | |
108 | // Nothing to do for now ("first" | |
109 | // version is 3.0) | |
110 | } else if (entryLName.endsWith(".info")) { | |
111 | // Info file | |
112 | IOUtils.write(zipIn, tmpInfo); | |
113 | } else if (imageEntry) { | |
114 | // Cover | |
115 | if (getCover()) { | |
116 | try { | |
595dfa7a | 117 | cover = IOUtils.toImage(zipIn); |
08fe2e33 NR |
118 | } catch (Exception e) { |
119 | Instance.syserr(e); | |
120 | } | |
121 | } | |
122 | } else if (entry.getName().equals(getDataPrefix() + "URL")) { | |
123 | // Do nothing | |
124 | } else if (entry.getName().equals(getDataPrefix() + "SUMMARY")) { | |
125 | // Do nothing | |
126 | } else { | |
127 | // Hopefully the data file | |
128 | IOUtils.write(zipIn, tmp); | |
129 | } | |
130 | } | |
131 | } | |
132 | ||
133 | if (requireInfo() && (!tmp.exists() || !tmpInfo.exists())) { | |
134 | throw new IOException( | |
135 | "file not supported (maybe not created with this program or corrupt)"); | |
136 | } | |
137 | ||
138 | if (tmp.exists()) { | |
68686a37 NR |
139 | this.fakeIn = new MarkableFileInputStream(new FileInputStream(tmp)); |
140 | } | |
141 | ||
142 | if (tmpInfo.exists()) { | |
57f02339 | 143 | meta = InfoReader.readMeta(tmpInfo, true); |
68686a37 NR |
144 | if (cover != null) { |
145 | meta.setCover(cover); | |
146 | } | |
147 | tmpInfo.delete(); | |
148 | } else { | |
149 | meta = new MetaData(); | |
150 | meta.setUuid(source.toString()); | |
151 | meta.setLang("EN"); | |
152 | meta.setTags(new ArrayList<String>()); | |
153 | meta.setSource(getSourceName()); | |
2206ef66 | 154 | meta.setUrl(source.toString()); |
08fe2e33 NR |
155 | } |
156 | } | |
157 | ||
158 | @Override | |
159 | protected void close() throws IOException { | |
68686a37 NR |
160 | if (tmp != null && tmp.exists()) { |
161 | if (!tmp.delete()) { | |
162 | tmp.deleteOnExit(); | |
08fe2e33 NR |
163 | } |
164 | } | |
165 | ||
166 | tmp = null; | |
08fe2e33 | 167 | |
68686a37 NR |
168 | fakeIn.close(); |
169 | super.close(); | |
08fe2e33 NR |
170 | } |
171 | ||
172 | protected String getDataPrefix() { | |
173 | return "DATA/"; | |
174 | } | |
175 | ||
176 | protected boolean requireInfo() { | |
177 | return true; | |
178 | } | |
179 | ||
180 | protected boolean getCover() { | |
181 | return true; | |
182 | } | |
08fe2e33 | 183 | } |