Fix various issues raised when adapting upper layers

[lgpl/argeo-commons.git] / org.argeo.cms / src / org / argeo / cms / auth / HttpRequestCallbackHandler.java

diff --git a/org.argeo.cms/src/org/argeo/cms/auth/HttpRequestCallbackHandler.java b/org.argeo.cms/src/org/argeo/cms/auth/HttpRequestCallbackHandler.java
index 0486d7479daa1ddbc14925b61e966ab427e08fdb..934fdd96bdf86ae7fa047507fc6c68acd3e58c25 100644 (file)
--- a/org.argeo.cms/src/org/argeo/cms/auth/HttpRequestCallbackHandler.java
+++ b/org.argeo.cms/src/org/argeo/cms/auth/HttpRequestCallbackHandler.java
@@ -4,26 +4,40 @@ import java.io.IOException;
 
 import javax.security.auth.callback.Callback;
 import javax.security.auth.callback.CallbackHandler;
+import javax.security.auth.callback.LanguageCallback;
 import javax.security.auth.callback.UnsupportedCallbackException;
-import javax.servlet.http.HttpServletRequest;
 
 /**
  * Callback handler populating {@link HttpRequestCallback}s with the provided
  * {@link HttpServletRequest}, and ignoring any other callback.
  */
 public class HttpRequestCallbackHandler implements CallbackHandler {
-       final private HttpServletRequest request;
+       final private HttpRequest request;
+       final private HttpResponse response;
+       final private HttpSession httpSession;
 
-       public HttpRequestCallbackHandler(HttpServletRequest request) {
+       public HttpRequestCallbackHandler(HttpRequest request, HttpResponse response) {
                this.request = request;
+               this.httpSession = request.getSession();
+               this.response = response;
+       }
+
+       public HttpRequestCallbackHandler(HttpSession httpSession) {
+               this.httpSession = httpSession;
+               this.request = null;
+               this.response = null;
        }
 
        @Override
-       public void handle(Callback[] callbacks) throws IOException,
-                       UnsupportedCallbackException {
+       public void handle(Callback[] callbacks) throws IOException, UnsupportedCallbackException {
                for (Callback callback : callbacks)
-                       if (callback instanceof HttpRequestCallback)
+                       if (callback instanceof HttpRequestCallback) {
                                ((HttpRequestCallback) callback).setRequest(request);
+                               ((HttpRequestCallback) callback).setResponse(response);
+                               ((HttpRequestCallback) callback).setHttpSession(httpSession);
+                       } else if (callback instanceof LanguageCallback) {
+                               ((LanguageCallback) callback).setLocale(request.getLocale());
+                       }
        }
 
 }