Subversion Repositories SmartDukaan

package in.shop2020.serving.controllers;

import in.shop2020.serving.interceptors.UserAware;

import in.shop2020.serving.interceptors.UserInterceptor;

import in.shop2020.serving.services.PageLoaderHandler;

import in.shop2020.serving.services.UserSessionInfo;

import in.shop2020.serving.utils.DesEncrypter;

import java.util.ArrayList;

import java.util.HashMap;

import java.util.List;

import java.util.Map;

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.interceptor.ServletRequestAware;

import org.apache.struts2.interceptor.ServletResponseAware;

import com.opensymphony.xwork2.ValidationAwareSupport;

/**

 * Base class for all user action handlers i.e. controllers

 * 

 * @author rajveer

 */

public abstract class BaseController extends ValidationAwareSupport implements

		ServletResponseAware, ServletRequestAware, UserAware {

	/**

	 * 

	 */

	private static final long serialVersionUID = 1L;

	protected Map<String, Cookie> cookiesMap = null;

	protected HttpServletResponse response;

	protected HttpServletRequest request;

	protected HttpSession session;

	protected String domainName;

	protected UserSessionInfo userinfo = null;

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

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

	protected Cookie userCookie = null;

	protected Map<String, String> htmlSnippets;

	PageLoaderHandler pageLoader = null;

	public BaseController() {

		pageLoader = new PageLoaderHandler();

		htmlSnippets = new HashMap<String, String>();

	}

	@Override

	public void setCookiesMap(Map<String, Cookie> cookiesMap) {

		log.info("Received cookiesMap.");

		this.cookiesMap = cookiesMap;

	}

	@Override

	public void setServletResponse(HttpServletResponse response) {

		this.response = response;

	}

	@Override

	public void setServletRequest(HttpServletRequest request) {

		this.request = request;

	}

	@Override

	public void setSession(HttpSession session) {

		this.session = session;

	}

	@Override

	public void setUserSessionInfo(UserSessionInfo userInfo) {

		this.userinfo = userInfo;

	}

	@Override

	public void setUserCookie(Cookie userCookie) {

		if(userCookie != null){

			log.info("cookie name is:" +  userCookie.getName() + "   value is:"+ userCookie.getValue() + " path : " + userCookie.getPath());

		}

		this.userCookie = userCookie;

	}

	@Override

    public void setCookieDomainName(String domainName) {

        this.domainName = domainName;

    }

	public String getHeaderSnippet() {

		String url = request.getQueryString();

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

		if (url == null) {

			url = "";

		} else {

			url = "?" + url;

		}

		url = request.getRequestURI() + url;

		return pageLoader.getHeaderHtml(userinfo.isLoggedIn(),

				userinfo.getNameOfUser(), url);

	}

	public String getSearchBarSnippet() {

		//FIXME From where it is called, need to pass category

		return pageLoader.getSearchBarHtml(userinfo.getTotalItems(), 10000);

	}

	public UserSessionInfo getUserInfo(){

		return this.userinfo;

	}

	@Override

	public List<Cookie> getCookies() {

		List<Cookie> cookies = new ArrayList<Cookie>();

		long userId = userinfo.getUserId();

		if(userId != -1){

			String newUserinfoCookieValue = UserSessionInfo.getUserinfoCookieValueFromUserSessionInfo(userinfo);

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

			if( userinfoCookie == null ||  !newUserinfoCookieValue.equals(userinfoCookie.getValue())){

				log.info("Set user session info cookie if it has been changed.");

				userinfoCookie = new Cookie(UserInterceptor.USER_INFO_COOKIE_NAME, newUserinfoCookieValue);

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

				userinfoCookie.setPath("/");

				userinfoCookie.setDomain(domainName);

				cookiesMap.put(UserInterceptor.USER_INFO_COOKIE_NAME, userinfoCookie);

				cookies.add(userinfoCookie);

			}

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

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

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

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

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

				userinfoCookie.setPath("/");

				userinfoCookie.setDomain(domainName);

				cookiesMap.put(UserInterceptor.USER_INFO_COOKIE_NAME, userinfoCookie);

				cookies.add(userinfoCookie);

			}

		}

		return cookies;

	}

	public void clearUserCookies(){

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

		uidCookie.setDomain(domainName);

		uidCookie.setPath("/");

		uidCookie.setValue("");

		uidCookie.setMaxAge(0);

		this.response.addCookie(uidCookie);

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

		uicCookie.setDomain(domainName);

		uicCookie.setPath("/");

		uicCookie.setValue("");

		uicCookie.setMaxAge(0);

		this.response.addCookie(uicCookie);

	}

}