package com.test.servlet;

import java.io.IOException;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

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

import com.test.bean.Student;
import com.test.dao.StudentDao;
import com.test.dao.impl.StudentDaoImpl;
import com.test.util.Pager;

public class StudentServlet extends HttpServlet {

	protected void doGet(HttpServletRequest req, HttpServletResponse resp)
			throws ServletException, IOException {
		String cmd = req.getParameter("cmd");
		String username = req.getParameter("username");
		String gender = req.getParameter("gender");
		String birth = req.getParameter("birth");
		String password = req.getParameter("password");
		String id = req.getParameter("id");
		StudentDao dao = new StudentDaoImpl();
		Student stu = new Student();
		stu.setBirth(birth);
		stu.setGender(gender);
		stu.setPassword(password);
		stu.setUsername(username);
		if (id != null) {

			stu.setId(Integer.parseInt(id));
		}
		if (cmd != null) {
			if (cmd.equals("login")) {
				int ret = dao.login(username, password);
				if (ret > 0) {
					Student student = dao.getStudentByName(username);
					req.getSession(true).setAttribute("stu", student);
					resp.sendRedirect("student?cmd=list");
				} else {
					resp.sendRedirect("index.jsp");
				}
			}
			if (cmd.equals("list")) {
				// 分頁處理

				Map map = new HashMap();
				int pageSize = 10;
				Pager pager = new Pager(pageSize, req.getQueryString());
				int total = dao.total();
				// 當(dāng)前頁
				String toPage = req.getParameter("toPage");
				if (toPage == null || toPage.equals("")) {
					toPage = "1";
				}
				int currentPage = Integer.parseInt(toPage);
				pager.setTotalRecord(total);
				pager.setCurrentPage(toPage);
				// 開始位置
				int start = (currentPage - 1) * pageSize;
				map.put("start", start);
				map.put("pageSize", pageSize);

				List<Student> list = dao.getStudentList(map);
				req.setAttribute("list", list);
				req.setAttribute("pager", pager);
				req.setAttribute("basePath", req.getRequestURI().substring(
						req.getRequestURI().lastIndexOf("/") + 1));
				req.getRequestDispatcher("list.jsp").forward(req, resp);
			} else if (cmd.equals("add")) {
				String rand = (String) req.getSession().getAttribute("rand");
				String checkCode = req.getParameter("checkCode");
				if (rand.equals(checkCode)) {
					dao.addStudent(stu);
					resp.sendRedirect("student?cmd=list");
				} else {
					req.getSession().setAttribute("stu", stu);
					req.getSession().setAttribute("error", "1");
					resp.sendRedirect("add.jsp");
				}
			} else if (cmd.equals("loadedit")) {
				Student s = dao.getStudent(Integer.valueOf(id));
				req.setAttribute("s", s);
				req.getRequestDispatcher("edit.jsp").forward(req, resp);
			} else if (cmd.equals("edit")) {
				dao.editStudent(stu);
				resp.sendRedirect("student?cmd=list");
			} else if (cmd.equals("get")) {
				Student student = dao.getStudent(Integer.valueOf(id));
			} else if (cmd.equals("delete")) {

				dao.deleteStudent(Integer.valueOf(id));
				resp.sendRedirect("student?cmd=list");
			} else if (cmd.equals("logout")) {
				req.getSession().invalidate();
				resp.sendRedirect("index.jsp");
			}
		}
	}

	protected void doPost(HttpServletRequest req, HttpServletResponse resp)
			throws ServletException, IOException {
		doGet(req, resp);
	}

}