Subversion Repositories SmartDukaan

package in.shop2020.serving.interceptors;

import java.util.HashMap;

import java.util.Map;

import in.shop2020.model.v1.user.User;

import in.shop2020.model.v1.user.UserContextService.Client;

import in.shop2020.serving.services.UserSessionInfo;

import in.shop2020.serving.utils.DesEncrypter;

import in.shop2020.thrift.clients.UserContextServiceClient;

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 com.opensymphony.xwork2.ActionInvocation;

import com.opensymphony.xwork2.interceptor.AbstractInterceptor;

public class CreateUserInterceptor extends AbstractInterceptor {

	private static final long serialVersionUID = -4125815700236506235L;

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

	private Map<String, Cookie> cookiesMap = null;

	private Cookie userCookie = null;

	private String cookieDomain = "";

	private DesEncrypter desEncrypter = new DesEncrypter(UserInterceptor.COOKIE_DECRYPTION_STRING);

	public void setCookieDomain(String cookieDomain) {

		this.cookieDomain = cookieDomain;

	}

	@Override

	public String intercept(ActionInvocation invocation) throws Exception {

		log.info("inside create user interceprot");

		HttpServletRequest request = ServletActionContext.getRequest();

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

        createCookiesMap(request);

        UserSessionInfo userInfo = null;

		Cookie userinfoCookie = cookiesMap.get(UserInterceptor.USER_INFO_COOKIE_NAME);

		//FIXME Remove the if part once we are done with solving the side effected of deleting anonymous users. 

		if(userinfoCookie!=null){

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

			UserContextServiceClient ucsc = new UserContextServiceClient();

			Client userClient = ucsc.getClient();

			User existingUser = userClient.getUserById(userInfo.getUserId());

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

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

	        	request.setAttribute(UserInterceptor.USER_INFO_COOKIE_NAME, userInfo);

	        	createUserCookie(userInfo.getUserId(), true);				

			}

		}else{

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

        	request.setAttribute(UserInterceptor.USER_INFO_COOKIE_NAME, userInfo);

        	createUserCookie(userInfo.getUserId(), true);

        }

        return invocation.invoke();

	}

	private void createUserCookie(long userId, boolean force) {

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

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

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

			userCookie = new Cookie(UserInterceptor.USER_ID_COOKIE_NAME, encryptedUserId);

			userCookie.setMaxAge(UserInterceptor.SECONDS_IN_YEAR); // one year

			userCookie.setPath("/");

			if(!cookieDomain.isEmpty()) {

				userCookie.setDomain(cookieDomain);

			}

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

			cookiesMap.put(UserInterceptor.USER_ID_COOKIE_NAME, userCookie);

	        HttpServletResponse response = ServletActionContext.getResponse();

	        response.addCookie(userCookie);

		}

	}

	private void createCookiesMap(HttpServletRequest request) {

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

		Cookie[] cookies = request.getCookies();

		if(cookies==null)

			return;

		for (Cookie cookie : cookies) {

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

		}

	}

}