issue #39 公众号添加oauth2的支持

2025-12-23 18:48:00 +08:00 · 2014-11-28 15:57:52 +08:00
parent 8e4077e061
commit 0d6712f709
11 changed files with 267 additions and 80 deletions
--- a/weixin-java-mp/src/test/java/me/chanjar/weixin/mp/demo/WxMpDemoServer.java
+++ b/weixin-java-mp/src/test/java/me/chanjar/weixin/mp/demo/WxMpDemoServer.java
@@ -1,20 +1,127 @@
 package me.chanjar.weixin.mp.demo;

+import me.chanjar.weixin.common.api.WxConsts;
+import me.chanjar.weixin.common.bean.result.WxMediaUploadResult;
+import me.chanjar.weixin.common.exception.WxErrorException;
+import me.chanjar.weixin.mp.api.*;
+import me.chanjar.weixin.mp.bean.WxMpMpXmlOutImageMessage;
+import me.chanjar.weixin.mp.bean.WxMpXmlMessage;
+import me.chanjar.weixin.mp.bean.WxMpXmlOutMessage;
+import me.chanjar.weixin.mp.bean.WxMpXmlOutTextMessage;
 import org.eclipse.jetty.server.Server;
-import org.eclipse.jetty.servlet.ServletHandler;
+import org.eclipse.jetty.servlet.*;
+
+import javax.xml.bind.JAXBException;
+import java.io.IOException;
+import java.io.InputStream;
+import java.util.Map;

 /**
 * @author Daniel Qian
 */
 public class WxMpDemoServer {
+
+  private static WxMpConfigStorage wxMpConfigStorage;
+  private static WxMpService wxMpService;
+  private static WxMpMessageRouter wxMpMessageRouter;
+
  public static void main(String[] args) throws Exception {
+    initWeixin();
+
    Server server = new Server(8080);

    ServletHandler handler = new ServletHandler();
-    server.setHandler(handler);

-    handler.addServletWithMapping(WxMpDemoServlet.class, "/*");
+    ServletHolder endpointServletHolder = new ServletHolder(new WxMpEndpointServlt(wxMpConfigStorage, wxMpService, wxMpMessageRouter));
+    handler.addServletWithMapping(endpointServletHolder, "/*");
+
+    ServletHolder oauthServletHolder = new ServletHolder(new WxMpOAuth2Servlet(wxMpService));
+    handler.addServletWithMapping(oauthServletHolder, "/oauth2/*");
+
+    server.setHandler(handler);
    server.start();
    server.join();
  }
+
+  private static void initWeixin() {
+    try {
+      InputStream is1 = ClassLoader.getSystemResourceAsStream("test-config.xml");
+      WxMpDemoInMemoryConfigStorage config = WxMpDemoInMemoryConfigStorage.fromXml(is1);
+
+      wxMpConfigStorage = config;
+      wxMpService = new WxMpServiceImpl();
+      wxMpService.setWxMpConfigStorage(config);
+
+      WxMpMessageHandler textHandler = new WxMpMessageHandler() {
+        @Override
+        public WxMpXmlOutMessage handle(WxMpXmlMessage wxMessage, Map<String, Object> context,
+            WxMpService wxMpService) {
+          WxMpXmlOutTextMessage m
+              = WxMpXmlOutMessage.TEXT().content("测试加密消息").fromUser(wxMessage.getToUserName())
+              .toUser(wxMessage.getFromUserName()).build();
+          return m;
+        }
+      };
+
+      WxMpMessageHandler imageHandler = new WxMpMessageHandler() {
+        @Override
+        public WxMpXmlOutMessage handle(WxMpXmlMessage wxMessage, Map<String, Object> context,
+            WxMpService wxMpService) {
+          try {
+            WxMediaUploadResult wxMediaUploadResult = wxMpService
+                .mediaUpload(WxConsts.MEDIA_IMAGE, WxConsts.FILE_JPG, ClassLoader.getSystemResourceAsStream("mm.jpeg"));
+            WxMpMpXmlOutImageMessage m
+                = WxMpXmlOutMessage
+                .IMAGE()
+                .mediaId(wxMediaUploadResult.getMediaId())
+                .fromUser(wxMessage.getToUserName())
+                .toUser(wxMessage.getFromUserName())
+                .build();
+            return m;
+          } catch (WxErrorException e) {
+            e.printStackTrace();
+          } catch (IOException e) {
+            e.printStackTrace();
+          }
+          return null;
+        }
+      };
+
+      WxMpMessageHandler oauth2handler = new WxMpMessageHandler() {
+        @Override
+        public WxMpXmlOutMessage handle(WxMpXmlMessage wxMessage, Map<String, Object> context,
+            WxMpService wxMpService) {
+          String href = "<a href=\"" + wxMpService.oauth2buildAuthorizationUrl(WxConsts.OAUTH2_SCOPE_USER_INFO, null)
+              + "\">测试oauth2</a>";
+          return WxMpXmlOutMessage
+              .TEXT()
+              .content(href)
+              .fromUser(wxMessage.getToUserName())
+              .toUser(wxMessage.getFromUserName()).build();
+        }
+      };
+
+      wxMpMessageRouter = new WxMpMessageRouter(wxMpService);
+      wxMpMessageRouter
+          .rule()
+          .async(false)
+          .content("哈哈") // 拦截内容为“哈哈”的消息
+          .handler(textHandler)
+          .end()
+          .rule()
+          .async(false)
+          .content("图片")
+          .handler(imageHandler)
+          .end()
+          .rule()
+          .async(false)
+          .content("oauth")
+          .handler(oauth2handler)
+          .end()
+      ;
+
+    } catch (JAXBException e) {
+      throw new RuntimeException(e);
+    }
+  }
 }
--- a/weixin-java-mp/src/test/java/me/chanjar/weixin/mp/demo/WxMpEndpointServlt.java
+++ b/weixin-java-mp/src/test/java/me/chanjar/weixin/mp/demo/WxMpEndpointServlt.java
@@ -22,71 +22,17 @@ import java.util.Map;
 /**
 * @author Daniel Qian
 */
-public class WxMpDemoServlet extends HttpServlet {
+public class WxMpEndpointServlt extends HttpServlet {

  protected WxMpConfigStorage wxMpConfigStorage;
  protected WxMpService wxMpService;
  protected WxMpMessageRouter wxMpMessageRouter;

-  @Override public void init() throws ServletException {
-    //
-    super.init();
-    try {
-      InputStream is1 = ClassLoader.getSystemResourceAsStream("test-config.xml");
-      WxMpDemoInMemoryConfigStorage config = WxMpDemoInMemoryConfigStorage.fromXml(is1);
-
-      wxMpConfigStorage = config;
-      wxMpService = new WxMpServiceImpl();
-      wxMpService.setWxMpConfigStorage(config);
-
-      WxMpMessageHandler textHandler = new WxMpMessageHandler() {
-        @Override public WxMpXmlOutMessage handle(WxMpXmlMessage wxMessage, Map<String, Object> context, WxMpService wxMpService) {
-          WxMpXmlOutTextMessage m
-              = WxMpXmlOutMessage.TEXT().content("测试加密消息").fromUser(wxMessage.getToUserName())
-              .toUser(wxMessage.getFromUserName()).build();
-          return m;
-        }
-      };
-
-      WxMpMessageHandler imageHandler = new WxMpMessageHandler() {
-        @Override public WxMpXmlOutMessage handle(WxMpXmlMessage wxMessage, Map<String, Object> context, WxMpService wxMpService) {
-          try {
-            WxMediaUploadResult wxMediaUploadResult = wxMpService
-                .mediaUpload(WxConsts.MEDIA_IMAGE, WxConsts.FILE_JPG, ClassLoader.getSystemResourceAsStream("mm.jpeg"));
-            WxMpMpXmlOutImageMessage m
-                = WxMpXmlOutMessage
-                .IMAGE()
-                .mediaId(wxMediaUploadResult.getMediaId())
-                .fromUser(wxMessage.getToUserName())
-                .toUser(wxMessage.getFromUserName())
-                .build();
-            return m;
-          } catch (WxErrorException e) {
-            e.printStackTrace();
-          } catch (IOException e) {
-            e.printStackTrace();
-          }
-          return null;
-        }
-      };
-
-      wxMpMessageRouter = new WxMpMessageRouter(wxMpService);
-      wxMpMessageRouter
-          .rule()
-            .async(false)
-            .content("哈哈") // 拦截内容为“哈哈”的消息
-            .handler(textHandler)
-          .end()
-          .rule()
-            .async(false)
-            .content("图片")
-            .handler(imageHandler)
-          .end()
-      ;
-
-    } catch (JAXBException e) {
-      throw new RuntimeException(e);
-    }
+  public WxMpEndpointServlt(WxMpConfigStorage wxMpConfigStorage, WxMpService wxMpService,
+      WxMpMessageRouter wxMpMessageRouter) {
+    this.wxMpConfigStorage = wxMpConfigStorage;
+    this.wxMpService = wxMpService;
+    this.wxMpMessageRouter = wxMpMessageRouter;
  }

  @Override protected void service(HttpServletRequest request, HttpServletResponse response)
--- a/weixin-java-mp/src/test/java/me/chanjar/weixin/mp/demo/WxMpOAuth2Servlet.java
+++ b/weixin-java-mp/src/test/java/me/chanjar/weixin/mp/demo/WxMpOAuth2Servlet.java
@@ -0,0 +1,68 @@
+package me.chanjar.weixin.mp.demo;
+
+import me.chanjar.weixin.common.api.WxConsts;
+import me.chanjar.weixin.common.bean.result.WxMediaUploadResult;
+import me.chanjar.weixin.common.exception.WxErrorException;
+import me.chanjar.weixin.common.util.StringUtils;
+import me.chanjar.weixin.mp.api.*;
+import me.chanjar.weixin.mp.bean.WxMpMpXmlOutImageMessage;
+import me.chanjar.weixin.mp.bean.WxMpXmlMessage;
+import me.chanjar.weixin.mp.bean.WxMpXmlOutMessage;
+import me.chanjar.weixin.mp.bean.WxMpXmlOutTextMessage;
+import me.chanjar.weixin.mp.bean.result.WxMpOAuth2AccessToken;
+import me.chanjar.weixin.mp.bean.result.WxMpUser;
+
+import javax.servlet.ServletContext;
+import javax.servlet.ServletException;
+import javax.servlet.http.HttpServlet;
+import javax.servlet.http.HttpServletRequest;
+import javax.servlet.http.HttpServletResponse;
+import javax.xml.bind.JAXBException;
+import java.io.IOException;
+import java.io.InputStream;
+import java.util.Map;
+
+/**
+ * @author Daniel Qian
+ */
+public class WxMpOAuth2Servlet extends HttpServlet {
+
+  protected WxMpService wxMpService;
+
+  public WxMpOAuth2Servlet(WxMpService wxMpService) {
+    this.wxMpService = wxMpService;
+  }
+
+  @Override protected void service(HttpServletRequest request, HttpServletResponse response)
+      throws ServletException, IOException {
+
+    response.setContentType("text/html;charset=utf-8");
+    response.setStatus(HttpServletResponse.SC_OK);
+
+    String code = request.getParameter("code");
+    try {
+      response.getWriter().println("<h1>code</h1>");
+      response.getWriter().println(code);
+
+      WxMpOAuth2AccessToken wxMpOAuth2AccessToken = wxMpService.oauth2getAccessToken(code);
+      response.getWriter().println("<h1>access token</h1>");
+      response.getWriter().println(wxMpOAuth2AccessToken.toString());
+
+      WxMpUser wxMpUser = wxMpService.oauth2getUserInfo(wxMpOAuth2AccessToken, null);
+      response.getWriter().println("<h1>user info</h1>");
+      response.getWriter().println(wxMpUser.toString());
+
+      wxMpOAuth2AccessToken = wxMpService.oauth2refreshAccessToken(wxMpOAuth2AccessToken.getRefreshToken());
+      response.getWriter().println("<h1>after refresh</h1>");
+      response.getWriter().println(wxMpOAuth2AccessToken.toString());
+
+    } catch (WxErrorException e) {
+      e.printStackTrace();
+    }
+
+    response.getWriter().flush();
+    response.getWriter().close();
+
+  }
+
+}
--- a/weixin-java-mp/src/test/resources/test-config.sample.xml
+++ b/weixin-java-mp/src/test/resources/test-config.sample.xml
@@ -6,4 +6,5 @@
    <accessToken>可以不填写</accessToken>
    <expiresIn>可以不填写</expiresIn>
    <openId>某个加你公众号的用户的openId</openId>
+    <oauth2redirectUri>网页授权获取用户信息回调地址</oauth2redirectUri>
 </xml>