97 lines
3.4 KiB
Java
97 lines
3.4 KiB
Java
|
package net.maku.system.controller;
|
|||
|
|
|
|||
|
|
import io.swagger.v3.oas.annotations.Operation;
|
|||
|
|
import io.swagger.v3.oas.annotations.tags.Tag;
|
|||
|
|
import jakarta.servlet.http.HttpServletResponse;
|
|||
|
|
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 me.zhyd.oauth.utils.AuthStateUtils;
|
|||
|
|
import net.maku.framework.common.utils.Result;
|
|||
|
|
import net.maku.framework.operatelog.annotations.OperateLog;
|
|||
|
|
import net.maku.framework.operatelog.enums.OperateTypeEnum;
|
|||
|
|
import net.maku.framework.security.user.SecurityUser;
|
|||
|
|
import net.maku.system.service.SysThirdLoginConfigService;
|
|||
|
|
import net.maku.system.service.SysThirdLoginService;
|
|||
|
|
import net.maku.system.vo.SysThirdCallbackVO;
|
|||
|
|
import net.maku.system.vo.SysThirdLoginVO;
|
|||
|
|
import org.springframework.web.bind.annotation.*;
|
|||
|
|
import org.springframework.web.servlet.ModelAndView;
|
|||
|
|
|
|||
|
|
import java.io.IOException;
|
|||
|
|
import java.util.HashMap;
|
|||
|
|
import java.util.List;
|
|||
|
|
import java.util.Map;
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 第三方账号登录
|
|||
|
|
*
|
|||
|
|
* @author 阿沐 babamu@126.com
|
|||
|
|
* <a href="https://maku.net">MAKU</a>
|
|||
|
|
*/
|
|||
|
|
@RestController
|
|||
|
|
@RequestMapping("sys/third")
|
|||
|
|
@Tag(name = "第三方账号")
|
|||
|
|
@AllArgsConstructor
|
|||
|
|
public class SysThirdLoginController {
|
|||
|
|
private final SysThirdLoginService sysThirdLoginService;
|
|||
|
|
private final SysThirdLoginConfigService sysThirdLoginConfigService;
|
|||
|
|
|
|||
|
|
@GetMapping("list")
|
|||
|
|
@Operation(summary = "列表")
|
|||
|
|
public Result<List<SysThirdLoginVO>> list() {
|
|||
|
|
List<SysThirdLoginVO> list = sysThirdLoginService.listByUserId(SecurityUser.getUserId());
|
|||
|
|
|
|||
|
|
return Result.ok(list);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
@RequestMapping("render/{source}")
|
|||
|
|
public void renderAuth(@PathVariable("source") String source, HttpServletResponse response) throws IOException {
|
|||
|
|
AuthRequest authRequest = sysThirdLoginConfigService.getAuthRequest(source);
|
|||
|
|
String authorizeUrl = authRequest.authorize(AuthStateUtils.createState());
|
|||
|
|
response.sendRedirect(authorizeUrl);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
@RequestMapping("/callback/{source}")
|
|||
|
|
public ModelAndView login(@PathVariable("source") String source, AuthCallback callback) {
|
|||
|
|
Map<String, Object> map = new HashMap<>();
|
|||
|
|
map.put("openType", source);
|
|||
|
|
map.put("state", callback.getState());
|
|||
|
|
map.put("code", callback.getCode());
|
|||
|
|
|
|||
|
|
return new ModelAndView("third_login", map);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
@PostMapping("bind")
|
|||
|
|
@Operation(summary = "绑定")
|
|||
|
|
@OperateLog(type = OperateTypeEnum.INSERT)
|
|||
|
|
public Result<String> bind(@RequestBody SysThirdCallbackVO vo) {
|
|||
|
|
AuthRequest authRequest = sysThirdLoginConfigService.getAuthRequest(vo.getOpenType());
|
|||
|
|
AuthCallback callback = AuthCallback.builder().code(vo.getCode()).state(vo.getState()).build();
|
|||
|
|
|
|||
|
|
// 根据code，获取用户信息
|
|||
|
|
AuthResponse<AuthUser> response = authRequest.login(callback);
|
|||
|
|
|
|||
|
|
// 判断是否成功
|
|||
|
|
if (!response.ok()) {
|
|||
|
|
throw new RuntimeException("第三方登录失败");
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 绑定用户信息
|
|||
|
|
sysThirdLoginService.bind(SecurityUser.getUserId(), vo.getOpenType(), response.getData());
|
|||
|
|
|
|||
|
|
return Result.ok();
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
@PutMapping("unbind/{openType}")
|
|||
|
|
@Operation(summary = "解绑")
|
|||
|
|
@OperateLog(type = OperateTypeEnum.UPDATE)
|
|||
|
|
public Result<String> unBind(@PathVariable("openType") String openType) {
|
|||
|
|
sysThirdLoginService.unBind(SecurityUser.getUserId(), openType);
|
|||
|
|
|
|||
|
|
return Result.ok();
|
|||
|
|
}
|
|||
|
|
}
|