¶Ô±ÈÐÂÎļþ |
| | |
| | | package com.smtaiserver.smtaiserver.control; |
| | | |
| | | import com.smtaiserver.smtaiserver.core.SMTAIServerApp; |
| | | import com.smtaiserver.smtaiserver.core.SMTAIServerRequest; |
| | | import com.smtaiserver.smtaiserver.database.SMTDatabase; |
| | | import com.smtaiserver.smtaiserver.database.SMTDatabase.DBRecords; |
| | | import com.smtaiserver.smtaiserver.javaai.SMTJavaAIChat; |
| | | import com.smtaiserver.smtaiserver.util.SMTWXSStatic; |
| | | import com.smtservlet.util.Json; |
| | | import com.smtservlet.util.SMTJsonWriter; |
| | | import com.smtservlet.util.SMTStatic; |
| | | import java.util.*; |
| | | import javax.servlet.http.HttpServletRequest; |
| | | import okhttp3.OkHttpClient; |
| | | import okhttp3.Request; |
| | | import okhttp3.Response; |
| | | import org.apache.logging.log4j.LogManager; |
| | | import org.apache.logging.log4j.Logger; |
| | | import org.jetbrains.annotations.NotNull; |
| | | import org.springframework.web.servlet.ModelAndView; |
| | | |
| | | public class SMTAIWeixinControl { |
| | | |
| | | private static final HashMap<String, String> asynchronousList = new HashMap<>(); |
| | | private static final String FROM_USER_NAME = "FromUserName"; |
| | | private static final String TO_USER_NAME = "ToUserName"; |
| | | private static final String CONTENT = "Content"; |
| | | private static final Logger _logger = LogManager.getLogger(SMTAIServerControl.class); |
| | | |
| | | private final Object _lockToken = new Object(); |
| | | |
| | | /** |
| | | * 微信éªè¯ |
| | | */ |
| | | public ModelAndView weChatNotify(SMTAIServerRequest tranReq) throws Exception { |
| | | String method = tranReq.getRequest().getMethod(); |
| | | if (method.equals("GET")) return SMTWXSStatic.getModelAndView(tranReq); |
| | | return reply(tranReq); |
| | | } |
| | | |
| | | /** 被å¨åå¤ */ |
| | | private ModelAndView reply(SMTAIServerRequest tranReq) { |
| | | |
| | | long l = System.currentTimeMillis() / 1000; |
| | | String createTimeStr = String.valueOf(l); |
| | | HttpServletRequest request = tranReq.getRequest(); |
| | | Map<String, String> requestMap = SMTWXSStatic.getWechatReqMap(request); |
| | | String fromUserName = requestMap.get(FROM_USER_NAME); |
| | | String toUserName = requestMap.get(TO_USER_NAME); |
| | | if (requestMap.isEmpty()) { |
| | | return null; |
| | | } |
| | | String reqContent = requestMap.get(CONTENT); |
| | | if (asynchronousList.get(fromUserName) != null && !reqContent.equals("忢è¾åº")) { |
| | | String dissuadeReturn = dissuadeReturn(fromUserName, toUserName, createTimeStr); |
| | | return tranReq.returnText(dissuadeReturn); |
| | | } |
| | | UUID randomUuid = UUID.randomUUID(); |
| | | asynchronousList.put(fromUserName, randomUuid.toString()); |
| | | String reply; |
| | | reply = getReply(reqContent); |
| | | String textContent = |
| | | "ææ£å¨æèå¦~请ç¨çâ¦â¦\n<a href=\"weixin://bizmsgmenu?msgmenucontent=忢è¾åº&msgmenuid=101\">忢è¾åº</a>"; |
| | | String baiduTextContent = "ææ¯ç¾åº¦\n<a href=\"www.baidu.com\">ç¾åº¦</a>"; |
| | | String result = |
| | | getString( |
| | | fromUserName, |
| | | toUserName, |
| | | createTimeStr, |
| | | reply, |
| | | null, |
| | | "https://pics0.baidu.com/feed/d31b0ef41bd5ad6e9388c7f8d8eda0d4b6fd3c60.png@f_auto?token=3da8e06f44a46832a7d0f50fa9e92c34", |
| | | "æ»ä¹¦è®°å¼ç¨çè¿äºå¤è¯è人寻å³", |
| | | "âæ²¡æè§ç©ï¼ä¸ææ¹åâãâ己䏿£ï¼çè½æ£äººâãâå°½å°è
å¤§ï¼æ
å¾®è
èââ¦â¦æ»ä¹¦è®°å¼ç¨çè¿äºå¤è¯è人寻å³ã", |
| | | "https://www.baidu.com/s?&wd=%E6%80%BB%E4%B9%A6%E8%AE%B0%E5%BC%95%E7%94%A8%E7%9A%84%E8%BF%99%E4%BA%9B%E5%8F%A4%E8%AF%AD%E8%80%90%E4%BA%BA%E5%AF%BB%E5%91%B3", |
| | | null); |
| | | if (reqContent.equals("忢è¾åº")) { |
| | | asynchronousList.remove(fromUserName); |
| | | _logger.info("ç¨æ·åæ¢è¾åº"); |
| | | return null; |
| | | } |
| | | |
| | | // 弿¥è°ç¨ aiReplyToTheUserASecondTime |
| | | // CompletableFuture.runAsync( |
| | | // () -> { |
| | | // try { |
| | | // SMTAIServerRequest threadTranReq = new SMTAIServerRequest(); |
| | | // threadTranReq.initInstance(null, "", null); |
| | | // String answer = callAIForAnswerQuestion(reqContent, threadTranReq); // Aiè°ç¨ è¿åç»æ |
| | | // aiReplyToTheUserASecondTime( |
| | | // answer, requestMap.get(FROM_USER_NAME), randomUuid.toString()); |
| | | // } catch (Exception e) { |
| | | // _logger.error("aiReplyToTheUserASecondTime error", e); |
| | | // asynchronousList.remove(request.getParameter("FromUserName")); |
| | | // } |
| | | // }); |
| | | _logger.info("å¾®ä¿¡æ¶æ¯è¿åï¼" + result); |
| | | // è¿å XML å符串 |
| | | return tranReq.returnText(result); |
| | | } |
| | | |
| | | // public ModelAndView weChatTest(SMTAIServerRequest tranReq) throws Exception { |
| | | // String question = "å°ç®±æ¯ä»ä¹"; |
| | | // tranReq.setAIQuestion(question); |
| | | // tranReq.setTextResultMode(); |
| | | // |
| | | // Set<String> setAgentGroup = new HashSet<>(); |
| | | // String weixinGroupId = |
| | | // (String) SMTAIServerApp.getApp().getGlobalConfig("weixin.group_id", false); |
| | | // setAgentGroup.add(weixinGroupId); |
| | | // |
| | | // SMTJavaAIChat.questionChat("aliyun", setAgentGroup, "ä¸å¡åºæ¯", question, null, false, tranReq); |
| | | // String text = tranReq.getTextResult(); |
| | | // |
| | | // return tranReq.returnText(text); |
| | | // } |
| | | |
| | | /** aiåå¤ */ |
| | | // private String callAIForAnswerQuestion(String question, SMTAIServerRequest tranReq) |
| | | // throws Exception { |
| | | // tranReq.setAIQuestion(question); |
| | | // tranReq.setTextResultMode(); |
| | | // |
| | | // Set<String> setAgentGroup = new HashSet<>(); |
| | | // String weixinGroupId = |
| | | // (String) SMTAIServerApp.getApp().getGlobalConfig("weixin.group_id", false); |
| | | // setAgentGroup.add(weixinGroupId); |
| | | // |
| | | // SMTJavaAIChat.questionChat("aliyun", setAgentGroup, "ä¸å¡åºæ¯", question, null, false, tranReq); |
| | | // return tranReq.getTextResult(); |
| | | // } |
| | | |
| | | /** äºæ¬¡åå¤ */ |
| | | public void aiReplyToTheUserASecondTime(String answer, String fromUserName, String abortID) |
| | | throws Exception { |
| | | String accessToken = getAccessToken(); |
| | | if (answer.isEmpty()) answer = "æ±æï¼æææ¶æ æ³çè§£æ¨çé®é¢ã"; |
| | | SMTJsonWriter jsonWr = new SMTJsonWriter(false); |
| | | jsonWr.addKeyValue("touser", fromUserName); |
| | | jsonWr.addKeyValue("msgtype", "text"); |
| | | jsonWr.beginMap("text"); |
| | | { |
| | | jsonWr.addKeyValue("content", answer); |
| | | } |
| | | jsonWr.endMap(); |
| | | String url = |
| | | String.format( |
| | | "https://api.weixin.qq.com/cgi-bin/message/custom/send?access_token=%s", accessToken); |
| | | if (abortID.equals(asynchronousList.get(fromUserName))) { |
| | | String s = SMTWXSStatic.sendPost(url, jsonWr.getRootJson()); |
| | | asynchronousList.remove(fromUserName); |
| | | _logger.info("ä¸ä¼ ç»æ: : " + s); |
| | | } else { |
| | | asynchronousList.remove(fromUserName); |
| | | _logger.info("弿¥è°ç¨è¢«åæ¶"); |
| | | } |
| | | } |
| | | |
| | | /** æ°æ®åºè·å access_koen */ |
| | | public String getAccessToken() throws Exception { |
| | | synchronized (this._lockToken) { |
| | | SMTDatabase db = SMTAIServerApp.getApp().allocDatabase(); |
| | | // try (SMTDatabase db = SMTAIServerApp.getApp().allocDatabase()) { |
| | | HashMap<String, String> weixinParam = SMTWXSStatic.getWeixinParam(); |
| | | Date curTime = new Date(); |
| | | String appId = weixinParam.get("appId"); |
| | | |
| | | // æ¥è¯¢æªè¿æç access_token |
| | | DBRecords dbRecord = |
| | | db.querySQL( |
| | | "SELECT app_id, access_token, expires_time " |
| | | + "FROM ai_weixin_token " |
| | | + "WHERE app_id = ? " |
| | | + " AND ? < expires_time", |
| | | new Object[] {appId, curTime}); |
| | | |
| | | // æ°æ®åºæ è®°å½ï¼ä»å¾®ä¿¡æå¡å¨è·å access_token |
| | | List<SMTDatabase.DBRecord> records = dbRecord.getRecords(); |
| | | |
| | | if (dbRecord.getRowCount() > 0) { |
| | | return records.get(0).getString("access_token"); |
| | | } |
| | | // 微信åï¼è¿åtokenå¹¶ä¸ä¿åæè¦çæ°æ® |
| | | else { |
| | | |
| | | Object[] accessToken = fetchAccessTokenFromWeixinServer(); // ä»å¾®ä¿¡æå¡å¨è·å access_token |
| | | |
| | | Date expiresTime = |
| | | SMTStatic.calculateTime( |
| | | curTime, SMTStatic.SMTCalcTime.ADD_SECOND, ((int) accessToken[1]) / 2); |
| | | |
| | | String sql = |
| | | "INSERT INTO ai_weixin_token (app_id, access_token, expires_time) " |
| | | + "VALUES (?, ?, ?) " |
| | | + "ON CONFLICT (app_id) " |
| | | + // 妿 app_id å²çª |
| | | "DO UPDATE SET " |
| | | + " access_token = EXCLUDED.access_token, " |
| | | + " expires_time = EXCLUDED.expires_time;"; |
| | | db.executeSQL(sql, new Object[] {appId, accessToken[0], expiresTime}); |
| | | return (String) accessToken[0]; |
| | | } |
| | | // } catch (Exception e) { |
| | | // throw new Exception("Failed to get access token", e); |
| | | // |
| | | // } |
| | | } |
| | | } |
| | | |
| | | /** ä»å¾®ä¿¡æå¡å¨è·å access_token */ |
| | | private Object[] fetchAccessTokenFromWeixinServer() throws Exception { |
| | | HashMap<String, String> weixinParam = SMTWXSStatic.getWeixinParam(); |
| | | OkHttpClient okHttpClient = new OkHttpClient(); |
| | | // åå»ºè¯·æ± |
| | | Request request = |
| | | new Request.Builder() |
| | | .url( |
| | | "https://api.weixin.qq.com/cgi-bin/token?grant_type=client_credential&appid=" |
| | | + weixinParam.get("appId") |
| | | + "&secret=" |
| | | + weixinParam.get("secret")) // 请æ±URL |
| | | .get() // 使ç¨GETæ¹æ³ |
| | | .build(); |
| | | Response response = okHttpClient.newCall(request).execute(); |
| | | if (!response.isSuccessful() || response.body() == null) { |
| | | throw new Exception("can't get weixin token"); |
| | | } |
| | | Json json = Json.read(response.body().string()); |
| | | String accessToken = json.safeGetStr("access_token", null); |
| | | String expiresIn = json.safeGetStr("expires_in", null); |
| | | |
| | | if (accessToken != null) { |
| | | return new Object[] {accessToken, SMTStatic.toInt(expiresIn)}; |
| | | } else { |
| | | _logger.info("can't get weixin token : " + json); |
| | | return null; |
| | | // throw new Exception("can't get weixin token : " + json); |
| | | } |
| | | } |
| | | |
| | | @NotNull |
| | | private static String getString(String fromUserName, String toUserName, String createTimeStr, String msgType, String content, String mediaId, String title, String description, String musicUrl, String hqMusicUrl) { |
| | | StringBuilder xmlBuilder = new StringBuilder(); |
| | | xmlBuilder.append("<xml>\n") |
| | | .append(" <ToUserName><![CDATA[").append(fromUserName).append("]]></ToUserName>\n") |
| | | .append(" <FromUserName><![CDATA[").append(toUserName).append("]]></FromUserName>\n") |
| | | .append(" <CreateTime>").append(createTimeStr).append("</CreateTime>\n") |
| | | .append(" <MsgType><![CDATA[").append(msgType).append("]]></MsgType>\n"); |
| | | |
| | | switch (msgType) { |
| | | case "text": |
| | | xmlBuilder.append(" <Content><![CDATA[").append(content).append("]]></Content>\n"); |
| | | break; |
| | | case "image": |
| | | xmlBuilder.append(" <Image><MediaId><![CDATA[").append(mediaId).append("]]></MediaId></Image>\n"); |
| | | break; |
| | | case "voice": |
| | | xmlBuilder.append(" <Voice><MediaId><![CDATA[").append(mediaId).append("]]></MediaId></Voice>\n"); |
| | | break; |
| | | case "video": |
| | | xmlBuilder.append(" <Video>\n") |
| | | .append(" <MediaId><![CDATA[").append(mediaId).append("]]></MediaId>\n") |
| | | .append(" <Title><![CDATA[").append(title).append("]]></Title>\n") |
| | | .append(" <Description><![CDATA[").append(description).append("]]></Description>\n") |
| | | .append(" </Video>\n"); |
| | | break; |
| | | case "music": |
| | | xmlBuilder.append(" <Music>\n") |
| | | .append(" <Title><![CDATA[").append(title).append("]]></Title>\n") |
| | | .append(" <Description><![CDATA[").append(description).append("]]></Description>\n") |
| | | .append(" <MusicUrl><![CDATA[").append(musicUrl).append("]]></MusicUrl>\n") |
| | | .append(" <HQMusicUrl><![CDATA[").append(hqMusicUrl).append("]]></HQMusicUrl>\n") |
| | | .append(" <ThumbMediaId><![CDATA[").append(mediaId).append("]]></ThumbMediaId>\n") |
| | | .append(" </Music>\n"); |
| | | break; |
| | | case "news": // å¾ææ¶æ¯ |
| | | xmlBuilder.append(" <ArticleCount>2</ArticleCount>\n") |
| | | .append(" <Articles>\n") |
| | | .append(" <item>\n") |
| | | .append(" <Title><![CDATA[").append(title).append("]]></Title>\n") |
| | | .append(" <Description><![CDATA[").append(description).append("]]></Description>\n") |
| | | .append(" <PicUrl><![CDATA[").append(mediaId).append("]]></PicUrl>\n") |
| | | .append(" <Url><![CDATA[").append(musicUrl).append("]]></Url>\n") |
| | | .append(" </item>\n") |
| | | .append(" </Articles>\n"); |
| | | break; |
| | | default: |
| | | xmlBuilder.append(" <Content><![CDATA[æªç¥çæ¶æ¯ç±»å]]></Content>\n"); |
| | | } |
| | | xmlBuilder.append("</xml>"); |
| | | return xmlBuilder.toString(); |
| | | } |
| | | |
| | | /** |
| | | * ç¨äºæµè¯ï¼æ£å¸¸å¯ä»¥ä¸ç¨ |
| | | * |
| | | * @param reqContent |
| | | * @return |
| | | */ |
| | | @NotNull |
| | | private static String getReply(String reqContent) { |
| | | String reply; |
| | | switch (reqContent) { |
| | | case "æå": |
| | | reply = "text"; |
| | | break; |
| | | case "å¾ç": |
| | | reply = "image"; |
| | | break; |
| | | case "è¯é³": |
| | | reply = "voice"; |
| | | break; |
| | | case "è§é¢": |
| | | reply = "video"; |
| | | break; |
| | | case "é³ä¹": |
| | | reply = "music"; |
| | | break; |
| | | case "徿": |
| | | reply = "news"; |
| | | break; |
| | | default: |
| | | reply = "text"; |
| | | break; |
| | | } |
| | | return reply; |
| | | } |
| | | |
| | | private static String dissuadeReturn( |
| | | String fromUserName, String toUserName, String createTimeStr) { |
| | | String xmltemp = |
| | | "<xml>\n" |
| | | + " <ToUserName><![CDATA[{{{toUser}}}]]></ToUserName>\n" |
| | | + " <FromUserName><![CDATA[{{{fromUser}}}]]></FromUserName>\n" |
| | | + " <CreateTime>{{{CreateTime}}}</CreateTime>\n" |
| | | + " <MsgType><![CDATA[text]]></MsgType>\n" |
| | | + " <Content><![CDATA[ä¸ä¸æ¡æ¶æ¯è¿å¨å è½½ä¸å¦ï¼è¯·ç¨çæç¹å»ä¸æ¬¡åæ¢ä¸ä¸è½®åå¤\n<a href=\"weixin://bizmsgmenu?msgmenucontent=忢è¾åº&msgmenuid=101\">忢è¾åº</a>]]></Content>\n" |
| | | + " <Content><![CDATA[]]></Content>\n" |
| | | + "</xml>"; |
| | | // æ¿æ¢å ä½ç¬¦ |
| | | return xmltemp |
| | | .replace("{{{toUser}}}", fromUserName) |
| | | .replace("{{{fromUser}}}", toUserName) |
| | | .replace("{{{CreateTime}}}", createTimeStr); |
| | | } |
| | | } |