WebSVN – SmartDukaan – Diff – /trunk/Website/src/main/java/in/shop2020/serving/controllers/LoginController.java

  */
 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.interceptors.UserInterceptor;
 import in.shop2020.serving.utils.DesEncrypter;
 import in.shop2020.serving.utils.UserMessage;
 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 javax.servlet.http.HttpServletResponse;
+import org.apache.commons.codec.binary.Base64;
+import org.apache.commons.lang.RandomStringUtils;
 import org.apache.log4j.Logger;
+import org.apache.struts2.ServletActionContext;
 import org.apache.struts2.convention.annotation.Action;
 import org.apache.struts2.convention.annotation.Actions;
 import org.apache.struts2.convention.annotation.InterceptorRef;
 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
+ *
 	private static Logger log = Logger.getLogger(Class.class);
 	private DesEncrypter desEncrypter = new DesEncrypter("saholic");
 	private String loginResult = "0";
 	private String redirectUrl = "/";
+	private String signed_request;
+	private Long userId = -1L;
 	public LoginController() {
 		super();
+	}
 	@Actions({
 		@Action(value="login", interceptorRefs={@InterceptorRef("myDefault")}),
+		}
+	}
 	private boolean loginUser() {
 		try {
-			String email, password;
+			String email, password = null, isFacebookUser;
-			email = this.request.getParameter("email");
-			password = this.request.getParameter("password");
+			isFacebookUser = (String)this.request.getAttribute("isFacebookUser");
-			if (email == null || password == null) {
+			if (isFacebookUser!=null && isFacebookUser.equals("true")) {
+				email = (String)this.request.getAttribute("email");
+			} else {
+				email = this.request.getParameter("email");
+				password = this.request.getParameter("password");
+			}
+			if (email == null || (password == null && (isFacebookUser==null || !isFacebookUser.equals("true")))) {
 				return false;
+			}
+			String encryptedPassword = null;
+			if(!(password==null)){
-			String encryptedPassword = desEncrypter.encrypt(password);
+				encryptedPassword = desEncrypter.encrypt(password);
+			}
 			UserClient userContextServiceClient = new UserClient();
 			in.shop2020.model.v1.user.UserContextService.Client userClient = userContextServiceClient.getClient();
+			User user = null;
+			boolean shouldChangeCookieExpiry = false;
+			boolean isNewFacebookUser = false;
+			if (isFacebookUser!=null && isFacebookUser.equals("true") && !email.equals("undefined")) {
+				String accessToken = (String)this.request.getAttribute("accessToken");
+				String facebookId = (String)this.request.getAttribute("facebookId");
+				if (userClient.userExists(email)){
+					shouldChangeCookieExpiry = true;
+					user = userClient.getUserByEmail(email);
+					userId = user.getUserId();
+					if(!user.isIsFacebookUser()) {
+						user.setIsFacebookUser(true);
+						user.setFacebookAccessToken(accessToken);
+						user.setFacebookId(facebookId);
+						userClient.updateUser(user);
+					} else if(user.getFacebookAccessToken() == null || (accessToken!=null && !accessToken.equals(user.getFacebookAccessToken()))){
+						user.setFacebookAccessToken(accessToken);
+						if(facebookId!=null) {
+							user.setFacebookId(facebookId);
+						}
+						userClient.updateUser(user);
+					}
+				} else {
+					isNewFacebookUser = createNewUserForFacebook(email);
+				}
+				if(shouldChangeCookieExpiry || isNewFacebookUser) {
+					Cookie uidCookie = (Cookie) cookiesMap.get(UserInterceptor.USER_ID_COOKIE_NAME);
+					Cookie uicCookie = (Cookie) cookiesMap.get(UserInterceptor.USER_INFO_COOKIE_NAME);
+					HttpServletResponse response = ServletActionContext.getResponse();
+					if (uidCookie == null) {
+						DesEncrypter cookieEncryter = new DesEncrypter(UserInterceptor.COOKIE_DECRYPTION_STRING);
+						uidCookie = new Cookie(UserInterceptor.USER_ID_COOKIE_NAME, cookieEncryter.encrypt(userId + ""));
+						uidCookie.setPath("/");
+						if(!domainName.isEmpty()) {
+							uidCookie.setDomain(domainName);
+						}
+					}
+					if (uicCookie != null) {
+						uicCookie.setMaxAge(UserInterceptor.FACEBOOK_USER_COOKIE_EXPIRY_TIME);
+						cookiesMap.put(UserInterceptor.USER_INFO_COOKIE_NAME, uicCookie);
+				        response.addCookie(uicCookie);
+					}
+					uidCookie.setMaxAge(UserInterceptor.FACEBOOK_USER_COOKIE_EXPIRY_TIME);
+					cookiesMap.put(UserInterceptor.USER_ID_COOKIE_NAME, uidCookie);
+			        response.addCookie(uidCookie);
+			        if(isNewFacebookUser) {
+			        	return true;
+			        }
+				}
+			} else {
-			User user = userClient.authenticateUser(email, encryptedPassword);
+				user = userClient.authenticateUser(email, encryptedPassword);
+			}
 			userClient.setUserAsLoggedIn(user.getUserId(),(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
 			log.error(UserMessage.USER_AUTHENTICATION_FAILURE, e);
 			return false;
+		}
+	}
+	private boolean createNewUserForFacebook(String email) {
+    	String password;
+    	password = RandomStringUtils.randomAlphabetic(8);
+    	String name = (String)this.request.getAttribute("name");
+    	String accessToken = (String)this.request.getAttribute("accessToken");
+    	String facebookId = (String)this.request.getAttribute("facebookId");
+		User user = new User();
+		user.setEmail(email);
+		String encryptedPassword = desEncrypter.encrypt(password);
+		user.setPassword(encryptedPassword);
+		user.setCommunicationEmail(email);
+		user.setName(name);
+		user.setFacebookAccessToken(accessToken);
+		if(user.getFacebookId()==null || user.getFacebookId().isEmpty()) {
+			user.setFacebookId(facebookId);
+		}
+		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);
+		user.setIsFacebookUser(true);
+		UserClient userContextServiceClient;
+		try {
+			userContextServiceClient = new UserClient();
+			in.shop2020.model.v1.user.UserContextService.Client userClient = userContextServiceClient.getClient();
+			try{
+				user = userClient.createUser(user);
+			}catch (UserContextException ux){
+				addActionError("User already exists with this email id.");
+				DataLogger.logData(EventType.REGISTER_FAILED_USER_EXISTS, getSessionId(), userinfo.getUserId(), email, "", email);
+				return false;
+			}
+			this.userId = user.getUserId();
+			long userId = user.getUserId();
+			userClient.setUserAsLoggedIn(userId, (new Date()).getTime());
+			String pincode = userClient.getDefaultPincode(user.getUserId());
+			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());
+			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);
+			return true;
+		} catch (TTransportException e) {
+			log.error("Unable to register user." + e);
+		} catch (UserContextException e) {
+			log.error("Unable to register user." + e);
+		} catch (TException e) {
+			log.error("Unable to register user." + e);
+		} catch (UserAffiliateException e) {
+			log.error("Unable to register user." + e);
+		} catch (ShoppingCartException e) {
+			log.error("Unable to register user." + e);
+		} catch (WidgetException e) {
+			log.error("Unable to register user." + e);
+		}
+		return false;
+	}
+	public String facebooklogin(){
+		try {
+			String dataResponse = signed_request.split("\\.")[1];
+			String encoded_json = dataResponse.replace("-", "+").replace("_", "/");
+			byte[] decoded_json  = Base64.decodeBase64(encoded_json);
+			JSONObject json_data = new JSONObject(new String(decoded_json, "UTF-8"));
+			this.request.setAttribute("name", (new JSONObject(json_data.get("registration").toString())).get("name"));
+			this.request.setAttribute("email", (new JSONObject(json_data.get("registration").toString())).get("email"));
+			this.request.setAttribute("accessToken", json_data.get("oauth_token").toString());
+			this.request.setAttribute("facebookId", json_data.get("user_id").toString());
+			this.request.setAttribute("isFacebookUser", "true");
+			if(loginUser()){
+				return "redirect";
+			} else {
+				log.error("Error in logging through facebook");
+				addActionError(UserMessage.FACEBOOK_USER_AUTH_FAILURE);
+				return "login";
+			}
+		} catch (Exception e ) {
+			log.error("Error in reading facebook response for login", e);
+			addActionError(UserMessage.FACEBOOK_USER_AUTH_FAILURE);
+			return "login";
+		}
+	}
 	public String authenticateUser() {
 		String email, password;
 		email = this.request.getParameter("email");
 		password = this.request.getParameter("password");
 	public void setRedirectUrl(String redirectUrl) {
 		this.redirectUrl = redirectUrl;
+	}
+	public String getSigned_request() {
+		return signed_request;
+	}
+	public void setSigned_request(String signed_request) {
+		this.signed_request = signed_request;
+	}
 	public static void main(String[] args) {
         DesEncrypter des = new DesEncrypter("saholic");
         System.out.println(des.decrypt("XvrWIvmYrUAdXqxhovZMSw"));
+    }
+}

Subversion Repositories SmartDukaan

(root)/trunk/Website/src/main/java/in/shop2020/serving/controllers/LoginController.java – Rev 7039 → 7825