| | |
| | | import com.ruoyi.framework.security.service.SysLoginService; |
| | | import com.ruoyi.framework.security.service.SysPermissionService; |
| | | import com.ruoyi.framework.security.service.TokenService; |
| | | import com.ruoyi.framework.web.controller.BaseController; |
| | | import com.ruoyi.framework.web.domain.R; |
| | | import com.ruoyi.framework.web.domain.AjaxResult; |
| | | import com.ruoyi.project.system.domain.SysDept; |
| | | import com.ruoyi.project.system.domain.SysMenu; |
| | | import com.ruoyi.project.system.domain.SysUser; |
| | |
| | | import com.ruoyi.project.system.service.ISysUserDeptService; |
| | | import com.ruoyi.project.system.service.ISysUserService; |
| | | import lombok.AllArgsConstructor; |
| | | import org.springframework.beans.factory.annotation.Autowired; |
| | | import org.springframework.util.ObjectUtils; |
| | | import org.springframework.web.bind.annotation.GetMapping; |
| | | import org.springframework.web.bind.annotation.PostMapping; |
| | |
| | | import org.springframework.web.bind.annotation.RestController; |
| | | |
| | | import java.util.ArrayList; |
| | | import java.util.HashMap; |
| | | import java.util.List; |
| | | import java.util.Map; |
| | | import java.util.Set; |
| | |
| | | */ |
| | | @RestController |
| | | @AllArgsConstructor |
| | | public class SysLoginController extends BaseController |
| | | public class SysLoginController |
| | | { |
| | | private SysLoginService loginService; |
| | | private ISysMenuService menuService; |
| | |
| | | * @return 结果 |
| | | */ |
| | | @PostMapping("/login") |
| | | public R<?> login(@RequestBody LoginBody loginBody) |
| | | public AjaxResult login(@RequestBody LoginBody loginBody) |
| | | { |
| | | AjaxResult ajax = AjaxResult.success(); |
| | | // 生成令牌 |
| | | String token = loginService.login(loginBody.getUsername(), loginBody.getPassword(), loginBody.getCode(), |
| | | loginBody.getUuid()); |
| | | Map<String, Object> map = new HashMap<>(); |
| | | map.put(Constants.TOKEN, token); |
| | | return R.ok(map); |
| | | ajax.put(Constants.TOKEN, token); |
| | | return ajax; |
| | | } |
| | | |
| | | /** |
| | |
| | | * @return 用户信息 |
| | | */ |
| | | @GetMapping("/getInfo") |
| | | public R<?> getInfo() |
| | | public AjaxResult getInfo() |
| | | { |
| | | LoginUser loginUser = SecurityUtils.getLoginUser(); |
| | | SysUser user = loginUser.getUser(); |
| | |
| | | loginUser.setPermissions(permissions); |
| | | tokenService.refreshToken(loginUser); |
| | | } |
| | | Map<String, Object> map = new HashMap<>(); |
| | | map.put("user", user); |
| | | map.put("aiEnabled", loginUser.getAiEnabled()); |
| | | map.put("roles", roles); |
| | | map.put("permissions", permissions); |
| | | return R.ok(map); |
| | | AjaxResult ajax = AjaxResult.success(); |
| | | ajax.put("user", user); |
| | | ajax.put("aiEnabled", loginUser.getAiEnabled()); |
| | | ajax.put("roles", roles); |
| | | ajax.put("permissions", permissions); |
| | | return ajax; |
| | | } |
| | | |
| | | /** |
| | |
| | | * @return 路由信息 |
| | | */ |
| | | @GetMapping("getRouters") |
| | | public R<?> getRouters() |
| | | public AjaxResult getRouters() |
| | | { |
| | | Long userId = SecurityUtils.getUserId(); |
| | | List<SysMenu> menus = menuService.selectMenuTreeByUserId(userId); |
| | | return R.ok(menuService.buildMenus(menus)); |
| | | return AjaxResult.success(menuService.buildMenus(menus)); |
| | | } |
| | | |
| | | @PostMapping("/loginCheck") |
| | | public R<?> loginCheck(@RequestBody LoginBody loginBody) |
| | | public AjaxResult loginCheck(@RequestBody LoginBody loginBody) |
| | | { |
| | | try { |
| | | Long userId = loginService.loginCheck(loginBody.getUsername(), loginBody.getPassword()); |
| | | return R.ok(userId); |
| | | return AjaxResult.success(userId); |
| | | }catch (Exception e) { |
| | | return R.fail(e.getMessage()); |
| | | return AjaxResult.error(e.getMessage()); |
| | | } |
| | | } |
| | | |
| | | @GetMapping("/userLoginFacotryList") |
| | | public R<?> userLoginFacotryList(SysUserDeptVo sysUserDeptVo){ |
| | | public AjaxResult userLoginFacotryList(SysUserDeptVo sysUserDeptVo){ |
| | | List<SysUserDeptVo> sysUserDeptVoList = userDeptService.userLoginFacotryList(sysUserDeptVo); |
| | | Map<Long, SysUserDeptVo> map = sysUserDeptVoList.stream() |
| | | .collect(Collectors.toMap( |
| | |
| | | (existing, replacement) -> existing // 如果重复,保留第一个 |
| | | )); |
| | | List<SysUserDeptVo> uniqueList = new ArrayList<>(map.values()); |
| | | return R.ok(uniqueList); |
| | | return AjaxResult.success(uniqueList); |
| | | } |
| | | |
| | | /** |
| | |
| | | * @return 结果 |
| | | */ |
| | | @PostMapping("/loginCheckFactory") |
| | | public R<?> loginCheckFactory(@RequestBody LoginBody loginBody) |
| | | public AjaxResult loginCheckFactory(@RequestBody LoginBody loginBody) |
| | | { |
| | | AjaxResult ajax = AjaxResult.success(); |
| | | // 生成令牌 |
| | | String token = loginService.loginCheckFactory(loginBody.getUsername(), loginBody.getPassword(),loginBody.getFactoryId()); |
| | | Map<String, Object> map = new HashMap<>(); |
| | | map.put(Constants.TOKEN, token); |
| | | return R.ok(map); |
| | | ajax.put(Constants.TOKEN, token); |
| | | return ajax; |
| | | } |
| | | } |