Commit | Line | Data |
---|---|---|
08fe2e33 NR |
1 | package be.nikiroo.fanfix.supported; |
2 | ||
595dfa7a | 3 | import java.awt.image.BufferedImage; |
08fe2e33 NR |
4 | import java.io.IOException; |
5 | import java.io.InputStream; | |
6 | import java.net.URL; | |
7 | import java.util.ArrayList; | |
8 | import java.util.List; | |
9 | import java.util.Map.Entry; | |
10 | import java.util.Scanner; | |
11 | ||
12 | import be.nikiroo.fanfix.Instance; | |
13 | import be.nikiroo.fanfix.data.Chapter; | |
68686a37 | 14 | import be.nikiroo.fanfix.data.MetaData; |
08fe2e33 | 15 | import be.nikiroo.fanfix.data.Story; |
3b2b638f | 16 | import be.nikiroo.utils.Progress; |
08fe2e33 NR |
17 | import be.nikiroo.utils.StringUtils; |
18 | ||
19 | /** | |
20 | * Support class for <a href="http://e621.net/">e621.net</a> and <a | |
21 | * href="http://e926.net/">e926.net</a>, a Furry website supporting comics, | |
22 | * including some of MLP. | |
23 | * <p> | |
24 | * <a href="http://e926.net/">e926.net</a> only shows the "clean" images and | |
25 | * comics, but it can be difficult to browse. | |
26 | * | |
27 | * @author niki | |
28 | */ | |
29 | class E621 extends BasicSupport { | |
30 | @Override | |
31 | public String getSourceName() { | |
32 | return "e621.net"; | |
33 | } | |
34 | ||
35 | @Override | |
68686a37 NR |
36 | protected MetaData getMeta(URL source, InputStream in) throws IOException { |
37 | MetaData meta = new MetaData(); | |
38 | ||
39 | meta.setTitle(getTitle(reset(in))); | |
40 | meta.setAuthor(getAuthor(source, reset(in))); | |
41 | meta.setDate(""); | |
42 | meta.setTags(new ArrayList<String>()); // TODDO ??? | |
43 | meta.setSource(getSourceName()); | |
2206ef66 | 44 | meta.setUrl(source.toString()); |
68686a37 NR |
45 | meta.setPublisher(getSourceName()); |
46 | meta.setUuid(source.toString()); | |
47 | meta.setLuid(""); | |
48 | meta.setLang("EN"); | |
a4143cd7 | 49 | meta.setSubject("Furry"); |
68686a37 NR |
50 | meta.setType(getType().toString()); |
51 | meta.setImageDocument(true); | |
595dfa7a | 52 | meta.setCover(getCover(source)); |
a9eb3f46 | 53 | meta.setFakeCover(true); |
68686a37 NR |
54 | |
55 | return meta; | |
08fe2e33 NR |
56 | } |
57 | ||
58 | @Override | |
92fb0719 | 59 | public Story process(URL url, Progress pg) throws IOException { |
08fe2e33 | 60 | // There is no chapters on e621, just pagination... |
92fb0719 | 61 | Story story = super.process(url, pg); |
08fe2e33 NR |
62 | |
63 | Chapter only = new Chapter(1, null); | |
64 | for (Chapter chap : story) { | |
65 | only.getParagraphs().addAll(chap.getParagraphs()); | |
66 | } | |
67 | ||
68 | story.getChapters().clear(); | |
69 | story.getChapters().add(only); | |
70 | ||
71 | return story; | |
72 | } | |
73 | ||
74 | @Override | |
75 | protected boolean supports(URL url) { | |
76 | String host = url.getHost(); | |
77 | if (host.startsWith("www.")) { | |
78 | host = host.substring("www.".length()); | |
79 | } | |
80 | ||
81 | return ("e621.net".equals(host) || "e926.net".equals(host)) | |
82 | && url.getPath().startsWith("/pool/"); | |
83 | } | |
84 | ||
85 | @Override | |
86 | protected boolean isHtml() { | |
87 | return true; | |
88 | } | |
89 | ||
595dfa7a NR |
90 | private BufferedImage getCover(URL source) throws IOException { |
91 | InputStream in = Instance.getCache().open(source, this, true); | |
92 | String images = getChapterContent(new URL(source.toString() + "?page=" | |
ed08c171 | 93 | + 1), in, 1, null); |
595dfa7a | 94 | if (!images.isEmpty()) { |
406447a4 | 95 | int pos = images.indexOf("<br/>"); |
595dfa7a NR |
96 | if (pos >= 0) { |
97 | images = images.substring(1, pos - 1); | |
98 | return getImage(this, null, images); | |
99 | } | |
100 | } | |
101 | ||
102 | return null; | |
103 | } | |
104 | ||
211f7ddb | 105 | private String getAuthor(URL source, InputStream in) { |
08fe2e33 NR |
106 | String author = getLine(in, "href=\"/post/show/", 0); |
107 | if (author != null) { | |
108 | String key = "href=\""; | |
109 | int pos = author.indexOf(key); | |
110 | if (pos >= 0) { | |
111 | author = author.substring(pos + key.length()); | |
112 | pos = author.indexOf("\""); | |
113 | if (pos >= 0) { | |
114 | author = author.substring(0, pos - 1); | |
115 | String page = source.getProtocol() + "://" | |
116 | + source.getHost() + author; | |
08fe2e33 | 117 | try { |
7d0d2be6 NR |
118 | InputStream pageIn = Instance.getCache().open( |
119 | new URL(page), this, false); | |
120 | try { | |
121 | key = "class=\"tag-type-artist\""; | |
122 | author = getLine(pageIn, key, 0); | |
123 | if (author != null) { | |
124 | pos = author.indexOf("<a href=\""); | |
08fe2e33 | 125 | if (pos >= 0) { |
7d0d2be6 NR |
126 | author = author.substring(pos); |
127 | pos = author.indexOf("</a>"); | |
128 | if (pos >= 0) { | |
129 | author = author.substring(0, pos); | |
130 | return StringUtils.unhtml(author); | |
131 | } | |
08fe2e33 NR |
132 | } |
133 | } | |
7d0d2be6 NR |
134 | } finally { |
135 | pageIn.close(); | |
08fe2e33 | 136 | } |
7d0d2be6 NR |
137 | } catch (Exception e) { |
138 | // No author found | |
08fe2e33 NR |
139 | } |
140 | } | |
141 | } | |
142 | } | |
143 | ||
144 | return null; | |
145 | } | |
146 | ||
211f7ddb | 147 | private String getTitle(InputStream in) { |
08fe2e33 NR |
148 | String title = getLine(in, "<title>", 0); |
149 | if (title != null) { | |
150 | int pos = title.indexOf('>'); | |
151 | if (pos >= 0) { | |
152 | title = title.substring(pos + 1); | |
153 | pos = title.indexOf('<'); | |
154 | if (pos >= 0) { | |
155 | title = title.substring(0, pos); | |
156 | } | |
157 | } | |
158 | ||
159 | if (title.startsWith("Pool:")) { | |
160 | title = title.substring("Pool:".length()); | |
161 | } | |
162 | ||
68686a37 | 163 | title = StringUtils.unhtml(title).trim(); |
08fe2e33 NR |
164 | } |
165 | ||
166 | return title; | |
167 | } | |
168 | ||
169 | @Override | |
170 | protected String getDesc(URL source, InputStream in) throws IOException { | |
171 | String desc = getLine(in, "margin-bottom: 2em;", 0); | |
172 | ||
173 | if (desc != null) { | |
174 | StringBuilder builder = new StringBuilder(); | |
175 | ||
176 | boolean inTags = false; | |
177 | for (char car : desc.toCharArray()) { | |
178 | if ((inTags && car == '>') || (!inTags && car == '<')) { | |
179 | inTags = !inTags; | |
180 | } | |
181 | ||
182 | if (inTags) { | |
183 | builder.append(car); | |
184 | } | |
185 | } | |
186 | ||
187 | return builder.toString().trim(); | |
188 | } | |
189 | ||
190 | return null; | |
191 | } | |
192 | ||
193 | @Override | |
ed08c171 NR |
194 | protected List<Entry<String, URL>> getChapters(URL source, InputStream in, |
195 | Progress pg) throws IOException { | |
08fe2e33 NR |
196 | List<Entry<String, URL>> urls = new ArrayList<Entry<String, URL>>(); |
197 | int last = 1; // no pool/show when only one page | |
198 | ||
199 | @SuppressWarnings("resource") | |
200 | Scanner scan = new Scanner(in, "UTF-8"); | |
201 | scan.useDelimiter("\\n"); | |
202 | while (scan.hasNext()) { | |
203 | String line = scan.next(); | |
204 | for (int pos = line.indexOf(source.getPath()); pos >= 0; pos = line | |
205 | .indexOf(source.getPath(), pos + source.getPath().length())) { | |
206 | int equalPos = line.indexOf("=", pos); | |
207 | int quotePos = line.indexOf("\"", pos); | |
208 | if (equalPos >= 0 && quotePos > equalPos) { | |
209 | String snum = line.substring(equalPos + 1, quotePos); | |
210 | try { | |
211 | int num = Integer.parseInt(snum); | |
212 | if (num > last) { | |
213 | last = num; | |
214 | } | |
215 | } catch (NumberFormatException e) { | |
216 | } | |
217 | } | |
218 | } | |
219 | } | |
220 | ||
221 | for (int i = 1; i <= last; i++) { | |
222 | final String key = Integer.toString(i); | |
223 | final URL value = new URL(source.toString() + "?page=" + i); | |
224 | urls.add(new Entry<String, URL>() { | |
211f7ddb | 225 | @Override |
08fe2e33 NR |
226 | public URL setValue(URL value) { |
227 | return null; | |
228 | } | |
229 | ||
211f7ddb | 230 | @Override |
08fe2e33 NR |
231 | public URL getValue() { |
232 | return value; | |
233 | } | |
234 | ||
211f7ddb | 235 | @Override |
08fe2e33 NR |
236 | public String getKey() { |
237 | return key; | |
238 | } | |
239 | }); | |
240 | } | |
241 | ||
242 | return urls; | |
243 | } | |
244 | ||
245 | @Override | |
ed08c171 NR |
246 | protected String getChapterContent(URL source, InputStream in, int number, |
247 | Progress pg) throws IOException { | |
08fe2e33 NR |
248 | StringBuilder builder = new StringBuilder(); |
249 | String staticSite = "https://static1.e621.net"; | |
250 | if (source.getHost().contains("e926")) { | |
251 | staticSite = staticSite.replace("e621", "e926"); | |
252 | } | |
253 | ||
254 | String key = staticSite + "/data/preview/"; | |
255 | ||
256 | @SuppressWarnings("resource") | |
257 | Scanner scan = new Scanner(in, "UTF-8"); | |
258 | scan.useDelimiter("\\n"); | |
259 | while (scan.hasNext()) { | |
260 | String line = scan.next(); | |
d98a2900 | 261 | if (line.contains("class=\"preview")) { |
08fe2e33 NR |
262 | for (int pos = line.indexOf(key); pos >= 0; pos = line.indexOf( |
263 | key, pos + key.length())) { | |
264 | int endPos = line.indexOf("\"", pos); | |
265 | if (endPos >= 0) { | |
266 | String id = line.substring(pos + key.length(), endPos); | |
267 | id = staticSite + "/data/" + id; | |
268 | ||
269 | int dotPos = id.lastIndexOf("."); | |
270 | if (dotPos >= 0) { | |
271 | id = id.substring(0, dotPos); | |
272 | builder.append("["); | |
273 | builder.append(id); | |
406447a4 | 274 | builder.append("]<br/>"); |
08fe2e33 NR |
275 | } |
276 | } | |
277 | } | |
278 | } | |
279 | } | |
280 | ||
281 | return builder.toString(); | |
282 | } | |
283 | } |