Subversion Repositories SmartDukaan

package com.spice.profitmandi.web.util;

import java.util.HashSet;

import java.util.Set;

import javax.servlet.http.Cookie;

import javax.servlet.http.HttpServletRequest;

import javax.servlet.http.HttpServletResponse;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

import org.springframework.stereotype.Component;

import com.spice.profitmandi.common.exception.ProfitMandiBusinessException;

import com.spice.profitmandi.common.model.ProfitMandiConstants;

import com.spice.profitmandi.dao.enumuration.dtr.RoleType;

import com.spice.profitmandi.web.model.LoginDetails;

@Component

public class CookiesProcessor {

	private static final Logger LOGGER = LoggerFactory.getLogger(CookiesProcessor.class);

	public LoginDetails getCookiesObject(HttpServletRequest request) throws ProfitMandiBusinessException{

		Cookie[] cookies = request.getCookies();

		if (cookies == null){

			throw new ProfitMandiBusinessException("", "", "");

		}

		String fofoIdFound = null, emailIdFound = null, roleNamesString = null, fofoFlagFound = null;

		for(Cookie cookie : cookies){

			LOGGER.info("Requested Cookie {}={}", cookie.getName(), cookie.getValue());

			if(cookie.getName().equals(ProfitMandiConstants.FOFO_ID) && cookie.getValue() != null && !cookie.getValue().isEmpty()){

				fofoIdFound = cookie.getValue();

			}

			if(cookie.getName().equals(ProfitMandiConstants.EMAIL_ID) && cookie.getValue() != null && !cookie.getValue().isEmpty()){

				emailIdFound = cookie.getValue();

			}

			if(cookie.getName().equals(ProfitMandiConstants.ROLE_NAMES) && cookie.getValue() != null && !cookie.getValue().isEmpty()){

				roleNamesString = cookie.getValue();

			}

			if(cookie.getName().equals(ProfitMandiConstants.FOFO_FLAG) && cookie.getValue() != null && !cookie.getValue().isEmpty()){

				fofoFlagFound = cookie.getValue();

			}

			if(fofoIdFound != null && emailIdFound != null && roleNamesString != null && fofoFlagFound != null){

				break;

			}

		}

		if(fofoIdFound == null || emailIdFound == null || roleNamesString == null || fofoFlagFound == null){

			LOGGER.error("Requested session is not valid");

			throw new ProfitMandiBusinessException("", "", "");

		}else{

			LoginDetails fofoDetails = new LoginDetails();

			fofoDetails.setFofoId(Integer.parseInt(fofoIdFound));

			fofoDetails.setEmailId(emailIdFound);

			String[] roleNames = roleNamesString.split(",");

			Set<RoleType> roleTypes = new HashSet<>();

			for(String roleName : roleNames){

				RoleType roleType = RoleType.valueOf(roleName);

				roleTypes.add(roleType);

			}

			fofoDetails.setRoleTypes(roleTypes);

			fofoDetails.setFofo(Boolean.parseBoolean(fofoFlagFound));

 			return fofoDetails;

		}

	}

	public void removeCookies(HttpServletRequest request, HttpServletResponse response) throws ProfitMandiBusinessException{

		Cookie[] cookies = request.getCookies();

		if (cookies == null){

			throw new ProfitMandiBusinessException("", "", "");

		}

		Cookie cookieFofoId = new Cookie(ProfitMandiConstants.FOFO_ID, "");

		cookieFofoId.setMaxAge(0);

		cookieFofoId.setPath(request.getContextPath());

		cookieFofoId.setDomain(request.getServerName());

		Cookie cookieEmailId = new Cookie(ProfitMandiConstants.EMAIL_ID, "");

		cookieEmailId.setMaxAge(0);

		cookieEmailId.setDomain(request.getServerName());

		cookieEmailId.setPath(request.getContextPath());

		Cookie cookieRoleNames = new Cookie(ProfitMandiConstants.ROLE_NAMES, "");

		cookieRoleNames.setMaxAge(0);

		cookieRoleNames.setDomain(request.getServerName());

		cookieRoleNames.setPath(request.getContextPath());

		Cookie cookieFofoFlag = new Cookie(ProfitMandiConstants.FOFO_FLAG, "");

		cookieFofoFlag.setMaxAge(0);

		cookieFofoFlag.setDomain(request.getServerName());

		cookieFofoFlag.setPath(request.getContextPath());

		response.addCookie(cookieFofoId);

		response.addCookie(cookieEmailId);

		response.addCookie(cookieRoleNames);

		response.addCookie(cookieFofoFlag);

	}

}