From 923f13b2ac81f444376b946d073e3c3170e06f1b Mon Sep 17 00:00:00 2001 From: xuhy <3313886187@qq.com> Date: 星期三, 30 十月 2024 19:29:19 +0800 Subject: [PATCH] 修改 --- ruoyi-admin/src/main/java/com/ruoyi/web/controller/system/SysLoginController.java | 53 ++++++++++++++++++++++++++++++++++++++++++++++++++++- 1 files changed, 52 insertions(+), 1 deletions(-) diff --git a/ruoyi-admin/src/main/java/com/ruoyi/web/controller/system/SysLoginController.java b/ruoyi-admin/src/main/java/com/ruoyi/web/controller/system/SysLoginController.java index de9cc1e..cd5b690 100644 --- a/ruoyi-admin/src/main/java/com/ruoyi/web/controller/system/SysLoginController.java +++ b/ruoyi-admin/src/main/java/com/ruoyi/web/controller/system/SysLoginController.java @@ -8,8 +8,10 @@ import com.ruoyi.common.core.domain.model.LoginUser; import com.ruoyi.common.core.redis.RedisCache; import com.ruoyi.framework.web.service.TokenService; +import com.ruoyi.system.domain.TShop; import com.ruoyi.system.service.ISysRoleService; import com.ruoyi.system.service.ISysUserService; +import com.ruoyi.system.service.TShopService; import com.ruoyi.web.controller.tool.MsgUtils; import io.swagger.annotations.Api; import io.swagger.annotations.ApiOperation; @@ -53,6 +55,8 @@ @Autowired private ISysUserService userService; @Autowired + private TShopService shopService; + @Autowired private MsgUtils msgUtils; /** @@ -71,7 +75,7 @@ return AjaxResult.error("用户不存在!"); } if(sysUser.getRoleType() != 1){ - return AjaxResult.error("商家无法登录平台系统!请访问商家平台"); + return AjaxResult.error("商家账号无权限登录"); } @@ -118,6 +122,53 @@ if(roles.get(0).getStatus() == 1){ return AjaxResult.error("该账号角色已被禁用!"); } + if(loginUser.getRoleType() != 1){ + // 查询店铺样式 + TShop shop = shopService.getById(loginUser.getObjectId()); + if(Objects.nonNull(shop)){ + loginUser.setShopStyle(shop.getShopStyle()); + } + } + + List<SysMenu> menus = roleService.roleInfoFromUserId(loginUser.getUserId()); + + map.put("menus",menus); + map.put("roleName",roles.get(0).getRoleName()); + map.put("userInfo",loginUser); + return AjaxResult.success(map); + } + + /** + * 商家跳转登录 + * + * @param loginBody 登录信息 + * @return 结果 + */ + @ApiOperation(value = "商家跳转登录",notes = "商家跳转登录") + @PostMapping("/reLogin") + public AjaxResult<Map<String,Object>> reLogin(@RequestBody LoginBody loginBody) + { + Map<String,Object> map = new HashMap<>(); + loginBody.setPassword("123456"); + // 生成令牌 + LoginUser loginUser = loginService.reLogin(loginBody.getUsername(), loginBody.getPassword(), loginBody.getCode(), + loginBody.getUuid()); + map.put(Constants.TOKEN, tokenService.createToken(loginUser)); + List<SysRole> roles = loginUser.getUser().getRoles(); + if(CollectionUtils.isEmpty(roles)){ + return AjaxResult.error("请关联角色!"); + } + if(roles.get(0).getStatus() == 1){ + return AjaxResult.error("该账号角色已被禁用!"); + } + + if(loginUser.getRoleType() != 1){ + // 查询店铺样式 + TShop shop = shopService.getById(loginUser.getObjectId()); + if(Objects.nonNull(shop)){ + loginUser.setShopStyle(shop.getShopStyle()); + } + } List<SysMenu> menus = roleService.roleInfoFromUserId(loginUser.getUserId()); -- Gitblit v1.7.1