package com.nitpro.school.servlet;

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

import com.nitpro.school.bean.Admin;
import com.nitpro.school.bean.Emp;
import com.nitpro.school.bean.Teacher;
import com.nitpro.school.bean.User;
import com.nitpro.school.model.AdminModel;
import com.nitpro.school.model.EmpModel;
import com.nitpro.school.model.TeacherModel;

public class LoginServlet extends HttpServlet {
	public void doPost(HttpServletRequest req,HttpServletResponse resp) throws ServletException, java.io.IOException{
		doIt(req, resp);
	}

	public void doGet(HttpServletRequest req, HttpServletResponse resp) throws ServletException, java.io.IOException{
		doIt(req, resp);
	}
	
	
	public void doIt(HttpServletRequest req, HttpServletResponse resp) throws ServletException, java.io.IOException{
        String method = req.getParameter("method");//通過標記參數進行分支判斷
        if("login".equals(method)){
        	login(req, resp);
        }
	}
	
	public void login(HttpServletRequest req, HttpServletResponse resp) throws ServletException, java.io.IOException{
		String userName = req.getParameter("userName");
		String pwd = req.getParameter("pwd");
		String type = req.getParameter("type");
		String nextUrl = req.getParameter("nextUrl");
		
		User user = new User();
		user.setUserName(userName);
		user.setPwd(pwd);
		user.setType(type);
		boolean success = false;
		
		HttpSession session = req.getSession();
		if("admin".equals(type)){
			AdminModel adminModel = new AdminModel();
			Admin admin = adminModel.login(user);
			if(admin != null){
				success = true;
				session.setAttribute("user", admin);
				if("".equals(nextUrl)){
					nextUrl = "/admin/adminindex.jsp";
				}
			}
		}
		
		if("emp".equals(type)){
			EmpModel empModel = new EmpModel();
			Emp emp = empModel.login(user);
			if(emp != null){
				success = true;
				session.setAttribute("user", emp);
				if("".equals(nextUrl)){
					nextUrl = "/emp/empindex.jsp";
				}
			}
		}
		
		if("teacher".equals(type)){
			TeacherModel teacherModel = new TeacherModel();
			Teacher teacher = teacherModel.login(user);
			if(teacher != null){
				success = true;
				session.setAttribute("user", teacher);
				if("".equals(nextUrl)){
					nextUrl = "/teacher/teacherindex.jsp";
				}
			}
		}

		RequestDispatcher rd = null;
		if(success){
			rd = req.getRequestDispatcher(nextUrl);
		}else{
			req.setAttribute("msg", "用戶名稱或密碼錯誤，請重新輸入！");
			req.setAttribute("nextUrl", nextUrl);
			rd = req.getRequestDispatcher("/index.jsp");
		}
		
		rd.forward(req, resp);
	}
}