新增微信授权登录

zhongrj

2023-11-14 2d7fc00f4c1b39f45ab4eb1d42582c45d6cad921

新增微信授权登录

 src/main/java/org/springblade/modules/auth/endpoint/BladeTokenEndPoint.java

@@ -48,6 +48,7 @@
import org.springblade.modules.loginrecord.entity.LoginRecord;
import org.springblade.modules.loginrecord.service.LoginRecordService;
import org.springblade.modules.system.entity.UserInfo;
import org.springblade.modules.system.service.IUserService;
import org.springblade.modules.zc.service.IZcService;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.context.request.RequestContextHolder;
@@ -85,6 +86,7 @@
    private final LoginRecordService loginRecordService;

    private final ServerConfig serverConfig;
    private final IUserService userService;

    /**
     * 登录
@@ -162,6 +164,14 @@
                    }
                }
            }
            //如果是app模拟考试登录
            if (loginType.equals(2)) {
                //判断是否已报名
                boolean flag = userService.getUserIsApply(userInfo.getUser().getId());
                if (!flag){
                    return authInfo.set("error_description", "没有报名不能进行刷题");
                }
            }
            //刷新 token 不新增登录记录
            if (!grantType.equals("refresh_token")){
                //新增登录记录

 src/main/java/org/springblade/modules/auth/granter/WxTokenGranter.java

@@ -16,25 +16,39 @@
 */
package org.springblade.modules.auth.granter;

import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONObject;
import lombok.AllArgsConstructor;
import me.zhyd.oauth.model.AuthCallback;
import me.zhyd.oauth.model.AuthResponse;
import me.zhyd.oauth.model.AuthUser;
import me.zhyd.oauth.request.AuthRequest;
import org.springblade.common.cache.CacheNames;
import org.springblade.common.utils.HttpClientUtils;
import org.springblade.core.log.exception.ServiceException;
import org.springblade.core.redis.cache.BladeRedis;
import org.springblade.core.tool.utils.DigestUtil;
import org.springblade.core.tool.utils.Func;
import org.springblade.core.tool.utils.StringUtil;
import org.springblade.core.tool.utils.WebUtil;
import org.springblade.core.social.props.SocialProperties;
import org.springblade.core.social.utils.SocialUtil;
import org.springblade.core.tool.api.R;
import org.springblade.core.tool.utils.*;
import org.springblade.modules.auth.enums.UserEnum;
import org.springblade.modules.auth.provider.ITokenGranter;
import org.springblade.modules.auth.provider.TokenParameter;
import org.springblade.modules.auth.utils.TokenUtil;
import org.springblade.modules.system.entity.Tenant;
import org.springblade.modules.system.entity.UserInfo;
import org.springblade.modules.system.entity.UserOauth;
import org.springblade.modules.system.service.ITenantService;
import org.springblade.modules.system.service.IUserService;
import org.springframework.stereotype.Component;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.client.RestTemplate;

import javax.servlet.http.HttpServletRequest;
import java.util.HashMap;
import java.util.LinkedHashMap;
import java.util.Map;
import java.util.Objects;

/**
 * 微信小程序 TokenGranter
@@ -48,16 +62,68 @@

    private final IUserService userService;

    //获取wxAccessToken
    private final String WX_GET_ACCESS_TOKEN = "https://api.weixin.qq.com/cgi-bin/token";
    private final String WX_GET_PHONE_NUMBER = "https://api.weixin.qq.com/wxa/business/getuserphonenumber";
    private final String WX_APP_ID = "wx11797b813d61cac1";
    private final String WX_SECRET = "c9bbdf03f4f98e2d8fc45aebd4e41eab";

    @Override
    public UserInfo grant(TokenParameter tokenParameter) {
        HttpServletRequest request = WebUtil.getRequest();
        String tenantId = tokenParameter.getArgs().getStr("tenantId");
        String username = tokenParameter.getArgs().getStr("username");
        String code = tokenParameter.getArgs().getStr("code");
        // 校验，获取手机号
        String phone = "15170720695";
        // 返回
        return userService.userInfoByWx(tenantId, phone, UserEnum.WEB);
        // 开放平台授权码
        String username = request.getParameter("username");
        String phone = getPhone(username);
        // 判断用户是否存在
        UserInfo userInfo = userService.getUserByPhone(phone);
        if (null==userInfo){
            throw new ServiceException("用户不存在!");
        }else {
            //判断是否为保安员
            //判断是否已报名
            boolean flag = userService.getUserIsApply(userInfo.getUser().getId());
            if (!flag){
                throw new ServiceException("没有报名不能进行刷题!");
            }
        }
        // 返回UserInfo
        return userInfo;
    }

    /**
     * 获取手机号
     * @param code
     * @return
     */
    public String getPhone(String code) {
        String wxAccessToken = getWxAccessToken();
        //参数
        String url = WX_GET_PHONE_NUMBER + "?access_token=" + wxAccessToken;
        //向微信接口发起请求获取手机号
        RestTemplate template = new RestTemplate();
        Map<String, Object> params = new LinkedHashMap<>();
        params.put("code", code);
        String result = template.postForObject(url, params, String.class);
        JSONObject resultObj = (JSONObject) JSONObject.parse(result);
        JSONObject phoneInfo = (JSONObject) resultObj.get("phone_info");
        String phoneNumber = phoneInfo.get("phoneNumber").toString();
        return phoneNumber;
    }

    /**
     * 获取 WxAccessToken
     * @return
     */
    public String getWxAccessToken() {
        //参数
        Map<String, String> params = new HashMap<>();
        params.put("grant_type", "client_credential");
        params.put("appid", WX_APP_ID);
        params.put("secret", WX_SECRET);
        String result = HttpClientUtils.doGet(WX_GET_ACCESS_TOKEN, params);
        JSONObject jsonObject = JSON.parseObject(result);
        String accessToken = jsonObject.getString("access_token");
        return accessToken;
    }

}

 src/main/java/org/springblade/modules/system/mapper/UserMapper.java

@@ -298,4 +298,10 @@
     */
    void updateLogout(@Param("list") List<String> userIds);

    /**
     * 判断是否报名
     * @param id
     * @return
     */
    User getUserIsApply(@Param("id") Long id);
}

 src/main/java/org/springblade/modules/system/mapper/UserMapper.xml

@@ -1137,7 +1137,7 @@
    <!--根据手机号获取用户信息-->
    <select id="getUserByPhone" resultType="org.springblade.modules.system.entity.User">
        select * from blade_user where is_deleted = 0 and status = 1
        and tenant_id = #{tenantId} and phone = #{phone}
        and tenant_id = #{tenantId} and phone = #{phone} limit 1
    </select>

    <!--按天查询 day 天未登录的人员ids集合信息-->
@@ -1183,5 +1183,13 @@
        </choose>
    </update>

    <!--判断是否报名-->
    <select id="getUserIsApply" resultType="org.springblade.modules.system.entity.User">
        select bu.* from blade_user bu
        left join blade_role br on br.id = bu.role_id
        where bu.is_deleted = 0 and bu.status = 1 and bu.id = #{id}
        and br.role_alias = '保安'
    </select>


</mapper>

 src/main/java/org/springblade/modules/system/service/IUserService.java

@@ -404,4 +404,18 @@
     * 12个月未登录人员，进行注销
     */
    void oneYearNotLoginHandle();

    /**
     * 判断用户是否存在(根据手机号）
     * @param phone
     * @return
     */
    UserInfo getUserByPhone(String phone);

    /**
     * 判断是否报名
     * @param id
     * @return
     */
    boolean getUserIsApply(Long id);
}

 src/main/java/org/springblade/modules/system/service/impl/UserServiceImpl.java

@@ -2036,4 +2036,29 @@
        // 统一注销操作
        baseMapper.updateLogout(userIds);
    }

    /**
     * 判断用户是否存在(根据手机号）
     * @param phone
     * @return
     */
    @Override
    public UserInfo getUserByPhone(String phone) {
        User user = baseMapper.getUserByPhone("000000", phone);
        return buildUserInfo(user, UserEnum.WEB);
    }

    /**
     * 判断是否报名
     * @param id
     * @return
     */
    @Override
    public boolean getUserIsApply(Long id) {
        User user = baseMapper.getUserIsApply(id);
        if (null!=user){
            return true;
        }
        return false;
    }
}

 src/main/resources/application-test.yml

@@ -60,12 +60,12 @@
#ftp 设置
ftp:
  sqlConnect: jdbc:mysql://106.225.193.35:3306/zhbaw-test?useSSL=false&useUnicode=true&characterEncoding=utf-8&zeroDateTimeBehavior=convertToNull&transformedBitIsBoolean=true&serverTimezone=GMT%2B8&nullCatalogMeansCurrent=true&allowPublicKeyRetrieval=true
  ftpHost: 117.40.91.118
  ftpHost: 10.141.11.2
  ftpPort: 21
  ftpUserName: zhbain
  ftpPassword: zhbain@123
  ftpPath: anbao/
  localPath: E:\\anbao\\
  ftpUserName: ewm_lg
  ftpPassword: ewm_lg_20230920
  ftpPath: hotel/
  localPath: E:\\hotel\\
  ip: http://61.131.136.25:2081
  jsonUrl: E:\\anbao\\
  username: root