Subversion Repositories SmartDukaan

package in.shop2020.serving.interceptors;

import in.shop2020.model.v1.catalog.InventoryService.Client;

import in.shop2020.model.v1.catalog.Source;

import in.shop2020.serving.utils.DesEncrypter;

import in.shop2020.thrift.clients.CatalogClient;

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 org.apache.log4j.Logger;

import org.apache.struts2.ServletActionContext;

import org.apache.thrift.TException;

import org.apache.thrift.transport.TTransportException;

import com.opensymphony.xwork2.ActionInvocation;

import com.opensymphony.xwork2.interceptor.AbstractInterceptor;

public class SourceInterceptor extends AbstractInterceptor {

    private static final long serialVersionUID = 1L;

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

    private static final int SECONDS_IN_DAY = 60 * 60 * 24;

    private String cookieDomain = "";

    public static final String SOURCE_ID_COOKIE = "src_idc";

    private Cookie sourceIdCookie = null;

    public static final String ENCRIPTION_STRING = "Saholic";

    public static List<Source> allSources = new ArrayList<Source>();

    private Map<String, Cookie> cookiesMap = null;

    private HttpServletRequest request;

    private HttpServletResponse response;

    static{

    	try {

			Client client = (new CatalogClient()).getClient();

			allSources = client.getAllSources();

		} catch (TTransportException e) {

			log.error("Unable to get all sources from thrift.", e);

		} catch (TException e) {

			log.error("Unable to get all sources from thrift.", e);

		}

    }

    public void setCookieDomain(String cookieDomain) {

        this.cookieDomain = cookieDomain;

    }

    @Override

    public String intercept(ActionInvocation invocation) throws Exception {

    	final Object action = invocation.getAction();

        request = ServletActionContext.getRequest();

        response = ServletActionContext.getResponse();

        createCookiesMap();

        String sourceIdString = updateSourceIdCookie();

        long sourceId = -1;

        if(sourceIdString != null){

        	try{

        		sourceId = Long.parseLong(sourceIdString);

        	}catch (NumberFormatException e) {

        		log.error("Invalid source cookie. Clear the cookie.");

        		//FIXME Clear the cookie, although it will work without it

			}

        }

		if (action instanceof SourceAware) {

			SourceAware sessionAction = (SourceAware) action;

			sessionAction.setSourceId(sourceId);

        }

        return invocation.invoke();

    }

    /**

     * Update source id cookie

     * 

     */

    private String updateSourceIdCookie() {

    	String sourceId = getSourceId();

        if(sourceId == null){

        	return sourceId;

        }

        DesEncrypter des = new DesEncrypter(ENCRIPTION_STRING);

        String sessionSourceCookieVal = des.encrypt(sourceId);

        //session source

        sourceIdCookie = new Cookie(SOURCE_ID_COOKIE,

                sessionSourceCookieVal);

        sourceIdCookie.setPath("/");

        sourceIdCookie.setMaxAge(SECONDS_IN_DAY);

        if (!cookieDomain.isEmpty()) {

            sourceIdCookie.setDomain(cookieDomain);

        }

        cookiesMap.put(SOURCE_ID_COOKIE, sourceIdCookie);

        response.addCookie(sourceIdCookie);

        return sourceId;

    }

    private void createCookiesMap() {

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

        Cookie[] cookies = request.getCookies();

        if (cookies == null)

            return;

        for (Cookie cookie : cookies) {

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

        }

    }

    /**

     * Check if referer is not saholic or null, try to match with all sources. return source id if matches else return null.

     * If referer is saholic, return sourceId from cookie if set, else return null.

     */

    private String getSourceId() {

    	String sourceId = null;

        String referer = request.getHeader("referer");

        if (referer != null && !referer.isEmpty() && !referer.contains("saholic.com" ) && !referer.contains("shop2020.in" )) {

        	for(Source source: allSources){

        		if(referer.matches(source.getIdentifier())){

        			return source.getId()+"";

        		}

        	}

        }else{

        	sourceIdCookie = (Cookie) cookiesMap.get(SOURCE_ID_COOKIE);

            if (sourceIdCookie != null) {

            	DesEncrypter des = new DesEncrypter(ENCRIPTION_STRING);

            	sourceId = des.decrypt(sourceIdCookie.getValue());

                return sourceId;

            }

        }

        return sourceId;

    }

}