package servlet;

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

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("/error.jsp");
        request.setAttribute("infor", "用戶的名字和密碼不正確,請<a href=\"login.jsp\">重新"
                             +"輸入</a>.");
        requestDispatcher.forward(request, response);
      }else{
        //創建用戶類
        User user = new User(String.valueOf(userId), userName, userPassword);
        //將用戶放入session對象
        session.setAttribute("user", user);
        //取得用戶的錢包記錄數組
        String[][] detail = stockManagementData.getWallet(user.getUserId());
        //取得錢包的余額
        double balance = stockManagementData.getBalance(user.getUserId());
        request.setAttribute("balance", new Double(balance));
        request.setAttribute("detail", detail);
        //創建頁面轉向類
        requestDispatcher = application.getRequestDispatcher("/walletview.jsp");
        requestDispatcher.forward(request, response);
      }
      return;
    }
  }
  public void destroy() {
  }
}