Subversion Repositories SmartDukaan

/**
 * 
 */
package in.shop2020.serving.controllers;


import in.shop2020.datalogger.EventType;
import in.shop2020.model.v1.user.Cart;
import in.shop2020.model.v1.user.Sex;
import in.shop2020.model.v1.user.ShoppingCartException;
import in.shop2020.model.v1.user.TrackLogType;
import in.shop2020.model.v1.user.User;
import in.shop2020.model.v1.user.UserAffiliateException;
import in.shop2020.model.v1.user.UserContextException;
import in.shop2020.model.v1.user.WidgetException;
import in.shop2020.serving.interceptors.TrackingInterceptor;
import in.shop2020.serving.utils.DesEncrypter;
import in.shop2020.serving.utils.Utils;
import in.shop2020.thrift.clients.UserClient;
import in.shop2020.utils.DataLogger;

import java.io.IOException;
import java.util.Date;
import java.util.List;

import javax.servlet.http.Cookie;

import org.apache.log4j.Logger;
import org.apache.struts2.convention.annotation.Result;
import org.apache.struts2.convention.annotation.Results;
import org.apache.thrift.TException;
import org.apache.thrift.transport.TTransportException;
import org.json.JSONObject;

/**
 * 
 * @author rajveer
 *
 */

@Results({
    @Result(name = "success", type = "redirectAction", params = {"actionName" , "home"}),
    @Result(name = "failure", type = "redirectAction", params = {"actionName" , "login"}),
    @Result(name = "redirect-to-login", type = "redirectAction", params = {"actionName" , "login"}),
    @Result(name = "redirect", location = "${redirectUrl}", type = "redirect")
})
public class RegisterController extends BaseController{
        
        private static final long serialVersionUID = 1L;
        private static Logger log = Logger.getLogger(Class.class);
        private DesEncrypter desEncrypter = new DesEncrypter("saholic");
        
        private String redirectUrl = "/";
        private String registerJson = "{}";
        public RegisterController(){
                super();
        }

    public String index() throws SecurityException, IOException {
        if(userinfo.isLoggedIn()){
                return "success";
        }
//      htmlSnippets.put("REGISTRATION_HEADER",pageLoader.getRegistrationHeaderHtml());
//      htmlSnippets.put("REGISTRATION_FORM",pageLoader.getRegistrationFormHtml());
        return "redirect-to-login";
    }

    public String create() throws SecurityException, Exception {
        log.info("in register");
        boolean result= registerUser();
        log.info("result "+result);
                return "register-response";
        //else
        //      return "failure";
    }
    
    public boolean registerUser(){
        String email, password;
        boolean isValid = true;
        JSONObject json = new JSONObject();
        String errorMessage = "{\"status\":\"error\",\"message\":\"Some Problem occurred. Please try again.\"}";
        email = this.request.getParameter("email"); 
        password = this.request.getParameter("password");
        try{
        if(!Utils.isValidEmail(email))  {
                addActionError("Please enter valid email address.");
                json.put("message", "Please enter valid email address.");
                json.put("status", "error");
                isValid = false;
        }
        if(password == null )   {
                addActionError("Please enter password.");
                json.put("message", "Please enter password.");
                json.put("status", "error");
                isValid = false;
        }
        
        if(!isValid){
                setRegisterJson(json.toString());
//          DataLogger.logData(EventType.REGISTER_DATA_INCOMPLETE, getSessionId(), userinfo.getUserId(), email, "", email);
                return isValid;
        }
        
                User user = new User();
                user.setEmail(email);
                String encryptedPassword = desEncrypter.encrypt(password);
                user.setPassword(encryptedPassword);
                user.setCommunicationEmail(email);
                Cookie sourceCookie = (Cookie) cookiesMap.get(TrackingInterceptor.SRC_COOKIE);
        if (sourceCookie != null) {
            DesEncrypter des = new DesEncrypter(TrackingInterceptor.ENCRIPTION_STRING);
            String sourceCookieVal = des.decrypt(sourceCookie.getValue());
            user.setSource(sourceCookieVal);
        }
        
        Cookie sourceTimeCookie = (Cookie) cookiesMap.get(TrackingInterceptor.SRC_TIME_COOKIE);
        long sourceTime = 0;
        if (sourceTimeCookie != null) {
            try {
                sourceTime = Long.parseLong(sourceTimeCookie.getValue());
            }
            catch (Exception e) {
                log.warn("Unable to parse session src time cookie.");
            }
            user.setSourceStartTime(sourceTime);
        }
                
                user.setSex(Sex.WONT_SAY);
                

                UserClient userContextServiceClient;
                try {
                        userContextServiceClient = new UserClient();
                
                        in.shop2020.model.v1.user.UserContextService.Client userClient = userContextServiceClient.getClient();
                        try{
                                log.info("trying to create user");
                                user = userClient.createUser(user);
                        }catch (UserContextException ux){
                                ux.printStackTrace();
                                log.info("exception in creating user");
                                addActionError("User already exists with this email id.");
                                json.put("message", "You are already a member. Please login.");
                                json.put("status", "error");
                                
                                setRegisterJson(json.toString());
//                              DataLogger.logData(EventType.REGISTER_FAILED_USER_EXISTS, getSessionId(), userinfo.getUserId(), email, "", email);
                                return false;                           
                        }
                        
                        long userId = user.getUserId();
                        userClient.setUserAsLoggedIn(userId, (new Date()).getTime());
                        String pincode = userClient.getDefaultPincode(user.getUserId());
                
                        // TODO: setTotalItems shouldn't be a method on userinfo. This allows
                        // for potentially updating the item count wrongly. The method setCartId
                        // should update the item count as well. Also, there can be a method
                        // called refreshItemCount() that automatically updates the number of
                        // items currently in the cart.
                        if(userinfo.getUserId() != -1){
                                userClient.mergeCart(userinfo.getCartId(), user.getActiveCartId());
                        
                                List<Long> items = userClient.getBrowseHistoryItems(userinfo.getUserId());
                                if(items != null){
                                        for(Long itemId: items){
                                                userClient.updateBrowseHistory(user.getUserId(), itemId);
                                        }
                                }
                        
                                items = userClient.getMyResearchItems(userinfo.getUserId());
                                if(items != null){
                                        for(Long itemId: items){
                                                userClient.updateMyResearch(user.getUserId(), itemId);
                                        }
                                }
                        }
                        
                        userinfo.setUserId(userId);
                        userinfo.setEmail(email);
                        userinfo.setLoggedIn(true);
                        userinfo.setPincode(pincode);
                        userinfo.setCartId(user.getActiveCartId());
                        userinfo.setPrivateDealUser(isPrivateDealUser());
                        Cart cart = userClient.getCart(userinfo.getCartId());
                        userinfo.setTotalItems(cart.getLinesSize());
                        userinfo.setTotalAmount(cart.getTotalPrice());
                        
                if (cookiesMap.containsKey(TrackingInterceptor.AFF_COOKIE)) {
                    long affId = Long.parseLong(cookiesMap.get(TrackingInterceptor.AFF_COOKIE).getValue());
                    userClient.addTrackLog(affId, userId, TrackLogType.NEW_REGISTRATION, "",email, (new Date()).getTime());
                }
//              DataLogger.logData(EventType.REGISTER_SUCCESS, getSessionId(), userinfo.getUserId(), email, "", email);
                json.put("message", "Register successful");
                        json.put("status", "success");
                        
                        setRegisterJson(json.toString());
                        return true;
                } catch (TTransportException e) {
                        log.error("Unable to register user." + e);
                        setRegisterJson(errorMessage);
                } catch (UserContextException e) {
                        log.error("Unable to register user." + e);
                        setRegisterJson(errorMessage);
                } catch (TException e) {
                        log.error("Unable to register user." + e);
                        setRegisterJson(errorMessage);
                } catch (UserAffiliateException e) {
                        log.error("Unable to register user." + e);
                        setRegisterJson(errorMessage);
                } catch (ShoppingCartException e) {
                        log.error("Unable to register user." + e);
                        setRegisterJson(errorMessage);
                } catch (WidgetException e) {
                        log.error("Unable to register user." + e);
                        setRegisterJson(errorMessage);
                }
        }catch(Exception e){
                log.error("Unable to register user." + e);
                setRegisterJson(errorMessage);
        }
//              DataLogger.logData(EventType.REGISTER_FAILED_USER_EXISTS, getSessionId(), userinfo.getUserId(), email, "", email);
                return false;
                
    }

        public String getRegistrationHeaderSnippet(){
                return htmlSnippets.get("REGISTRATION_HEADER");
        }
        
        public String getRegistrationFormSnippet(){
                return htmlSnippets.get("REGISTRATION_FORM");
        }
        public String getRedirectUrl() {
                return redirectUrl;
        }

        public void setRedirectUrl(String redirectUrl) {
                this.redirectUrl = redirectUrl;
        }
        
        
        
        public String getRegisterJson() {
                return registerJson;
        }

        public void setRegisterJson(String registerJson) {
                this.registerJson = registerJson;
        }

        private boolean isPrivateDealUser() throws TTransportException, TException{
                try {
                        in.shop2020.model.v1.user.UserContextService.Client uc = new UserClient().getClient();
                        return uc.isPrivateDealUser(userinfo.getUserId());
                } catch (TTransportException e) {
                        log.error("Unable to get user service client.", e);     
                }
                return false;
        }
}