| 1 | package be.nikiroo.fanfix.library; |
| 2 | |
| 3 | import java.io.ByteArrayInputStream; |
| 4 | import java.io.File; |
| 5 | import java.io.FileInputStream; |
| 6 | import java.io.IOException; |
| 7 | import java.io.InputStream; |
| 8 | import java.security.KeyStore; |
| 9 | import java.util.ArrayList; |
| 10 | import java.util.HashMap; |
| 11 | import java.util.LinkedList; |
| 12 | import java.util.List; |
| 13 | import java.util.Map; |
| 14 | |
| 15 | import javax.net.ssl.KeyManagerFactory; |
| 16 | import javax.net.ssl.SSLServerSocketFactory; |
| 17 | |
| 18 | import org.json.JSONArray; |
| 19 | import org.json.JSONObject; |
| 20 | |
| 21 | import be.nikiroo.fanfix.Instance; |
| 22 | import be.nikiroo.fanfix.bundles.Config; |
| 23 | import be.nikiroo.fanfix.bundles.UiConfig; |
| 24 | import be.nikiroo.fanfix.data.Chapter; |
| 25 | import be.nikiroo.fanfix.data.JsonIO; |
| 26 | import be.nikiroo.fanfix.data.MetaData; |
| 27 | import be.nikiroo.fanfix.data.Paragraph; |
| 28 | import be.nikiroo.fanfix.data.Paragraph.ParagraphType; |
| 29 | import be.nikiroo.fanfix.data.Story; |
| 30 | import be.nikiroo.fanfix.library.web.WebLibraryServerIndex; |
| 31 | import be.nikiroo.fanfix.reader.TextOutput; |
| 32 | import be.nikiroo.utils.CookieUtils; |
| 33 | import be.nikiroo.utils.IOUtils; |
| 34 | import be.nikiroo.utils.Image; |
| 35 | import be.nikiroo.utils.NanoHTTPD; |
| 36 | import be.nikiroo.utils.NanoHTTPD.IHTTPSession; |
| 37 | import be.nikiroo.utils.NanoHTTPD.Response; |
| 38 | import be.nikiroo.utils.NanoHTTPD.Response.Status; |
| 39 | import be.nikiroo.utils.TraceHandler; |
| 40 | import be.nikiroo.utils.Version; |
| 41 | |
| 42 | public class WebLibraryServer implements Runnable { |
| 43 | static private String VIEWER_URL_BASE = "/view/story/"; |
| 44 | static private String VIEWER_URL = VIEWER_URL_BASE + "{luid}/{chap}/{para}"; |
| 45 | static private String STORY_URL_BASE = "/story/"; |
| 46 | static private String STORY_URL = STORY_URL_BASE + "{luid}/{chap}/{para}"; |
| 47 | static private String STORY_URL_COVER = STORY_URL_BASE + "{luid}/cover"; |
| 48 | static private String LIST_URL = "/list/"; |
| 49 | |
| 50 | private class LoginResult { |
| 51 | private boolean success; |
| 52 | private boolean rw; |
| 53 | private boolean wl; |
| 54 | private String wookie; |
| 55 | private String token; |
| 56 | private boolean badLogin; |
| 57 | private boolean badToken; |
| 58 | |
| 59 | public LoginResult(String who, String key, String subkey, |
| 60 | boolean success, boolean rw, boolean wl) { |
| 61 | this.success = success; |
| 62 | this.rw = rw; |
| 63 | this.wl = wl; |
| 64 | this.wookie = CookieUtils.generateCookie(who + key, 0); |
| 65 | |
| 66 | String opts = ""; |
| 67 | if (rw) |
| 68 | opts += "|rw"; |
| 69 | if (!wl) |
| 70 | opts += "|wl"; |
| 71 | |
| 72 | this.token = wookie + "~" |
| 73 | + CookieUtils.generateCookie(wookie + subkey + opts, 0) |
| 74 | + "~" + opts; |
| 75 | this.badLogin = !success; |
| 76 | } |
| 77 | |
| 78 | public LoginResult(String token, String who, String key, |
| 79 | List<String> subkeys) { |
| 80 | |
| 81 | if (token != null) { |
| 82 | String hashes[] = token.split("~"); |
| 83 | if (hashes.length >= 2) { |
| 84 | String wookie = hashes[0]; |
| 85 | String rehashed = hashes[1]; |
| 86 | String opts = hashes.length > 2 ? hashes[2] : ""; |
| 87 | |
| 88 | if (CookieUtils.validateCookie(who + key, wookie)) { |
| 89 | if (subkeys == null) { |
| 90 | subkeys = new ArrayList<String>(); |
| 91 | } |
| 92 | subkeys = new ArrayList<String>(subkeys); |
| 93 | subkeys.add(""); |
| 94 | |
| 95 | for (String subkey : subkeys) { |
| 96 | if (CookieUtils.validateCookie( |
| 97 | wookie + subkey + opts, rehashed)) { |
| 98 | this.wookie = wookie; |
| 99 | this.token = token; |
| 100 | this.success = true; |
| 101 | |
| 102 | this.rw = opts.contains("|rw"); |
| 103 | this.wl = !opts.contains("|wl"); |
| 104 | } |
| 105 | } |
| 106 | } |
| 107 | } |
| 108 | |
| 109 | this.badToken = !success; |
| 110 | } |
| 111 | |
| 112 | // No token -> no bad token |
| 113 | } |
| 114 | |
| 115 | public boolean isSuccess() { |
| 116 | return success; |
| 117 | } |
| 118 | |
| 119 | public boolean isRw() { |
| 120 | return rw; |
| 121 | } |
| 122 | |
| 123 | public boolean isWl() { |
| 124 | return wl; |
| 125 | } |
| 126 | |
| 127 | public String getToken() { |
| 128 | return token; |
| 129 | } |
| 130 | |
| 131 | public boolean isBadLogin() { |
| 132 | return badLogin; |
| 133 | } |
| 134 | |
| 135 | public boolean isBadToken() { |
| 136 | return badToken; |
| 137 | } |
| 138 | } |
| 139 | |
| 140 | private NanoHTTPD server; |
| 141 | private Map<String, Story> storyCache = new HashMap<String, Story>(); |
| 142 | private LinkedList<String> storyCacheOrder = new LinkedList<String>(); |
| 143 | private long storyCacheSize = 0; |
| 144 | private long maxStoryCacheSize; |
| 145 | private TraceHandler tracer = new TraceHandler(); |
| 146 | |
| 147 | public WebLibraryServer(boolean secure) throws IOException { |
| 148 | Integer port = Instance.getInstance().getConfig() |
| 149 | .getInteger(Config.SERVER_PORT); |
| 150 | if (port == null) { |
| 151 | throw new IOException( |
| 152 | "Cannot start web server: port not specified"); |
| 153 | } |
| 154 | |
| 155 | int cacheMb = Instance.getInstance().getConfig() |
| 156 | .getInteger(Config.SERVER_MAX_CACHE_MB, 100); |
| 157 | maxStoryCacheSize = cacheMb * 1024 * 1024; |
| 158 | |
| 159 | setTraceHandler(Instance.getInstance().getTraceHandler()); |
| 160 | |
| 161 | SSLServerSocketFactory ssf = null; |
| 162 | if (secure) { |
| 163 | String keystorePath = Instance.getInstance().getConfig() |
| 164 | .getString(Config.SERVER_SSL_KEYSTORE, ""); |
| 165 | String keystorePass = Instance.getInstance().getConfig() |
| 166 | .getString(Config.SERVER_SSL_KEYSTORE_PASS); |
| 167 | |
| 168 | if (secure && keystorePath.isEmpty()) { |
| 169 | throw new IOException( |
| 170 | "Cannot start a secure web server: no keystore.jks file povided"); |
| 171 | } |
| 172 | |
| 173 | if (!keystorePath.isEmpty()) { |
| 174 | File keystoreFile = new File(keystorePath); |
| 175 | try { |
| 176 | KeyStore keystore = KeyStore |
| 177 | .getInstance(KeyStore.getDefaultType()); |
| 178 | InputStream keystoreStream = new FileInputStream( |
| 179 | keystoreFile); |
| 180 | try { |
| 181 | keystore.load(keystoreStream, |
| 182 | keystorePass.toCharArray()); |
| 183 | KeyManagerFactory keyManagerFactory = KeyManagerFactory |
| 184 | .getInstance(KeyManagerFactory |
| 185 | .getDefaultAlgorithm()); |
| 186 | keyManagerFactory.init(keystore, |
| 187 | keystorePass.toCharArray()); |
| 188 | ssf = NanoHTTPD.makeSSLSocketFactory(keystore, |
| 189 | keyManagerFactory); |
| 190 | } finally { |
| 191 | keystoreStream.close(); |
| 192 | } |
| 193 | } catch (Exception e) { |
| 194 | throw new IOException(e.getMessage()); |
| 195 | } |
| 196 | } |
| 197 | } |
| 198 | |
| 199 | server = new NanoHTTPD(port) { |
| 200 | @Override |
| 201 | public Response serve(final IHTTPSession session) { |
| 202 | super.serve(session); |
| 203 | |
| 204 | String query = session.getQueryParameterString(); // a=a%20b&dd=2 |
| 205 | Method method = session.getMethod(); // GET, POST.. |
| 206 | String uri = session.getUri(); // /home.html |
| 207 | |
| 208 | // need them in real time (not just those sent by the UA) |
| 209 | Map<String, String> cookies = new HashMap<String, String>(); |
| 210 | for (String cookie : session.getCookies()) { |
| 211 | cookies.put(cookie, session.getCookies().read(cookie)); |
| 212 | } |
| 213 | |
| 214 | List<String> whitelist = Instance.getInstance().getConfig() |
| 215 | .getList(Config.SERVER_WHITELIST); |
| 216 | if (whitelist == null) { |
| 217 | whitelist = new ArrayList<String>(); |
| 218 | } |
| 219 | |
| 220 | LoginResult login = null; |
| 221 | Map<String, String> params = session.getParms(); |
| 222 | String who = session.getRemoteHostName() |
| 223 | + session.getRemoteIpAddress(); |
| 224 | if (params.get("login") != null) { |
| 225 | login = login(who, params.get("password"), |
| 226 | params.get("login"), whitelist); |
| 227 | } else { |
| 228 | String token = cookies.get("token"); |
| 229 | login = login(who, token, Instance.getInstance().getConfig() |
| 230 | .getList(Config.SERVER_ALLOWED_SUBKEYS)); |
| 231 | } |
| 232 | |
| 233 | if (login.isSuccess()) { |
| 234 | // refresh token |
| 235 | session.getCookies().set(new Cookie("token", |
| 236 | login.getToken(), "30; path=/")); |
| 237 | |
| 238 | // set options |
| 239 | String optionName = params.get("optionName"); |
| 240 | if (optionName != null && !optionName.isEmpty()) { |
| 241 | String optionValue = params.get("optionValue"); |
| 242 | if (optionValue == null || optionValue.isEmpty()) { |
| 243 | session.getCookies().delete(optionName); |
| 244 | cookies.remove(optionName); |
| 245 | } else { |
| 246 | session.getCookies().set(new Cookie(optionName, |
| 247 | optionValue, "; path=/")); |
| 248 | cookies.put(optionName, optionValue); |
| 249 | } |
| 250 | } |
| 251 | } |
| 252 | |
| 253 | Response rep = null; |
| 254 | if (!login.isSuccess() && (uri.equals("/") // |
| 255 | || uri.startsWith(STORY_URL_BASE) // |
| 256 | || uri.startsWith(VIEWER_URL_BASE) // |
| 257 | || uri.startsWith(LIST_URL))) { |
| 258 | rep = loginPage(login, uri); |
| 259 | } |
| 260 | |
| 261 | if (rep == null) { |
| 262 | try { |
| 263 | if (uri.equals("/")) { |
| 264 | rep = root(session, cookies, whitelist); |
| 265 | } else if (uri.startsWith(LIST_URL)) { |
| 266 | rep = getList(uri, whitelist); |
| 267 | } else if (uri.startsWith(STORY_URL_BASE)) { |
| 268 | rep = getStoryPart(uri, whitelist); |
| 269 | } else if (uri.startsWith(VIEWER_URL_BASE)) { |
| 270 | rep = getViewer(cookies, uri, whitelist); |
| 271 | } else if (uri.equals("/logout")) { |
| 272 | session.getCookies().delete("token"); |
| 273 | cookies.remove("token"); |
| 274 | rep = loginPage(login, uri); |
| 275 | } else { |
| 276 | if (uri.startsWith("/")) |
| 277 | uri = uri.substring(1); |
| 278 | InputStream in = IOUtils.openResource( |
| 279 | WebLibraryServerIndex.class, uri); |
| 280 | if (in != null) { |
| 281 | String mimeType = MIME_PLAINTEXT; |
| 282 | if (uri.endsWith(".css")) { |
| 283 | mimeType = "text/css"; |
| 284 | } else if (uri.endsWith(".html")) { |
| 285 | mimeType = "text/html"; |
| 286 | } else if (uri.endsWith(".js")) { |
| 287 | mimeType = "text/javascript"; |
| 288 | } |
| 289 | rep = newChunkedResponse(Status.OK, mimeType, |
| 290 | in); |
| 291 | } else { |
| 292 | getTraceHandler().trace("404: " + uri); |
| 293 | } |
| 294 | } |
| 295 | |
| 296 | if (rep == null) { |
| 297 | rep = newFixedLengthResponse(Status.NOT_FOUND, |
| 298 | NanoHTTPD.MIME_PLAINTEXT, "Not Found"); |
| 299 | } |
| 300 | } catch (Exception e) { |
| 301 | Instance.getInstance().getTraceHandler().error( |
| 302 | new IOException("Cannot process web request", |
| 303 | e)); |
| 304 | rep = newFixedLengthResponse(Status.INTERNAL_ERROR, |
| 305 | NanoHTTPD.MIME_PLAINTEXT, "An error occured"); |
| 306 | } |
| 307 | } |
| 308 | |
| 309 | return rep; |
| 310 | |
| 311 | // Get status: for story, use "luid" + active map of current |
| 312 | // luids |
| 313 | // map must use a addRef/removeRef and delete at 0 |
| 314 | |
| 315 | // http://localhost:2000/?token=ok |
| 316 | |
| 317 | // |
| 318 | // MetaData meta = new MetaData(); |
| 319 | // meta.setTitle("Title"); |
| 320 | // meta.setLuid("000"); |
| 321 | // |
| 322 | // JSONObject json = new JSONObject(); |
| 323 | // json.put("", MetaData.class.getName()); |
| 324 | // json.put("title", meta.getTitle()); |
| 325 | // json.put("luid", meta.getLuid()); |
| 326 | // |
| 327 | // return newFixedLengthResponse(json.toString()); |
| 328 | } |
| 329 | }; |
| 330 | |
| 331 | if (ssf != null) { |
| 332 | getTraceHandler().trace("Install SSL on the web server..."); |
| 333 | server.makeSecure(ssf, null); |
| 334 | getTraceHandler().trace("Done."); |
| 335 | } |
| 336 | } |
| 337 | |
| 338 | @Override |
| 339 | public void run() { |
| 340 | try { |
| 341 | server.start(NanoHTTPD.SOCKET_READ_TIMEOUT, false); |
| 342 | } catch (IOException e) { |
| 343 | tracer.error(new IOException("Cannot start the web server", e)); |
| 344 | } |
| 345 | } |
| 346 | |
| 347 | /** |
| 348 | * Start the server (listen on the network for new connections). |
| 349 | * <p> |
| 350 | * Can only be called once. |
| 351 | * <p> |
| 352 | * This call is asynchronous, and will just start a new {@link Thread} on |
| 353 | * itself (see {@link WebLibraryServer#run()}). |
| 354 | */ |
| 355 | public void start() { |
| 356 | new Thread(this).start(); |
| 357 | } |
| 358 | |
| 359 | /** |
| 360 | * The traces handler for this {@link WebLibraryServer}. |
| 361 | * |
| 362 | * @return the traces handler |
| 363 | */ |
| 364 | public TraceHandler getTraceHandler() { |
| 365 | return tracer; |
| 366 | } |
| 367 | |
| 368 | /** |
| 369 | * The traces handler for this {@link WebLibraryServer}. |
| 370 | * |
| 371 | * @param tracer |
| 372 | * the new traces handler |
| 373 | */ |
| 374 | public void setTraceHandler(TraceHandler tracer) { |
| 375 | if (tracer == null) { |
| 376 | tracer = new TraceHandler(false, false, false); |
| 377 | } |
| 378 | |
| 379 | this.tracer = tracer; |
| 380 | } |
| 381 | |
| 382 | private LoginResult login(String who, String token, List<String> subkeys) { |
| 383 | String realKey = Instance.getInstance().getConfig() |
| 384 | .getString(Config.SERVER_KEY); |
| 385 | realKey = realKey == null ? "" : realKey; |
| 386 | return new LoginResult(token, who, realKey, subkeys); |
| 387 | } |
| 388 | |
| 389 | // allow rw/wl |
| 390 | private LoginResult login(String who, String key, String subkey, |
| 391 | List<String> whitelist) { |
| 392 | String realKey = Instance.getInstance().getConfig() |
| 393 | .getString(Config.SERVER_KEY); |
| 394 | |
| 395 | // I don't like NULLs... |
| 396 | realKey = realKey == null ? "" : realKey; |
| 397 | key = key == null ? "" : key; |
| 398 | subkey = subkey == null ? "" : subkey; |
| 399 | |
| 400 | if (!realKey.equals(key)) { |
| 401 | return new LoginResult(null, null, null, false, false, false); |
| 402 | } |
| 403 | |
| 404 | // defaults are positive (as previous versions without the feature) |
| 405 | boolean rw = true; |
| 406 | boolean wl = true; |
| 407 | |
| 408 | if (whitelist.isEmpty()) { |
| 409 | wl = false; |
| 410 | } |
| 411 | |
| 412 | rw = Instance.getInstance().getConfig().getBoolean(Config.SERVER_RW, |
| 413 | rw); |
| 414 | if (!subkey.isEmpty()) { |
| 415 | List<String> allowed = Instance.getInstance().getConfig() |
| 416 | .getList(Config.SERVER_ALLOWED_SUBKEYS); |
| 417 | if (allowed != null && allowed.contains(subkey)) { |
| 418 | if ((subkey + "|").contains("|rw|")) { |
| 419 | rw = true; |
| 420 | } |
| 421 | if ((subkey + "|").contains("|wl|")) { |
| 422 | wl = false; // |wl| = bypass whitelist |
| 423 | } |
| 424 | } else { |
| 425 | return new LoginResult(null, null, null, false, false, false); |
| 426 | } |
| 427 | } |
| 428 | |
| 429 | return new LoginResult(who, key, subkey, true, rw, wl); |
| 430 | } |
| 431 | |
| 432 | private Response loginPage(LoginResult login, String uri) { |
| 433 | StringBuilder builder = new StringBuilder(); |
| 434 | |
| 435 | appendPreHtml(builder, true); |
| 436 | |
| 437 | if (login.isBadLogin()) { |
| 438 | builder.append("<div class='error'>Bad login or password</div>"); |
| 439 | } else if (login.isBadToken()) { |
| 440 | builder.append("<div class='error'>Your session timed out</div>"); |
| 441 | } |
| 442 | |
| 443 | if (uri.equals("/logout")) { |
| 444 | uri = "/"; |
| 445 | } |
| 446 | |
| 447 | builder.append( |
| 448 | "<form method='POST' action='" + uri + "' class='login'>\n"); |
| 449 | builder.append( |
| 450 | "<p>You must be logged into the system to see the stories.</p>"); |
| 451 | builder.append("\t<input type='text' name='login' />\n"); |
| 452 | builder.append("\t<input type='password' name='password' />\n"); |
| 453 | builder.append("\t<input type='submit' value='Login' />\n"); |
| 454 | builder.append("</form>\n"); |
| 455 | |
| 456 | appendPostHtml(builder); |
| 457 | |
| 458 | return NanoHTTPD.newFixedLengthResponse(Status.FORBIDDEN, |
| 459 | NanoHTTPD.MIME_HTML, builder.toString()); |
| 460 | } |
| 461 | |
| 462 | protected Response getList(String uri, List<String> whitelist) |
| 463 | throws IOException { |
| 464 | if (uri.equals("/list/luids")) { |
| 465 | BasicLibrary lib = Instance.getInstance().getLibrary(); |
| 466 | List<MetaData> metas = lib.getList().filter(whitelist, null, null); |
| 467 | List<JSONObject> jsons = new ArrayList<JSONObject>(); |
| 468 | for (MetaData meta : metas) { |
| 469 | jsons.add(JsonIO.toJson(meta)); |
| 470 | } |
| 471 | |
| 472 | return newInputStreamResponse("application/json", |
| 473 | new ByteArrayInputStream( |
| 474 | new JSONArray(jsons).toString().getBytes())); |
| 475 | } |
| 476 | |
| 477 | return NanoHTTPD.newFixedLengthResponse(Status.BAD_REQUEST, |
| 478 | NanoHTTPD.MIME_PLAINTEXT, null); |
| 479 | } |
| 480 | |
| 481 | private Response root(IHTTPSession session, Map<String, String> cookies, |
| 482 | List<String> whitelist) throws IOException { |
| 483 | BasicLibrary lib = Instance.getInstance().getLibrary(); |
| 484 | MetaResultList result = lib.getList(); |
| 485 | result = new MetaResultList(result.filter(whitelist, null, null)); |
| 486 | StringBuilder builder = new StringBuilder(); |
| 487 | |
| 488 | appendPreHtml(builder, true); |
| 489 | |
| 490 | String filter = cookies.get("filter"); |
| 491 | if (filter == null) { |
| 492 | filter = ""; |
| 493 | } |
| 494 | |
| 495 | Map<String, String> params = session.getParms(); |
| 496 | String browser = params.get("browser") == null ? "" |
| 497 | : params.get("browser"); |
| 498 | String browser2 = params.get("browser2") == null ? "" |
| 499 | : params.get("browser2"); |
| 500 | String browser3 = params.get("browser3") == null ? "" |
| 501 | : params.get("browser3"); |
| 502 | |
| 503 | String filterSource = null; |
| 504 | String filterAuthor = null; |
| 505 | String filterTag = null; |
| 506 | |
| 507 | // TODO: javascript in realtime, using visible=false + hide [submit] |
| 508 | |
| 509 | builder.append("<form class='browser'>\n"); |
| 510 | builder.append("<div class='breadcrumbs'>\n"); |
| 511 | |
| 512 | builder.append("\t<select name='browser'>"); |
| 513 | appendOption(builder, 2, "", "", browser); |
| 514 | appendOption(builder, 2, "Sources", "sources", browser); |
| 515 | appendOption(builder, 2, "Authors", "authors", browser); |
| 516 | appendOption(builder, 2, "Tags", "tags", browser); |
| 517 | builder.append("\t</select>\n"); |
| 518 | |
| 519 | if (!browser.isEmpty()) { |
| 520 | builder.append("\t<select name='browser2'>"); |
| 521 | if (browser.equals("sources")) { |
| 522 | filterSource = browser2.isEmpty() ? filterSource : browser2; |
| 523 | // TODO: if 1 group -> no group |
| 524 | appendOption(builder, 2, "", "", browser2); |
| 525 | Map<String, List<String>> sources = result.getSourcesGrouped(); |
| 526 | for (String source : sources.keySet()) { |
| 527 | appendOption(builder, 2, source, source, browser2); |
| 528 | } |
| 529 | } else if (browser.equals("authors")) { |
| 530 | filterAuthor = browser2.isEmpty() ? filterAuthor : browser2; |
| 531 | // TODO: if 1 group -> no group |
| 532 | appendOption(builder, 2, "", "", browser2); |
| 533 | Map<String, List<String>> authors = result.getAuthorsGrouped(); |
| 534 | for (String author : authors.keySet()) { |
| 535 | appendOption(builder, 2, author, author, browser2); |
| 536 | } |
| 537 | } else if (browser.equals("tags")) { |
| 538 | filterTag = browser2.isEmpty() ? filterTag : browser2; |
| 539 | appendOption(builder, 2, "", "", browser2); |
| 540 | for (String tag : result.getTags()) { |
| 541 | appendOption(builder, 2, tag, tag, browser2); |
| 542 | } |
| 543 | } |
| 544 | builder.append("\t</select>\n"); |
| 545 | } |
| 546 | |
| 547 | if (!browser2.isEmpty()) { |
| 548 | if (browser.equals("sources")) { |
| 549 | filterSource = browser3.isEmpty() ? filterSource : browser3; |
| 550 | Map<String, List<String>> sourcesGrouped = result |
| 551 | .getSourcesGrouped(); |
| 552 | List<String> sources = sourcesGrouped.get(browser2); |
| 553 | if (sources != null && !sources.isEmpty()) { |
| 554 | // TODO: single empty value |
| 555 | builder.append("\t<select name='browser3'>"); |
| 556 | appendOption(builder, 2, "", "", browser3); |
| 557 | for (String source : sources) { |
| 558 | appendOption(builder, 2, source, source, browser3); |
| 559 | } |
| 560 | builder.append("\t</select>\n"); |
| 561 | } |
| 562 | } else if (browser.equals("authors")) { |
| 563 | filterAuthor = browser3.isEmpty() ? filterAuthor : browser3; |
| 564 | Map<String, List<String>> authorsGrouped = result |
| 565 | .getAuthorsGrouped(); |
| 566 | List<String> authors = authorsGrouped.get(browser2); |
| 567 | if (authors != null && !authors.isEmpty()) { |
| 568 | // TODO: single empty value |
| 569 | builder.append("\t<select name='browser3'>"); |
| 570 | appendOption(builder, 2, "", "", browser3); |
| 571 | for (String author : authors) { |
| 572 | appendOption(builder, 2, author, author, browser3); |
| 573 | } |
| 574 | builder.append("\t</select>\n"); |
| 575 | } |
| 576 | } |
| 577 | } |
| 578 | |
| 579 | builder.append("\t<input type='submit' value='Select'/>\n"); |
| 580 | builder.append("</div>\n"); |
| 581 | |
| 582 | // TODO: javascript in realtime, using visible=false + hide [submit] |
| 583 | builder.append("<div class='filter'>\n"); |
| 584 | builder.append("\tFilter: \n"); |
| 585 | builder.append( |
| 586 | "\t<input name='optionName' type='hidden' value='filter' />\n"); |
| 587 | builder.append("\t<input name='optionValue' type='text' value='" |
| 588 | + filter + "' place-holder='...' />\n"); |
| 589 | builder.append( |
| 590 | "\t<input name='submit' type='submit' value='Filter' />\n"); |
| 591 | builder.append("</div>\n"); |
| 592 | builder.append("</form>\n"); |
| 593 | |
| 594 | builder.append("\t<div class='books'>"); |
| 595 | for (MetaData meta : result.getMetas()) { |
| 596 | if (!filter.isEmpty() && !meta.getTitle().toLowerCase() |
| 597 | .contains(filter.toLowerCase())) { |
| 598 | continue; |
| 599 | } |
| 600 | |
| 601 | // TODO Sub sources |
| 602 | if (filterSource != null |
| 603 | && !filterSource.equals(meta.getSource())) { |
| 604 | continue; |
| 605 | } |
| 606 | |
| 607 | // TODO: sub authors |
| 608 | if (filterAuthor != null |
| 609 | && !filterAuthor.equals(meta.getAuthor())) { |
| 610 | continue; |
| 611 | } |
| 612 | |
| 613 | if (filterTag != null && !meta.getTags().contains(filterTag)) { |
| 614 | continue; |
| 615 | } |
| 616 | |
| 617 | builder.append("<div class='book_line'>"); |
| 618 | builder.append("<a href='"); |
| 619 | builder.append(getViewUrl(meta.getLuid(), 0, null)); |
| 620 | builder.append("'"); |
| 621 | builder.append(" class='link'>"); |
| 622 | |
| 623 | if (lib.isCached(meta.getLuid())) { |
| 624 | // â—‰ = ◉ |
| 625 | builder.append( |
| 626 | "<span class='cache_icon cached'>◉</span>"); |
| 627 | } else { |
| 628 | // â—‹ = ○ |
| 629 | builder.append( |
| 630 | "<span class='cache_icon uncached'>○</span>"); |
| 631 | } |
| 632 | builder.append("<span class='luid'>"); |
| 633 | builder.append(meta.getLuid()); |
| 634 | builder.append("</span>"); |
| 635 | builder.append("<span class='title'>"); |
| 636 | builder.append(meta.getTitle()); |
| 637 | builder.append("</span>"); |
| 638 | builder.append("<span class='author'>"); |
| 639 | if (meta.getAuthor() != null && !meta.getAuthor().isEmpty()) { |
| 640 | builder.append("(").append(meta.getAuthor()).append(")"); |
| 641 | } |
| 642 | builder.append("</span>"); |
| 643 | builder.append("</a></div>\n"); |
| 644 | } |
| 645 | builder.append("</div>"); |
| 646 | |
| 647 | appendPostHtml(builder); |
| 648 | return NanoHTTPD.newFixedLengthResponse(builder.toString()); |
| 649 | } |
| 650 | |
| 651 | // /story/luid/chapter/para <-- text/image |
| 652 | // /story/luid/cover <-- image |
| 653 | // /story/luid/metadata <-- json |
| 654 | private Response getStoryPart(String uri, List<String> whitelist) { |
| 655 | String[] cover = uri.split("/"); |
| 656 | int off = 2; |
| 657 | |
| 658 | if (cover.length < off + 2) { |
| 659 | return NanoHTTPD.newFixedLengthResponse(Status.BAD_REQUEST, |
| 660 | NanoHTTPD.MIME_PLAINTEXT, null); |
| 661 | } |
| 662 | |
| 663 | String luid = cover[off + 0]; |
| 664 | String chapterStr = cover[off + 1]; |
| 665 | String imageStr = cover.length < off + 3 ? null : cover[off + 2]; |
| 666 | |
| 667 | // 1-based (0 = desc) |
| 668 | int chapter = 0; |
| 669 | if (chapterStr != null && !"cover".equals(chapterStr) |
| 670 | && !"metadata".equals(chapterStr)) { |
| 671 | try { |
| 672 | chapter = Integer.parseInt(chapterStr); |
| 673 | if (chapter < 0) { |
| 674 | throw new NumberFormatException(); |
| 675 | } |
| 676 | } catch (NumberFormatException e) { |
| 677 | return NanoHTTPD.newFixedLengthResponse(Status.BAD_REQUEST, |
| 678 | NanoHTTPD.MIME_PLAINTEXT, "Chapter is not valid"); |
| 679 | } |
| 680 | } |
| 681 | |
| 682 | // 1-based |
| 683 | int paragraph = 1; |
| 684 | if (imageStr != null) { |
| 685 | try { |
| 686 | paragraph = Integer.parseInt(imageStr); |
| 687 | if (paragraph < 0) { |
| 688 | throw new NumberFormatException(); |
| 689 | } |
| 690 | } catch (NumberFormatException e) { |
| 691 | return NanoHTTPD.newFixedLengthResponse(Status.BAD_REQUEST, |
| 692 | NanoHTTPD.MIME_PLAINTEXT, "Paragraph is not valid"); |
| 693 | } |
| 694 | } |
| 695 | |
| 696 | String mimeType = NanoHTTPD.MIME_PLAINTEXT; |
| 697 | InputStream in = null; |
| 698 | try { |
| 699 | if ("cover".equals(chapterStr)) { |
| 700 | Image img = getCover(luid, whitelist); |
| 701 | if (img != null) { |
| 702 | in = img.newInputStream(); |
| 703 | } |
| 704 | } else if ("metadata".equals(chapterStr)) { |
| 705 | MetaData meta = meta(luid, whitelist); |
| 706 | JSONObject json = JsonIO.toJson(meta); |
| 707 | mimeType = "application/json"; |
| 708 | in = new ByteArrayInputStream(json.toString().getBytes()); |
| 709 | } else { |
| 710 | Story story = story(luid, whitelist); |
| 711 | if (story != null) { |
| 712 | if (chapter == 0) { |
| 713 | StringBuilder builder = new StringBuilder(); |
| 714 | for (Paragraph p : story.getMeta().getResume()) { |
| 715 | if (builder.length() == 0) { |
| 716 | builder.append("\n"); |
| 717 | } |
| 718 | builder.append(p.getContent()); |
| 719 | } |
| 720 | |
| 721 | in = new ByteArrayInputStream( |
| 722 | builder.toString().getBytes("utf-8")); |
| 723 | } else { |
| 724 | Paragraph para = story.getChapters().get(chapter - 1) |
| 725 | .getParagraphs().get(paragraph - 1); |
| 726 | Image img = para.getContentImage(); |
| 727 | if (para.getType() == ParagraphType.IMAGE) { |
| 728 | // TODO: get correct image type |
| 729 | mimeType = "image/png"; |
| 730 | in = img.newInputStream(); |
| 731 | } else { |
| 732 | in = new ByteArrayInputStream( |
| 733 | para.getContent().getBytes("utf-8")); |
| 734 | } |
| 735 | } |
| 736 | } |
| 737 | } |
| 738 | } catch (IndexOutOfBoundsException e) { |
| 739 | return NanoHTTPD.newFixedLengthResponse(Status.NOT_FOUND, |
| 740 | NanoHTTPD.MIME_PLAINTEXT, |
| 741 | "Chapter or paragraph does not exist"); |
| 742 | } catch (IOException e) { |
| 743 | Instance.getInstance().getTraceHandler() |
| 744 | .error(new IOException("Cannot get image: " + uri, e)); |
| 745 | return NanoHTTPD.newFixedLengthResponse(Status.INTERNAL_ERROR, |
| 746 | NanoHTTPD.MIME_PLAINTEXT, "Error when processing request"); |
| 747 | } |
| 748 | |
| 749 | return newInputStreamResponse(mimeType, in); |
| 750 | } |
| 751 | |
| 752 | private Response getViewer(Map<String, String> cookies, String uri, |
| 753 | List<String> whitelist) { |
| 754 | String[] cover = uri.split("/"); |
| 755 | int off = 2; |
| 756 | |
| 757 | if (cover.length < off + 2) { |
| 758 | return NanoHTTPD.newFixedLengthResponse(Status.BAD_REQUEST, |
| 759 | NanoHTTPD.MIME_PLAINTEXT, null); |
| 760 | } |
| 761 | |
| 762 | String type = cover[off + 0]; |
| 763 | String luid = cover[off + 1]; |
| 764 | String chapterStr = cover.length < off + 3 ? null : cover[off + 2]; |
| 765 | String paragraphStr = cover.length < off + 4 ? null : cover[off + 3]; |
| 766 | |
| 767 | // 1-based (0 = desc) |
| 768 | int chapter = -1; |
| 769 | if (chapterStr != null) { |
| 770 | try { |
| 771 | chapter = Integer.parseInt(chapterStr); |
| 772 | if (chapter < 0) { |
| 773 | throw new NumberFormatException(); |
| 774 | } |
| 775 | } catch (NumberFormatException e) { |
| 776 | return NanoHTTPD.newFixedLengthResponse(Status.BAD_REQUEST, |
| 777 | NanoHTTPD.MIME_PLAINTEXT, "Chapter is not valid"); |
| 778 | } |
| 779 | } |
| 780 | |
| 781 | // 1-based |
| 782 | int paragraph = 0; |
| 783 | if (paragraphStr != null) { |
| 784 | try { |
| 785 | paragraph = Integer.parseInt(paragraphStr); |
| 786 | if (paragraph <= 0) { |
| 787 | throw new NumberFormatException(); |
| 788 | } |
| 789 | } catch (NumberFormatException e) { |
| 790 | return NanoHTTPD.newFixedLengthResponse(Status.BAD_REQUEST, |
| 791 | NanoHTTPD.MIME_PLAINTEXT, "Paragraph is not valid"); |
| 792 | } |
| 793 | } |
| 794 | |
| 795 | try { |
| 796 | Story story = story(luid, whitelist); |
| 797 | if (story == null) { |
| 798 | return NanoHTTPD.newFixedLengthResponse(Status.NOT_FOUND, |
| 799 | NanoHTTPD.MIME_PLAINTEXT, "Story not found"); |
| 800 | } |
| 801 | |
| 802 | StringBuilder builder = new StringBuilder(); |
| 803 | appendPreHtml(builder, false); |
| 804 | |
| 805 | if (chapter < 0) { |
| 806 | builder.append(story); |
| 807 | } else { |
| 808 | if (chapter == 0) { |
| 809 | // TODO: description |
| 810 | chapter = 1; |
| 811 | } |
| 812 | |
| 813 | Chapter chap = null; |
| 814 | try { |
| 815 | chap = story.getChapters().get(chapter - 1); |
| 816 | } catch (IndexOutOfBoundsException e) { |
| 817 | return NanoHTTPD.newFixedLengthResponse(Status.NOT_FOUND, |
| 818 | NanoHTTPD.MIME_PLAINTEXT, "Chapter not found"); |
| 819 | } |
| 820 | |
| 821 | if (story.getMeta().isImageDocument() && paragraph <= 0) { |
| 822 | paragraph = 1; |
| 823 | } |
| 824 | |
| 825 | String first, previous, next, last; |
| 826 | String content; |
| 827 | |
| 828 | if (paragraph <= 0) { |
| 829 | first = getViewUrl(luid, 1, null); |
| 830 | previous = getViewUrl(luid, (Math.max(chapter - 1, 1)), |
| 831 | null); |
| 832 | next = getViewUrl(luid, |
| 833 | (Math.min(chapter + 1, story.getChapters().size())), |
| 834 | null); |
| 835 | last = getViewUrl(luid, story.getChapters().size(), null); |
| 836 | |
| 837 | content = "<div class='viewer text'>\n" |
| 838 | + new TextOutput(false).convert(chap, true) |
| 839 | + "</div>\n"; |
| 840 | } else { |
| 841 | first = getViewUrl(luid, chapter, 1); |
| 842 | previous = getViewUrl(luid, chapter, |
| 843 | (Math.max(paragraph - 1, 1))); |
| 844 | next = getViewUrl(luid, chapter, (Math.min(paragraph + 1, |
| 845 | chap.getParagraphs().size()))); |
| 846 | last = getViewUrl(luid, chapter, |
| 847 | chap.getParagraphs().size()); |
| 848 | |
| 849 | Paragraph para = null; |
| 850 | try { |
| 851 | para = chap.getParagraphs().get(paragraph - 1); |
| 852 | } catch (IndexOutOfBoundsException e) { |
| 853 | return NanoHTTPD.newFixedLengthResponse( |
| 854 | Status.NOT_FOUND, NanoHTTPD.MIME_PLAINTEXT, |
| 855 | "Paragraph not found"); |
| 856 | } |
| 857 | |
| 858 | if (para.getType() == ParagraphType.IMAGE) { |
| 859 | String zoomStyle = "max-width: 100%;"; |
| 860 | String zoomOption = cookies.get("zoom"); |
| 861 | if (zoomOption != null && !zoomOption.isEmpty()) { |
| 862 | if (zoomOption.equals("real")) { |
| 863 | zoomStyle = ""; |
| 864 | } else if (zoomOption.equals("width")) { |
| 865 | zoomStyle = "max-width: 100%;"; |
| 866 | } else if (zoomOption.equals("height")) { |
| 867 | // see height of navbar + optionbar |
| 868 | zoomStyle = "max-height: calc(100% - 128px);"; |
| 869 | } |
| 870 | } |
| 871 | content = String.format("" // |
| 872 | + "<a class='viewer link' href='%s'>" // |
| 873 | + "<img class='viewer img' style='%s' src='%s'/>" |
| 874 | + "</a>", // |
| 875 | next, // |
| 876 | zoomStyle, // |
| 877 | getStoryUrl(luid, chapter, paragraph)); |
| 878 | } else { |
| 879 | content = para.getContent(); |
| 880 | } |
| 881 | |
| 882 | } |
| 883 | |
| 884 | builder.append(String.format("" // |
| 885 | + "<div class='bar navbar'>\n" // |
| 886 | + "\t<a class='button first' href='%s'><<</a>\n"// |
| 887 | + "\t<a class='button previous' href='%s'><</a>\n"// |
| 888 | + "\t<a class='button next' href='%s'>></a>\n"// |
| 889 | + "\t<a class='button last' href='%s'>>></a>\n"// |
| 890 | + "</div>\n" // |
| 891 | + "%s", // |
| 892 | first, // |
| 893 | previous, // |
| 894 | next, // |
| 895 | last, // |
| 896 | content // |
| 897 | )); |
| 898 | |
| 899 | builder.append("<div class='bar optionbar "); |
| 900 | if (paragraph > 0) { |
| 901 | builder.append("s4"); |
| 902 | } else { |
| 903 | builder.append("s1"); |
| 904 | } |
| 905 | builder.append("'>\n"); |
| 906 | builder.append( |
| 907 | " <a class='button back' href='/'>BACK</a>\n"); |
| 908 | |
| 909 | if (paragraph > 0) { |
| 910 | builder.append(String.format("" // |
| 911 | + "\t<a class='button zoomreal' href='%s'>REAL</a>\n"// |
| 912 | + "\t<a class='button zoomwidth' href='%s'>WIDTH</a>\n"// |
| 913 | + "\t<a class='button zoomheight' href='%s'>HEIGHT</a>\n"// |
| 914 | + "</div>\n", // |
| 915 | uri + "?optionName=zoom&optionValue=real", // |
| 916 | uri + "?optionName=zoom&optionValue=width", // |
| 917 | uri + "?optionName=zoom&optionValue=height" // |
| 918 | )); |
| 919 | } |
| 920 | } |
| 921 | |
| 922 | appendPostHtml(builder); |
| 923 | return NanoHTTPD.newFixedLengthResponse(Status.OK, |
| 924 | NanoHTTPD.MIME_HTML, builder.toString()); |
| 925 | } catch (IOException e) { |
| 926 | Instance.getInstance().getTraceHandler() |
| 927 | .error(new IOException("Cannot get image: " + uri, e)); |
| 928 | return NanoHTTPD.newFixedLengthResponse(Status.INTERNAL_ERROR, |
| 929 | NanoHTTPD.MIME_PLAINTEXT, "Error when processing request"); |
| 930 | } |
| 931 | } |
| 932 | |
| 933 | private Response newInputStreamResponse(String mimeType, InputStream in) { |
| 934 | if (in == null) { |
| 935 | return NanoHTTPD.newFixedLengthResponse(Status.NO_CONTENT, "", |
| 936 | null); |
| 937 | } |
| 938 | return NanoHTTPD.newChunkedResponse(Status.OK, mimeType, in); |
| 939 | } |
| 940 | |
| 941 | private String getContentOf(String file) { |
| 942 | InputStream in = IOUtils.openResource(WebLibraryServerIndex.class, |
| 943 | file); |
| 944 | if (in != null) { |
| 945 | try { |
| 946 | return IOUtils.readSmallStream(in); |
| 947 | } catch (IOException e) { |
| 948 | Instance.getInstance().getTraceHandler().error( |
| 949 | new IOException("Cannot get file: index.pre.html", e)); |
| 950 | } |
| 951 | } |
| 952 | |
| 953 | return ""; |
| 954 | } |
| 955 | |
| 956 | private String getViewUrl(String luid, int chap, Integer para) { |
| 957 | return VIEWER_URL // |
| 958 | .replace("{luid}", luid) // |
| 959 | .replace("{chap}", Integer.toString(chap)) // |
| 960 | .replace("/{para}", |
| 961 | para == null ? "" : "/" + Integer.toString(para)); |
| 962 | } |
| 963 | |
| 964 | private String getStoryUrl(String luid, int chap, Integer para) { |
| 965 | return STORY_URL // |
| 966 | .replace("{luid}", luid) // |
| 967 | .replace("{chap}", Integer.toString(chap)) // |
| 968 | .replace("{para}", para == null ? "" : Integer.toString(para)); |
| 969 | } |
| 970 | |
| 971 | private String getStoryUrlCover(String luid) { |
| 972 | return STORY_URL_COVER // |
| 973 | .replace("{luid}", luid); |
| 974 | } |
| 975 | |
| 976 | private MetaData meta(String luid, List<String> whitelist) |
| 977 | throws IOException { |
| 978 | BasicLibrary lib = Instance.getInstance().getLibrary(); |
| 979 | MetaData meta = lib.getInfo(luid); |
| 980 | if (!whitelist.isEmpty() && !whitelist.contains(meta.getSource())) { |
| 981 | return null; |
| 982 | } |
| 983 | |
| 984 | return meta; |
| 985 | } |
| 986 | |
| 987 | private Image getCover(String luid, List<String> whitelist) |
| 988 | throws IOException { |
| 989 | MetaData meta = meta(luid, whitelist); |
| 990 | if (meta != null) { |
| 991 | return meta.getCover(); |
| 992 | } |
| 993 | |
| 994 | return null; |
| 995 | } |
| 996 | |
| 997 | // NULL if not whitelist OK or if not found |
| 998 | private Story story(String luid, List<String> whitelist) |
| 999 | throws IOException { |
| 1000 | synchronized (storyCache) { |
| 1001 | if (storyCache.containsKey(luid)) { |
| 1002 | Story story = storyCache.get(luid); |
| 1003 | if (!whitelist.isEmpty() |
| 1004 | && !whitelist.contains(story.getMeta().getSource())) { |
| 1005 | return null; |
| 1006 | } |
| 1007 | |
| 1008 | return story; |
| 1009 | } |
| 1010 | } |
| 1011 | |
| 1012 | Story story = null; |
| 1013 | MetaData meta = meta(luid, whitelist); |
| 1014 | if (meta != null) { |
| 1015 | BasicLibrary lib = Instance.getInstance().getLibrary(); |
| 1016 | story = lib.getStory(luid, null); |
| 1017 | long size = sizeOf(story); |
| 1018 | |
| 1019 | synchronized (storyCache) { |
| 1020 | // Could have been added by another request |
| 1021 | if (!storyCache.containsKey(luid)) { |
| 1022 | while (!storyCacheOrder.isEmpty() |
| 1023 | && storyCacheSize + size > maxStoryCacheSize) { |
| 1024 | String oldestLuid = storyCacheOrder.removeFirst(); |
| 1025 | Story oldestStory = storyCache.remove(oldestLuid); |
| 1026 | maxStoryCacheSize -= sizeOf(oldestStory); |
| 1027 | } |
| 1028 | |
| 1029 | storyCacheOrder.add(luid); |
| 1030 | storyCache.put(luid, story); |
| 1031 | } |
| 1032 | } |
| 1033 | } |
| 1034 | |
| 1035 | return story; |
| 1036 | } |
| 1037 | |
| 1038 | private long sizeOf(Story story) { |
| 1039 | long size = 0; |
| 1040 | for (Chapter chap : story) { |
| 1041 | for (Paragraph para : chap) { |
| 1042 | if (para.getType() == ParagraphType.IMAGE) { |
| 1043 | size += para.getContentImage().getSize(); |
| 1044 | } else { |
| 1045 | size += para.getContent().length(); |
| 1046 | } |
| 1047 | } |
| 1048 | } |
| 1049 | |
| 1050 | return size; |
| 1051 | } |
| 1052 | |
| 1053 | private void appendPreHtml(StringBuilder builder, boolean banner) { |
| 1054 | String favicon = "favicon.ico"; |
| 1055 | String icon = Instance.getInstance().getUiConfig() |
| 1056 | .getString(UiConfig.PROGRAM_ICON); |
| 1057 | if (icon != null) { |
| 1058 | favicon = "icon_" + icon.replace("-", "_") + ".png"; |
| 1059 | } |
| 1060 | |
| 1061 | builder.append( |
| 1062 | getContentOf("index.pre.html").replace("favicon.ico", favicon)); |
| 1063 | |
| 1064 | if (banner) { |
| 1065 | builder.append("<div class='banner'>\n"); |
| 1066 | builder.append("\t<img class='ico' src='") // |
| 1067 | .append(favicon) // |
| 1068 | .append("'/>\n"); |
| 1069 | builder.append("\t<h1>Fanfix</h1>\n"); |
| 1070 | builder.append("\t<h2>") // |
| 1071 | .append(Version.getCurrentVersion()) // |
| 1072 | .append("</h2>\n"); |
| 1073 | builder.append("</div>\n"); |
| 1074 | } |
| 1075 | } |
| 1076 | |
| 1077 | private void appendPostHtml(StringBuilder builder) { |
| 1078 | builder.append(getContentOf("index.post.html")); |
| 1079 | } |
| 1080 | |
| 1081 | private void appendOption(StringBuilder builder, int depth, String name, |
| 1082 | String value, String selected) { |
| 1083 | for (int i = 0; i < depth; i++) { |
| 1084 | builder.append("\t"); |
| 1085 | } |
| 1086 | builder.append("<option value='").append(value).append("'"); |
| 1087 | if (value.equals(selected)) { |
| 1088 | builder.append(" selected='selected'"); |
| 1089 | } |
| 1090 | builder.append(">").append(name).append("</option>\n"); |
| 1091 | } |
| 1092 | } |