lbqms.git - Gitblit

			@@ -1,18 +1,29 @@
			package com.ruoyi.web.controller;

			import cn.dev33.satoken.annotation.SaIgnore;
			import com.ruoyi.common.constant.Constants;
			import cn.hutool.core.collection.CollUtil;
			import com.ruoyi.common.core.domain.R;
			import com.ruoyi.system.domain.SysMenu;
			import com.ruoyi.system.domain.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.common.core.utils.MapstructUtils;
			import com.ruoyi.common.core.utils.StringUtils;
			import com.ruoyi.common.satoken.utils.LoginHelper;
			import com.ruoyi.common.tenant.helper.TenantHelper;
			import com.ruoyi.system.domain.SysMenu;
			import com.ruoyi.system.domain.bo.SysTenantBo;
			import com.ruoyi.system.domain.vo.RouterVo;
			import com.ruoyi.system.domain.vo.SysTenantVo;
			import com.ruoyi.system.domain.vo.SysUserVo;
			import com.ruoyi.system.service.ISysMenuService;
			import com.ruoyi.system.service.ISysTenantService;
			import com.ruoyi.system.service.ISysUserService;
			import com.ruoyi.system.service.SysLoginService;
			import com.ruoyi.web.domain.vo.LoginVo;
			import com.ruoyi.web.domain.vo.TenantListVo;
			import com.ruoyi.web.domain.vo.UserInfoVo;
			import com.ruoyi.web.service.SysLoginService;
			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.GetMapping;
			@@ -20,10 +31,8 @@
			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.net.URL;
			import java.util.List;
			import java.util.Map;

			/**
			* 登录验证
			@@ -38,38 +47,41 @@
			private final SysLoginService loginService;
			private final ISysMenuService menuService;
			private final ISysUserService userService;
			private final ISysTenantService tenantService;

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

			/**
			* 短信登录(示例)
			*
			* @param smsLoginBody 登录信息
			* @param body 登录信息
			* @return 结果
			*/
			@SaIgnore
			@PostMapping("/smsLogin")
			public R<Map<String, Object>> smsLogin(@Validated @RequestBody SmsLoginBody smsLoginBody) {
			Map<String, Object> ajax = new HashMap<>();
			public R<LoginVo> smsLogin(@Validated @RequestBody SmsLoginBody body) {
			LoginVo loginVo = new LoginVo();
			// 生成令牌
			String token = loginService.smsLogin(smsLoginBody.getPhonenumber(), smsLoginBody.getSmsCode());
			ajax.put(Constants.TOKEN, token);
			return R.ok(ajax);
			String token = loginService.smsLogin(body.getTenantId(), body.getPhonenumber(), body.getSmsCode());
			loginVo.setToken(token);
			return R.ok(loginVo);
			}

			/**
			@@ -80,12 +92,12 @@
			*/
			@SaIgnore
			@PostMapping("/xcxLogin")
			public R<Map<String, Object>> xcxLogin(@NotBlank(message = "{xcx.code.not.blank}") String xcxCode) {
			Map<String, Object> ajax = new HashMap<>();
			public R<LoginVo> xcxLogin(@NotBlank(message = "{xcx.code.not.blank}") String xcxCode) {
			LoginVo loginVo = new LoginVo();
			// 生成令牌
			String token = loginService.xcxLogin(xcxCode);
			ajax.put(Constants.TOKEN, token);
			return R.ok(ajax);
			loginVo.setToken(token);
			return R.ok(loginVo);
			}

			/**
			@@ -99,19 +111,40 @@
			}

			/**
			* 登录页面租户下拉框
			*
			* @return 租户列表
			*/
			@SaIgnore
			@GetMapping("/tenant/list")
			public R<List<TenantListVo>> tenantList(HttpServletRequest request) throws Exception {
			List<SysTenantVo> tenantList = tenantService.queryList(new SysTenantBo());
			List<TenantListVo> voList = MapstructUtils.convert(tenantList, TenantListVo.class);
			// 获取域名
			String host = new URL(request.getRequestURL().toString()).getHost();
			// 根据域名进行筛选
			List<TenantListVo> list = voList.stream().filter(vo -> StringUtils.equals(vo.getDomain(), host)).toList();
			return R.ok(CollUtil.isNotEmpty(list) ? list : voList);
			}

			/**
			* 获取用户信息
			*
			* @return 用户信息
			*/
			@GetMapping("getInfo")
			public R<Map<String, Object>> getInfo() {
			public R<UserInfoVo> getInfo() {
			UserInfoVo userInfoVo = new UserInfoVo();
			LoginUser loginUser = LoginHelper.getLoginUser();
			SysUser user = userService.selectUserById(loginUser.getUserId());
			Map<String, Object> ajax = new HashMap<>();
			ajax.put("user", user);
			ajax.put("roles", loginUser.getRolePermission());
			ajax.put("permissions", loginUser.getMenuPermission());
			return R.ok(ajax);
			if (TenantHelper.isEnable() && LoginHelper.isSuperAdmin()) {
			// 超级管理员如果重新加载用户信息需清除动态租户
			TenantHelper.clearDynamic();
			}
			SysUserVo user = userService.selectUserById(loginUser.getUserId());
			userInfoVo.setUser(user);
			userInfoVo.setPermissions(loginUser.getMenuPermission());
			userInfoVo.setRoles(loginUser.getRolePermission());
			return R.ok(userInfoVo);
			}

			/**
			@@ -121,8 +154,7 @@
			*/
			@GetMapping("getRouters")
			public R<List<RouterVo>> getRouters() {
			Long userId = LoginHelper.getUserId();
			List<SysMenu> menus = menuService.selectMenuTreeByUserId(userId);
			List<SysMenu> menus = menuService.selectMenuTreeByUserId(LoginHelper.getUserId());
			return R.ok(menuService.buildMenus(menus));
			}
			}