Subversion Repositories SmartDukaan

package in.shop2020.serving.interceptors;

import java.util.HashMap;

import java.util.Map;

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

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

import in.shop2020.serving.utils.DesEncrypter;

import in.shop2020.thrift.clients.UserContextServiceClient;

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_TWO_MONTHS = 60*60*24*60;

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

	public static final String TRACKER = "tracker";

	public static final String AFF_COOKIE = "uafc";

	public static final String SRC_COOKIE = "usrcc";

	private String cookieDomain = "";

	private Cookie trackerCookie = null;

	private Cookie affCookie = null;

	private Cookie firstSourceCookie = null;

    private Map<String, Cookie> cookiesMap = null;

    public void setCookieDomain(String cookieDomain) {

        this.cookieDomain = cookieDomain;

    }

    @Override

	public String intercept(ActionInvocation invocation) throws Exception {

		HttpServletRequest request = ServletActionContext.getRequest();

		String affId = request.getParameter("afid");

		createCookiesMap(request);

		cleanTrackerCookie();

		updateFirstSourceCookie();

		if(affId != null && !affId.isEmpty()) {

		    updateAffCookie(affId);

		}

        return invocation.invoke();

	}

    /**

     * Check afid, referer

     * 

     */

    private void updateFirstSourceCookie() {

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

        if (firstSourceCookie != null) {

            return;

        }

        HttpServletRequest request = ServletActionContext.getRequest();

        String src = "";

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

        if (referer == null || referer.isEmpty() || referer.contains("saholic") || referer.contains("shop2020")) {

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

        }

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

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

        }

        else {

            src = referer;

        }

        DesEncrypter des = new DesEncrypter("Saholic");

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

        HttpServletResponse response = ServletActionContext.getResponse();

        response.addCookie(firstSourceCookie);

    }

    private void createCookiesMap(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 cleanTrackerCookie() {

        trackerCookie = (Cookie) cookiesMap.get(TRACKER);

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

        if (trackerCookie != null) {

            if (affCookie == null) {

                try {

                    UserContextServiceClient userContextServiceClient = new UserContextServiceClient();

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

                    Tracker tracker = userClient.getTrackerById(Long.parseLong(trackerCookie.getValue()));

                    affCookie = new Cookie(AFF_COOKIE, String.valueOf(tracker.getAffiliateId()));

                    affCookie.setMaxAge(SECONDS_IN_TWO_MONTHS);

                    affCookie.setPath("/");

                    if (!cookieDomain.isEmpty()) {

                        affCookie.setDomain(cookieDomain);

                    }

                    cookiesMap.put(AFF_COOKIE, trackerCookie);

                    HttpServletResponse response = ServletActionContext.getResponse();

                    response.addCookie(affCookie);

                } catch (Exception e) {

                    // TODO Auto-generated catch block

                    e.printStackTrace();

                }

            }

            if (!cookieDomain.isEmpty()) {

                trackerCookie.setDomain(cookieDomain);

            }

            trackerCookie.setPath("/");

            trackerCookie.setMaxAge(0);

            HttpServletResponse response = ServletActionContext.getResponse();

            response.addCookie(trackerCookie);

        }

    }

	private void updateAffCookie(String affId) {

        try {

            UserContextServiceClient userContextServiceClient = new UserContextServiceClient();

            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_TWO_MONTHS);

                newAffCookie.setPath("/");

                if (!cookieDomain.isEmpty()) {

                    newAffCookie.setDomain(cookieDomain);

                }

                cookiesMap.put(AFF_COOKIE, newAffCookie);

                HttpServletResponse response = ServletActionContext.getResponse();

                response.addCookie(newAffCookie);

            }

        } catch (Exception e) {

            log.error(e);

        }

	}

}