fixup

openmrs · Oct 9, 2017 · 8bc1b8d · 8bc1b8d
1 parent 596217f
commit 8bc1b8d
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/...va/org/openmrs/module/webservices/rest/web/controller/SwaggerSpecificationController.java b/...va/org/openmrs/module/webservices/rest/web/controller/SwaggerSpecificationController.java
@@ -9,6 +9,7 @@
  */
 package org.openmrs.module.webservices.rest.web.controller;
 
+import com.google.common.net.HttpHeaders;
 import io.swagger.models.Scheme;
 import org.openmrs.module.webservices.docs.swagger.SwaggerSpecificationCreator;
 import org.springframework.stereotype.Controller;
@@ -31,10 +32,15 @@ String getSwaggerSpecification(HttpServletRequest request) throws Exception {
 			host += ":" + request.getServerPort();
 		}
 
+		String scheme = request.getHeader(HttpHeaders.X_FORWARDED_PROTO);
+		if (scheme == null) {
+			scheme = request.getScheme();
+		}
+
 		return new SwaggerSpecificationCreator()
 		        .host(host)
 		        .basePath(request.getContextPath() + "/ws/rest/v1")
-		        .scheme(Scheme.forValue(request.getScheme()))
+		        .scheme(Scheme.forValue(scheme))
 
 		        .BuildJSON();
 	}