package org.webslinger.servlet;

 import java.util.HashMap;

 import java.util.Map;

 import javax.servlet.http.HttpServletRequest;

 import javax.servlet.http.HttpServletResponse;

 import javax.servlet.http.HttpSession;

 public class EmbeddedHttpSessionManager implements HttpSessionIdManager, HttpSessionStore {

     public static final String SESSION_KEY = EmbeddedHttpSessionManager.class.getName() + ":sessions";

     protected final String id;

     public EmbeddedHttpSessionManager(String id) {

         this.id = id;

     }

     public HttpSessionId extractHttpSessionId(HttpServletRequest[] request, HttpServletResponse[] response) {

         HttpSessionId.Location location;

         if (request[0].isRequestedSessionIdFromCookie()) {

             location = HttpSessionId.Location.COOKIE;

         } else if (request[0].isRequestedSessionIdFromURL()) {

             location = HttpSessionId.Location.URL;

         } else {

             location = HttpSessionId.Location.OTHER;

         }

         return new HttpSessionId(id, location);

     }

     public HttpSessionId createHttpSessionId() {

         throw new UnsupportedOperationException();

     }

     public String encodeURL(String path, HttpSessionId id) {

         return path;

     }

     public String encodeRedirectURL(String path, HttpSessionId id) {

         return path;

     }

     public HttpSessionData getSessionData(HttpServletRequest request, HttpSessionId id, boolean create) {

         HttpSession session = request.getSession(create);

         if (session == null) return null;

         return getSessionData(request, session, id, create);

     }

     public static HttpSessionData getSessionData(HttpServletRequest request, HttpSession session, HttpSessionId id, boolean create) {

         Map<String, HttpSessionData> nestedSessions;

         synchronized (session) {

             nestedSessions = (Map<String, HttpSessionData>) session.getAttribute(SESSION_KEY);

             if (nestedSessions == null) {

                 if (!create) return null;

                 nestedSessions = new HashMap<String, HttpSessionData>();

                 session.setAttribute(SESSION_KEY, nestedSessions);

             }

         }

         HttpSessionData data;

         synchronized (nestedSessions) {

             data = nestedSessions.get(id.getId());

             if (data == null) {

                 if (!create) return null;

                 data = new HttpSessionData(id.getId(), request.getRemoteAddr());

                 nestedSessions.put(id.getId(), data);

             }

         }

         return data;

     }

 }