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.apache.commons.lang3.StringUtils;

import org.apache.logging.log4j.LogManager;

import org.apache.logging.log4j.Logger;

import org.springframework.stereotype.Component;

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

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

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

@Component

public class CookiesProcessor {

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

	public LoginDetails getCookiesObject(HttpServletRequest request) throws ProfitMandiBusinessException{

		Cookie[] cookies = request.getCookies();

		if (cookies == null){

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

		}

		String fofoIdFound = null, emailIdFound = null, roleIdsString = 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_IDS) && cookie.getValue() != null){

				roleIdsString = cookie.getValue();

				//LOGGER.info("roleNameString is {}", roleNamesString);

			}

			if(fofoIdFound != null && emailIdFound != null && roleIdsString != null){

				break;

			}

		}

		if(fofoIdFound == null || emailIdFound == null || roleIdsString == null){

			LOGGER.info("roleIdString is {}", roleIdsString);

			LOGGER.info("fofoIdFound is {}", fofoIdFound);

			LOGGER.info("emailIdFound is {}", emailIdFound);

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

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

		}else {

			LoginDetails fofoDetails = new LoginDetails();

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

			fofoDetails.setEmailId(emailIdFound);

			String[] roleIdStrings = StringUtils.split(roleIdsString, ",");

			Set<Integer> roleIds = new HashSet<>();

			for(String roleId : roleIdStrings){

				roleIds.add(Integer.valueOf(roleId));

			}

			fofoDetails.setRoleIds(roleIds);

			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_IDS, "");

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

	}

}