Subversion Repositories SmartDukaan

package in.shop2020.serving.interceptors;

import java.util.HashMap;

import java.util.List;

import java.util.Map;

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.log4j.Logger;

import org.apache.struts2.ServletActionContext;

import org.apache.struts2.StrutsStatics;

import com.opensymphony.xwork2.ActionContext;

import com.opensymphony.xwork2.ActionInvocation;

import com.opensymphony.xwork2.interceptor.AbstractInterceptor;

import com.opensymphony.xwork2.interceptor.PreResultListener;

public class UserInterceptor extends AbstractInterceptor implements PreResultListener{

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

	private static final long serialVersionUID = -4125815700236506235L;

	private static Logger log = Logger.getLogger(UserInterceptor.class);

	public static final String USER_INFO_COOKIE_NAME = "uic";

	public static final String USER_ID_COOKIE_NAME = "uid";

	private Map<String, Cookie> cookiesMap = null;

	private Cookie userCookie = null;

	public static DesEncrypter desEncrypter = new DesEncrypter("shop2020");

	private Cookie userinfoCookie = null;

	private String cookieDomain = "";

	public void setCookieDomain(String cookieDomain) {

		this.cookieDomain = cookieDomain;

	}

	@Override

	public String intercept(ActionInvocation invocation) throws Exception {

		final Object action = invocation.getAction();

		log.debug("inside user intercepror");

        HttpServletRequest request = ServletActionContext.getRequest();

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

		createCookiesMap(request);

		UserSessionInfo userInfo = (UserSessionInfo) request.getAttribute(USER_INFO_COOKIE_NAME);

		userCookie = cookiesMap.get(UserInterceptor.USER_ID_COOKIE_NAME);

		userinfoCookie = cookiesMap.get(USER_INFO_COOKIE_NAME);

		if(userInfo == null ){

			if(userinfoCookie!=null){

				userInfo = UserSessionInfo.getUserSessionInfoFromCookieValue(userinfoCookie.getValue());

			}else{

				userInfo = createAndGetSessionFromUIDCookie(session);

			}

		}

		request.setAttribute(USER_INFO_COOKIE_NAME, userInfo);

		if (action instanceof UserAware) {

        	UserAware sessionAction = (UserAware) action;

        	sessionAction.setSession(session);

        	sessionAction.setUserSessionInfo(userInfo);

        	sessionAction.setCookiesMap(cookiesMap);

        	sessionAction.setUserCookie(userCookie);

        	sessionAction.setCookieDomainName(cookieDomain);

        }

		invocation.addPreResultListener(this);

		return invocation.invoke();

	}

	@Override

	public void beforeResult(ActionInvocation invocation, String resultCode) {

		ActionContext ac = invocation.getInvocationContext();

		HttpServletResponse response = (HttpServletResponse) ac.get(StrutsStatics.HTTP_RESPONSE);

		addCookiesToResponse(invocation.getAction(), response);

	}	

	private void addCookiesToResponse(Object action, HttpServletResponse response) {

		if (action instanceof UserAware) {

			List<Cookie> cookies = ((UserAware) action).getCookies();

			if (cookies != null) {

				for (Cookie cookie : cookies) {

					response.addCookie(cookie);

				}

			}

		}

	}

	private void createCookiesMap(HttpServletRequest request) {

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

		Cookie[] cookies = request.getCookies();

		if(cookies==null)

			return;

		for (Cookie cookie : cookies) {

			if (cookie.getName().equals(UserInterceptor.USER_ID_COOKIE_NAME)) {

				if (cookie.getDomain() == null || cookie.getDomain().isEmpty()

						|| !cookie.getDomain().equals(this.cookieDomain)) 

				{

					if (!cookieDomain.isEmpty()) {

						cookie.setMaxAge(0);

						Cookie newUserCookie = new Cookie(UserInterceptor.USER_ID_COOKIE_NAME, cookie.getValue());

						newUserCookie.setMaxAge(SECONDS_IN_YEAR); // one year

						newUserCookie.setPath("/");

						newUserCookie.setDomain(cookieDomain);

						HttpServletResponse response = ServletActionContext.getResponse();

						response.addCookie(newUserCookie);

						response.addCookie(cookie);

					}

					else {

					    log.error("cookieDomain not set");

					}

				}

			}

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

		}

	}

	private UserSessionInfo createAndGetSessionFromUIDCookie(HttpSession session) {

		userCookie = (Cookie) cookiesMap.get(UserInterceptor.USER_ID_COOKIE_NAME);

		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());

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

					    log.error("The User for the UID cookie has been deleted in our database. So cleaning up the UID cookie.");

					    expireUidCookie();

					}

				} catch (NumberFormatException nfe) {

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

					expireUidCookie();

					userInfo = new UserSessionInfo();

				}

			}

		}

		else{

			userInfo = new UserSessionInfo();

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

		}

		return userInfo;

	}

    private void expireUidCookie() {

        Cookie newUserCookie = new Cookie(UserInterceptor.USER_ID_COOKIE_NAME, "-1"); //The value here is immaterial

        newUserCookie.setMaxAge(0);                     // Expire this cookie now

        newUserCookie.setPath("/");

        newUserCookie.setDomain(cookieDomain);

        HttpServletResponse response = ServletActionContext.getResponse();

        response.addCookie(newUserCookie);

    }

}