Subversion Repositories SmartDukaan

package in.shop2020.serving.interceptors;

import java.util.HashMap;

import java.util.Map;

import in.shop2020.serving.controllers.LoginController;

import in.shop2020.serving.services.UserSessionInfo;

import in.shop2020.serving.utils.DesEncrypter;

import javax.servlet.http.Cookie;

import javax.servlet.http.HttpServletRequest;

import javax.servlet.http.HttpServletResponse;

import javax.servlet.http.HttpSession;

import org.apache.juli.logging.Log;

import org.apache.juli.logging.LogFactory;

import org.apache.struts2.ServletActionContext;

import com.opensymphony.xwork2.ActionInvocation;

import com.opensymphony.xwork2.interceptor.AbstractInterceptor;

public class UserInterceptor extends AbstractInterceptor {

	/**

	 * 

	 */

	private static final int SECONDS_IN_YEAR = 60*60*24*365; 

	private static final long serialVersionUID = -4125815700236506235L;

	private static Log log = LogFactory.getLog(UserInterceptor.class);

	public static final String USER_INFO = "userinfo";

	private Map<String, Cookie> cookiesMap = null;

	private Cookie userCookie = null;

	private DesEncrypter desEncrypter = new DesEncrypter("shop2020");

	@Override

	public String intercept(ActionInvocation invocation) throws Exception {

		final Object action = invocation.getAction();

        HttpServletRequest request = ServletActionContext.getRequest();

        HttpSession session = request.getSession(); // Get the existing session or create a new one

        //getCookiesMap(request);

		String requestedSessionId = request.getRequestedSessionId();

		UserSessionInfo userInfo = null;

		createCookiesMap(request);

		// Check if this is a brand new request with no prior cookies set; OR

		// If the request is for an active session.

		if(requestedSessionId == null || request.isRequestedSessionIdValid()){

			log.info("Request received for valid session: " + requestedSessionId);

			// Set the userinfo and the uid cookie if they're not already set.

			userInfo = (UserSessionInfo) session.getAttribute(USER_INFO);

			if(userInfo == null || userInfo.getUserId() == -1) {

				userInfo = new UserSessionInfo(session.getId());

				session.setAttribute(USER_INFO, userInfo);

			}

			createUserCookie(userInfo.getUserId(), false);

		} else {

			log.info("Request received for invalid session: " + requestedSessionId);

			// If the requested session is inactive, do the following:

			// 1. Retrieve the user for the requested session from the user cookie

			// 2. Add the retrieved user to the newly created session above.

			// 3. Update the uid cookie to ensure that a valid user is set in the session

			userInfo = createAndGetSessionFromUIDCookie(session);

			session.setAttribute(USER_INFO, userInfo);

			createUserCookie(userInfo.getUserId(), true);

		}

		if (action instanceof UserAware) {

        	UserAware sessionAction = (UserAware) action;

        	sessionAction.setSession(session);

        	sessionAction.setUserSessionInfo(userInfo);

        	sessionAction.setCookiesMap(cookiesMap);

        	sessionAction.setUserCookie(userCookie);

        }

		if (!(action instanceof LoginController)) {

			setRedirectUrl(request);

		}

		return invocation.invoke();

	}

	private void createCookiesMap(HttpServletRequest request) {

		cookiesMap  = new HashMap<String, Cookie>();

		Cookie[] cookies = request.getCookies();

		// This check is necessary for the first request when no cookies are

		// sent.

		if(cookies==null)

			return;

		for (Cookie cookie : cookies)

			cookiesMap.put(cookie.getName(), cookie);

	}

	private void createUserCookie(long userId, boolean force) {

		userCookie = (Cookie) cookiesMap.get("uid");

		String encryptedUserId = desEncrypter.encrypt(userId + "");

		if(force || userCookie == null || !(encryptedUserId + "").equals(userCookie.getValue())){

			userCookie = new Cookie("uid", encryptedUserId);

			userCookie.setMaxAge(SECONDS_IN_YEAR); // one year

			userCookie.setPath("/");

			log.info("Created new cookie.");

			cookiesMap.put("uid", userCookie);

			HttpServletResponse response = ServletActionContext.getResponse();

	        response.addCookie(userCookie);

		}

	}

	private UserSessionInfo createAndGetSessionFromUIDCookie(HttpSession session) {

		userCookie = (Cookie) cookiesMap.get("uid");

		UserSessionInfo userInfo = null;

		if(userCookie != null){

			String uidString = userCookie.getValue();

			if(uidString != null){

				try {

					Long receivedUID = Long.parseLong(desEncrypter.decrypt(uidString));

                    log.info("Invalid session with user cookie : " + receivedUID);

					userInfo = new UserSessionInfo(receivedUID, session.getId());

					session.setAttribute(USER_INFO, userInfo);

				} catch (NumberFormatException nfe) {

					log.error("The UID cookie contains an unparseable userID");

				}

			}

		}

		else{

			userInfo = new UserSessionInfo(session.getId());

			session.setAttribute(USER_INFO, userInfo);

			log.info("Invalid session without user cookie.");

		}

		return userInfo;

	}

	public void setRedirectUrl(HttpServletRequest request) {

		String queryString = request.getQueryString();

		log.info("Query String is: " + queryString);

		if (queryString == null) {

			queryString = "";

		} else {

			queryString = "?" + queryString;

		}

		log.info("Redirect Url : " + request.getRequestURI() + queryString);

		request.getSession().setAttribute(LoginInterceptor.REDIRECT_URL,

				request.getRequestURI() + queryString);

	}

}