mirror of
				https://github.com/ZSCNetSupportDept/WechatTicketSystem.git
				synced 2025-11-01 02:46:19 +08:00 
			
		
		
		
	use student id as primary key instead.
This commit is contained in:
		| @@ -52,6 +52,7 @@ public class WxMpServlet extends HttpServlet { | ||||
| 		wxMpMessageRouter = new WxMpMessageRouter(wxMpService); | ||||
| 		wxMpMessageRouter.rule() | ||||
| 				.async(false) | ||||
| 				.msgType("text") | ||||
| 				.matcher(new CommandMatcher(Command.REGISTER)) | ||||
| 				.handler(new RegisterHandler()) | ||||
| 				.interceptor(checkSpamInterceptor) | ||||
|   | ||||
| @@ -1,6 +1,7 @@ | ||||
| package love.sola.netsupport.wechat.handler; | ||||
|  | ||||
| import love.sola.netsupport.pojo.User; | ||||
| import love.sola.netsupport.sql.TableUser; | ||||
| import love.sola.netsupport.wechat.Command; | ||||
| import love.sola.netsupport.wechat.matcher.CommandMatcher; | ||||
| import me.chanjar.weixin.common.exception.WxErrorException; | ||||
| @@ -22,24 +23,69 @@ import java.util.Map; | ||||
|  */ | ||||
| public class RegisterHandler implements WxMpMessageHandler { | ||||
|  | ||||
| 	public static final String STUDENT_ID_REGEX = "^(2012|2013|2014|2015)[0-9]{9}"; | ||||
|  | ||||
| 	Map<String, User> pre_confirm = new HashMap<>(); | ||||
| 	Map<String, User> confirmed = new HashMap<>(); | ||||
|  | ||||
| 	@Override | ||||
| 	public WxMpXmlOutMessage handle(WxMpXmlMessage wxMessage, Map<String, Object> context, WxMpService wxMpService, WxSessionManager sessionManager) | ||||
| 			throws WxErrorException { | ||||
| 		TextBuilder out = WxMpXmlOutMessage.TEXT().fromUser(wxMessage.getToUserName()).toUser(wxMessage.getFromUserName()); | ||||
| 		String in = wxMessage.getContent(); | ||||
| 		String userName = wxMessage.getFromUserName(); | ||||
| 		if (in.matches(Command.REGISTER.getRegex())) { | ||||
| 			out.content("Welcome, please type your student identification number."); | ||||
| 		} else if (pre_confirm.containsKey(userName)) { | ||||
| 			//TODO | ||||
|  | ||||
| 		} else { | ||||
| 			out.content("Illegal Operation."); | ||||
| 			CommandMatcher.inCmdUsers.remove(userName); | ||||
| 		} | ||||
| 		handle(wxMessage.getContent(), wxMessage.getFromUserName(), out); | ||||
| 		return out.build(); | ||||
| 	} | ||||
|  | ||||
| 	private void handle(String in, String userName, TextBuilder out) { | ||||
| 		if (in.matches(Command.REGISTER.getRegex())) { | ||||
| 			out.content("Welcome, please type your student identification number."); | ||||
| 			return; | ||||
| 		} | ||||
| 		if (pre_confirm.containsKey(userName)) { | ||||
| 			User user = pre_confirm.get(userName); | ||||
| 			if (in.trim().equals(user.getName())) { | ||||
| 				pre_confirm.remove(userName); | ||||
| 				confirmed.put(userName, user); | ||||
| 				out.content("Confirmed Success.\n" + | ||||
| 						"Please enter your ISP."); | ||||
| 			} else { | ||||
| 				out.content("Validate failed.\n" + | ||||
| 						"if you have any issue, please contact administrator.\n" + | ||||
| 						"Type 'q' to cancel this operation."); | ||||
| 			} | ||||
| 			return; | ||||
| 		} | ||||
| 		if (confirmed.containsKey(userName)) { | ||||
| 			User user = pre_confirm.get(userName); | ||||
| 			if (user.getIsp() == null) { | ||||
| 				//TODO | ||||
| 			} | ||||
| 			return; | ||||
| 		} | ||||
| 		if (in.equalsIgnoreCase("q")) { | ||||
| 			CommandMatcher.inCmdUsers.remove(userName); | ||||
| 			pre_confirm.remove(userName); | ||||
| 			confirmed.remove(userName); | ||||
| 			out.content("Operation Canceled."); | ||||
| 			return; | ||||
| 		} | ||||
| 		Long sid = checkStudentId(in); | ||||
| 		User user; | ||||
| 		if (sid == null || (user = TableUser.getUserById(sid)) == null) { | ||||
| 			out.content("Invalid student identification number. Type 'q' to cancel this operation."); | ||||
| 			return; | ||||
| 		} | ||||
| 		pre_confirm.put(userName, user); | ||||
| 		out.content("Please type your real name to validate."); | ||||
| 	} | ||||
|  | ||||
| 	private Long checkStudentId(String studentId) { | ||||
| 		if (studentId.matches(STUDENT_ID_REGEX)) { | ||||
| 			try { | ||||
| 				return Long.parseLong(studentId); | ||||
| 			} catch (NumberFormatException ignored) { } | ||||
| 		} | ||||
| 		return null; | ||||
| 	} | ||||
|  | ||||
| } | ||||
|   | ||||
		Reference in New Issue
	
	Block a user
	 Sola
					Sola