package com.allan.ums.action;

import java.io.IOException;
import java.util.Collection;

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

import com.allan.ums.entity.Administrator;
import com.allan.ums.entity.User;
import com.allan.ums.service.AdministratorService;
import com.allan.ums.service.UserService;
import com.allan.ums.util.RequestUtil;
import com.allan.ums.util.TokenUtil;
import com.allan.ums.util.UmsUtil;

public class ActionServlet extends HttpServlet {
	private UserService userService = new UserService();
	private AdministratorService adminService = new AdministratorService();

	public void execute(HttpServletRequest request, HttpServletResponse response)
			throws ServletException, IOException {
		String path = request.getServletPath();
		path = path.substring(0, path.indexOf("."));
		String nextPage = null;
		if (path.equals("/user/search")) {
			HttpSession session = request.getSession(false);
			if(session == null){
				nextPage = "/index.html";
			}else if(session.getAttribute("admin") == null){
				nextPage = "/index.html";
			} else {
				nextPage = searchUsers(request, response);
			}
		} else if (path.equals("/user/add")) {
			boolean isLegal = TokenUtil.isTokenValid(request);
			if (isLegal) {
				nextPage = addUser(request, response);
			} else {
				nextPage = "/user/add.html";
			}
		} else if (path.equals("/user/modify")) {
			nextPage = modifyUser(request, response);
		} else if (path.equals("/user/remove")) {
			nextPage = removeUser(request, response);
		} else if (path.equals("/user/load")) {
			nextPage = loadUser(request, response);
		} else if (path.equals("/login")) {
		  boolean isLegal = TokenUtil.isTokenValid(request);
			if (isLegal) {
				nextPage = login(request, response);
			} else {
				nextPage = "/index.html";
			}			
		} else if (path.equals("/logout")) {
			nextPage = logout(request, response);
		} else {
			response.sendError(HttpServletResponse.SC_NOT_FOUND, path
					+ " is not a valid request url");
		}
		RequestUtil.forward(getServletContext(), request, response, nextPage);
	}

	private String searchUsers(HttpServletRequest request,
			HttpServletResponse response) throws ServletException, IOException {
		User user = new User();
		String username = request.getParameter("username");
		if (username != null && username.trim().length() > 0) {
			user.setUsername(username);
		}
		String gender = request.getParameter("gender");
		if (gender != null && !gender.equals("all")) {
			user.setGender(gender);
		}

		String email = request.getParameter("email");
		if (email != null && email.trim().length() > 0) {
			user.setEmail(email);
		}
		// 獲取當前的會話,在登錄時開始的.(加入會話)
		HttpSession session = request.getSession();
		// 將查詢條件保存的會話空間中,供以后的功能使用
		session.setAttribute("searchCondition", user);
		try {
			Collection<User> users = userService.find(user);
			request.setAttribute("users", users);
			return "/user/list";
		} catch (Exception e) {
			e.printStackTrace();
		}
		return "/error";
	}

	private String removeUser(HttpServletRequest request,
			HttpServletResponse response) throws ServletException, IOException {
		String id = request.getParameter("id");
		try {
			userService.remove(Integer.parseInt(id));
			HttpSession session = request.getSession();
			User user = (User) session.getAttribute("searchCondition");
			Collection<User> users = userService.find(user);
			request.setAttribute("users", users);
			return "/user/list";
		} catch (Exception e) {
			e.printStackTrace();
		}
		return "/error";
	}

	private String loadUser(HttpServletRequest request,
			HttpServletResponse response) throws ServletException, IOException {
		String id = request.getParameter("id");
		try {
			User user = userService.find(Integer.parseInt(id));
			request.setAttribute("user", user);
			return "/user/edit";
		} catch (Exception e) {
			e.printStackTrace();
		}
		return "/error";
	}

	private String addUser(HttpServletRequest request,
			HttpServletResponse response) throws ServletException, IOException {
		String username = request.getParameter("username");
		String password = request.getParameter("password");
		String gender = request.getParameter("gender");
		String email = request.getParameter("email");
		String[] hobbies = request.getParameterValues("hobbies");
		User user = new User();
		user.setUsername(username);
		user.setPassword(password);
		user.setGender(gender);
		user.setEmail(email);
		user.setHobbies(UmsUtil.join(hobbies, ','));
		try {
			userService.add(user);
			Collection<User> users = userService.find(new User());
			request.setAttribute("users", users);
			return "/user/list";
		} catch (Exception e) {
			e.printStackTrace();
		}
		return "/error";
	}

	private String modifyUser(HttpServletRequest request,
			HttpServletResponse response) throws ServletException, IOException {
		String id = request.getParameter("id");
		String username = request.getParameter("username");
		String password = request.getParameter("password");
		String gender = request.getParameter("gender");
		String email = request.getParameter("email");
		String[] hobbies = request.getParameterValues("hobbies");
		User user = new User();
		user.setId(Integer.parseInt(id));
		user.setUsername(username);
		user.setPassword(password);
		user.setGender(gender);
		user.setEmail(email);
		user.setHobbies(UmsUtil.join(hobbies, ','));
		try {
			userService.modify(user);
			Collection<User> users = userService.find(new User());
			request.setAttribute("users", users);
			return "/user/list";
		} catch (Exception e) {
			e.printStackTrace();
		}
		return "/error";
	}

	private String login(HttpServletRequest request,
			HttpServletResponse response) throws ServletException, IOException {
		String username = request.getParameter("username");
		String password = request.getParameter("password");

		try {
			Administrator admin = adminService.find(username, password);
			if (admin != null) {
			  //處理Cookie信息
			  int period = Integer.parseInt(request.getParameter("period"));
			  if(period > 0){
			    Cookie u = new Cookie("username", username);
			    Cookie p = new Cookie("password", password);
			    u.setMaxAge(60 * 60 * 24 * 30 * period);
			    p.setMaxAge(60 * 60 * 24 * 30 * period);
			    response.addCookie(u);
			    response.addCookie(p);
			  }
				// 創建會話對象（開始會話）
				HttpSession session = request.getSession();
				session.setAttribute("admin", admin);
				return "/user/search.html";
			} else {
				return "/index.html";
			}
		} catch (Exception e) {
			e.printStackTrace();
		}
		return "/error";
	}
	
	private String logout(HttpServletRequest request,
			HttpServletResponse response) throws ServletException, IOException {
		HttpSession session = request.getSession();
		session.invalidate();
		return "/index.html";
	}

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

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