lbqms.git - Gitblit

			@@ -2,6 +2,15 @@

			import cn.dev33.satoken.annotation.SaIgnore;
			import cn.hutool.core.collection.CollUtil;
			import cn.hutool.core.util.ObjectUtil;
			import jakarta.servlet.http.HttpServletRequest;
			import jakarta.validation.constraints.NotBlank;
			import lombok.RequiredArgsConstructor;
			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 org.dromara.common.core.domain.R;
			import org.dromara.common.core.domain.model.EmailLoginBody;
			import org.dromara.common.core.domain.model.LoginBody;
			@@ -10,9 +19,13 @@
			import org.dromara.common.core.utils.MapstructUtils;
			import org.dromara.common.core.utils.StreamUtils;
			import org.dromara.common.core.utils.StringUtils;
			import org.dromara.common.social.config.properties.SocialLoginConfigProperties;
			import org.dromara.common.social.config.properties.SocialProperties;
			import org.dromara.common.social.utils.SocialUtils;
			import org.dromara.common.tenant.helper.TenantHelper;
			import org.dromara.system.domain.bo.SysTenantBo;
			import org.dromara.system.domain.vo.SysTenantVo;
			import org.dromara.system.service.ISysSocialService;
			import org.dromara.system.service.ISysConfigService;
			import org.dromara.system.service.ISysTenantService;
			import org.dromara.web.domain.vo.LoginTenantVo;
			@@ -20,9 +33,6 @@
			import org.dromara.web.domain.vo.TenantListVo;
			import org.dromara.web.service.SysLoginService;
			import org.dromara.web.service.SysRegisterService;
			import jakarta.servlet.http.HttpServletRequest;
			import jakarta.validation.constraints.NotBlank;
			import lombok.RequiredArgsConstructor;
			import org.springframework.validation.annotation.Validated;
			import org.springframework.web.bind.annotation.*;

			@@ -41,10 +51,13 @@
			@RequestMapping("/auth")
			public class AuthController {

			private final SocialProperties socialProperties;
			private final SysLoginService loginService;
			private final SysRegisterService registerService;
			private final ISysConfigService configService;
			private final ISysTenantService tenantService;
			private final ISysSocialService socialUserService;


			/**
			* 登录方法
			@@ -57,9 +70,9 @@
			LoginVo loginVo = new LoginVo();
			// 生成令牌
			String token = loginService.login(
			body.getTenantId(),
			body.getUsername(), body.getPassword(),
			body.getCode(), body.getUuid());
			body.getTenantId(),
			body.getUsername(), body.getPassword(),
			body.getCode(), body.getUuid());
			loginVo.setToken(token);
			return R.ok(loginVo);
			}
			@@ -74,7 +87,10 @@
			public R<LoginVo> smsLogin(@Validated @RequestBody SmsLoginBody body) {
			LoginVo loginVo = new LoginVo();
			// 生成令牌
			String token = loginService.smsLogin(body.getTenantId(), body.getPhonenumber(), body.getSmsCode());
			String token = loginService.smsLogin(
			body.getTenantId(),
			body.getPhonenumber(),
			body.getSmsCode());
			loginVo.setToken(token);
			return R.ok(loginVo);
			}
			@@ -89,7 +105,10 @@
			public R<LoginVo> emailLogin(@Validated @RequestBody EmailLoginBody body) {
			LoginVo loginVo = new LoginVo();
			// 生成令牌
			String token = loginService.emailLogin(body.getTenantId(), body.getEmail(), body.getEmailCode());
			String token = loginService.emailLogin(
			body.getTenantId(),
			body.getEmail(),
			body.getEmailCode());
			loginVo.setToken(token);
			return R.ok(loginVo);
			}
			@@ -108,6 +127,61 @@
			loginVo.setToken(token);
			return R.ok(loginVo);
			}


			/**
			* 认证授权
			*
			* @param source 登录来源
			* @return 结果
			*/
			@GetMapping("/binding/{source}")
			public R<String> authBinding(@PathVariable("source") String source) {
			SocialLoginConfigProperties obj = socialProperties.getType().get(source);
			if (ObjectUtil.isNull(obj)) {
			return R.fail(source + "平台账号暂不支持");
			}
			AuthRequest authRequest = SocialUtils.getAuthRequest(source,
			obj.getClientId(),
			obj.getClientSecret(),
			obj.getRedirectUri());
			String authorizeUrl = authRequest.authorize(AuthStateUtils.createState());
			return R.ok(authorizeUrl);
			}

			/**
			* 第三方登录回调业务处理
			*
			* @param source 登录来源
			* @param callback 授权响应实体
			* @return 结果
			*/
			@SuppressWarnings("unchecked")
			@GetMapping("/social-login/{source}")
			public R<String> socialLogin(@PathVariable("source") String source, AuthCallback callback) {
			SocialLoginConfigProperties obj = socialProperties.getType().get(source);
			if (ObjectUtil.isNull(obj)) {
			return R.fail(source + "平台账号暂不支持");
			}
			AuthRequest authRequest = SocialUtils.getAuthRequest(source,
			obj.getClientId(),
			obj.getClientSecret(),
			obj.getRedirectUri());
			AuthResponse<AuthUser> response = authRequest.login(callback);
			return loginService.socialLogin(source, response);
			}

			/**
			* 取消授权
			*
			* @param socialId socialId
			*/
			@DeleteMapping(value = "/unlock/{socialId}")
			public R<Void> unlockSocial(@PathVariable Long socialId) {
			Boolean rows = socialUserService.deleteWithValidById(socialId);
			return rows ? R.ok() : R.fail("取消授权失败");
			}


			/**
			* 退出登录
			@@ -140,9 +214,17 @@
			List<SysTenantVo> tenantList = tenantService.queryList(new SysTenantBo());
			List<TenantListVo> voList = MapstructUtils.convert(tenantList, TenantListVo.class);
			// 获取域名
			String host = new URL(request.getRequestURL().toString()).getHost();
			String host;
			String referer = request.getHeader("referer");
			if (StringUtils.isNotBlank(referer)) {
			// 这里从referer中取值是为了本地使用hosts添加虚拟域名，方便本地环境调试
			host = referer.split("//")[1].split("/")[0];
			} else {
			host = new URL(request.getRequestURL().toString()).getHost();
			}
			// 根据域名进行筛选
			List<TenantListVo> list = StreamUtils.filter(voList, vo -> StringUtils.equals(vo.getDomain(), host));
			List<TenantListVo> list = StreamUtils.filter(voList, vo ->
			StringUtils.equals(vo.getDomain(), host));
			// 返回对象
			LoginTenantVo vo = new LoginTenantVo();
			vo.setVoList(CollUtil.isNotEmpty(list) ? list : voList);