package org.webslinger.servlet;

 import javax.servlet.ServletContext;

 import javax.servlet.http.HttpServletRequest;

 import javax.servlet.http.HttpServletRequestWrapper;

 import javax.servlet.http.HttpServletResponse;

 import javax.servlet.http.HttpSession;

 public class OverrideHttpSessionRequest extends HttpServletRequestWrapper {

     protected final ServletContext context;

     protected final HttpSessionIdManager manager;

     protected final HttpSessionStore store;

     protected HttpSessionId sessionId;

     protected HttpSession session;

     public OverrideHttpSessionRequest(HttpServletRequest request, ServletContext context, HttpSessionIdManager manager, HttpSessionStore store, HttpSessionId sessionId) {

         super(request);

         this.context = context;

         this.manager = manager;

         this.store = store;

         this.sessionId = sessionId;

     }

     public HttpServletRequest getRequest() {

         return (HttpServletRequest) super.getRequest();

     }

     protected HttpSessionIdManager getManager() {

         return manager;

     }

     protected HttpSessionId getSessionId(boolean create) {

         if (sessionId == null && create) sessionId = manager.createHttpSessionId();

         return sessionId != null ? sessionId : HttpSessionId.INVALID;

     }

     public HttpSession getSession() {

         return getSession(true);

     }

     public HttpSession getSession(boolean create) {

         if (session != null) return session;

         HttpSessionId sessionId = getSessionId(create);

         HttpSessionData data = store.getSessionData(getRequest(), sessionId, create);

         if (data == null) return null;

         if (data.isNew()) data.sentId();

         session = new HttpSessionImpl(context, data);

         return session;

     }

     public String requestedSessionId(HttpServletRequest request) {

         return getSessionId(false).getId();

     }

     public boolean isRequestedSessionIdFromCookie(HttpServletRequest request) {

         return getSessionId(false).getLocation() == HttpSessionId.Location.COOKIE;

     }

     public boolean isRequestedSessionIdFromURL(HttpServletRequest request) {

         return getSessionId(false).getLocation() == HttpSessionId.Location.URL;

     }

     /** @deprecated */

     public boolean isRequestedSessionIdFromUrl(HttpServletRequest request) {

         return getSessionId(false).getLocation() == HttpSessionId.Location.URL;

     }

     public boolean isRequestedSessionIdValid(HttpServletRequest request) {

         return getSessionId(false).getLocation().isValid();

     }

 };