package Control;

import data.CupcakeLoginService;
import data.DataMapper;
import java.io.IOException;
import java.io.PrintWriter;
import java.sql.SQLException;
import java.util.logging.Level;
import java.util.logging.Logger;
import javax.servlet.RequestDispatcher;
import javax.servlet.ServletException;
import javax.servlet.annotation.WebServlet;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import model.User;

/**
 *
 * @author Arne
 */
@WebServlet(urlPatterns
        = {
            "/Cupcake"
        })
public class CupcakeServlet extends HttpServlet {

    DataMapper dm = new DataMapper();

    protected void processRequest(HttpServletRequest request, HttpServletResponse response)
            throws ServletException, IOException {

        User u = dm.getUser(request.getParameter("username"));

        request.setAttribute("user", u);

        CupcakeLoginService loginService = new CupcakeLoginService();
        response.setContentType("text/html;charset=UTF-8");
        String action = request.getParameter("action");
        if ("login".equals(action)) {
            String username = request.getParameter("username");
            String password = request.getParameter("password");
            User user = loginService.login(username, password);
            if (user == null) {

                sendLoginForm(request, response);
                return;
            }
            sendFrontPage(request, response);
            sendMainPage(request, response);
            return;
        }

        if ("logout".equals(action)) {
            request.removeAttribute("username");
            request.getSession().invalidate();
        }
//            sendFrontPage(request,response);

            if ("FrontPage".equals(action)) {
                sendFrontPage(request, response);
                return;
            }

            // register user
            User user = (User) request.getSession().getAttribute("user");
            if ("register".equals(action)) {
                String username = request.getParameter("registerUsername");
                String password = request.getParameter("registerPassword");
                boolean admin = Boolean.parseBoolean(request.getParameter("admin"));
                int userId = Integer.parseInt(request.getParameter("registeruserId"));
                int balance = Integer.parseInt(request.getParameter("balance"));
                String email = request.getParameter("registerEmail");

                dm.checkUsername(username);
                dm.registerUser(username, password, admin, userId, balance, email);
                user = dm.getUser(username);
                request.setAttribute("user", user);

                if (user == null) {
                    //registration fail
                    sendRegisterForm(request, response);
                } else {
//                    request.getSession().setAttribute("username", user.getUsername());
//                    sendFrontPage(request, response);
                    sendMainPage(request, response);
                }

            }
            if ("Sign up".equals(action)) {
                request.removeAttribute("username");
                request.getSession().invalidate();
            }
            if (user == null) {
                //Login Fail
                sendLoginForm(request, response);
            }
            if (action == null) {
                sendFrontPage(request, response);
            }
        }


    // <editor-fold defaultstate="collapsed" desc="HttpServlet methods. Click on the + sign on the left to edit the code.">
    /**
     * Handles the HTTP <code>GET</code> method.
     *
     * @param request servlet request
     * @param response servlet response
     * @throws ServletException if a servlet-specific error occurs
     * @throws IOException if an I/O error occurs
     */
    @Override
    protected void doGet(HttpServletRequest request, HttpServletResponse response)
            throws ServletException, IOException {
        processRequest(request, response);
    }

    /**
     * Handles the HTTP <code>POST</code> method.
     *
     * @param request servlet request
     * @param response servlet response
     * @throws ServletException if a servlet-specific error occurs
     * @throws IOException if an I/O error occurs
     */
    @Override
    protected void doPost(HttpServletRequest request, HttpServletResponse response)
            throws ServletException, IOException {
        processRequest(request, response);
    }

    /**
     * Returns a short description of the servlet.
     *
     * @return a String containing servlet description
     */
    @Override
    public String getServletInfo() {
        return "Short description";
    }// </editor-fold>

    private void sendLoginForm(HttpServletRequest request, HttpServletResponse response) throws IOException {
        try (PrintWriter out = response.getWriter()) {
            try {
                RequestDispatcher rd = request.getRequestDispatcher("login.jsp");
                rd.forward(request, response);

            } catch (IOException | ServletException e) {
                e.getMessage();
            }
        }
    }

    private void sendFrontPage(HttpServletRequest request, HttpServletResponse response) {
        try {

            RequestDispatcher rd = request.getRequestDispatcher("shop.jsp");
            rd.forward(request, response);

        } catch (IOException | ServletException e) {
            e.getMessage();
        }
    }

    private void sendRegisterForm(HttpServletRequest request, HttpServletResponse response) {
        try {
            RequestDispatcher rd = request.getRequestDispatcher("signup.jsp");
            rd.forward(request, response);

        } catch (IOException | ServletException e) {
            e.getMessage();
        }
    }

    private void sendMainPage(HttpServletRequest request, HttpServletResponse response) throws IOException {
        try (PrintWriter out = response.getWriter()) {
            try {
                RequestDispatcher rd = request.getRequestDispatcher("login.jsp");
                rd.forward(request, response);
                //request.getSession().invalidate();

            } catch (IOException | ServletException e) {
                e.getMessage();
            }
        }
    }
}