AiDOPWarehouse/server/Admin.NET.Core/SeedData/SysRoleMenuSeedData.cs

// Admin.NET 项目的版权、商标、专利和其他相关权利均受相应法律法规的保护。使用本项目应遵守相关法律法规和许可证的要求。
//
// 本项目主要遵循 MIT 许可证和 Apache 许可证（版本 2.0）进行分发和使用。许可证位于源代码树根目录中的 LICENSE-MIT 和 LICENSE-APACHE 文件。
//
// 不得利用本项目从事危害国家安全、扰乱社会秩序、侵犯他人合法权益等法律法规禁止的活动！任何基于本项目二次开发而产生的一切法律纠纷和责任，我们不承担任何责任！

namespace Admin.NET.Core;

/// <summary>
/// 系统角色菜单表种子数据
/// </summary>
public class SysRoleMenuSeedData : ISqlSugarEntitySeedData<SysRoleMenu>
{
    private const long CollisionFallbackBaseId = 9_000_000_000_000_000L;
    private const long CollisionFallbackRoleStride = 1_000_000_000L;

    /// <summary>
    /// 种子数据
    /// </summary>
    /// <returns></returns>
    public IEnumerable<SysRoleMenu> HasData()
    {
        var roleMenuList = new List<SysRoleMenu>();
        var roleMenuPairs = new HashSet<(long RoleId, long MenuId)>();

        var roleList = new SysRoleSeedData().HasData().ToList();
        var menuList = new SysMenuSeedData().HasData().ToList();
        var defaultMenuList = new SysTenantMenuSeedData().HasData().ToList();

        // 第一个角色拥有全部默认租户菜单
        AddRoleMenus(roleMenuList, roleMenuPairs, defaultMenuList.Select(u => new SysRoleMenu
        {
            Id = u.MenuId + (roleList[0].Id % 1300000000000),
            RoleId = roleList[0].Id,
            MenuId = u.MenuId
        }));

        // 其他角色权限：工作台、帮助文档、关于项目、个人中心（整棵子树）
        // 系统管理：仅挂「机构管理」子树（与租户基线一致），避免仅有空目录导致侧栏有「系统管理」却无子路由、访问未授权路径时应用内 404
        var otherRoleMenuList = menuList.ToChildList(u => u.Id, u => u.Pid, u => new[] { "工作台", "帮助文档", "关于项目", "个人中心" }.Contains(u.Title)).ToList();
        var systemDir = menuList.FirstOrDefault(u => u.Type == MenuTypeEnum.Dir && u.Title == "系统管理");
        if (systemDir != null)
        {
            if (!otherRoleMenuList.Any(u => u.Id == systemDir.Id))
                otherRoleMenuList.Add(systemDir);
            otherRoleMenuList.AddRange(menuList.ToChildList(u => u.Id, u => u.Pid, u => u.Pid == systemDir.Id && u.Title == "机构管理"));
        }
        foreach (var role in roleList.Skip(1))
        {
            AddRoleMenus(roleMenuList, roleMenuPairs, otherRoleMenuList.Select(u => new SysRoleMenu
            {
                Id = u.Id + (role.Id % 1300000000000),
                RoleId = role.Id,
                MenuId = u.Id
            }));
        }

        // 演示一般账户：在「其他角色」基线之上追加 Ai-DOP 全子树（智慧运营首页等）
        var demoGeneralRole = roleList.FirstOrDefault(r => r.Code == "demo_general");
        if (demoGeneralRole != null)
        {
            var allFlat = App.GetService<SysTenantService>().GetAllSeedMenusFlat();
            var aidopRoot = allFlat.FirstOrDefault(u =>
                u.Name == "aidopRoot" || string.Equals(u.Path?.Trim(), "/aidop", StringComparison.Ordinal));
            if (aidopRoot != null)
            {
                // ToChildList(..., topParentIdValue) 默认包含根节点，勿再手动 Add(aidopRoot)，否则 SysRoleMenu 主键重复导致启动失败
                var subtree = allFlat.ToChildList(u => u.Id, u => u.Pid, aidopRoot.Id).ToList();
                AddRoleMenus(roleMenuList, roleMenuPairs, subtree.Select(m => new SysRoleMenu
                {
                    Id = m.Id + (demoGeneralRole.Id % 1300000000000),
                    RoleId = demoGeneralRole.Id,
                    MenuId = m.Id
                }));
            }
        }

        EnsureUniqueIds(roleMenuList);
        return roleMenuList;
    }

    private static void AddRoleMenus(List<SysRoleMenu> roleMenuList, HashSet<(long RoleId, long MenuId)> roleMenuPairs, IEnumerable<SysRoleMenu> menus)
    {
        foreach (var menu in menus)
        {
            if (!roleMenuPairs.Add((menu.RoleId, menu.MenuId))) continue;
            roleMenuList.Add(menu);
        }
    }

    private static void EnsureUniqueIds(List<SysRoleMenu> roleMenuList)
    {
        var usedIds = new HashSet<long>();
        foreach (var roleMenu in roleMenuList.OrderBy(u => u.RoleId).ThenBy(u => u.MenuId))
        {
            if (usedIds.Add(roleMenu.Id)) continue;
            roleMenu.Id = BuildCollisionFallbackId(roleMenu.RoleId, roleMenu.MenuId, usedIds);
        }
    }

    private static long BuildCollisionFallbackId(long roleId, long menuId, HashSet<long> usedIds)
    {
        var candidate = CollisionFallbackBaseId
            + (roleId % 1_000_000L) * CollisionFallbackRoleStride
            + (menuId % CollisionFallbackRoleStride);

        while (!usedIds.Add(candidate)) candidate++;
        return candidate;
    }
}