Subversion Repositories SmartDukaan

package in.shop2020.serving.controllers;

import in.shop2020.serving.services.PageLoaderHandler;

import in.shop2020.serving.services.UserSessionInfo;

import in.shop2020.serving.utils.DesEncrypter;

import in.shop2020.serving.utils.Utils;

import java.util.Enumeration;

import java.util.HashMap;

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 javax.servlet.http.HttpUtils;

import org.apache.juli.logging.Log;

import org.apache.juli.logging.LogFactory;

import org.apache.struts2.interceptor.CookiesAware;

import org.apache.struts2.interceptor.ServletRequestAware;

import org.apache.struts2.interceptor.ServletResponseAware;

import org.apache.velocity.VelocityContext;

import com.opensymphony.xwork2.ValidationAware;

import com.opensymphony.xwork2.ValidationAwareSupport;

/**

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

 * 

 * @author rajveer

 */

public abstract class BaseController extends ValidationAwareSupport implements  CookiesAware, ServletResponseAware, ServletRequestAware {

	/**

	 * 

	 */

	private static final long serialVersionUID = 1L;

	protected Map<String, Cookie> cookiesMap = null;

    protected HttpServletResponse response;

    protected HttpServletRequest request;

    protected HttpSession session;

    protected UserSessionInfo userinfo = null;

	private static Log log = LogFactory.getLog(BaseController.class);

    private DesEncrypter desEncrypter = new DesEncrypter("shop2020");

    protected Cookie userCookie = null;

    protected Map<String,String> htmlSnippets;

    PageLoaderHandler pageLoader = null;

	public BaseController() {

		pageLoader = new PageLoaderHandler();

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

	}

	public Map getCookiesMap() {

		return cookiesMap;

	}

	@Override

	public void setCookiesMap(Map cookiesMap) {

		log.info("Received cookiesMap and it is " + cookiesMap);

		this.cookiesMap = cookiesMap;

	}

	@Override

	public void setServletResponse(HttpServletResponse response)

	{

		this.response = response;

		if(userCookie!=null)

			response.addCookie(userCookie);

	}

	@Override

	public void setServletRequest(HttpServletRequest request){

		this.request = request;

//		log.debug("REQUESTED URL: " + request.getRequestURL().toString());

//		log.debug("Remote host "+ request.getRemoteHost());

//		log.debug("requested Session Id "+ request.getRequestedSessionId());

//		log.debug("Session Id "+ request.getSession().getId());

//		log.debug("QUERY STRING IS: " + this.request.getQueryString());

//		

//		Enumeration names = request.getHeaderNames();

//	    while (names.hasMoreElements()) {

//	      String name = (String) names.nextElement();

//	      Enumeration values = request.getHeaders(name);  // support multiple values

//	      if (values != null) {

//	        while (values.hasMoreElements()) {

//	          String value = (String) values.nextElement();

//	          log.debug(name + ": " + value);

//	        }

//	      }

//	    }

//		

//		for(Object param: request.getParameterMap().keySet()){

//			log.debug("PARAMS: " + param + " = "+ request.getParameter((String)param));

//		}

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

		getCookiesMap(request);

		String requestedSessionId = request.getRequestedSessionId();

		// Check if this is a brand new request with no prior cookies set; OR

		// If the request is for an active session.

		if(requestedSessionId == null || request.isRequestedSessionIdValid()){

			log.info("Request received for valid session: " + requestedSessionId);

			// Set the userinfo and the uid cookie if they're not already set.

			this.session = request.getSession();

			setUserSessionInfo(this.session.getId());

			createUserCookie(this.userinfo.getUserId(), false);

		} else {

			log.info("Request received for invalid session: " + requestedSessionId);

			// If the requested session is inactive, do the following:

			// 1. Retrieve the user for the requested session from the user cookie

			// 2. Add the retrieved user to the newly created session above.

			// 3. Update the uid cookie to ensure that a valid user is set in the session

			recreateSessionFromUIDCookie(this.session.getId());

			createUserCookie(this.userinfo.getUserId(), true);

		}

	}

	private void getCookiesMap(HttpServletRequest request) {

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

		Cookie[] cookies = request.getCookies();

		// This check is necessary for the first request when no cookies are

		// sent.

		if(cookies==null)

			return;

		for (Cookie cookie : cookies)

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

	}

	private void setUserSessionInfo(String jsessionid){

		this.userinfo = (UserSessionInfo) this.session.getAttribute("userinfo");

		if(this.userinfo == null || this.userinfo.getUserId() == -1){

			this.userinfo = new UserSessionInfo(jsessionid);

			this.session.setAttribute("userinfo", this.userinfo);

		}

	}

	protected void createUserCookie(long userId, boolean force) {

		userCookie = (Cookie) cookiesMap.get("uid");

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

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

			userCookie = new Cookie("uid", encryptedUserId);

		}

	}

	private void recreateSessionFromUIDCookie(String jsessionid) {

		Cookie userCookie = (Cookie) cookiesMap.get("uid");

		if(userCookie != null){

			String uidString = userCookie.getValue();

			if(uidString != null){

				try {

					Long receivedUID = Long.parseLong(desEncrypter.decrypt(uidString));

					this.userinfo = new UserSessionInfo(receivedUID, jsessionid);

					this.session.setAttribute("userinfo", this.userinfo);

				} catch (NumberFormatException nfe) {

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

				}

			}

		}

		if(this.userinfo==null)

			setUserSessionInfo(jsessionid);

	}

	public UserSessionInfo getUserInfo(){

		return this.userinfo;

	}

	public String getHeaderSnippet(){

		return pageLoader.getHeaderHtml(userinfo.isLoggedIn(), userinfo.getNameOfUser());

	}

	public String getMainMenuSnippet(){

		return pageLoader.getMainMenuHtml();

	}

	public String getSearchBarSnippet(){

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

	}

	public String getCustomerServiceSnippet(){

		return pageLoader.getCustomerServiceHtml();

	}

	public String getMyResearchSnippet(){

		return pageLoader.getMyResearchHtml(userinfo.getUserId(), userinfo.isLoggedIn());

	}

	public String getBrowseHistorySnippet(){

		return pageLoader.getBrowseHistoryHtml(userinfo.getUserId(), userinfo.isLoggedIn());

	}

	public String getFooterSnippet(){

		return pageLoader.getFooterHtml();

	}

	public String getRedirectUrl(){

		return (String)this.request.getSession().getAttribute("REDIRECT_URL");

	}

	public void setRedirectUrl(){

		String queryString = this.request.getQueryString();

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

		if(queryString==null){

			queryString="";

		}else{

			queryString = "?" + queryString;

		}

		this.request.getSession().setAttribute("REDIRECT_URL", this.request.getRequestURI() + queryString);

	}

	public void resetRedirectUrl(){

		this.request.getSession().removeAttribute("REDIRECT_URL");

	}

}