Subversion Repositories SmartDukaan

package com.spice.profitmandi.web.controller;

import java.io.InputStream;

import java.util.Properties;

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.context.annotation.PropertySource;

import org.springframework.core.io.Resource;

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.web.config.AppConfig;

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

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

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

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

	GoogleLoginUtil googleLoginUtil;

	@Autowired

	MVCResponseSender mvcResponseSender;

	@Autowired

	CookiesProcessor cookiesProcessor;

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

	public String loginPage(HttpServletRequest request, @ModelAttribute("model") ModelMap model) throws Exception{

		try{

			cookiesProcessor.getCookiesObject(request);

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

			return "redirect:/dashboard";

		}catch(Exception | ProfitMandiBusinessException profitMandiBusinessException){

			Resource resource = AppConfig.getResource();

			Properties properties = new Properties();

			properties.load(resource.getInputStream());

			model.addAttribute("googleApiKey", properties.getProperty("google.api.key"));

			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{

			FofoDetails fofoDetails = googleLoginUtil.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());

			response.addCookie(cookieFofoId);

			response.addCookie(cookieEmailId);

			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, "/profitmandi-fofo/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 "login";

		}catch(Exception | ProfitMandiBusinessException profitMandiBusinessException){

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

			return "login";

		}

	}

}