Subversion Repositories SmartDukaan

package in.shop2020.serving.interceptors;

import in.shop2020.datalogger.EventType;

import in.shop2020.model.v1.user.Affiliate;

import in.shop2020.serving.services.UserSessionInfo;

import in.shop2020.serving.utils.DesEncrypter;

import in.shop2020.thrift.clients.UserClient;

import in.shop2020.utils.DataLogger;

import java.util.Date;

import java.util.HashMap;

import java.util.HashSet;

import java.util.Map;

import java.util.Set;

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 com.opensymphony.xwork2.ActionInvocation;

import com.opensymphony.xwork2.interceptor.AbstractInterceptor;

public class TrackingInterceptor extends AbstractInterceptor {

    private static final long serialVersionUID = 1L;

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

    private static final int SECONDS_IN_YEAR = 60 * 60 * 24 * 365;

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

    public static final String TRACKER = "tracker";

    public static final String AFF_COOKIE = "uafc";

    public static final String SRC_COOKIE = "usrcc";

    public static final String SESSION_SRC_COOKIE = "sess_srcc";

    public static final String SRC_TIME_COOKIE = "usrctc";

    public static final String SESSION_SRC_TIME_COOKIE = "sess_srctc";

    public static final String SESSION_ID_COOKIE = "sess_idc";

    public static final String ENCRIPTION_STRING = "Saholic";

    private static final Set<String> paidAffIds;

    private String cookieDomain = "";

    private Cookie affCookie = null;

    private Cookie firstSourceCookie = null;

    private Cookie sessionSourceCookie = null;

    private Cookie firstSourceTimeCookie = null;

    private Cookie sessionSourceTimeCookie = null;

    private Cookie sessionIdCookie = null;

    private Map<String, Cookie> cookiesMap = null;

    private HttpServletRequest request;

    private HttpServletResponse response;

    private String affId = null;

    static {

        paidAffIds = new HashSet<String>();

        paidAffIds.add("6");

        paidAffIds.add("38");

    }

    public void setCookieDomain(String cookieDomain) {

        this.cookieDomain = cookieDomain;

    }

    @Override

    public String intercept(ActionInvocation invocation) throws Exception {

        request = ServletActionContext.getRequest();

        response = ServletActionContext.getResponse();

        affId = request.getParameter("afid");

        createCookiesMap();

        updateFirstSourceCookie();

        updateSessionSourceCookie();

        updateAffCookie();

        return invocation.invoke();

    }

    /**

     * Update first source cookie

     * 

     */

    private void updateFirstSourceCookie() {

        firstSourceCookie = (Cookie) cookiesMap.get(SRC_COOKIE);

        if (firstSourceCookie != null) {

            return;

        }

        String src = getSource();

        DesEncrypter des = new DesEncrypter(ENCRIPTION_STRING);

        String sourceCookieVal = des.encrypt(src);

        firstSourceCookie = new Cookie(SRC_COOKIE, sourceCookieVal);

        firstSourceCookie.setMaxAge(SECONDS_IN_YEAR);

        firstSourceCookie.setPath("/");

        if (!cookieDomain.isEmpty()) {

            firstSourceCookie.setDomain(cookieDomain);

        }

        cookiesMap.put(SRC_COOKIE, firstSourceCookie);

        firstSourceTimeCookie = new Cookie(SRC_TIME_COOKIE,

                Long.toString((new Date()).getTime()));

        firstSourceTimeCookie.setMaxAge(SECONDS_IN_YEAR);

        firstSourceTimeCookie.setPath("/");

        if (!cookieDomain.isEmpty()) {

            firstSourceTimeCookie.setDomain(cookieDomain);

        }

        cookiesMap.put(SRC_TIME_COOKIE, firstSourceTimeCookie);

        response.addCookie(firstSourceCookie);

        response.addCookie(firstSourceTimeCookie);

    }

    /**

     * Update first source cookie

     * 

     */

    private void updateSessionSourceCookie() {

        sessionSourceCookie = (Cookie) cookiesMap.get(SESSION_SRC_COOKIE);

        if (sessionSourceCookie != null) {

            return;

        }

        String src = getSource();

        DesEncrypter des = new DesEncrypter(ENCRIPTION_STRING);

        String sessionSourceCookieVal = des.encrypt(src);

        //session source

        sessionSourceCookie = new Cookie(SESSION_SRC_COOKIE,

                sessionSourceCookieVal);

        sessionSourceCookie.setPath("/");

        if (!cookieDomain.isEmpty()) {

            sessionSourceCookie.setDomain(cookieDomain);

        }

        cookiesMap.put(SESSION_SRC_COOKIE, sessionSourceCookie);

        //session time

        sessionSourceTimeCookie = new Cookie(SESSION_SRC_TIME_COOKIE,

                Long.toString((new Date()).getTime()));

        sessionSourceTimeCookie.setPath("/");

        if (!cookieDomain.isEmpty()) {

            sessionSourceTimeCookie.setDomain(cookieDomain);

        }

        cookiesMap.put(SESSION_SRC_TIME_COOKIE, sessionSourceTimeCookie);

        //session id

        String sessionId = request.getSession().getId();

        sessionIdCookie = new Cookie(SESSION_ID_COOKIE, sessionId);

        sessionIdCookie.setPath("/");

        if (!cookieDomain.isEmpty()) {

            sessionIdCookie.setDomain(cookieDomain);

        }

        cookiesMap.put(SESSION_ID_COOKIE, sessionIdCookie);

        response.addCookie(sessionSourceCookie);

        response.addCookie(sessionSourceTimeCookie);

        response.addCookie(sessionIdCookie);

        UserSessionInfo userInfo = (UserSessionInfo) request

                .getAttribute(UserInterceptor.USER_INFO_COOKIE_NAME);

        String firstSrc = "";

        firstSourceCookie = (Cookie) cookiesMap.get(SRC_COOKIE);

        if (firstSourceCookie != null) {

            firstSrc = des.decrypt(firstSourceCookie.getValue());

        }

        DataLogger.logData(EventType.NEW_SESSION, cookiesMap.get(SESSION_ID_COOKIE).getValue(),

                userInfo.getUserId(), userInfo.getEmail(), src, firstSrc);

    }

    private void createCookiesMap() {

        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 updateAffCookie() {

        try {

            if(affId == null || affId.isEmpty()) {

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

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

                    Cookie affCookie = cookiesMap.get(AFF_COOKIE);

                    if(affCookie != null) {

                        affCookie.setDomain(cookieDomain);

                        affCookie.setPath("/");

                        affCookie.setValue("");

                        affCookie.setMaxAge(0);

                        response.addCookie(affCookie);

                    }

                }

                return;

            }

            UserClient userContextServiceClient = new UserClient();

            in.shop2020.model.v1.user.UserContextService.Client userClient = userContextServiceClient.getClient();

            long affiliateId = Long.parseLong(affId);

            Affiliate affiliate = userClient.getAffiliateById(affiliateId);

            if (affiliate != null) {

                affCookie = (Cookie) cookiesMap.get(AFF_COOKIE);

                if (affCookie != null) {

                    Long.parseLong(affCookie.getValue());

                    if (affiliateId == Long.parseLong(affCookie.getValue())) {

                        return;

                    }

                }

                // create new aff cookie if affCookie ==null or affiliateId() != affCookie.

                Cookie newAffCookie = new Cookie(AFF_COOKIE, String.valueOf(affiliateId));

                newAffCookie.setMaxAge(SECONDS_IN_DAY * 5);

                newAffCookie.setPath("/");

                if (!cookieDomain.isEmpty()) {

                    newAffCookie.setDomain(cookieDomain);

                }

                cookiesMap.put(AFF_COOKIE, newAffCookie);

                response.addCookie(newAffCookie);

            }

        } catch (Exception e) {

            log.error("Failed to update affcookie.", e);

        }

	}

    private String getSource() {

        String src = "";

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

        if (referer == null || referer.isEmpty()) {

            String queryString = request.getQueryString();

            queryString = queryString == null? "" : "?" + queryString;

            src = "DIRECT : " + "(" + request.getRequestURL() + "/" + queryString + ")";

        } 

        else if (referer.startsWith("http://www.saholic.com") || referer.startsWith("http://www.shop2020.in")) {

            src = "DIRECT : " + "(" + referer + ")";

        } 

        else if(paidAffIds.contains(affId)) {

            src = "PAID : " + "(" + referer + ")";

        }

        else if (referer.contains("google.co")) {

            src = "ORGANIC : " + "(" + referer + ")";

        }

        else {

            src = referer;

        }

        return src;

    }

}