From 27069b82d6c5ff3f70f4b58df871f3d52ab8f698 Mon Sep 17 00:00:00 2001
From: Ito Zapata <ito.zapata@gmail.com>
Date: Sat, 19 Oct 2013 17:58:35 -0700
Subject: [PATCH] #112 - Added support for X-Forwarded-Ssl header during link
 construction.

---
 .../hateoas/mvc/ControllerLinkBuilder.java    |  9 ++++-
 .../mvc/ControllerLinkBuilderUnitTest.java    | 37 +++++++++++++++++++
 2 files changed, 45 insertions(+), 1 deletion(-)

diff --git a/src/main/java/org/springframework/hateoas/mvc/ControllerLinkBuilder.java b/src/main/java/org/springframework/hateoas/mvc/ControllerLinkBuilder.java
index 1a926248..43b7138d 100755
--- a/src/main/java/org/springframework/hateoas/mvc/ControllerLinkBuilder.java
+++ b/src/main/java/org/springframework/hateoas/mvc/ControllerLinkBuilder.java
@@ -158,7 +158,8 @@ public class ControllerLinkBuilder extends LinkBuilderSupport<ControllerLinkBuil
 
 	/**
 	 * Returns a {@link UriComponentsBuilder} obtained from the current servlet mapping with the host tweaked in case the
-	 * request contains an {@code X-Forwarded-Host} header.
+	 * request contains an {@code X-Forwarded-Host} header and the scheme tweaked in case the request contains an
+	 * {@code X-Forwarded-Ssl} header
 	 * 
 	 * @return
 	 */
@@ -167,6 +168,12 @@ public class ControllerLinkBuilder extends LinkBuilderSupport<ControllerLinkBuil
 		HttpServletRequest request = getCurrentRequest();
 		ServletUriComponentsBuilder builder = ServletUriComponentsBuilder.fromServletMapping(request);
 
+		String forwardedSsl = request.getHeader("X-Forwarded-Ssl");
+
+		if (StringUtils.hasText(forwardedSsl) && forwardedSsl.equalsIgnoreCase("on")) {
+			builder.scheme("https");
+		}
+
 		String header = request.getHeader("X-Forwarded-Host");
 
 		if (!StringUtils.hasText(header)) {
diff --git a/src/test/java/org/springframework/hateoas/mvc/ControllerLinkBuilderUnitTest.java b/src/test/java/org/springframework/hateoas/mvc/ControllerLinkBuilderUnitTest.java
index cbac99b7..4ad27d30 100644
--- a/src/test/java/org/springframework/hateoas/mvc/ControllerLinkBuilderUnitTest.java
+++ b/src/test/java/org/springframework/hateoas/mvc/ControllerLinkBuilderUnitTest.java
@@ -149,6 +149,43 @@ public class ControllerLinkBuilderUnitTest extends TestUtils {
 		assertThat(link.getHref(), startsWith("http://somethingDifferent"));
 	}
 
+	/**
+	 * @see #112
+	 */
+	@Test
+	public void usesForwardedSslIfHeaderIsSet() {
+
+		request.addHeader("X-Forwarded-Ssl", "on");
+
+		Link link = linkTo(PersonControllerImpl.class).withSelfRel();
+		assertThat(link.getHref(), startsWith("https://"));
+	}
+
+	/**
+	 * @see #112
+	 */
+	@Test
+	public void usesForwardedSslIfHeaderIsSetOff() {
+
+		request.addHeader("X-Forwarded-Ssl", "off");
+
+		Link link = linkTo(PersonControllerImpl.class).withSelfRel();
+		assertThat(link.getHref(), startsWith("http://"));
+	}
+
+	/**
+	 * @see #112
+	 */
+	@Test
+	public void usesForwardedSslAndHostIfHeaderIsSet() {
+
+		request.addHeader("X-Forwarded-Host", "somethingDifferent");
+		request.addHeader("X-Forwarded-Ssl", "on");
+
+		Link link = linkTo(PersonControllerImpl.class).withSelfRel();
+		assertThat(link.getHref(), startsWith("https://somethingDifferent"));
+	}
+
 	/**
 	 * @see #26, #39
 	 */