| | |
| | | package com.ruoyi.web.controller.system; |
| | | |
| | | import cn.dev33.satoken.exception.NotLoginException; |
| | | import cn.dev33.satoken.stp.StpUtil; |
| | | import cn.dev33.satoken.annotation.SaIgnore; |
| | | import com.ruoyi.common.constant.Constants; |
| | | import com.ruoyi.common.core.domain.R; |
| | | import com.ruoyi.common.core.domain.entity.SysMenu; |
| | | import com.ruoyi.common.core.domain.entity.SysUser; |
| | | import com.ruoyi.common.core.domain.model.LoginBody; |
| | | import com.ruoyi.common.core.domain.model.LoginUser; |
| | | import com.ruoyi.common.core.domain.model.SmsLoginBody; |
| | | import com.ruoyi.common.helper.LoginHelper; |
| | | import com.ruoyi.system.domain.vo.RouterVo; |
| | | import com.ruoyi.system.service.ISysMenuService; |
| | | import com.ruoyi.system.service.ISysUserService; |
| | | import com.ruoyi.system.service.SysLoginService; |
| | | import com.ruoyi.system.service.SysPermissionService; |
| | | import io.swagger.annotations.Api; |
| | | import io.swagger.annotations.ApiOperation; |
| | | import lombok.RequiredArgsConstructor; |
| | | import org.springframework.validation.annotation.Validated; |
| | | import org.springframework.web.bind.annotation.GetMapping; |
| | |
| | | import org.springframework.web.bind.annotation.RequestBody; |
| | | import org.springframework.web.bind.annotation.RestController; |
| | | |
| | | import jakarta.validation.constraints.NotBlank; |
| | | import java.util.HashMap; |
| | | import java.util.List; |
| | | import java.util.Map; |
| | | import java.util.Set; |
| | | |
| | | /** |
| | | * 登录验证 |
| | |
| | | * @author Lion Li |
| | | */ |
| | | @Validated |
| | | @Api(value = "登录验证控制器", tags = {"登录验证管理"}) |
| | | @RequiredArgsConstructor |
| | | @RestController |
| | | public class SysLoginController { |
| | |
| | | private final SysLoginService loginService; |
| | | private final ISysMenuService menuService; |
| | | private final ISysUserService userService; |
| | | private final SysPermissionService permissionService; |
| | | |
| | | /** |
| | | * 登录方法 |
| | |
| | | * @param loginBody 登录信息 |
| | | * @return 结果 |
| | | */ |
| | | @ApiOperation("登录方法") |
| | | @SaIgnore |
| | | @PostMapping("/login") |
| | | public R<Map<String, Object>> login(@Validated @RequestBody LoginBody loginBody) { |
| | | Map<String, Object> ajax = new HashMap<>(); |
| | |
| | | return R.ok(ajax); |
| | | } |
| | | |
| | | @ApiOperation("登出方法") |
| | | /** |
| | | * 短信登录(示例) |
| | | * |
| | | * @param smsLoginBody 登录信息 |
| | | * @return 结果 |
| | | */ |
| | | @SaIgnore |
| | | @PostMapping("/smsLogin") |
| | | public R<Map<String, Object>> smsLogin(@Validated @RequestBody SmsLoginBody smsLoginBody) { |
| | | Map<String, Object> ajax = new HashMap<>(); |
| | | // 生成令牌 |
| | | String token = loginService.smsLogin(smsLoginBody.getPhonenumber(), smsLoginBody.getSmsCode()); |
| | | ajax.put(Constants.TOKEN, token); |
| | | return R.ok(ajax); |
| | | } |
| | | |
| | | /** |
| | | * 小程序登录(示例) |
| | | * |
| | | * @param xcxCode 小程序code |
| | | * @return 结果 |
| | | */ |
| | | @SaIgnore |
| | | @PostMapping("/xcxLogin") |
| | | public R<Map<String, Object>> xcxLogin(@NotBlank(message = "{xcx.code.not.blank}") String xcxCode) { |
| | | Map<String, Object> ajax = new HashMap<>(); |
| | | // 生成令牌 |
| | | String token = loginService.xcxLogin(xcxCode); |
| | | ajax.put(Constants.TOKEN, token); |
| | | return R.ok(ajax); |
| | | } |
| | | |
| | | /** |
| | | * 退出登录 |
| | | */ |
| | | @SaIgnore |
| | | @PostMapping("/logout") |
| | | public R<Void> logout() { |
| | | try { |
| | | StpUtil.logout(); |
| | | } catch (NotLoginException e) { |
| | | } |
| | | loginService.logout(); |
| | | return R.ok("退出成功"); |
| | | } |
| | | |
| | |
| | | * |
| | | * @return 用户信息 |
| | | */ |
| | | @ApiOperation("获取用户信息") |
| | | @GetMapping("getInfo") |
| | | public R<Map<String, Object>> getInfo() { |
| | | SysUser user = userService.selectUserById(LoginHelper.getUserId()); |
| | | // 角色集合 |
| | | Set<String> roles = permissionService.getRolePermission(user); |
| | | // 权限集合 |
| | | Set<String> permissions = permissionService.getMenuPermission(user); |
| | | LoginUser loginUser = LoginHelper.getLoginUser(); |
| | | SysUser user = userService.selectUserById(loginUser.getUserId()); |
| | | Map<String, Object> ajax = new HashMap<>(); |
| | | ajax.put("user", user); |
| | | ajax.put("roles", roles); |
| | | ajax.put("permissions", permissions); |
| | | ajax.put("roles", loginUser.getRolePermission()); |
| | | ajax.put("permissions", loginUser.getMenuPermission()); |
| | | return R.ok(ajax); |
| | | } |
| | | |
| | |
| | | * |
| | | * @return 路由信息 |
| | | */ |
| | | @ApiOperation("获取路由信息") |
| | | @GetMapping("getRouters") |
| | | public R<List<RouterVo>> getRouters() { |
| | | Long userId = LoginHelper.getUserId(); |