package com.chaowen;
import java.io.IOException;
import java.io.InputStream;
import java.io.PrintWriter;
import java.io.StringReader;
import java.io.Writer;
import java.net.MalformedURLException;
import java.security.MessageDigest;
import java.security.NoSuchAlgorithmException;
import java.util.Arrays;
import java.util.Date;
import java.util.Scanner;
import javax.servlet.ServletException;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import javax.xml.bind.JAXBContext;
import javax.xml.bind.JAXBException;
import javax.xml.bind.Marshaller;
import javax.xml.bind.Unmarshaller;
import com.chaowen.bean.WeChatReqBean;
import com.chaowen.bean.WeChatRespBean;
import com.chaowen.common.Constants;
import com.sun.xml.bind.marshaller.CharacterEscapeHandler;
/**
* Servlet implementation class WeiXinServlet
*/
public class WeiXinServlet extends HttpServlet {
private static final long serialVersionUID = 1L;
public static final String Token = "token";
protected void doGet(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
response.setCharacterEncoding("UTF-8");
PrintWriter out = response.getWriter();
String signature = request.getParameter("signature");
String timestamp = request.getParameter("timestamp");
String nonce = request.getParameter("nonce");
String[] ArrTmp = { Token, timestamp, nonce };
Arrays.sort(ArrTmp);
StringBuffer sb = new StringBuffer();
for (int i = 0; i < ArrTmp.length; i++) {
sb.append(ArrTmp[i]);
}
String pwd = Encrypt(sb.toString());
String echostr = request.getParameter("echostr");
System.out.println("pwd=="+pwd);
System.out.println("echostr=="+echostr);
if(pwd.equals(signature)){
if(!"".equals(echostr) && echostr != null){
response.getWriter().print(echostr);
}
}
out.flush();
out.close();
//this.doPost(request, response);
}
protected void doPost(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
Scanner scanner = new Scanner(request.getInputStream());
response.setContentType("application/xml");
response.setCharacterEncoding("UTF-8");
PrintWriter out = response.getWriter();
try {
// 1、获取用户发送的信息
InputStream is = request.getInputStream();
// 取HTTP请求流长度
int size = request.getContentLength();
// 用于缓存每次读取的数据
byte[] buffer = new byte[size];
// 用于存放结果的数组
byte[] xmldataByte = new byte[size];
int count = 0;
int rbyte = 0;
// 循环读取
while (count < size) {
// 每次实际读取长度存于rbyte中
rbyte = is.read(buffer);
for(int i=0;i<rbyte;i++) {
xmldataByte[count + i] = buffer[i];
}
count += rbyte;
}
is.close();
String requestStr = new String(xmldataByte, "UTF-8");
// 2、解析用户的信息
JAXBContext jc = JAXBContext.newInstance(WeChatReqBean.class);
Unmarshaller u = jc.createUnmarshaller();
WeChatReqBean reqBean = (WeChatReqBean) u.unmarshal(new StringReader(requestStr));
// 3、判定用户了解动作
String content = getContent(reqBean);
// 4、创建一个回复消息
jc = JAXBContext.newInstance(WeChatRespBean.class);
Marshaller m = jc.createMarshaller();
m.setProperty(Marshaller.JAXB_FORMATTED_OUTPUT, Boolean.TRUE);
m.setProperty(CharacterEscapeHandler.class.getName(), new CharacterEscapeHandler() {
@Override
public void escape(char[] arg0, int arg1, int arg2, boolean arg3,
Writer arg4) throws IOException {
arg4.write(arg0, arg1, arg2);
}
});
m.setProperty(Marshaller.JAXB_FRAGMENT, Boolean.TRUE);
WeChatRespBean respBean = createRespBean(reqBean, content);
m.marshal(respBean, out);
out.flush();
}catch (JAXBException e) {
e.printStackTrace();
} finally {
if (scanner != null) {
scanner.close();
scanner = null;
}
if (out != null) {
out.close();
out = null;
}
}
}
/**
* @param reqBean
* @param content
* @return
*/
private WeChatRespBean createRespBean(WeChatReqBean reqBean, String content) {
WeChatRespBean respBean = new WeChatRespBean();
respBean.setFromUserName(reqBean.getToUserName());
respBean.setToUserName(reqBean.getFromUserName());
respBean.setMsgType("text");
respBean.setCreateTime(new Date().getTime());
respBean.setContent(content);
return respBean;
}
public String Encrypt(String strSrc) {
MessageDigest md = null;
String strDes = null;
byte[] bt = strSrc.getBytes();
try {
md = MessageDigest.getInstance("SHA-1");
md.update(bt);
strDes = bytes2Hex(md.digest()); //to HexString
} catch (NoSuchAlgorithmException e) {
System.out.println("Invalid algorithm.");
return null;
}
return strDes;
}
public String bytes2Hex(byte[] bts) {
String des = "";
String tmp = null;
for (int i = 0; i < bts.length; i++) {
tmp = (Integer.toHexString(bts[i] & 0xFF));
if (tmp.length() == 1) {
des += "0";
}
des += tmp;
}
return des;
}
/**
* @param reqBean
* @throws JAXBException
* @throws MalformedURLException
*/
private String getContent(WeChatReqBean reqBean) throws JAXBException, MalformedURLException {
StringBuffer content = new StringBuffer("");
switch (Constants.getType(reqBean.getMsgType())){
case text:
if (reqBean.getContent().contains("成长档案")) {
// 这是新用户关注时默认发的一条信息。可以做一个欢迎处理。
content.append("这是成长档案的介绍!"+"\r\n");
content.append(""+"\r\n");
content.append(""+"\r\n");
content.append("回复h获取更多内容!"+"\r\n");
}else if(reqBean.getContent().contains("h")){
content.append("这是帮助信息!"+"\r\n");
content.append("如果你还有疑问或建议:请以#开头发送文字内容，我们的客服收到留言后会尽快跟你联系。"+"\r\n");
}else if(reqBean.getContent().contains("cp")){
content.append("产品介绍:"+"\r\n");
content.append("001.家园共育平台"+"\r\n");
content.append("002.运营分析系统"+"\r\n");
content.append("003.互动学堂"+"\r\n");
content.append(""+"\r\n");
content.append(""+"\r\n");
content.append("回复数字获取更多内容!"+"\r\n");
}
else {
content.append("你的信息已收到，我们会第一时间处理.");
}
break;
case event:
if("subscribe".equals(reqBean.getEvent())){
// 这是新用户关注时默认发的一条信息。可以做一个欢迎处理。
content.append("非常感谢你的添加，我们定时为你提供最新的儿童内容服务哟!"+"\r\n");
content.append("1.想了解我们的产品请直接将产品名称回复给我(如\"成长档案\"）"+"\r\n");
content.append("2.想知道已经上线了哪些产品请回复\"cp\"或直接点击链接www.yey.com进行查看"+"\r\n");
content.append("3.如需帮助或了解请回复\"h\"。"+"\r\n");
}
break;
case image:
break;
default:
content.append("这是一个测试!!!");
break;
}
return content.toString();
}
}