137 lines
4.8 KiB
Java
137 lines
4.8 KiB
Java
package com.cnbm.admin.controller;
|
||
|
||
import com.cnbm.admin.annotation.LogOperation;
|
||
import com.cnbm.admin.dto.SysMenuDTO;
|
||
import com.cnbm.admin.entity.LoginUser;
|
||
import com.cnbm.admin.entity.SysUserEntity;
|
||
import com.cnbm.admin.enums.MenuTypeEnum;
|
||
import com.cnbm.admin.service.SysMenuService;
|
||
import com.cnbm.common.exception.ErrorCode;
|
||
import com.cnbm.common.utils.Result;
|
||
import com.cnbm.common.validator.AssertUtils;
|
||
import com.cnbm.common.validator.ValidatorUtils;
|
||
import com.cnbm.common.validator.group.DefaultGroup;
|
||
import io.swagger.annotations.Api;
|
||
import io.swagger.annotations.ApiImplicitParam;
|
||
import io.swagger.annotations.ApiOperation;
|
||
import org.springframework.beans.factory.annotation.Autowired;
|
||
import org.springframework.security.access.prepost.PreAuthorize;
|
||
import org.springframework.security.authentication.UsernamePasswordAuthenticationToken;
|
||
import org.springframework.security.core.context.SecurityContextHolder;
|
||
import org.springframework.web.bind.annotation.*;
|
||
|
||
import java.util.HashSet;
|
||
import java.util.List;
|
||
import java.util.Set;
|
||
|
||
/**
|
||
* @Author weihongyang
|
||
* @Date 2022/6/10 2:32 PM
|
||
* @Version 1.0
|
||
*/
|
||
@RestController
|
||
@RequestMapping("/sys/menu")
|
||
@Api(tags="菜单管理")
|
||
public class SysMenuController {
|
||
@Autowired
|
||
private SysMenuService sysMenuService;
|
||
|
||
|
||
@GetMapping("nav")
|
||
@ApiOperation("导航")
|
||
public Result<List<SysMenuDTO>> nav(){
|
||
UsernamePasswordAuthenticationToken authentication = (UsernamePasswordAuthenticationToken) SecurityContextHolder.getContext().getAuthentication();
|
||
LoginUser loginUser = (LoginUser) authentication.getPrincipal();
|
||
SysUserEntity user = loginUser.getSysUserEntity();
|
||
List<SysMenuDTO> list = sysMenuService.getUserMenuList(user, MenuTypeEnum.MENU.value());
|
||
|
||
return new Result<List<SysMenuDTO>>().ok(list);
|
||
}
|
||
|
||
@GetMapping("permissions")
|
||
@ApiOperation("权限标识")
|
||
public Result<Set<String>> permissions(){
|
||
UsernamePasswordAuthenticationToken authentication = (UsernamePasswordAuthenticationToken) SecurityContextHolder.getContext().getAuthentication();
|
||
LoginUser loginUser = (LoginUser) authentication.getPrincipal();
|
||
SysUserEntity user = loginUser.getSysUserEntity();
|
||
Set<String> set = sysMenuService.getUserPermissions(user);
|
||
return new Result<Set<String>>().ok(set);
|
||
}
|
||
|
||
@GetMapping("list")
|
||
@ApiOperation("列表")
|
||
@ApiImplicitParam(name = "type", value = "菜单类型 0：菜单 1：按钮 null：全部", paramType = "query", dataTypeClass=Integer.class)
|
||
@PreAuthorize("@ex.hasAuthority('sys:menu:list')")
|
||
public Result<List<SysMenuDTO>> list(Integer type){
|
||
List<SysMenuDTO> list = sysMenuService.getAllMenuList(type);
|
||
|
||
return new Result<List<SysMenuDTO>>().ok(list);
|
||
}
|
||
|
||
@GetMapping("{id}")
|
||
@ApiOperation("信息")
|
||
@PreAuthorize("@ex.hasAuthority('sys:menu:info')")
|
||
public Result<SysMenuDTO> get(@PathVariable("id") Long id){
|
||
SysMenuDTO data = sysMenuService.get(id);
|
||
|
||
return new Result<SysMenuDTO>().ok(data);
|
||
}
|
||
|
||
@PostMapping
|
||
@ApiOperation("保存")
|
||
@LogOperation("保存")
|
||
@PreAuthorize("@ex.hasAuthority('sys:menu:save')")
|
||
public Result save(@RequestBody SysMenuDTO dto){
|
||
//效验数据
|
||
ValidatorUtils.validateEntity(dto, DefaultGroup.class);
|
||
|
||
sysMenuService.save(dto);
|
||
|
||
return new Result();
|
||
}
|
||
|
||
@PutMapping
|
||
@ApiOperation("修改")
|
||
@LogOperation("修改")
|
||
@PreAuthorize("@ex.hasAuthority('sys:menu:update')")
|
||
public Result update(@RequestBody SysMenuDTO dto){
|
||
//效验数据
|
||
ValidatorUtils.validateEntity(dto, DefaultGroup.class);
|
||
|
||
sysMenuService.update(dto);
|
||
|
||
return new Result();
|
||
}
|
||
|
||
@DeleteMapping("{id}")
|
||
@ApiOperation("删除")
|
||
@LogOperation("删除")
|
||
@PreAuthorize("@ex.hasAuthority('sys:menu:delete')")
|
||
public Result delete(@PathVariable("id") Long id){
|
||
//效验数据
|
||
AssertUtils.isNull(id, "id");
|
||
|
||
//判断是否有子菜单或按钮
|
||
List<SysMenuDTO> list = sysMenuService.getListPid(id);
|
||
if(list.size() > 0){
|
||
return new Result().error(ErrorCode.SUB_MENU_EXIST);
|
||
}
|
||
|
||
sysMenuService.delete(id);
|
||
|
||
return new Result();
|
||
}
|
||
|
||
@GetMapping("select")
|
||
@ApiOperation("角色菜单权限")
|
||
@PreAuthorize("@ex.hasAuthority('sys:menu:select')")
|
||
public Result<List<SysMenuDTO>> select(){
|
||
UsernamePasswordAuthenticationToken authentication = (UsernamePasswordAuthenticationToken) SecurityContextHolder.getContext().getAuthentication();
|
||
LoginUser loginUser = (LoginUser) authentication.getPrincipal();
|
||
SysUserEntity user = loginUser.getSysUserEntity();
|
||
List<SysMenuDTO> list = sysMenuService.getUserMenuList(user, null);
|
||
|
||
return new Result<List<SysMenuDTO>>().ok(list);
|
||
}
|
||
}
|