微信网页授权移植

jeecg-module-system/jeecg-system-start/src/main/java/org/jeecg/modules/contoller/h5/WxAuthController.java

@@ -0,0 +1,130 @@
+package org.jeecg.modules.contoller.h5;
+
+import io.swagger.annotations.Api;
+import io.swagger.annotations.ApiOperation;
+import lombok.RequiredArgsConstructor;
+import me.chanjar.weixin.common.bean.WxOAuth2UserInfo;
+import me.chanjar.weixin.common.bean.oauth2.WxOAuth2AccessToken;
+import me.chanjar.weixin.common.error.WxErrorException;
+import me.chanjar.weixin.mp.api.WxMpService;
+import org.jeecg.common.api.vo.Result;
+import org.jeecg.modules.utils.CookieUtil;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.util.StringUtils;
+import org.springframework.web.bind.annotation.*;
+
+import javax.servlet.http.HttpServletRequest;
+import javax.servlet.http.HttpServletResponse;
+import java.util.Map;
+import java.util.TreeMap;
+import java.util.UUID;
+
+/**
+ * 微信网页授权相关
+ */
+@RestController
+@RequestMapping("/wxAuth")
+@Api(tags = {"微信网页授权"})
+@RequiredArgsConstructor
+public class WxAuthController {
+    Logger logger = LoggerFactory.getLogger(this.getClass());
+    @Autowired
+    private final WxMpService wxMpService;
+
+    /**
+     * 使用微信授权code换取openid
+     *
+     * @param request
+     * @param response
+     * @param form
+     * @return
+     */
+    @PostMapping("/codeToOpenid")
+    @CrossOrigin
+    @ApiOperation(value = "网页登录-code换取openid",notes = "scope为snsapi_base")
+    public Result<?> codeToOpenid(HttpServletRequest request, HttpServletResponse response,
+                                  @CookieValue String appid, @RequestBody WxH5OuthrizeForm form) {
+        try {
+            this.wxMpService.switchoverTo(appid);
+            WxOAuth2AccessToken token = wxMpService.getOAuth2Service().getAccessToken(form.getCode());
+            String openid = token.getOpenId();
+            CookieUtil.setCookie(response, "openid", openid, 365 * 24 * 60 * 60);
+            String openidToken = MD5Util.getMd5AndSalt(openid);
+            CookieUtil.setCookie(response, "openidToken", openidToken, 365 * 24 * 60 * 60);
+            return Result.ok(openid);
+        } catch (WxErrorException e) {
+            logger.error("code换取openid失败", e);
+            return Result.error(e.getError().getErrorMsg());
+        }
+    }
+
+    /**
+     * 使用微信授权code换取用户信息(需scope为 snsapi_userinfo)
+     *
+     * @param request
+     * @param response
+     * @param form
+     * @return
+     */
+    @PostMapping("/codeToUserInfo")
+    @CrossOrigin
+    @ApiOperation(value = "网页登录-code换取用户信息",notes = "需scope为 snsapi_userinfo")
+    public Result codeToUserInfo(HttpServletRequest request, HttpServletResponse response,
+                                 @CookieValue String appid, @RequestBody WxH5OuthrizeForm form) {
+        try {
+            this.wxMpService.switchoverTo(appid);
+            WxOAuth2AccessToken token = wxMpService.getOAuth2Service().getAccessToken(form.getCode());
+            WxOAuth2UserInfo userInfo = wxMpService.getOAuth2Service().getUserInfo(token,"zh_CN");
+            String openid = userInfo.getOpenid();
+            CookieUtil.setCookie(response, "openid", openid, 365 * 24 * 60 * 60);
+            String openidToken = MD5Util.getMd5AndSalt(openid);
+            CookieUtil.setCookie(response, "openidToken", openidToken, 365 * 24 * 60 * 60);
+            return Result.ok(new WxUser(userInfo,appid));
+        } catch (WxErrorException e) {
+            logger.error("code换取用户信息失败", e);
+            return Result.error(e.getError().getErrorMsg());
+        }
+    }
+
+    /**
+     * 获取微信分享的签名配置
+     * 允许跨域（只有微信公众号添加了js安全域名的网站才能加载微信分享，故这里不对域名进行校验）
+     *
+     * @param request
+     * @param response
+     * @return
+     */
+    @GetMapping("/getShareSignature")
+    @ApiOperation(value = "获取微信分享的签名配置",notes = "微信公众号添加了js安全域名的网站才能加载微信分享")
+    public Result getShareSignature(HttpServletRequest request, HttpServletResponse response, @CookieValue String appid) throws WxErrorException {
+        this.wxMpService.switchoverTo(appid);
+        // 1.拼接url（当前网页的URL，不包含#及其后面部分）
+        String wxShareUrl = request.getHeader(Constant.WX_CLIENT_HREF_HEADER);
+        if (!StringUtils.hasText(wxShareUrl)) {
+            return Result.error("header中缺少"+Constant.WX_CLIENT_HREF_HEADER+"参数，微信分享加载失败");
+        }
+        wxShareUrl = wxShareUrl.split("#")[0];
+        Map<String, String> wxMap = new TreeMap<>();
+        String wxNoncestr = UUID.randomUUID().toString();
+        String wxTimestamp = (System.currentTimeMillis() / 1000) + "";
+        wxMap.put("noncestr", wxNoncestr);
+        wxMap.put("timestamp", wxTimestamp);
+        wxMap.put("jsapi_ticket", wxMpService.getJsapiTicket());
+        wxMap.put("url", wxShareUrl);
+
+        // 加密获取signature
+        StringBuilder wxBaseString = new StringBuilder();
+        wxMap.forEach((key, value) -> wxBaseString.append(key).append("=").append(value).append("&"));
+        String wxSignString = wxBaseString.substring(0, wxBaseString.length() - 1);
+        // signature
+        String wxSignature = SHA1Util.sha1(wxSignString);
+        Map<String, String> resMap = new TreeMap<>();
+        resMap.put("appId", appid);
+        resMap.put("wxTimestamp", wxTimestamp);
+        resMap.put("wxNoncestr", wxNoncestr);
+        resMap.put("wxSignature", wxSignature);
+        return Result.ok(resMap);
+    }
+}

jeecg-module-system/jeecg-system-start/src/main/java/org/jeecg/modules/entity/WxH5OuthrizeForm.java

@@ -0,0 +1,15 @@
+package org.jeecg.modules.entity;
+
+import lombok.Data;
+
+import javax.validation.constraints.NotEmpty;
+@Data
+public class WxH5OuthrizeForm {
+    @NotEmpty(message = "code不得为空")
+    private String code;
+
+    @Override
+    public String toString() {
+        return Json.toJsonString(this);
+    }
+}

jeecg-module-system/jeecg-system-start/src/main/java/org/jeecg/modules/utils/CookieUtil.java

@@ -0,0 +1,64 @@
+package org.jeecg.modules.utils;
+
+import javax.servlet.http.Cookie;
+import javax.servlet.http.HttpServletRequest;
+import javax.servlet.http.HttpServletResponse;
+
+/**
+ * Cookie工具类
+ */
+public class CookieUtil {
+
+    /**
+     * 设置cookie
+     *
+     * @param maxAge 秒
+     */
+    public static void setCookie(HttpServletResponse response, String cookieName, String cookieValue,
+                                 int maxAge) {
+        Cookie cookie = new Cookie(cookieName, cookieValue);
+        maxAge = Math.max(maxAge, 0);
+        cookie.setMaxAge(maxAge);
+        cookie.setPath("/");
+        response.addCookie(cookie);
+    }
+
+    public static void clearCookie(HttpServletResponse response, String cookieName, String domain) {
+        Cookie cookie = new Cookie(cookieName, "");
+        cookie.setMaxAge(0);
+        cookie.setDomain(domain);
+        cookie.setPath("/");
+        response.addCookie(cookie);
+    }
+
+    public static void refreshCookie(HttpServletRequest request, HttpServletResponse response, String cookieName,
+                                     String domain, int maxAge) {
+        Cookie cookie = getCookie(request, cookieName);
+        if (cookie != null) {
+            cookie.setMaxAge(maxAge);
+            cookie.setDomain(domain);
+            cookie.setPath("/");
+            response.addCookie(cookie);
+        }
+    }
+
+    public static String getCookieValue(HttpServletRequest request, String cookieName) {
+        Cookie cookie = getCookie(request, cookieName);
+        if (cookie != null) {
+            return cookie.getValue();
+        }
+        return null;
+    }
+
+    private static Cookie getCookie(HttpServletRequest request, String cookieName) {
+        Cookie[] cookies = request.getCookies();
+        if (cookies != null) {
+            for (int i = 0; i < cookies.length; i++) {
+                if (cookieName.equals(cookies[i].getName())) {
+                    return cookies[i];
+                }
+            }
+        }
+        return null;
+    }
+}

pom.xml

@@ -77,6 +77,7 @@
 		<logback.version>1.2.9</logback.version>
 		<!-- seata -->
 		<seata.version>1.5.2</seata.version>
+		<weixin-java.version>4.5.6.B</weixin-java.version>
 	</properties>
 
 	<modules>
@@ -116,6 +117,11 @@
 	</repositories>
 
 	<dependencies>
+		<dependency>
+			<groupId>com.github.binarywang</groupId>
+			<artifactId>weixin-java-mp</artifactId>
+			<version>${weixin-java.version}</version>
+		</dependency>
 		<dependency>
 			<groupId>org.springframework.boot</groupId>
 			<artifactId>spring-boot-starter-test</artifactId>