Subversion Repositories SmartDukaan

package in.shop2020.serving.controllers;

import java.util.Date;

import javax.servlet.http.Cookie;

import org.apache.juli.logging.Log;
import org.apache.juli.logging.LogFactory;
import org.apache.struts2.convention.annotation.Result;
import org.apache.struts2.convention.annotation.Results;
import org.apache.struts2.rest.DefaultHttpHeaders;
import org.apache.struts2.rest.HttpHeaders;
import org.apache.thrift.TException;

import in.shop2020.model.v1.user.AuthenticationException;
import in.shop2020.model.v1.user.UserContextException;
import in.shop2020.model.v1.user.UserContextService.Client;
import in.shop2020.serving.utils.Utils;
import in.shop2020.thrift.clients.UserContextServiceClient;

import com.opensymphony.xwork2.ModelDriven;

/**
 * Authantication class to authenticate the user.
 * 
 * @author rajveer
 *
 */

public class AuthController extends BaseController implements ModelDriven<Object>{

        private Auth auth = new Auth();
        
        private UserContextServiceClient client = null;
        
        private int errorCode = 0;
        private String errorMessage;
        
        private String id;
        
        private String nameOfUser;
        
        private static Log log = LogFactory.getLog(AuthController.class);
        
        public AuthController(){
                try {
                        client = new UserContextServiceClient();
                } catch (Exception e) {
                        e.printStackTrace();
                }
        }
        
        @Override
        public Object getModel() {
                return this.auth;
        }
        
        // POST /auth
        public HttpHeaders create(){
                log.info(auth);
                Client c = client.getClient();
                try {
                        if (c.authenticateUser(auth.getUserName(), auth.getPassword(), true)){
                                long userId = c.getContext(auth.getUserName(), auth.getPassword()).getId();
                                c.setUserAsLoggedIn(userId, (new Date()).getTime());

                                userinfo.setUserId(userId);
                                userinfo.setLoggedIn(true);
                                userinfo.setEmail(auth.getUserName());
                                userinfo.setNameOfUser(Utils.getNameOfUser(userId));
                                this.nameOfUser = Utils.getNameOfUser(userId);
                                if(userinfo.getCartId() == -1){
                                        userinfo.setCartId(Utils.getCartId(userId));
                                        userinfo.setTotalItems(Utils.getNumberOfItemsInCart(userinfo.getCartId()));
                                }else{
                                        Utils.mergeCarts(userinfo.getCartId(), Utils.getCartId(userId));
                                        userinfo.setCartId(Utils.getCartId(userId));
                                        userinfo.setTotalItems(Utils.getNumberOfItemsInCart(userinfo.getCartId()));
                                }
                                this.session.setAttribute("userinfo", userinfo);

                                Cookie cookie1 = new Cookie("userid", userId+"");
                        this.response.addCookie(cookie1);
                                return new DefaultHttpHeaders("lsuccess");
                        }
                } catch (AuthenticationException e) {
                        
                        errorCode = e.getErrorCode();
                        errorMessage = e.getMessage();
                        log.error("Exception number is: " +  errorCode + "Exception message is:" + errorMessage);
                } catch (TException e) {
                        errorCode = -1;
                        e.printStackTrace();
                        log.error("Exception number is: " +  errorCode + "Exception message is:" + errorMessage);
                } catch (UserContextException e) {
                        errorCode = e.getErrorCode();
                        e.printStackTrace();
                        log.error("Exception number is: " +  errorCode + "Exception message is:" + errorMessage);
                }
                return new DefaultHttpHeaders("lfail");
        }

        public int getErrorCode() {
                return errorCode;
        }

        public String getErrorMessage() {
                return errorMessage;
        }
        
        public String getId(){
                return id;
        }
        
        public void setId(String id){
                this.id = id;
        }

        public Auth getAuth() {
                return auth;
        }

        public void setAuth(Auth auth) {
                this.auth = auth;
        }
        
        public String getUserName(){
                return nameOfUser;
        }
}