lbqms.git - Gitblit

			@@ -1,94 +1,128 @@
			package com.ruoyi.web.controller.system;

			import java.util.List;
			import java.util.Set;
			import org.springframework.beans.factory.annotation.Autowired;
			import org.springframework.web.bind.annotation.GetMapping;
			import org.springframework.web.bind.annotation.PostMapping;
			import org.springframework.web.bind.annotation.RequestBody;
			import org.springframework.web.bind.annotation.RestController;
			import cn.dev33.satoken.annotation.SaIgnore;
			import com.ruoyi.common.constant.Constants;
			import com.ruoyi.common.core.domain.AjaxResult;
			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.utils.ServletUtils;
			import com.ruoyi.framework.web.service.SysLoginService;
			import com.ruoyi.framework.web.service.SysPermissionService;
			import com.ruoyi.framework.web.service.TokenService;
			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 lombok.RequiredArgsConstructor;
			import org.springframework.validation.annotation.Validated;
			import org.springframework.web.bind.annotation.GetMapping;
			import org.springframework.web.bind.annotation.PostMapping;
			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;

			/**
			* 登录验证
			*
			* @author ruoyi
			*
			* @author Lion Li
			*/
			@Validated
			@RequiredArgsConstructor
			@RestController
			public class SysLoginController
			{
			@Autowired
			private SysLoginService loginService;
			public class SysLoginController {

			@Autowired
			private ISysMenuService menuService;

			@Autowired
			private SysPermissionService permissionService;

			@Autowired
			private TokenService tokenService;
			private final SysLoginService loginService;
			private final ISysMenuService menuService;
			private final ISysUserService userService;

			/**
			* 登录方法
			*
			*
			* @param loginBody 登录信息
			* @return 结果
			*/
			@SaIgnore
			@PostMapping("/login")
			public AjaxResult login(@RequestBody LoginBody loginBody)
			{
			AjaxResult ajax = AjaxResult.success();
			public R<Map<String, Object>> login(@Validated @RequestBody LoginBody loginBody) {
			Map<String, Object> ajax = new HashMap<>();
			// 生成令牌
			String token = loginService.login(loginBody.getUsername(), loginBody.getPassword(), loginBody.getCode(),
			loginBody.getUuid());
			loginBody.getUuid());
			ajax.put(Constants.TOKEN, token);
			return ajax;
			return R.ok(ajax);
			}

			/**
			* 短信登录(示例)
			*
			* @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() {
			loginService.logout();
			return R.ok("退出成功");
			}

			/**
			* 获取用户信息
			*
			*
			* @return 用户信息
			*/
			@GetMapping("getInfo")
			public AjaxResult getInfo()
			{
			LoginUser loginUser = tokenService.getLoginUser(ServletUtils.getRequest());
			SysUser user = loginUser.getUser();
			// 角色集合
			Set<String> roles = permissionService.getRolePermission(user);
			// 权限集合
			Set<String> permissions = permissionService.getMenuPermission(user);
			AjaxResult ajax = AjaxResult.success();
			public R<Map<String, Object>> getInfo() {
			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);
			return ajax;
			ajax.put("roles", loginUser.getRolePermission());
			ajax.put("permissions", loginUser.getMenuPermission());
			return R.ok(ajax);
			}

			/**
			* 获取路由信息
			*
			*
			* @return 路由信息
			*/
			@GetMapping("getRouters")
			public AjaxResult getRouters()
			{
			LoginUser loginUser = tokenService.getLoginUser(ServletUtils.getRequest());
			// 用户信息
			SysUser user = loginUser.getUser();
			List<SysMenu> menus = menuService.selectMenuTreeByUserId(user.getUserId());
			return AjaxResult.success(menuService.buildMenus(menus));
			public R<List<RouterVo>> getRouters() {
			Long userId = LoginHelper.getUserId();
			List<SysMenu> menus = menuService.selectMenuTreeByUserId(userId);
			return R.ok(menuService.buildMenus(menus));
			}
			}