Merge pull request #48 from kgujral/master

Https config for issue #44

Author: greengerong

README.md

@@ -54,6 +54,9 @@ example: someproxy,someproxy1
 Important for servers behind reverse proxy that need the public url to be used for pre-rendering.
 We usually set the original url in an http header which is added by the reverse proxy (similar to the more standard `x-forwarded-proto` and `x-forwarded-for`)
 
+### protocol
+If you specifically want to make sure that the Prerender service queries using https or http protocol, you can set the init-param `protocol` to `https` or `http` respectively.
+
 ### Using your own prerender service
 
 If you've deployed the prerender service on your own, set the `PRERENDER_SERVICE_URL` environment variable so that this package points there instead. Otherwise, it will default to the service already deployed at `http://service.prerender.io/`
@@ -109,4 +112,4 @@ project demo test url:
 
 ## License
 
-The MIT License (MIT)
+The MIT License (MIT)

src/main/java/com/github/greengerong/PreRenderSEOFilter.java

@@ -1,20 +1,23 @@
 package com.github.greengerong;
 
-import com.google.common.collect.Lists;
-import com.google.common.collect.Maps;
-
-import javax.servlet.*;
-import javax.servlet.http.HttpServletRequest;
-import javax.servlet.http.HttpServletResponse;
 import java.io.IOException;
-import java.util.HashMap;
 import java.util.List;
 import java.util.Map;
+import javax.servlet.Filter;
+import javax.servlet.FilterChain;
+import javax.servlet.FilterConfig;
+import javax.servlet.ServletException;
+import javax.servlet.ServletRequest;
+import javax.servlet.ServletResponse;
+import javax.servlet.http.HttpServletRequest;
+import javax.servlet.http.HttpServletResponse;
+import com.google.common.collect.Lists;
+import com.google.common.collect.Maps;
 
 public class PreRenderSEOFilter implements Filter {
     public static final List<String> PARAMETER_NAMES = Lists.newArrayList("preRenderEventHandler", "proxy", "proxyPort",
             "prerenderToken", "forwardedURLHeader", "crawlerUserAgents", "extensionsToIgnore", "whitelist",
-            "blacklist", "prerenderServiceUrl");
+            "blacklist", "prerenderServiceUrl", "protocol");
     private PrerenderSeoService prerenderSeoService;
 
     @Override

src/main/java/com/github/greengerong/PrerenderConfig.java

@@ -79,6 +79,10 @@ public String getPrerenderToken() {
     public String getForwardedURLHeader() {
         return config.get("forwardedURLHeader");
     }
+    
+    public String getProtocol() {
+        return config.get("protocol");
+    }
 
     public List<String> getCrawlerUserAgents() {
         List<String> crawlerUserAgents = Lists.newArrayList("baiduspider",

src/main/java/com/github/greengerong/PrerenderSeoService.java

@@ -179,12 +179,16 @@ private void copyRequestHeaders(HttpServletRequest servletRequest, HttpRequest p
     }
 
     private String getRequestURL(HttpServletRequest request) {
-        if (prerenderConfig.getForwardedURLHeader() != null) {
+        if (StringUtils.isNotEmpty(prerenderConfig.getForwardedURLHeader())) {
             String url = request.getHeader(prerenderConfig.getForwardedURLHeader());
-            if (url != null) {
+            if (StringUtils.isNotEmpty(url)) {
                 return url;
             }
         }
+        if (StringUtils.isNotEmpty(prerenderConfig.getProtocol())) {
+            String url = request.getRequestURL().toString();
+            return url.replace(request.getScheme(), prerenderConfig.getProtocol());
+        }
         return request.getRequestURL().toString();
     }