Subversion Repositories SmartDukaan

package com.spice.profitmandi.web.processor;

import java.io.IOException;

import java.time.LocalDateTime;

import java.util.HashMap;

import java.util.HashSet;

import java.util.Iterator;

import java.util.Map;

import java.util.Set;

import org.springframework.beans.factory.annotation.Autowired;

import org.springframework.stereotype.Component;

import com.fasterxml.jackson.core.JsonProcessingException;

import com.fasterxml.jackson.databind.JsonNode;

import com.fasterxml.jackson.databind.ObjectMapper;

import com.spice.profitmandi.common.exception.ProfitMandiBusinessException;

import com.spice.profitmandi.common.model.ProfitMandiConstants;

import com.spice.profitmandi.common.util.JWTUtil;

import com.spice.profitmandi.dao.entity.Role;

import com.spice.profitmandi.dao.entity.SocialUser;

import com.spice.profitmandi.dao.entity.User;

import com.spice.profitmandi.dao.enumuration.Gender;

import com.spice.profitmandi.dao.enumuration.SocialType;

import com.spice.profitmandi.dao.repository.SocialUserRepository;

import com.spice.profitmandi.dao.repository.UserRepository;

import com.spice.profitmandi.web.client.RestClient;

import com.spice.profitmandi.web.enumuration.SchemeType;

@Component

public class GoogleLoginProcessor {

	private static final String HOST_NAME = "content.googleapis.com";

	private static final String URI = "/plus/v1/people/me";

	private static final int PORT_NUMBER = 443;

	private final ObjectMapper objectMapper = new ObjectMapper();

	@Autowired

	SocialUserRepository socialUserRepository;

	@Autowired

	UserRepository userRepository;

	public Map<String, Object> process(Map<String, Object> map) throws ProfitMandiBusinessException{

		RestClient restClient = new RestClient(SchemeType.HTTPS, HOST_NAME, PORT_NUMBER);

		Map<String, String> params = new HashMap<>();

		params.put(ProfitMandiConstants.ACCESS_TOKEN, map.get(ProfitMandiConstants.TOKEN).toString());

		String responseString = restClient.get(URI, params);

		try {

			JsonNode rootNode = objectMapper.readTree(responseString);

			SocialUser socialUser = new SocialUser();

			if(rootNode.has("emails")){

				JsonNode emails = rootNode.get("emails");

				if(emails.isArray()){

					Iterator<JsonNode> emailsIterator = emails.elements();

					if(emailsIterator.hasNext()){

						JsonNode email = emailsIterator.next();

						if(email.has("value")){

							socialUser.setEmailId(email.get("value").asText());

						}

					}

				}

				//socialUser.setEmailId(rootNode.get("email").asText());

			}

			if(rootNode.has("displayName")){

				socialUser.setName(rootNode.get("displayName").asText());

			}

			if(rootNode.has("gender")){

				String genderName = rootNode.get("gender").asText();

				switch(genderName){

					case "male":{

						socialUser.setGender(Gender.MALE);

						break;

					}case "female":{

						socialUser.setGender(Gender.FEMALE);

						break;

					}

				}

			}

			socialUser.setCreateTimestamp(LocalDateTime.now());

			socialUser.setType(SocialType.GOOGLE);

			socialUser.setUpdateTimestamp(LocalDateTime.now());

			socialUserRepository.persist(socialUser);

			Map<String, Object> responseMap = new HashMap<>(2);

			try{

				User user = userRepository.selectByEmailId(socialUser.getEmailId());

				//Set<Role> roles = user.getRoles();

				Set<Role> roles = new HashSet<>();

				String[] roleNames = new String[roles.size()];

				int index = 0;

				for(Role role : roles){

					roleNames[index++] = role.getName();

				}

				responseMap.put(ProfitMandiConstants.TOKEN, JWTUtil.create(user.getId(), roleNames));

				responseMap.put(ProfitMandiConstants.REGISTERED, true);

			}catch (ProfitMandiBusinessException profitMandiBusinessException) {

				responseMap.put(ProfitMandiConstants.TOKEN, JWTUtil.create(socialUser.getEmailId()));

				responseMap.put(ProfitMandiConstants.REGISTERED, false);

			}

			return responseMap;

		}catch (JsonProcessingException jsonProcessingException) {

			//LOGGER.error("Json parse exception of "+json,jsonProcessingException);

			throw new ProfitMandiBusinessException("", "", "VE_1001");

		}catch (IOException ioException) {

			//LOGGER.error("IO Exception occurred while parsing json String");

			throw new ProfitMandiBusinessException("", "", "VE_1001");

		}

	}

}