package com.genersoft.iot.vmp.vmanager.user; 
 | 
  
 | 
import com.genersoft.iot.vmp.conf.exception.ControllerException; 
 | 
import com.genersoft.iot.vmp.conf.security.JwtUtils; 
 | 
import com.genersoft.iot.vmp.conf.security.SecurityUtils; 
 | 
import com.genersoft.iot.vmp.service.IRoleService; 
 | 
import com.genersoft.iot.vmp.storager.dao.dto.Role; 
 | 
import com.genersoft.iot.vmp.utils.DateUtil; 
 | 
import com.genersoft.iot.vmp.vmanager.bean.ErrorCode; 
 | 
import io.swagger.v3.oas.annotations.Operation; 
 | 
import io.swagger.v3.oas.annotations.Parameter; 
 | 
import io.swagger.v3.oas.annotations.security.SecurityRequirement; 
 | 
import io.swagger.v3.oas.annotations.tags.Tag; 
 | 
import org.springframework.beans.factory.annotation.Autowired; 
 | 
import org.springframework.web.bind.annotation.*; 
 | 
  
 | 
import java.util.List; 
 | 
  
 | 
@Tag(name  = "角色管理") 
 | 
  
 | 
@RestController 
 | 
@RequestMapping("/api/role") 
 | 
public class RoleController { 
 | 
  
 | 
    @Autowired 
 | 
    private IRoleService roleService; 
 | 
  
 | 
    @PostMapping("/add") 
 | 
    @Operation(summary = "添加角色", security = @SecurityRequirement(name = JwtUtils.HEADER)) 
 | 
    @Parameter(name = "name", description = "角色名", required = true) 
 | 
    @Parameter(name = "authority", description = "权限(自行定义内容,目前未使用)", required = true) 
 | 
    public void add(@RequestParam String name, 
 | 
                                                  @RequestParam(required = false) String authority){ 
 | 
        // 获取当前登录用户id 
 | 
        int currenRoleId = SecurityUtils.getUserInfo().getRole().getId(); 
 | 
        if (currenRoleId != 1) { 
 | 
            // 只用角色id为1才可以删除和添加用户 
 | 
            throw new ControllerException(ErrorCode.ERROR403); 
 | 
        } 
 | 
  
 | 
        Role role = new Role(); 
 | 
        role.setName(name); 
 | 
        role.setAuthority(authority); 
 | 
        role.setCreateTime(DateUtil.getNow()); 
 | 
        role.setUpdateTime(DateUtil.getNow()); 
 | 
  
 | 
        int addResult = roleService.add(role); 
 | 
        if (addResult <= 0) { 
 | 
            throw new ControllerException(ErrorCode.ERROR100); 
 | 
        } 
 | 
    } 
 | 
  
 | 
    @DeleteMapping("/delete") 
 | 
    @Operation(summary = "删除角色", security = @SecurityRequirement(name = JwtUtils.HEADER)) 
 | 
    @Parameter(name = "id", description = "用户Id", required = true) 
 | 
    public void delete(@RequestParam Integer id){ 
 | 
        // 获取当前登录用户id 
 | 
        int currenRoleId = SecurityUtils.getUserInfo().getRole().getId(); 
 | 
        if (currenRoleId != 1) { 
 | 
            // 只用角色id为0才可以删除和添加用户 
 | 
            throw new ControllerException(ErrorCode.ERROR403); 
 | 
        } 
 | 
        int deleteResult = roleService.delete(id); 
 | 
  
 | 
        if (deleteResult <= 0) { 
 | 
            throw new ControllerException(ErrorCode.ERROR100); 
 | 
        } 
 | 
    } 
 | 
  
 | 
    @GetMapping("/all") 
 | 
    @Operation(summary = "查询角色", security = @SecurityRequirement(name = JwtUtils.HEADER)) 
 | 
    public List<Role> all(){ 
 | 
        // 获取当前登录用户id 
 | 
        List<Role> allRoles = roleService.getAll(); 
 | 
        return roleService.getAll(); 
 | 
    } 
 | 
} 
 |