Subversion Repositories SmartDukaan

package in.shop2020.serving.controllers;

import java.util.Date;

import javax.servlet.http.Cookie;

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.thrift.clients.UserContextServiceClient;

import in.shop2020.thrift.clients.UserProfileClient;

import com.opensymphony.xwork2.ModelDriven;

@Results({

    @Result(name="success", type="redirectAction", params = {"actionName" , 

    		"entity"})

})

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;

	public AuthController(){

		try {

			client = new UserContextServiceClient();

		} catch (Exception e) {

			e.printStackTrace();

		}

	}

	@Override

	public Object getModel() {

		return this.auth;

	}

	public HttpHeaders create(){

		System.out.println(auth);

		//return new DefaultHttpHeaders("lsuccess");

		//AUTH service to be invoked from here

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

				this.session.removeAttribute("useid");

				this.session.removeAttribute("loggedin");

				this.session.removeAttribute("issessionid");

				this.session.setAttribute("userid", userId+"");

				this.session.setAttribute("loggedin", "true");

				this.session.setAttribute("issessionid", "false");

				Cookie cookie1 = new Cookie("userid", userId+"");

		    	this.response.addCookie(cookie1);

				return new DefaultHttpHeaders("lsuccess");

			}

		} catch (AuthenticationException e) {

			errorCode = e.getErrorCode();

			errorMessage = e.getMessage();

		} catch (TException e) {

			// TODO Auto-generated catch block

			errorCode = -1;

			e.printStackTrace();

		} catch (UserContextException e) {

			// TODO Auto-generated catch block

			errorCode = -1;

			e.printStackTrace();

		}

		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;

	}

}