From 818973b2baece0ff3245de10091f5ca8bdffebc6 Mon Sep 17 00:00:00 2001
From: James Moger <james.moger@gitblit.com>
Date: Thu, 4 Sep 2014 13:04:20 -0400
Subject: Do not stamp raw servlet responses with cache-control headers

---
 .../java/com/gitblit/servlet/PagesServlet.java     | 27 ++++++++++++++++++++++
 1 file changed, 27 insertions(+)

(limited to 'src/main/java/com/gitblit/servlet/PagesServlet.java')

diff --git a/src/main/java/com/gitblit/servlet/PagesServlet.java b/src/main/java/com/gitblit/servlet/PagesServlet.java
index f578f86f..a8e83418 100644
--- a/src/main/java/com/gitblit/servlet/PagesServlet.java
+++ b/src/main/java/com/gitblit/servlet/PagesServlet.java
@@ -15,10 +15,19 @@
  */
 package com.gitblit.servlet;
 
+import java.io.IOException;
+import java.io.InputStream;
+import java.util.Date;
+
+import javax.servlet.ServletException;
 import javax.servlet.http.HttpServletRequest;
 import javax.servlet.http.HttpServletResponse;
 
+import org.eclipse.jgit.lib.Repository;
+import org.eclipse.jgit.revwalk.RevCommit;
+
 import com.gitblit.Constants;
+import com.gitblit.utils.JGitUtils;
 
 /**
  * Serves the content of a gh-pages branch.
@@ -73,4 +82,22 @@ public class PagesServlet extends RawServlet {
 	protected void setContentType(HttpServletResponse response, String contentType) {
 		response.setContentType(contentType);;
 	}
+
+	@Override
+	protected void streamFromRepo(HttpServletResponse response, Repository repository,
+			RevCommit commit, String requestedPath) throws IOException {
+
+		response.setDateHeader("Last-Modified", JGitUtils.getCommitDate(commit).getTime());
+		response.setHeader("Cache-Control", "public, max-age=3600, must-revalidate");
+
+		super.streamFromRepo(response, repository, commit, requestedPath);
+	}
+
+	@Override
+	protected void sendContent(HttpServletResponse response, Date date, InputStream is) throws ServletException, IOException {
+		response.setDateHeader("Last-Modified", date.getTime());
+		response.setHeader("Cache-Control", "public, max-age=3600, must-revalidate");
+
+		super.sendContent(response, date, is);
+	}
 }
-- 
cgit v1.2.3