Subversion Repositories SmartDukaan

package com.spice.profitmandi.web.controller;

import java.util.HashSet;

import java.util.List;

import java.util.Set;

import javax.servlet.http.Cookie;

import javax.servlet.http.HttpServletRequest;

import javax.servlet.http.HttpServletResponse;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

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

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

import org.springframework.stereotype.Controller;

import org.springframework.transaction.annotation.Transactional;

import org.springframework.ui.Model;

import org.springframework.ui.ModelMap;

import org.springframework.web.bind.annotation.ModelAttribute;

import org.springframework.web.bind.annotation.RequestMapping;

import org.springframework.web.bind.annotation.RequestMethod;

import org.springframework.web.bind.annotation.RequestParam;

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

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

import com.spice.profitmandi.dao.entity.dtr.Retailer;

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

import com.spice.profitmandi.dao.entity.dtr.UserAccounts;

import com.spice.profitmandi.dao.entity.dtr.UserRole;

import com.spice.profitmandi.dao.enumuration.dtr.AccountType;

import com.spice.profitmandi.dao.enumuration.dtr.RoleType;

import com.spice.profitmandi.dao.repository.dtr.RetailerRepository;

import com.spice.profitmandi.dao.repository.dtr.UserAccountRepository;

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

import com.spice.profitmandi.dao.repository.dtr.UserRoleRepository;

import com.spice.profitmandi.web.model.FofoDetails;

import com.spice.profitmandi.web.util.CookiesProcessor;

import com.spice.profitmandi.web.util.GoogleTokenUtil;

import com.spice.profitmandi.web.util.MVCResponseSender;

@Controller

@Transactional(rollbackFor=Throwable.class)

public class LoginController {

	private static final Logger LOGGER = LoggerFactory.getLogger(LoginController.class);

	@Autowired

	GoogleTokenUtil googleTokenUtil;

	@Autowired

	RetailerRepository retailerRepository;

	@Autowired

	UserRepository userRepository;

	@Autowired

	UserAccountRepository userAccountRepository;

	@Autowired

	UserRoleRepository userRoleRepository;

	@Autowired

	MVCResponseSender mvcResponseSender;

	@Autowired

	CookiesProcessor cookiesProcessor;

	@Value("${google.api.key}")

	private String googleApiKey;

	@RequestMapping(value = "/login", method = RequestMethod.GET)

	public String loginPage(HttpServletRequest request, Model model) throws Exception{

		LOGGER.info("Context Path is {}", request.getContextPath());

		try{

			cookiesProcessor.getCookiesObject(request);

			LOGGER.info("Request session is already exist, should be redirect to /dashboard");

			return "redirect:/dashboard";

		}catch(Exception | ProfitMandiBusinessException profitMandiBusinessException){

			model.addAttribute("googleApiKey", googleApiKey);

			model.addAttribute("appContextPath", request.getContextPath());

			return "login";

		}

	}

	@RequestMapping(value = "/login", method = RequestMethod.POST)

	public String login(HttpServletRequest request, HttpServletResponse response, @RequestParam(name = ProfitMandiConstants.TOKEN) String token, Model model) throws Exception{

		try{

			String emailId = googleTokenUtil.getEmailId(token);

			User user = null;

			try{

				user = userRepository.selectByEmailId(emailId);

			}catch(ProfitMandiBusinessException profitMandiBusinessException){

				LOGGER.error("User not found with given emailId", profitMandiBusinessException);

				model.addAttribute("loginResponse", mvcResponseSender.createResponseString("RTLR_OK_1002", true, request.getContextPath() + "/register"));

				return "response";

			}

			UserAccounts userAccounts = userAccountRepository.getUserAccountByType(user.getId(), AccountType.saholic);

			Retailer retailer = retailerRepository.selectById(Integer.parseInt(userAccounts.getAccount_key()));

			if(!retailer.isFofo()){

				LOGGER.error("Retailer is not fofo, should be registered");

				model.addAttribute("loginResponse", mvcResponseSender.createResponseString("RTLR_OK_1002", true, request.getContextPath() + "/register"));

				return "response";

			}

			List<UserRole> userRoles = userRoleRepository.selectByUserId(user.getId());

			Set<RoleType> roleTypes = new HashSet<>();

			StringBuilder roleNames = new StringBuilder();

			for(int index = 0; index < userRoles.size(); index++){

				roleTypes.add(userRoles.get(index).getRoleType());

				roleNames.append(userRoles.get(index).getRoleType().toString());

				if(index + 1 != userRoles.size()){

					roleNames.append(",");

				}

			}

			FofoDetails fofoDetails = new FofoDetails();

			fofoDetails.setFofoId(retailer.getId());

			fofoDetails.setEmailId(emailId);

			fofoDetails.setRoleTypes(roleTypes);

			//FofoDetails fofoDetails = googleTokenUtil.getFofoDetail(token);

			Cookie cookieFofoId = new Cookie(ProfitMandiConstants.FOFO_ID, String.valueOf(fofoDetails.getFofoId()));

			cookieFofoId.setDomain(request.getServerName());

			cookieFofoId.setPath(request.getContextPath());

			Cookie cookieEmailId = new Cookie(ProfitMandiConstants.EMAIL_ID, fofoDetails.getEmailId());

			cookieEmailId.setDomain(request.getServerName());

			cookieEmailId.setPath(request.getContextPath());

			Cookie cookieRoleNames = new Cookie(ProfitMandiConstants.ROLE_NAMES, roleNames.toString());

			response.addCookie(cookieFofoId);

			response.addCookie(cookieEmailId);

			response.addCookie(cookieRoleNames);

			LOGGER.info("Requested token email_id is valid, user login to system, shoud be redirect to /dashboard");

			model.addAttribute("loginResponse", mvcResponseSender.createResponseString("RTLR_OK_1002", true, request.getContextPath() + "/dashboard"));

			return "response";

//			return mvcResponseSender.createResponseString("RTLR_OK_1002", true, "/profitmandi-fofo/dashboard");

		}catch(ProfitMandiBusinessException profitMandiBusinessException){

			LOGGER.error("Error : ", profitMandiBusinessException);

			model.addAttribute("loginResponse", mvcResponseSender.createResponseString(profitMandiBusinessException.getCode(), false, "/error"));

			return "response";

		}

	}

	@RequestMapping(value = "/logout", method = RequestMethod.GET)

	public String logout(HttpServletRequest request, @ModelAttribute("model") ModelMap model, HttpServletResponse response) throws Exception{

		try{

			cookiesProcessor.removeCookies(request, response);

			LOGGER.info("Logout is successfull, should be redirect to /login");

			return "redirect:/login";

		}catch(Exception | ProfitMandiBusinessException profitMandiBusinessException){

			LOGGER.info("Error occured while removing requested cookies, should be redirect to /login");

			return "redirect:/login";

		}

	}

}