WechatTicketSystem/src/main/java/love/sola/netsupport/api/Register.java

package love.sola.netsupport.api;

import love.sola.netsupport.config.Settings;
import love.sola.netsupport.enums.Block;
import love.sola.netsupport.enums.ISP;
import love.sola.netsupport.pojo.User;
import love.sola.netsupport.sql.TableUser;

import javax.servlet.ServletException;
import javax.servlet.annotation.WebServlet;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.io.PrintWriter;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;

import static love.sola.netsupport.config.Lang.lang;

/**
 * ***********************************************
 * Created by Sola on 2015/11/29.
 * Don't modify this source without my agreement
 * ***********************************************
 */
@WebServlet(name = "Register", urlPatterns = "/api/register", loadOnStartup = 22)
public class Register extends HttpServlet {

	public static Map<String, Long> authorized = new ConcurrentHashMap<>();

	public static final String STUDENT_ID_REGEX = "^(2010|2012|2013|2014|2015)[0-9]{9}$";
	public static final String PHONE_NUMBER_REGEX = "^1[34578][0-9]{9}$";

	protected void doPost(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
		request.setCharacterEncoding("utf-8");
		response.setCharacterEncoding("utf-8");
		response.addHeader("Content-type", "text/plain;charset=utf-8");
		PrintWriter out = response.getWriter();

		ISP isp = checkISP(request.getParameter("isp"));
		Block block = checkBlock(request.getParameter("block"));
		out.println(
				register(
						checkStudentId(request.getParameter("sid")),
						request.getParameter("name"),
						isp,
						checkNetAccount(request.getParameter("username"), isp),
						block,
						checkRoom(request.getParameter("room"), block),
						checkPhoneNumber(request.getParameter("phone")),
						checkWechat(request.getParameter("wechatid"))
				)
		);
		out.close();
	}

	@SuppressWarnings("Duplicates")
	protected void doGet(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
		request.setCharacterEncoding("utf-8");
		response.setCharacterEncoding("utf-8");
		response.addHeader("Content-type", "text/plain;charset=utf-8");
		PrintWriter out = response.getWriter();
		out.println(lang("Illegal_Request"));
		out.close();
	}

	private String register(long sid, String name, ISP isp, String netAccount, Block block, int room, long phone, String wechat) {
		if (wechat == null) return lang("Illegal_Request");
		if (sid == -1) return lang("Invalid_Student_Id");
		if (name == null) return lang("Invalid_Name");
		if (isp == null) return lang("Invalid_ISP");
		if (netAccount == null) return lang("Invalid_Account");
		if (block == null) return lang("Invalid_Block");
		if (room == -1) return lang("Invalid_Room");
		if (phone == -1) return lang("Invalid_Phone_Number");
		User user = TableUser.getUserById(sid);
		if (user == null) return lang("Invalid_Student_Id");
		if (!user.getName().equals(name)) return lang("Invalid_Name");
		if (user.getWechatId() != null) return lang("User_Already_Registered");
		user.setIsp(isp);
		user.setNetAccount(netAccount);
		user.setBlock(block);
		user.setRoom(room);
		user.setPhone(phone);
		user.setWechatId(wechat);
		return lang("Register_Success");
	}


	private long checkStudentId(String studentId) {
		if (studentId == null) return -1;
		if (studentId.matches(STUDENT_ID_REGEX)) {
			try {
				return Long.parseLong(studentId);
			} catch (NumberFormatException ignored) {
			}
		}
		return -1;
	}

	private long checkPhoneNumber(String phone) {
		if (phone == null) return -1;
		if (phone.matches(PHONE_NUMBER_REGEX)) {
			try {
				return Long.parseLong(phone);
			} catch (NumberFormatException ignored) {
			}
		}
		return -1;
	}

	private ISP checkISP(String isp) {
		if (isp == null) return null;
		try {
			return ISP.fromId(Integer.parseInt(isp));
		} catch (NumberFormatException ignored) {
		}
		return null;
	}

	private String checkNetAccount(String account, ISP isp) {
		if (isp == null) return null;
		if (account == null) return null;
		if (!account.matches(isp.accountRegex)) return null;
		return account;
	}

	private Block checkBlock(String block) {
		if (block == null) return null;
		try {
			return Block.fromId(Integer.parseInt(block));
		} catch (NumberFormatException ignored) {
		}
		return null;
	}

	private int checkRoom(String room, Block block) {
		if (block == null) return -1;
		if (room == null) return -1;
		try {
			Integer i = Integer.parseInt(room);
			if (i <= 100 || i >= 1300) return -1;
			return i;
		} catch (NumberFormatException ignored) {
		}
		return -1;
	}

	private String checkWechat(String wechat) {
		if (wechat == null) return null;
		Long l = authorized.remove(wechat);
		return l == null ? null : l < System.currentTimeMillis() - Settings.I.User_Register_Timeout ? null : wechat;
	}

}