package local.cmu.qatar.db;

import java.io.*;
import javax.servlet.*;
import javax.servlet.http.*;
import java.util.*;

public class Login extends HttpServlet
{
        public void doGet(HttpServletRequest request, HttpServletResponse response)
                                     throws ServletException, IOException
        {
                boolean newUser = true;
                Cookie unameCookie = null;

                String username = request.getParameter("username");
                String password = request.getParameter("password");

                /* Returns an array of all the cookies ever sent by the
                   web-server to the user */
                Cookie allCookies[] = request.getCookies();

                /* Check if a username cookie has already been sent by the
                   server. If so, do not create and send such a cookie */
                if(allCookies != null && allCookies.length != 0)
                {
                        for (Cookie c : allCookies)
                        {
                                if(c.getName().equals("username"))
                                {
                                        newUser = false;
                                        unameCookie = c;
                                }
                        }
                }

                /* Create a new username cookie to store the username */
                if((newUser == true) && (username.length() != 0))
                {
                        unameCookie = new Cookie("username", username);
                        unameCookie.setMaxAge(86400); /* one day in s */

                        /* addCookie must precede getWriter!!!*/
                        response.addCookie(unameCookie);
                }

                /* Send the HTML body to welcome the user if password !empty */
                response.setContentType("text/html");
                PrintWriter outstream = response.getWriter();

                outstream.println("<head>");
                outstream.println("<title> Welcome " + unameCookie.getValue() + "! </title>");
                outstream.println("</head>");

                outstream.println("<body>");

                if(password.length() == 0)
                {
                        outstream.println("<h2> Error 101: Registration Failed </h2>");
                        outstream.println("Your password can not be empty!");
                }

                else
                {
                        outstream.println("<h2> Welcome " + username + "</h2>");
                        outstream.println("Your registration is successful!");
                }

                outstream.println("</body>");
        }

        public void  doPost(HttpServletRequest request, HttpServletResponse response)
                                       throws ServletException, IOException
        {
                doGet(request,response);
        }
}