Posts in category: Java

				
					package com.brains;

import java.io.IOException;
import java.sql.Connection;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.sql.Statement;

import javax.servlet.ServletException;
import javax.servlet.ServletRequest;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import javax.servlet.http.HttpSession;

import org.apache.logging.log4j.LogManager;
import org.apache.logging.log4j.Logger;

import com.brains.models.Employee;
import com.google.api.client.googleapis.auth.oauth2.GoogleIdToken;

/**
 * Servlet implementation class LoginServlet
 */
public class LoginServlet extends HttpServlet {
	private static final long serialVersionUID = 1L;
	private Connection conn;

	private static final Logger LOGGER = LogManager.getLogger(LoginServlet.class);
	/**
	 * @see HttpServlet#doPost(HttpServletRequest request, HttpServletResponse response)
	 */
	protected void doPost(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {
		// TODO Auto-generated method stub
		
		checkActiveSession(req, resp);
		
		resp.setContentType("text/html");
		try {
            String idToken = req.getParameter("id_token");
            GoogleIdToken.Payload payLoad = IdTokenVerifierAndParser.getPayload(idToken);
            String name = (String) payLoad.get("name");
            String email = payLoad.getEmail();

            HttpSession session = req.getSession(true);
            session.setAttribute("ubName", name);
            req.setAttribute("email", email);
            session.setAttribute("picture", payLoad.get("picture"));
            
            getUserDetailsFromDB(req);
            
            req.getServletContext()
               .getRequestDispatcher("/welcome-page.jsp").forward(req, resp);

        } catch (Exception e) {
            throw new RuntimeException(e);
        }
	}

	private void getUserDetailsFromDB(ServletRequest req) {
		if(conn == null) {
			conn = (Connection) this.getServletContext().getAttribute("con");			
		}
		String loginEmail = (String) req.getAttribute("email");
		
		HttpSession session = ((HttpServletRequest) req).getSession(true);
		String str = "SELECT * FROM employee where email = '"+loginEmail+"'";

		try (
             Statement stmt = conn.createStatement();
             ResultSet rs = stmt.executeQuery(str)) {
            // Extract data from result set
            while (rs.next()) {
                //Retrieve by column name
                String name = rs.getString("name");
                String email = rs.getString("email");
                String mobile = rs.getString("mobile");
                // Add item
                Employee emp = new Employee(name, email, mobile);
                session.setAttribute("emp", emp);
            }

            session.setAttribute("title", "Top 5 fruits");
//            req.getRequestDispatcher("/WEB-INF/views/index.jsp").forward(req, resp);
        } catch (SQLException se) {
            se.printStackTrace();
        } catch (Exception e) {
            e.printStackTrace();
        }
	}
	
	@Override
	protected void doGet(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {
		checkActiveSession(req, resp);
		req.getServletContext()
        .getRequestDispatcher("/welcome-page.jsp").forward(req, resp);
	}
	
	private void checkActiveSession(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException{
		HttpSession session = req.getSession();
        String userName = (String) session.getAttribute("ubName");
        LOGGER.warn("checkActiveSession: "+ userName);
        if(userName == null) {
        	req.getRequestDispatcher("index.jsp").include(req, resp);
        }
	}
}