package com.library.controller;

import java.io.File;
import java.io.IOException;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpSession;

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Scope;
import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.multipart.MultipartFile;
import org.springframework.web.multipart.MultipartHttpServletRequest;

import com.library.model.User;
import com.library.service.UserService;

/**
 * 登录/注册
 */
@Controller
@Scope(value = "prototype")
public class LoginController {

	@Autowired
	private UserService userService;

	@RequestMapping("login")
	public String login(HttpServletRequest request) {
		String username = request.getParameter("username");
		String password = request.getParameter("password");
		if (username == null || username.isEmpty()) {
			return "login";
		}
		if (password == null || password.isEmpty()) {
			request.setAttribute("username", username);
			request.setAttribute("errorMessage", "密码不能为空");
			return "login";
		}
		User user = userService.login(username, password);
		if (user != null) {
			HttpSession session = request.getSession();
			session.setAttribute("user", user);
			if ("1".equals(user.getIdent())) {
				return "redirect:admin_home.cc";
			} else if ("0".equals(user.getIdent())) {
				return "redirect:user_home.cc";
			} else {
				return "login";
			}
		} else {
			request.setAttribute("username", username);
			request.setAttribute("errorMessage", "用户名或密码错误");
			return "login";
		}
	}

	@RequestMapping("register")
	public String register(HttpServletRequest request) {
		String username = request.getParameter("username");
		String password = request.getParameter("password");
		String sex = request.getParameter("sex");
		String telephone = request.getParameter("telephone");
		String address = request.getParameter("address");
		String headUrl = "/static/images/touxiang/01.jpg";
		if (username == null || username.isEmpty()) {
			return "register";
		}
		if (userService.checkUsername(username)) {
			request.setAttribute("errorMessage", "用户名已存在，立即登录");
			return "login";
		}
		MultipartHttpServletRequest multipartRequest = (MultipartHttpServletRequest) (request);
		MultipartFile head = multipartRequest.getFile("head");
		if (!head.getOriginalFilename().isEmpty()) {
			String path = request.getServletContext().getRealPath("/") + "page/static/images/touxiang/";
			String name = String.valueOf(System.currentTimeMillis() + "_" + head.getOriginalFilename());
			File destFile = new File(path, name);
			try {
				head.transferTo(destFile);
			} catch (IllegalStateException | IOException e) {
				e.printStackTrace();
			}
			headUrl = "images/touxiang/" + name;
		}
		User user = new User();
		user.setUsername(username);
		user.setPassword(password);
		user.setSex(sex);
		user.setTelephone(telephone);
		user.setAddress(address);
		user.setIdent("0");
		user.setPortrait(headUrl);
		userService.addUser(user);
		request.setAttribute("errorMessage", "注册成功，立即登录");
		return "login";
	}

	@RequestMapping("logout")
	public String logout(HttpSession session) {
		session.removeAttribute("user");
		return "login";
	}

}