update 优化 三方登录不同域名问题 采用新方案

疯狂的狮子Li

2024-06-19 7c82b4c370aa63c630e7b0cbbef5d2757d46089d

 ruoyi-admin/src/main/java/org/dromara/web/controller/AuthController.java

@@ -1,6 +1,7 @@
package org.dromara.web.controller;

import cn.dev33.satoken.annotation.SaIgnore;
import cn.hutool.core.codec.Base64;
import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.util.ObjectUtil;
import jakarta.servlet.http.HttpServletRequest;
@@ -42,7 +43,10 @@
import org.springframework.web.bind.annotation.*;

import java.net.URL;
import java.nio.charset.StandardCharsets;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.concurrent.ScheduledExecutorService;
import java.util.concurrent.TimeUnit;

@@ -91,7 +95,7 @@
            return R.fail(MessageUtils.message("auth.grant.type.blocked"));
        }
        // 校验租户
        SysTenantVo tenant = loginService.checkTenant(loginBody.getTenantId());
        loginService.checkTenant(loginBody.getTenantId());
        // 登录
        LoginVo loginVo = IAuthStrategy.login(body, client, grantType);

@@ -102,10 +106,6 @@
            dto.setSessionKeys(List.of(userId));
            WebSocketUtils.publishMessage(dto);
        }, 3, TimeUnit.SECONDS);

        if (StringUtils.isNotBlank(tenant.getTenantId())) {
            loginVo.setDomain(tenant.getDomain());
        }
        return R.ok(loginVo);
    }

@@ -116,13 +116,18 @@
     * @return 结果
     */
    @GetMapping("/binding/{source}")
    public R<String> authBinding(@PathVariable("source") String source) {
    public R<String> authBinding(@PathVariable("source") String source,
                                 @RequestParam String tenantId, @RequestParam String domain) {
        SocialLoginConfigProperties obj = socialProperties.getType().get(source);
        if (ObjectUtil.isNull(obj)) {
            return R.fail(source + "平台账号暂不支持");
        }
        AuthRequest authRequest = SocialUtils.getAuthRequest(source, socialProperties);
        String authorizeUrl = authRequest.authorize(AuthStateUtils.createState());
        Map<String, String> map = new HashMap<>();
        map.put("tenantId", tenantId);
        map.put("domain", domain);
        map.put("state", AuthStateUtils.createState());
        String authorizeUrl = authRequest.authorize(Base64.encode(JsonUtils.toJsonString(map), StandardCharsets.UTF_8));
        return R.ok("操作成功", authorizeUrl);
    }

@@ -133,7 +138,7 @@
     * @return 结果
     */
    @PostMapping("/social/callback")
    public R<LoginVo> socialCallback(@RequestBody SocialLoginBody loginBody) {
    public R<Void> socialCallback(@RequestBody SocialLoginBody loginBody) {
        // 获取第三方登录信息
        AuthResponse<AuthUser> response = SocialUtils.loginAuth(
                loginBody.getSource(), loginBody.getSocialCode(),
@@ -144,13 +149,7 @@
            return R.fail(response.getMsg());
        }
        loginService.socialRegister(authUserData);

        LoginVo loginVo = new LoginVo();
        SysTenantVo tenant = tenantService.queryByTenantId(LoginHelper.getTenantId());
        if (StringUtils.isNotBlank(tenant.getTenantId())) {
            loginVo.setDomain(tenant.getDomain());
        }
        return R.ok(loginVo);
        return R.ok();
    }