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

    public void setCookieDomain(String cookieDomain) {
        this.cookieDomain = cookieDomain;
    }

    @Override
    public String intercept(ActionInvocation invocation) throws Exception {
        request = ServletActionContext.getRequest();
        response = ServletActionContext.getResponse();

        String affId = request.getParameter("afid");
        createCookiesMap();
        updateFirstSourceCookie();
        updateSessionSourceCookie();

        updateAffCookie(affId);
        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(String affId) {
        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() || referer.contains("saholic")
                || referer.contains("shop2020")) {
            String queryString = request.getQueryString();
            queryString = queryString == null? "" : "?" + queryString;
            src = "DIRECT : " + "(" + request.getRequestURL() + queryString + ")";
        } else if (referer.contains("google.co")) {
            if(request.getQueryString().contains("afid=6")) {
                src = "PAID : " + "(" + referer + ")";
            }
            else {
                src = "ORGANIC : " + "(" + referer + ")";
            }
        } else {
            if(request.getQueryString().contains("afid=6")) {
                src = "PAID : " + "(" + referer + ")";
            }
            else {
                src = referer;
            }
        }
        return src;
    }
}