From 13417cf9c6eec555b51da49742e47939d2f5715b Mon Sep 17 00:00:00 2001 From: James Moger <james.moger@gitblit.com> Date: Fri, 19 Oct 2012 22:47:33 -0400 Subject: [PATCH] Exclude submodules from zip downloads (issue 151) --- src/com/gitblit/wicket/GitBlitWebSession.java | 68 +++++++++++++++++++++++++++++++++- 1 files changed, 66 insertions(+), 2 deletions(-) diff --git a/src/com/gitblit/wicket/GitBlitWebSession.java b/src/com/gitblit/wicket/GitBlitWebSession.java index 2238660..015d97a 100644 --- a/src/com/gitblit/wicket/GitBlitWebSession.java +++ b/src/com/gitblit/wicket/GitBlitWebSession.java @@ -15,10 +15,17 @@ */ package com.gitblit.wicket; +import java.util.Map; import java.util.TimeZone; +import java.util.concurrent.atomic.AtomicBoolean; +import org.apache.wicket.Page; +import org.apache.wicket.PageParameters; +import org.apache.wicket.RedirectToUrlException; import org.apache.wicket.Request; import org.apache.wicket.Session; +import org.apache.wicket.protocol.http.RequestUtils; +import org.apache.wicket.protocol.http.WebRequestCycle; import org.apache.wicket.protocol.http.WebSession; import org.apache.wicket.protocol.http.request.WebClientInfo; @@ -33,14 +40,59 @@ private UserModel user; private String errorMessage; - + + private String requestUrl; + + private AtomicBoolean isForking; + public GitBlitWebSession(Request request) { super(request); + isForking = new AtomicBoolean(); } public void invalidate() { super.invalidate(); user = null; + } + + /** + * Cache the requested protected resource pending successful authentication. + * + * @param pageClass + */ + public void cacheRequest(Class<? extends Page> pageClass) { + // build absolute url with correctly encoded parameters?! + Request req = WebRequestCycle.get().getRequest(); + Map<String, ?> params = req.getRequestParameters().getParameters(); + PageParameters pageParams = new PageParameters(params); + String relativeUrl = WebRequestCycle.get().urlFor(pageClass, pageParams).toString(); + requestUrl = RequestUtils.toAbsolutePath(relativeUrl); + if (isTemporary()) + { + // we must bind the temporary session into the session store + // so that we can re-use this session for reporting an error message + // on the redirected page and continuing the request after + // authentication. + bind(); + } + } + + /** + * Continue any cached request. This is used when a request for a protected + * resource is aborted/redirected pending proper authentication. Gitblit + * no longer uses Wicket's built-in mechanism for this because of Wicket's + * failure to properly handle parameters with forward-slashes. This is a + * constant source of headaches with Wicket. + * + * @return false if there is no cached request to process + */ + public boolean continueRequest() { + if (requestUrl != null) { + String url = requestUrl; + requestUrl = null; + throw new RedirectToUrlException(url); + } + return false; } public boolean isLoggedIn() { @@ -51,7 +103,11 @@ if (user == null) { return false; } - return user.canAdmin; + return user.canAdmin(); + } + + public String getUsername() { + return user == null ? "anonymous" : user.username; } public UserModel getUser() { @@ -82,6 +138,14 @@ errorMessage = null; return msg; } + + public boolean isForking() { + return isForking.get(); + } + + public void isForking(boolean val) { + isForking.set(val); + } public static GitBlitWebSession get() { return (GitBlitWebSession) Session.get(); -- Gitblit v1.9.1