package servlet;

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

public class Login extends HttpServlet {
  private static final String CONTENT_TYPE = "text/html; charset=GBK";
  public void init() throws ServletException {
  }

  public void doPost(HttpServletRequest request, HttpServletResponse response) throws
      ServletException, IOException {
    //設置取得字符串的編碼機制
    request.setCharacterEncoding("GBK");
    //設置字符輸出的編碼機制
    response.setContentType(CONTENT_TYPE);
    //取得html代碼輸出類
    PrintWriter out = response.getWriter();
    //通過HttpServlet類的getServletContext方法取得application的對象
    ServletContext application = getServletContext();
    //創建session對象
    HttpSession session = request.getSession();
    //聲明頁面轉向類
    RequestDispatcher requestDispatcher = null;
    //取得按鈕的字符
    String action = request.getParameter("action");
    //創建數據類
    StockManagementData stockManagementData = new StockManagementData();
    if(action.equals("登陸")){
      //取得用戶名字和密碼
      String userName = request.getParameter("userName");
      String userPassword = request.getParameter("userPassword");
      int[] result = stockManagementData.checkUser(userName, userPassword);
      if( result[0] == -1){
        //創建頁面轉向類
        requestDispatcher = application.getRequestDispatcher("/infor.jsp");
        request.setAttribute("infor", "用戶的名字和密碼不正確，請<a href=\"javascript:"
                             + "history.back()\">重新輸入</a>.");
        requestDispatcher.forward(request, response);
      }else{
        //創建用戶類
        User user = new User(userName, userPassword, result[0], result[1], result[2],
                        result[3]);
        //將用戶放入session對象
        session.setAttribute("user", user);
        //創建頁面轉向類
        requestDispatcher = application.getRequestDispatcher("/infor.jsp");
        request.setAttribute("infor", "成功登陸，請返回主頁進行相關信息瀏覽.");
        requestDispatcher.forward(request, response);
      }
    }
  }
  public void destroy() {
  }
}