Added Android code

[wl-app.git] / Android / r2-streamer / r2-server / src / main / java / org / readium / r2_streamer / server / handler / SearchQueryHandler.java

diff --git a/Android/r2-streamer/r2-server/src/main/java/org/readium/r2_streamer/server/handler/SearchQueryHandler.java b/Android/r2-streamer/r2-server/src/main/java/org/readium/r2_streamer/server/handler/SearchQueryHandler.java
new file mode 100755 (executable)
index 0000000..f5ec986
--- /dev/null
+++ b/Android/r2-streamer/r2-server/src/main/java/org/readium/r2_streamer/server/handler/SearchQueryHandler.java
@@ -0,0 +1,153 @@
+package org.readium.r2_streamer.server.handler;
+
+//import android.support.annotation.Nullable;
+//import android.util.Log;
+
+import org.readium.r2_streamer.fetcher.EpubFetcher;
+import org.readium.r2_streamer.fetcher.EpubFetcherException;
+import org.readium.r2_streamer.model.publication.link.Link;
+import org.readium.r2_streamer.model.searcher.SearchQueryResults;
+import org.readium.r2_streamer.model.searcher.SearchResult;
+import org.readium.r2_streamer.server.ResponseStatus;
+import com.fasterxml.jackson.core.JsonProcessingException;
+import com.fasterxml.jackson.databind.ObjectMapper;
+
+import org.jsoup.Jsoup;
+import org.jsoup.nodes.Document;
+import org.jsoup.nodes.Element;
+
+import java.util.Map;
+import java.util.regex.Matcher;
+import java.util.regex.Pattern;
+
+import fi.iki.elonen.NanoHTTPD;
+import fi.iki.elonen.NanoHTTPD.IHTTPSession;
+import fi.iki.elonen.NanoHTTPD.Method;
+import fi.iki.elonen.NanoHTTPD.Response;
+import fi.iki.elonen.NanoHTTPD.Response.IStatus;
+import fi.iki.elonen.NanoHTTPD.Response.Status;
+import fi.iki.elonen.router.RouterNanoHTTPD.DefaultHandler;
+import fi.iki.elonen.router.RouterNanoHTTPD.UriResource;
+
+/**
+ * Created by Shrikant Badwaik on 17-Feb-17.
+ */
+
+public class SearchQueryHandler extends DefaultHandler {
+    private static final String TAG = "SearchQueryHandler";
+    private Response response;
+
+    public SearchQueryHandler() {
+    }
+
+    @Override
+    public String getText() {
+        return ResponseStatus.FAILURE_RESPONSE;
+    }
+
+    @Override
+    public String getMimeType() {
+        return "application/json";
+    }
+
+    @Override
+    public IStatus getStatus() {
+        return Status.NOT_ACCEPTABLE;
+    }
+
+    @Override
+    public Response get(UriResource uriResource, Map<String, String> urlParams, IHTTPSession session) {
+        Method method = session.getMethod();
+        String uri = session.getUri();
+
+        //Log.d(TAG, "Method: " + method + ", Url: " + uri);
+
+        try {
+            EpubFetcher fetcher = uriResource.initParameter(EpubFetcher.class);
+
+            String queryParameter = session.getQueryParameterString();
+            int startIndex = queryParameter.indexOf("=");
+            String searchQueryPath = queryParameter.substring(startIndex + 1);
+
+            SearchQueryResults searchQueryResults = new SearchQueryResults();
+            for (Link link : fetcher.publication.spines) {
+                String htmlText = fetcher.getData(link.getHref());
+                if (searchQueryPath.contains("%20")) {
+                    searchQueryPath = searchQueryPath.replaceAll("%20", " ");
+                }
+                Pattern pattern = Pattern.compile(searchQueryPath, Pattern.CASE_INSENSITIVE);
+                Matcher matcher = pattern.matcher(htmlText);
+                while (matcher.find()) {
+                    int start = matcher.start();
+
+                    String prev = getTextBefore(start, htmlText);
+                    //String prev = htmlText.substring(start - 20, start);
+                    String next = getTextAfter(start, searchQueryPath, htmlText);
+                    //String next = htmlText.substring(start + searchQueryPath.length(), (start + searchQueryPath.length()) + 20);
+                    String match = prev.concat(searchQueryPath).concat(next);
+
+                    SearchResult searchResult = new SearchResult();
+                    searchResult.setSearchIndex(start);
+                    searchResult.setResource(link.getHref());
+                    searchResult.setSearchQuery(searchQueryPath);
+                    searchResult.setMatchString(match);
+                    searchResult.setTextBefore(prev);
+                    searchResult.setTextAfter(next);
+
+                    String title = parseChapterTitle(htmlText);
+                    if (title != null) {
+                        searchResult.setTitle(title);
+                    } else {
+                        searchResult.setTitle("Title not available");
+                    }
+                    searchQueryResults.searchResultList.add(searchResult);
+                }
+            }
+            ObjectMapper objectMapper = new ObjectMapper();
+            String json = objectMapper.writeValueAsString(searchQueryResults);
+            response = NanoHTTPD.newFixedLengthResponse(Status.OK, getMimeType(), json);
+            return response;
+        } catch (EpubFetcherException | JsonProcessingException e) {
+            e.printStackTrace();
+            return NanoHTTPD.newFixedLengthResponse(Status.INTERNAL_ERROR, getMimeType(), ResponseStatus.FAILURE_RESPONSE);
+        }
+    }
+
+    //@Nullable
+    private String parseChapterTitle(String searchData) {
+        Document document = Jsoup.parse(searchData);
+        Element h1Element = document.select("h1").first();
+        if (h1Element != null) {
+            return h1Element.text();
+        } else {
+            Element h2Element = document.select("h2").first();
+            if (h2Element != null) {
+                return h2Element.text();
+            } else {
+                Element titleElement = document.select("title").first();
+                if (titleElement != null) {
+                    return titleElement.text();
+                }
+            }
+        }
+        return null;
+    }
+
+    private String getTextBefore(int start, String htmlString) {
+        int beginIndex = start - 20;
+        if (beginIndex >= 0 && beginIndex < htmlString.length()) {
+            return htmlString.substring(beginIndex, start);
+        } else {
+            return htmlString.substring(0, start);
+        }
+    }
+
+    private String getTextAfter(int start, String searchQueryPath, String htmlString) {
+        int beginIndex = start + searchQueryPath.length();
+        if ((beginIndex + 20) > htmlString.length()) {
+            return htmlString.substring(beginIndex);
+        } else {
+            return htmlString.substring(beginIndex, beginIndex + 20);
+        }
+    }
+}
\ No newline at end of file