| | |
| | | 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; |
| | |
| | | |
| | | 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; |
| | | |
| | | /** |
| | |
| | | private UserLoginService userLoginService; |
| | | @Resource |
| | | private RoleMenuService roleMenuService; |
| | | @Resource |
| | | private MenuService menuService; |
| | | @Resource |
| | | private TenantService tenantService; |
| | | @Autowired |
| | |
| | | |
| | | @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())) { |
| | |
| | | |
| | | // ---------------------------------------------------- |
| | | |
| | | /** |
| | | * 页面级菜单:若存在带 :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; |