cl
2 天以前 f21530d33ca23aeeba2ef073ac6b64422e4e26ca
rsf-server/src/main/java/com/vincent/rsf/server/system/controller/AuthController.java
@@ -20,12 +20,14 @@
import com.vincent.rsf.server.system.controller.result.LoginResult;
import com.vincent.rsf.server.system.controller.result.MenuVo;
import com.vincent.rsf.server.system.controller.result.SystemInfoVo;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.vincent.rsf.server.system.entity.Menu;
import com.vincent.rsf.server.system.entity.Tenant;
import com.vincent.rsf.server.system.entity.User;
import com.vincent.rsf.server.system.entity.UserLogin;
import com.vincent.rsf.server.system.enums.EmailType;
import com.vincent.rsf.server.system.enums.StatusType;
import com.vincent.rsf.server.system.service.MenuService;
import com.vincent.rsf.server.system.service.RoleMenuService;
import com.vincent.rsf.server.system.service.TenantService;
import com.vincent.rsf.server.system.service.UserLoginService;
@@ -35,7 +37,9 @@
import javax.annotation.Resource;
import javax.servlet.http.HttpServletRequest;
import java.util.HashSet;
import java.util.List;
import java.util.Set;
import java.util.stream.Collectors;
/**
@@ -53,6 +57,8 @@
    private UserLoginService userLoginService;
    @Resource
    private RoleMenuService roleMenuService;
    @Resource
    private MenuService menuService;
    @Resource
    private TenantService tenantService;
    @Autowired
@@ -164,7 +170,13 @@
    @GetMapping("/auth/menu")
    public R userMenu() {
        List<Menu> menus = roleMenuService.listMenuByUserId(getLoginUserId(), Menu.TYPE_MENU);
        Long uid = getLoginUserId();
        List<Menu> grantedAll = roleMenuService.listMenuByUserId(uid, null);
        Set<Long> grantedMenuIds = grantedAll.stream().map(Menu::getId).collect(Collectors.toCollection(HashSet::new));
        List<Menu> allMenus = menuService.list(new LambdaQueryWrapper<Menu>().eq(Menu::getDeleted, 0));
        List<Menu> menus = roleMenuService.listMenuByUserId(uid, Menu.TYPE_MENU).stream()
                .filter(m -> menuPageGranted(m, grantedMenuIds, allMenus))
                .collect(Collectors.toList());
        List<MenuVo> voList = menus.stream().map(this::convertToVo).collect(Collectors.toList());
        // exclude tenant
        if (!configProperties.getSuperUserList().contains(getLoginUser().getUsername())) {
@@ -225,6 +237,32 @@
    // ----------------------------------------------------
    /**
     * 页面级菜单:若存在带 :list 的按钮权限子节点,则必须同时勾选至少一个该类按钮,才展示该页(与接口 @PreAuthorize 一致)
     */
    private boolean menuPageGranted(Menu m, Set<Long> grantedMenuIds, List<Menu> allMenus) {
        if (m == null || !Integer.valueOf(Menu.TYPE_MENU).equals(m.getType())) {
            return true;
        }
        if (Cools.isEmpty(m.getComponent())) {
            return true;
        }
        List<Menu> btnChildren = allMenus.stream()
                .filter(c -> m.getId().equals(c.getParentId()) && Integer.valueOf(Menu.TYPE_BTN).equals(c.getType()))
                .collect(Collectors.toList());
        if (btnChildren.isEmpty()) {
            return true;
        }
        boolean needListBtn = btnChildren.stream().anyMatch(c ->
                c.getAuthority() != null && c.getAuthority().contains(":list"));
        if (!needListBtn) {
            return true;
        }
        return btnChildren.stream()
                .filter(c -> c.getAuthority() != null && c.getAuthority().contains(":list"))
                .anyMatch(c -> grantedMenuIds.contains(c.getId()));
    }
    private MenuVo convertToVo(Menu menu) {
        if (menu == null) {
            return null;