Commit | Line | Data |
---|---|---|
f433d153 NR |
1 | package be.nikiroo.fanfix.library; |
2 | ||
3 | import java.io.ByteArrayInputStream; | |
f433d153 NR |
4 | import java.io.IOException; |
5 | import java.io.InputStream; | |
f70bcacf | 6 | import java.net.URL; |
acb25f06 | 7 | import java.net.URLDecoder; |
f433d153 | 8 | import java.util.ArrayList; |
5c4ce687 | 9 | import java.util.Arrays; |
f433d153 NR |
10 | import java.util.HashMap; |
11 | import java.util.LinkedList; | |
12 | import java.util.List; | |
13 | import java.util.Map; | |
14 | ||
f433d153 NR |
15 | import org.json.JSONArray; |
16 | import org.json.JSONObject; | |
17 | ||
18 | import be.nikiroo.fanfix.Instance; | |
19 | import be.nikiroo.fanfix.bundles.Config; | |
f433d153 NR |
20 | import be.nikiroo.fanfix.data.Chapter; |
21 | import be.nikiroo.fanfix.data.JsonIO; | |
22 | import be.nikiroo.fanfix.data.MetaData; | |
23 | import be.nikiroo.fanfix.data.Paragraph; | |
24 | import be.nikiroo.fanfix.data.Paragraph.ParagraphType; | |
25 | import be.nikiroo.fanfix.data.Story; | |
f433d153 | 26 | import be.nikiroo.utils.Image; |
fce0a73f | 27 | import be.nikiroo.utils.LoginResult; |
f433d153 | 28 | import be.nikiroo.utils.NanoHTTPD; |
f433d153 NR |
29 | import be.nikiroo.utils.NanoHTTPD.Response; |
30 | import be.nikiroo.utils.NanoHTTPD.Response.Status; | |
f70bcacf | 31 | import be.nikiroo.utils.Progress; |
f433d153 | 32 | |
33d40b9f NR |
33 | public class WebLibraryServer extends WebLibraryServerHtml { |
34 | class WLoginResult extends LoginResult { | |
fce0a73f NR |
35 | public WLoginResult(boolean badLogin, boolean badCookie) { |
36 | super(badLogin, badCookie); | |
37 | } | |
38 | ||
39 | public WLoginResult(String who, String key, String subkey, boolean rw, | |
40 | boolean wl, boolean bl) { | |
41 | super(who, key, subkey, (rw ? "|rw" : "") + (wl ? "|wl" : "") | |
42 | + (bl ? "|bl" : "") + "|"); | |
f433d153 NR |
43 | } |
44 | ||
fce0a73f | 45 | public WLoginResult(String cookie, String who, String key, |
f433d153 | 46 | List<String> subkeys) { |
fce0a73f NR |
47 | super(cookie, who, key, subkeys, |
48 | subkeys == null || subkeys.isEmpty()); | |
f433d153 NR |
49 | } |
50 | ||
51 | public boolean isRw() { | |
fce0a73f | 52 | return getOption().contains("|rw|"); |
f433d153 NR |
53 | } |
54 | ||
55 | public boolean isWl() { | |
fce0a73f | 56 | return getOption().contains("|wl|"); |
f433d153 NR |
57 | } |
58 | ||
d11fb35b | 59 | public boolean isBl() { |
fce0a73f | 60 | return getOption().contains("|bl|"); |
f433d153 NR |
61 | } |
62 | } | |
63 | ||
f433d153 NR |
64 | private Map<String, Story> storyCache = new HashMap<String, Story>(); |
65 | private LinkedList<String> storyCacheOrder = new LinkedList<String>(); | |
66 | private long storyCacheSize = 0; | |
67 | private long maxStoryCacheSize; | |
f433d153 | 68 | |
d11fb35b NR |
69 | private List<String> whitelist; |
70 | private List<String> blacklist; | |
71 | ||
f70bcacf NR |
72 | private Map<String, Progress> imprts = new HashMap<String, Progress>(); |
73 | ||
09c2396e NR |
74 | private boolean exiting; |
75 | ||
f433d153 | 76 | public WebLibraryServer(boolean secure) throws IOException { |
33d40b9f | 77 | super(secure); |
f433d153 NR |
78 | |
79 | int cacheMb = Instance.getInstance().getConfig() | |
80 | .getInteger(Config.SERVER_MAX_CACHE_MB, 100); | |
81 | maxStoryCacheSize = cacheMb * 1024 * 1024; | |
82 | ||
83 | setTraceHandler(Instance.getInstance().getTraceHandler()); | |
84 | ||
d11fb35b NR |
85 | whitelist = Instance.getInstance().getConfig() |
86 | .getList(Config.SERVER_WHITELIST, new ArrayList<String>()); | |
87 | blacklist = Instance.getInstance().getConfig() | |
88 | .getList(Config.SERVER_BLACKLIST, new ArrayList<String>()); | |
f433d153 NR |
89 | } |
90 | ||
91 | /** | |
92 | * Start the server (listen on the network for new connections). | |
93 | * <p> | |
94 | * Can only be called once. | |
95 | * <p> | |
96 | * This call is asynchronous, and will just start a new {@link Thread} on | |
97 | * itself (see {@link WebLibraryServer#run()}). | |
98 | */ | |
99 | public void start() { | |
100 | new Thread(this).start(); | |
101 | } | |
102 | ||
09c2396e NR |
103 | @Override |
104 | protected Response stop(WLoginResult login) { | |
105 | if (!login.isRw()) { | |
106 | return NanoHTTPD.newFixedLengthResponse(Status.FORBIDDEN, | |
107 | NanoHTTPD.MIME_PLAINTEXT, "Exit not allowed"); | |
108 | } | |
109 | ||
110 | if (exiting) { | |
111 | return NanoHTTPD.newFixedLengthResponse(Status.SERVICE_UNAVAILABLE, | |
112 | NanoHTTPD.MIME_PLAINTEXT, "Server is already exiting..."); | |
113 | } | |
cab46763 | 114 | |
09c2396e NR |
115 | exiting = true; |
116 | Instance.getInstance().getTraceHandler().trace("Exiting"); | |
117 | ||
118 | boolean ok; | |
119 | do { | |
120 | synchronized (imprts) { | |
121 | ok = imprts.isEmpty(); | |
122 | } | |
123 | if (!ok) { | |
124 | try { | |
125 | Thread.sleep(2000); | |
126 | } catch (InterruptedException e) { | |
127 | Instance.getInstance().getTraceHandler() | |
128 | .trace("Waiting to exit..."); | |
129 | } | |
130 | } | |
131 | } while (!ok); | |
132 | ||
133 | doStop(); | |
134 | ||
cab46763 NR |
135 | new Thread(new Runnable() { |
136 | @Override | |
137 | public void run() { | |
138 | try { | |
139 | Thread.sleep(1500); | |
140 | } catch (InterruptedException e) { | |
141 | } | |
142 | ||
143 | Instance.getInstance().getTraceHandler() | |
144 | .trace("Exit timeout: force-quit"); | |
145 | System.exit(0); | |
146 | } | |
147 | }, "Exit program after timeout of 1500 ms").start(); | |
148 | ||
09c2396e NR |
149 | return NanoHTTPD.newFixedLengthResponse(Status.OK, |
150 | NanoHTTPD.MIME_PLAINTEXT, "Exited"); | |
151 | } | |
152 | ||
33d40b9f NR |
153 | @Override |
154 | protected WLoginResult login(boolean badLogin, boolean badCookie) { | |
155 | return new WLoginResult(false, false); | |
f433d153 NR |
156 | } |
157 | ||
33d40b9f NR |
158 | @Override |
159 | protected WLoginResult login(String who, String cookie) { | |
fce0a73f NR |
160 | List<String> subkeys = Instance.getInstance().getConfig() |
161 | .getList(Config.SERVER_ALLOWED_SUBKEYS); | |
f433d153 | 162 | String realKey = Instance.getInstance().getConfig() |
fce0a73f | 163 | .getString(Config.SERVER_KEY); |
d11fb35b | 164 | |
fce0a73f | 165 | return new WLoginResult(cookie, who, realKey, subkeys); |
f433d153 NR |
166 | } |
167 | ||
168 | // allow rw/wl | |
33d40b9f NR |
169 | @Override |
170 | protected WLoginResult login(String who, String key, String subkey) { | |
f433d153 | 171 | String realKey = Instance.getInstance().getConfig() |
d11fb35b | 172 | .getString(Config.SERVER_KEY, ""); |
f433d153 NR |
173 | |
174 | // I don't like NULLs... | |
f433d153 NR |
175 | key = key == null ? "" : key; |
176 | subkey = subkey == null ? "" : subkey; | |
177 | ||
178 | if (!realKey.equals(key)) { | |
fce0a73f | 179 | return new WLoginResult(true, false); |
f433d153 NR |
180 | } |
181 | ||
d11fb35b | 182 | // defaults are true (as previous versions without the feature) |
f433d153 NR |
183 | boolean rw = true; |
184 | boolean wl = true; | |
d11fb35b | 185 | boolean bl = true; |
f433d153 | 186 | |
599b05c7 NR |
187 | rw = Instance.getInstance().getConfig().getBoolean(Config.SERVER_RW, |
188 | rw); | |
fce0a73f NR |
189 | |
190 | List<String> allowed = Instance.getInstance().getConfig().getList( | |
191 | Config.SERVER_ALLOWED_SUBKEYS, new ArrayList<String>()); | |
192 | ||
193 | if (!allowed.isEmpty()) { | |
194 | if (!allowed.contains(subkey)) { | |
195 | return new WLoginResult(true, false); | |
196 | } | |
197 | ||
198 | if ((subkey + "|").contains("|rw|")) { | |
199 | rw = true; | |
200 | } | |
201 | if ((subkey + "|").contains("|wl|")) { | |
202 | wl = false; // |wl| = bypass whitelist | |
203 | } | |
204 | if ((subkey + "|").contains("|bl|")) { | |
205 | bl = false; // |bl| = bypass blacklist | |
f433d153 NR |
206 | } |
207 | } | |
208 | ||
fce0a73f | 209 | return new WLoginResult(who, key, subkey, rw, wl, bl); |
f433d153 NR |
210 | } |
211 | ||
33d40b9f | 212 | @Override |
fce0a73f | 213 | protected Response getList(String uri, WLoginResult login) |
f433d153 | 214 | throws IOException { |
5ee0fc14 | 215 | if (WebLibraryUrls.LIST_URL_METADATA.equals(uri)) { |
f433d153 | 216 | List<JSONObject> jsons = new ArrayList<JSONObject>(); |
d11fb35b | 217 | for (MetaData meta : metas(login)) { |
f433d153 NR |
218 | jsons.add(JsonIO.toJson(meta)); |
219 | } | |
220 | ||
bd743ca2 NR |
221 | return NanoHTTPD.newFixedLengthResponse(Status.OK, |
222 | "application/json", new JSONArray(jsons).toString()); | |
f433d153 NR |
223 | } |
224 | ||
225 | return NanoHTTPD.newFixedLengthResponse(Status.BAD_REQUEST, | |
226 | NanoHTTPD.MIME_PLAINTEXT, null); | |
227 | } | |
228 | ||
f433d153 NR |
229 | // /story/luid/chapter/para <-- text/image |
230 | // /story/luid/cover <-- image | |
231 | // /story/luid/metadata <-- json | |
c5103223 | 232 | // /story/luid/json <-- json, whole chapter (no images) |
33d40b9f NR |
233 | @Override |
234 | protected Response getStoryPart(String uri, WLoginResult login) { | |
6673ec59 | 235 | String[] uriParts = uri.split("/"); |
f433d153 NR |
236 | int off = 2; |
237 | ||
6673ec59 | 238 | if (uriParts.length < off + 2) { |
f433d153 NR |
239 | return NanoHTTPD.newFixedLengthResponse(Status.BAD_REQUEST, |
240 | NanoHTTPD.MIME_PLAINTEXT, null); | |
241 | } | |
242 | ||
6673ec59 NR |
243 | String luid = uriParts[off + 0]; |
244 | String chapterStr = uriParts[off + 1]; | |
245 | String imageStr = uriParts.length < off + 3 ? null : uriParts[off + 2]; | |
f433d153 NR |
246 | |
247 | // 1-based (0 = desc) | |
248 | int chapter = 0; | |
249 | if (chapterStr != null && !"cover".equals(chapterStr) | |
599b05c7 NR |
250 | && !"metadata".equals(chapterStr) |
251 | && !"json".equals(chapterStr)) { | |
f433d153 NR |
252 | try { |
253 | chapter = Integer.parseInt(chapterStr); | |
254 | if (chapter < 0) { | |
255 | throw new NumberFormatException(); | |
256 | } | |
257 | } catch (NumberFormatException e) { | |
258 | return NanoHTTPD.newFixedLengthResponse(Status.BAD_REQUEST, | |
259 | NanoHTTPD.MIME_PLAINTEXT, "Chapter is not valid"); | |
260 | } | |
261 | } | |
262 | ||
263 | // 1-based | |
264 | int paragraph = 1; | |
265 | if (imageStr != null) { | |
266 | try { | |
267 | paragraph = Integer.parseInt(imageStr); | |
268 | if (paragraph < 0) { | |
269 | throw new NumberFormatException(); | |
270 | } | |
271 | } catch (NumberFormatException e) { | |
272 | return NanoHTTPD.newFixedLengthResponse(Status.BAD_REQUEST, | |
273 | NanoHTTPD.MIME_PLAINTEXT, "Paragraph is not valid"); | |
274 | } | |
275 | } | |
276 | ||
277 | String mimeType = NanoHTTPD.MIME_PLAINTEXT; | |
278 | InputStream in = null; | |
279 | try { | |
280 | if ("cover".equals(chapterStr)) { | |
6673ec59 | 281 | Image img = storyCover(luid, login); |
f433d153 NR |
282 | if (img != null) { |
283 | in = img.newInputStream(); | |
284 | } | |
3f468ac7 NR |
285 | // TODO: get correct image type |
286 | mimeType = "image/png"; | |
f433d153 | 287 | } else if ("metadata".equals(chapterStr)) { |
d11fb35b | 288 | MetaData meta = meta(luid, login); |
f433d153 NR |
289 | JSONObject json = JsonIO.toJson(meta); |
290 | mimeType = "application/json"; | |
291 | in = new ByteArrayInputStream(json.toString().getBytes()); | |
3fbc084c | 292 | } else if ("json".equals(chapterStr)) { |
d11fb35b | 293 | Story story = story(luid, login); |
c5103223 NR |
294 | JSONObject json = JsonIO.toJson(story); |
295 | mimeType = "application/json"; | |
296 | in = new ByteArrayInputStream(json.toString().getBytes()); | |
f433d153 | 297 | } else { |
d11fb35b | 298 | Story story = story(luid, login); |
f433d153 NR |
299 | if (story != null) { |
300 | if (chapter == 0) { | |
301 | StringBuilder builder = new StringBuilder(); | |
302 | for (Paragraph p : story.getMeta().getResume()) { | |
303 | if (builder.length() == 0) { | |
304 | builder.append("\n"); | |
305 | } | |
306 | builder.append(p.getContent()); | |
307 | } | |
308 | ||
599b05c7 NR |
309 | in = new ByteArrayInputStream( |
310 | builder.toString().getBytes("utf-8")); | |
f433d153 NR |
311 | } else { |
312 | Paragraph para = story.getChapters().get(chapter - 1) | |
313 | .getParagraphs().get(paragraph - 1); | |
314 | Image img = para.getContentImage(); | |
315 | if (para.getType() == ParagraphType.IMAGE) { | |
316 | // TODO: get correct image type | |
317 | mimeType = "image/png"; | |
318 | in = img.newInputStream(); | |
319 | } else { | |
599b05c7 NR |
320 | in = new ByteArrayInputStream( |
321 | para.getContent().getBytes("utf-8")); | |
f433d153 NR |
322 | } |
323 | } | |
324 | } | |
325 | } | |
326 | } catch (IndexOutOfBoundsException e) { | |
327 | return NanoHTTPD.newFixedLengthResponse(Status.NOT_FOUND, | |
328 | NanoHTTPD.MIME_PLAINTEXT, | |
329 | "Chapter or paragraph does not exist"); | |
330 | } catch (IOException e) { | |
331 | Instance.getInstance().getTraceHandler() | |
332 | .error(new IOException("Cannot get image: " + uri, e)); | |
333 | return NanoHTTPD.newFixedLengthResponse(Status.INTERNAL_ERROR, | |
334 | NanoHTTPD.MIME_PLAINTEXT, "Error when processing request"); | |
335 | } | |
336 | ||
337 | return newInputStreamResponse(mimeType, in); | |
338 | } | |
339 | ||
089e354e NR |
340 | // /story/luid/source |
341 | // /story/luid/title | |
342 | // /story/luid/author | |
343 | @Override | |
344 | protected Response setStoryPart(String uri, String value, | |
345 | WLoginResult login) throws IOException { | |
346 | String[] uriParts = uri.split("/"); | |
347 | int off = 2; // "" and "story" | |
348 | ||
349 | if (uriParts.length < off + 2) { | |
350 | return NanoHTTPD.newFixedLengthResponse(Status.BAD_REQUEST, | |
351 | NanoHTTPD.MIME_PLAINTEXT, "Invalid story part request"); | |
352 | } | |
353 | ||
a1226ce0 NR |
354 | if (!login.isRw()) { |
355 | return NanoHTTPD.newFixedLengthResponse(Status.FORBIDDEN, | |
356 | NanoHTTPD.MIME_PLAINTEXT, "SET story part not allowed"); | |
357 | } | |
cab46763 | 358 | |
09c2396e NR |
359 | if (exiting) { |
360 | return NanoHTTPD.newFixedLengthResponse(Status.SERVICE_UNAVAILABLE, | |
361 | NanoHTTPD.MIME_PLAINTEXT, "Server is exiting..."); | |
362 | } | |
a1226ce0 | 363 | |
089e354e NR |
364 | String luid = uriParts[off + 0]; |
365 | String type = uriParts[off + 1]; | |
366 | ||
367 | if (!Arrays.asList("source", "title", "author").contains(type)) { | |
368 | return NanoHTTPD.newFixedLengthResponse(Status.BAD_REQUEST, | |
369 | NanoHTTPD.MIME_PLAINTEXT, | |
370 | "Invalid SET story part: " + type); | |
371 | } | |
372 | ||
373 | if (meta(luid, login) != null) { | |
374 | BasicLibrary lib = Instance.getInstance().getLibrary(); | |
375 | if ("source".equals(type)) { | |
376 | lib.changeSource(luid, value, null); | |
377 | } else if ("title".equals(type)) { | |
378 | lib.changeTitle(luid, value, null); | |
379 | } else if ("author".equals(type)) { | |
380 | lib.changeAuthor(luid, value, null); | |
381 | } | |
382 | } | |
383 | ||
384 | return newInputStreamResponse(NanoHTTPD.MIME_PLAINTEXT, null); | |
385 | } | |
386 | ||
6673ec59 NR |
387 | @Override |
388 | protected Response getCover(String uri, WLoginResult login) | |
389 | throws IOException { | |
390 | String[] uriParts = uri.split("/"); | |
391 | int off = 2; // "" and "cover" | |
392 | ||
393 | if (uriParts.length < off + 2) { | |
394 | return NanoHTTPD.newFixedLengthResponse(Status.BAD_REQUEST, | |
395 | NanoHTTPD.MIME_PLAINTEXT, "Invalid cover request"); | |
396 | } | |
397 | ||
398 | String type = uriParts[off + 0]; | |
acb25f06 | 399 | String id = URLDecoder.decode(uriParts[off + 1], "UTF-8"); |
6673ec59 NR |
400 | |
401 | InputStream in = null; | |
402 | ||
e4b1b70c | 403 | if ("story".equals(type)) { |
6673ec59 NR |
404 | Image img = storyCover(id, login); |
405 | if (img != null) { | |
406 | in = img.newInputStream(); | |
407 | } | |
e4b1b70c NR |
408 | } else if ("source".equals(type)) { |
409 | Image img = sourceCover(id, login); | |
6673ec59 NR |
410 | if (img != null) { |
411 | in = img.newInputStream(); | |
412 | } | |
e4b1b70c NR |
413 | } else if ("author".equals(type)) { |
414 | Image img = authorCover(id, login); | |
6673ec59 NR |
415 | if (img != null) { |
416 | in = img.newInputStream(); | |
417 | } | |
418 | } else { | |
419 | return NanoHTTPD.newFixedLengthResponse(Status.BAD_REQUEST, | |
e4b1b70c NR |
420 | NanoHTTPD.MIME_PLAINTEXT, |
421 | "Invalid GET cover type: " + type); | |
6673ec59 NR |
422 | } |
423 | ||
424 | // TODO: get correct image type | |
425 | return newInputStreamResponse("image/png", in); | |
426 | } | |
427 | ||
e4b1b70c NR |
428 | @Override |
429 | protected Response setCover(String uri, String luid, WLoginResult login) | |
430 | throws IOException { | |
431 | String[] uriParts = uri.split("/"); | |
432 | int off = 2; // "" and "cover" | |
433 | ||
434 | if (uriParts.length < off + 2) { | |
435 | return NanoHTTPD.newFixedLengthResponse(Status.BAD_REQUEST, | |
436 | NanoHTTPD.MIME_PLAINTEXT, "Invalid cover request"); | |
437 | } | |
438 | ||
a1226ce0 NR |
439 | if (!login.isRw()) { |
440 | return NanoHTTPD.newFixedLengthResponse(Status.FORBIDDEN, | |
441 | NanoHTTPD.MIME_PLAINTEXT, "Cover request not allowed"); | |
442 | } | |
cab46763 | 443 | |
09c2396e NR |
444 | if (exiting) { |
445 | return NanoHTTPD.newFixedLengthResponse(Status.SERVICE_UNAVAILABLE, | |
446 | NanoHTTPD.MIME_PLAINTEXT, "Server is exiting..."); | |
447 | } | |
a1226ce0 | 448 | |
e4b1b70c | 449 | String type = uriParts[off + 0]; |
acb25f06 | 450 | String id = URLDecoder.decode(uriParts[off + 1], "UTF-8"); |
e4b1b70c NR |
451 | |
452 | if ("source".equals(type)) { | |
453 | sourceCover(id, login, luid); | |
454 | } else if ("author".equals(type)) { | |
455 | authorCover(id, login, luid); | |
456 | } else { | |
457 | return NanoHTTPD.newFixedLengthResponse(Status.BAD_REQUEST, | |
458 | NanoHTTPD.MIME_PLAINTEXT, | |
459 | "Invalid SET cover type: " + type); | |
460 | } | |
461 | ||
462 | return newInputStreamResponse(NanoHTTPD.MIME_PLAINTEXT, null); | |
463 | } | |
464 | ||
f70bcacf NR |
465 | @Override |
466 | protected Response imprt(String uri, String urlStr, WLoginResult login) | |
467 | throws IOException { | |
468 | final BasicLibrary lib = Instance.getInstance().getLibrary(); | |
469 | ||
a1226ce0 NR |
470 | if (!login.isRw()) { |
471 | return NanoHTTPD.newFixedLengthResponse(Status.FORBIDDEN, | |
472 | NanoHTTPD.MIME_PLAINTEXT, "Import not allowed"); | |
473 | } | |
cab46763 | 474 | |
09c2396e NR |
475 | if (exiting) { |
476 | return NanoHTTPD.newFixedLengthResponse(Status.SERVICE_UNAVAILABLE, | |
477 | NanoHTTPD.MIME_PLAINTEXT, "Server is exiting..."); | |
478 | } | |
a1226ce0 | 479 | |
f70bcacf NR |
480 | final URL url = new URL(urlStr); |
481 | final Progress pg = new Progress(); | |
482 | final String luid = lib.getNextId(); | |
483 | ||
484 | synchronized (imprts) { | |
485 | imprts.put(luid, pg); | |
486 | } | |
487 | ||
488 | new Thread(new Runnable() { | |
489 | @Override | |
490 | public void run() { | |
491 | try { | |
acbec0d2 | 492 | lib.imprt(url, luid, pg); |
f70bcacf NR |
493 | } catch (IOException e) { |
494 | Instance.getInstance().getTraceHandler().error(e); | |
495 | } finally { | |
496 | synchronized (imprts) { | |
497 | imprts.remove(luid); | |
498 | } | |
499 | } | |
500 | } | |
501 | }, "Import story: " + urlStr).start(); | |
502 | ||
f70bcacf NR |
503 | return NanoHTTPD.newFixedLengthResponse(Status.OK, |
504 | NanoHTTPD.MIME_PLAINTEXT, luid); | |
505 | } | |
506 | ||
507 | @Override | |
508 | protected Response imprtProgress(String uri, WLoginResult login) { | |
509 | String[] uriParts = uri.split("/"); | |
510 | int off = 2; // "" and "import" | |
511 | ||
512 | if (uriParts.length < off + 1) { | |
513 | return NanoHTTPD.newFixedLengthResponse(Status.BAD_REQUEST, | |
514 | NanoHTTPD.MIME_PLAINTEXT, "Invalid cover request"); | |
515 | } | |
516 | ||
517 | String luid = uriParts[off + 0]; | |
518 | ||
519 | Progress pg = null; | |
520 | synchronized (imprts) { | |
521 | pg = imprts.get(luid); | |
522 | } | |
523 | if (pg != null) { | |
524 | return NanoHTTPD.newFixedLengthResponse(Status.OK, | |
525 | "application/json", JsonIO.toJson(pg).toString()); | |
526 | } | |
527 | ||
528 | return newInputStreamResponse(NanoHTTPD.MIME_PLAINTEXT, null); | |
529 | } | |
530 | ||
a1226ce0 NR |
531 | @Override |
532 | protected Response delete(String uri, WLoginResult login) | |
533 | throws IOException { | |
534 | String[] uriParts = uri.split("/"); | |
535 | int off = 2; // "" and "delete" | |
536 | ||
537 | if (uriParts.length < off + 1) { | |
538 | return NanoHTTPD.newFixedLengthResponse(Status.BAD_REQUEST, | |
539 | NanoHTTPD.MIME_PLAINTEXT, "Invalid delete request"); | |
540 | } | |
541 | ||
542 | if (!login.isRw()) { | |
543 | return NanoHTTPD.newFixedLengthResponse(Status.FORBIDDEN, | |
544 | NanoHTTPD.MIME_PLAINTEXT, "Delete not allowed"); | |
545 | } | |
cab46763 | 546 | |
09c2396e NR |
547 | if (exiting) { |
548 | return NanoHTTPD.newFixedLengthResponse(Status.SERVICE_UNAVAILABLE, | |
549 | NanoHTTPD.MIME_PLAINTEXT, "Server is exiting..."); | |
550 | } | |
a1226ce0 NR |
551 | |
552 | String luid = uriParts[off + 0]; | |
553 | ||
554 | BasicLibrary lib = Instance.getInstance().getLibrary(); | |
555 | lib.delete(luid); | |
556 | ||
557 | return newInputStreamResponse(NanoHTTPD.MIME_PLAINTEXT, null); | |
558 | } | |
559 | ||
33d40b9f NR |
560 | @Override |
561 | protected List<MetaData> metas(WLoginResult login) throws IOException { | |
d11fb35b NR |
562 | BasicLibrary lib = Instance.getInstance().getLibrary(); |
563 | List<MetaData> metas = new ArrayList<MetaData>(); | |
564 | for (MetaData meta : lib.getList().getMetas()) { | |
565 | if (isAllowed(meta, login)) { | |
566 | metas.add(meta); | |
567 | } | |
568 | } | |
569 | ||
570 | return metas; | |
571 | } | |
572 | ||
f433d153 | 573 | // NULL if not whitelist OK or if not found |
33d40b9f NR |
574 | @Override |
575 | protected Story story(String luid, WLoginResult login) throws IOException { | |
f433d153 NR |
576 | synchronized (storyCache) { |
577 | if (storyCache.containsKey(luid)) { | |
578 | Story story = storyCache.get(luid); | |
d11fb35b | 579 | if (!isAllowed(story.getMeta(), login)) |
f433d153 | 580 | return null; |
f433d153 NR |
581 | |
582 | return story; | |
583 | } | |
584 | } | |
585 | ||
586 | Story story = null; | |
d11fb35b | 587 | MetaData meta = meta(luid, login); |
f433d153 NR |
588 | if (meta != null) { |
589 | BasicLibrary lib = Instance.getInstance().getLibrary(); | |
590 | story = lib.getStory(luid, null); | |
591 | long size = sizeOf(story); | |
592 | ||
593 | synchronized (storyCache) { | |
594 | // Could have been added by another request | |
595 | if (!storyCache.containsKey(luid)) { | |
596 | while (!storyCacheOrder.isEmpty() | |
597 | && storyCacheSize + size > maxStoryCacheSize) { | |
598 | String oldestLuid = storyCacheOrder.removeFirst(); | |
599 | Story oldestStory = storyCache.remove(oldestLuid); | |
600 | maxStoryCacheSize -= sizeOf(oldestStory); | |
601 | } | |
602 | ||
603 | storyCacheOrder.add(luid); | |
604 | storyCache.put(luid, story); | |
605 | } | |
606 | } | |
607 | } | |
608 | ||
609 | return story; | |
610 | } | |
611 | ||
33d40b9f NR |
612 | private MetaData meta(String luid, WLoginResult login) throws IOException { |
613 | BasicLibrary lib = Instance.getInstance().getLibrary(); | |
614 | MetaData meta = lib.getInfo(luid); | |
615 | if (!isAllowed(meta, login)) | |
616 | return null; | |
f433d153 | 617 | |
33d40b9f | 618 | return meta; |
f433d153 NR |
619 | } |
620 | ||
6673ec59 NR |
621 | private Image storyCover(String luid, WLoginResult login) |
622 | throws IOException { | |
33d40b9f NR |
623 | MetaData meta = meta(luid, login); |
624 | if (meta != null) { | |
625 | BasicLibrary lib = Instance.getInstance().getLibrary(); | |
626 | return lib.getCover(meta.getLuid()); | |
f433d153 | 627 | } |
f433d153 | 628 | |
33d40b9f | 629 | return null; |
f433d153 NR |
630 | } |
631 | ||
6673ec59 NR |
632 | private Image authorCover(String author, WLoginResult login) |
633 | throws IOException { | |
634 | Image img = null; | |
635 | ||
636 | List<MetaData> metas = new MetaResultList(metas(login)).filter(null, | |
637 | author, null); | |
638 | if (metas.size() > 0) { | |
639 | BasicLibrary lib = Instance.getInstance().getLibrary(); | |
640 | img = lib.getCustomAuthorCover(author); | |
641 | if (img == null) | |
642 | img = lib.getCover(metas.get(0).getLuid()); | |
643 | } | |
644 | ||
645 | return img; | |
646 | ||
647 | } | |
648 | ||
e4b1b70c NR |
649 | private void authorCover(String author, WLoginResult login, String luid) |
650 | throws IOException { | |
651 | if (meta(luid, login) != null) { | |
652 | List<MetaData> metas = new MetaResultList(metas(login)).filter(null, | |
653 | author, null); | |
654 | if (metas.size() > 0) { | |
655 | BasicLibrary lib = Instance.getInstance().getLibrary(); | |
656 | lib.setAuthorCover(author, luid); | |
657 | } | |
658 | } | |
659 | } | |
660 | ||
6673ec59 NR |
661 | private Image sourceCover(String source, WLoginResult login) |
662 | throws IOException { | |
663 | Image img = null; | |
664 | ||
665 | List<MetaData> metas = new MetaResultList(metas(login)).filter(source, | |
666 | null, null); | |
667 | if (metas.size() > 0) { | |
668 | BasicLibrary lib = Instance.getInstance().getLibrary(); | |
669 | img = lib.getCustomSourceCover(source); | |
670 | if (img == null) | |
671 | img = lib.getCover(metas.get(0).getLuid()); | |
672 | } | |
673 | ||
674 | return img; | |
675 | } | |
676 | ||
e4b1b70c NR |
677 | private void sourceCover(String source, WLoginResult login, String luid) |
678 | throws IOException { | |
679 | if (meta(luid, login) != null) { | |
680 | List<MetaData> metas = new MetaResultList(metas(login)) | |
681 | .filter(source, null, null); | |
682 | if (metas.size() > 0) { | |
683 | BasicLibrary lib = Instance.getInstance().getLibrary(); | |
684 | lib.setSourceCover(source, luid); | |
685 | } | |
686 | } | |
687 | } | |
688 | ||
33d40b9f | 689 | private boolean isAllowed(MetaData meta, WLoginResult login) { |
5c4ce687 NR |
690 | MetaResultList one = new MetaResultList(Arrays.asList(meta)); |
691 | if (login.isWl() && !whitelist.isEmpty()) { | |
692 | if (one.filter(whitelist, null, null).isEmpty()) { | |
693 | return false; | |
694 | } | |
f433d153 | 695 | } |
5c4ce687 NR |
696 | if (login.isBl() && !blacklist.isEmpty()) { |
697 | if (!one.filter(blacklist, null, null).isEmpty()) { | |
698 | return false; | |
699 | } | |
599b05c7 NR |
700 | } |
701 | ||
33d40b9f | 702 | return true; |
599b05c7 NR |
703 | } |
704 | ||
33d40b9f NR |
705 | private long sizeOf(Story story) { |
706 | long size = 0; | |
707 | for (Chapter chap : story) { | |
708 | for (Paragraph para : chap) { | |
709 | if (para.getType() == ParagraphType.IMAGE) { | |
710 | size += para.getContentImage().getSize(); | |
711 | } else { | |
712 | size += para.getContent().length(); | |
713 | } | |
714 | } | |
6b89e45c NR |
715 | } |
716 | ||
33d40b9f | 717 | return size; |
6b89e45c NR |
718 | } |
719 | ||
3fbc084c NR |
720 | public static void main(String[] args) throws IOException { |
721 | Instance.init(); | |
722 | WebLibraryServer web = new WebLibraryServer(false); | |
723 | web.run(); | |
724 | } | |
f433d153 | 725 | } |