You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
224 lines
7.9 KiB
Java
224 lines
7.9 KiB
Java
/*
|
|
*
|
|
* * | Licensed 未经许可不能去掉「OPENIITA」相关版权
|
|
* * +----------------------------------------------------------------------
|
|
* * | Author: xw2sy@163.com
|
|
* * +----------------------------------------------------------------------
|
|
*
|
|
* Copyright [2024] [OPENIITA]
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
* /
|
|
*/
|
|
|
|
package cc.iotkit.web.controller;
|
|
|
|
import cc.iotkit.common.api.Request;
|
|
import cc.iotkit.common.constant.Constants;
|
|
import cc.iotkit.common.constant.GlobalConstants;
|
|
import cc.iotkit.common.exception.BizException;
|
|
import cc.iotkit.common.redis.utils.RedisUtils;
|
|
import cc.iotkit.common.utils.StreamUtils;
|
|
import cc.iotkit.common.utils.StringUtils;
|
|
import cc.iotkit.data.manager.IDeviceInfoData;
|
|
import cc.iotkit.system.dto.LoginBody;
|
|
import cc.iotkit.system.dto.RegisterBody;
|
|
import cc.iotkit.system.dto.bo.SysTenantBo;
|
|
import cc.iotkit.system.dto.bo.XcxLoginBo;
|
|
import cc.iotkit.system.dto.vo.SysTenantVo;
|
|
import cc.iotkit.system.service.ISysConfigService;
|
|
import cc.iotkit.system.service.ISysTenantService;
|
|
import cc.iotkit.web.domain.vo.LoginTenantVo;
|
|
import cc.iotkit.web.domain.vo.LoginVo;
|
|
import cc.iotkit.web.domain.vo.TenantListVo;
|
|
import cc.iotkit.web.service.SysLoginService;
|
|
import cc.iotkit.web.service.SysRegisterService;
|
|
import cn.dev33.satoken.annotation.SaIgnore;
|
|
import cn.hutool.core.collection.CollUtil;
|
|
import cn.hutool.core.util.RandomUtil;
|
|
import io.swagger.annotations.Api;
|
|
import io.swagger.annotations.ApiOperation;
|
|
import lombok.RequiredArgsConstructor;
|
|
import org.dromara.sms4j.api.SmsBlend;
|
|
import org.dromara.sms4j.api.entity.SmsResponse;
|
|
import org.dromara.sms4j.core.factory.SmsFactory;
|
|
import org.dromara.sms4j.provider.enumerate.SupplierType;
|
|
import org.springframework.validation.annotation.Validated;
|
|
import org.springframework.web.bind.annotation.*;
|
|
|
|
import javax.servlet.http.HttpServletRequest;
|
|
import java.net.URL;
|
|
import java.time.Duration;
|
|
import java.util.LinkedHashMap;
|
|
import java.util.List;
|
|
import java.util.Map;
|
|
import java.util.stream.Collectors;
|
|
|
|
/**
|
|
* 认证
|
|
*
|
|
* @author Lion Li
|
|
*/
|
|
@Api(tags = "认证")
|
|
@SaIgnore
|
|
@Validated
|
|
@RequiredArgsConstructor
|
|
@RestController
|
|
@RequestMapping("/auth")
|
|
public class AuthController {
|
|
|
|
private final SysLoginService loginService;
|
|
private final SysRegisterService registerService;
|
|
private final ISysConfigService configService;
|
|
private final ISysTenantService tenantService;
|
|
|
|
/**
|
|
* 登录方法
|
|
*
|
|
* @param body 登录信息
|
|
* @return 结果
|
|
*/
|
|
@ApiOperation("登录")
|
|
@PostMapping("/login")
|
|
public LoginVo login(@Validated @RequestBody Request<LoginBody> body) {
|
|
LoginBody loginBody = body.getData();
|
|
LoginVo loginVo = new LoginVo();
|
|
// 生成令牌
|
|
String token = loginService.login(
|
|
loginBody.getTenantId(),
|
|
loginBody.getUsername(), loginBody.getPassword(),
|
|
loginBody.getCode(), loginBody.getUuid());
|
|
loginVo.setToken(token);
|
|
return loginVo;
|
|
}
|
|
/**
|
|
* 邮箱验证码
|
|
*
|
|
* @param email 邮箱
|
|
*/
|
|
// @RateLimiter(key = "#email", time = 60, count = 1)
|
|
/* @GetMapping("/resource/email/code")
|
|
public R<Void> emailCode(@NotBlank(message = "{user.email.not.blank}") String email) {
|
|
if (!mailProperties.getEnabled()) {
|
|
return R.fail("当前系统没有开启邮箱功能！");
|
|
}
|
|
String key = GlobalConstants.CAPTCHA_CODE_KEY + email;
|
|
String code = RandomUtil.randomNumbers(4);
|
|
RedisUtils.setCacheObject(key, code, Duration.ofMinutes(Constants.CAPTCHA_EXPIRATION));
|
|
try {
|
|
MailUtils.sendText(email, "登录验证码", "您本次验证码为：" + code + "，有效性为" + Constants.CAPTCHA_EXPIRATION + "分钟，请尽快填写。");
|
|
} catch (Exception e) {
|
|
log.error("验证码短信发送异常 => {}", e.getMessage());
|
|
return R.fail(e.getMessage());
|
|
}
|
|
return R.ok();
|
|
}*/
|
|
|
|
/**
|
|
* 短信验证码
|
|
*
|
|
* @param body 用户手机号
|
|
*/
|
|
// @RateLimiter(key = "#phonenumber", time = 60, count = 1)
|
|
@PostMapping("/smscode")
|
|
public boolean smsCode(@Validated @RequestBody Request<String> body) {
|
|
String key = GlobalConstants.CAPTCHA_CODE_KEY + body.getData();
|
|
String code = RandomUtil.randomNumbers(4);
|
|
RedisUtils.setCacheObject(key, code, Duration.ofMinutes(Constants.CAPTCHA_EXPIRATION));
|
|
// 验证码模板id 自行处理 (查数据库或写死均可)
|
|
String templateId = "SMS_235815708";
|
|
LinkedHashMap<String, String> map = new LinkedHashMap<>(1);
|
|
map.put("code", code);
|
|
SmsBlend smsBlend = SmsFactory.createSmsBlend(SupplierType.ALIBABA);
|
|
SmsResponse smsResponse = smsBlend.sendMessage(body.getData(), templateId, map);
|
|
System.out.println(smsResponse);
|
|
if (!"OK".equals(smsResponse.getCode())) {
|
|
return false;
|
|
}
|
|
return true;
|
|
}
|
|
/**
|
|
* 小程序登录(示例)
|
|
*
|
|
* @param body 小程序appid,code
|
|
* @return 结果
|
|
*/
|
|
@ApiOperation("小程序登录")
|
|
@PostMapping("/xcxLogin")
|
|
public LoginVo xcxLogin(@Validated @RequestBody Request<XcxLoginBo> body) {
|
|
LoginVo loginVo = new LoginVo();
|
|
XcxLoginBo data = body.getData();
|
|
// 生成令牌
|
|
loginVo = loginService.xcxLogin(data.getAppId(), data.getCode(),data.getUsername(),data.getPassword());
|
|
return loginVo;
|
|
}
|
|
@ApiOperation("小程序注册")
|
|
@PostMapping("/xcxRegister")
|
|
public LoginVo xcxRegister(@Validated @RequestBody Request<XcxLoginBo> body) {
|
|
LoginVo loginVo = new LoginVo();
|
|
XcxLoginBo data = body.getData();
|
|
// 生成令牌
|
|
String token = loginService.xcxRegister(data.getAppId(), data.getCode(),data.getUsername(),data.getPassword(),data.getSmsCode());
|
|
loginVo.setToken(token);
|
|
return loginVo;
|
|
}
|
|
|
|
/**
|
|
* 退出登录
|
|
*/
|
|
@ApiOperation("退出登录")
|
|
@PostMapping("/logout")
|
|
public void logout() {
|
|
loginService.logout();
|
|
|
|
}
|
|
|
|
/**
|
|
* 用户注册
|
|
*/
|
|
@ApiOperation("用户注册")
|
|
@PostMapping("/register")
|
|
public void register(@Validated @RequestBody RegisterBody user) {
|
|
if (!configService.selectRegisterEnabled(user.getTenantId())) {
|
|
throw new BizException("当前租户不允许注册");
|
|
}
|
|
registerService.register(user);
|
|
}
|
|
|
|
/**
|
|
* 登录页面租户下拉框
|
|
*
|
|
* @return 租户列表
|
|
*/
|
|
@ApiOperation("登录页面租户下拉框")
|
|
@PostMapping("/tenant/list")
|
|
public LoginTenantVo tenantList(HttpServletRequest request) throws Exception {
|
|
List<SysTenantVo> tenantList = tenantService.queryList(new SysTenantBo());
|
|
List<TenantListVo> voList = tenantList.stream().map(t -> TenantListVo.builder()
|
|
.tenantId(t.getTenantId())
|
|
.companyName(t.getCompanyName())
|
|
.domain(t.getDomain())
|
|
.build()).collect(Collectors.toList());
|
|
// 获取域名
|
|
String host = new URL(request.getRequestURL().toString()).getHost();
|
|
// 根据域名进行筛选
|
|
List<TenantListVo> list = StreamUtils.filter(voList, vo -> StringUtils.equals(vo.getDomain(), host));
|
|
// 返回对象
|
|
LoginTenantVo vo = new LoginTenantVo();
|
|
vo.setVoList(CollUtil.isNotEmpty(list) ? list : voList);
|
|
vo.setTenantEnabled(true);
|
|
return vo;
|
|
}
|
|
|
|
}
|