using System.Windows.Controls; using CommunityToolkit.Mvvm.Messaging; using iNKORE.UI.WPF.Modern.Common.IconKeys; using PMSWPF.Data.Entities; using PMSWPF.Enums; using PMSWPF.Extensions; using PMSWPF.Helper; using PMSWPF.Message; using PMSWPF.Models; using SqlSugar; namespace PMSWPF.Data.Repositories; public class MenuRepository { private readonly SqlSugarClient _db; public MenuRepository() { _db = DbContext.GetInstance(); } public async Task> GetMenu() { // //无主键用法新:5.1.4.110 // db.Queryable().ToTree(it=>it.Child,it=>it.ParentId,0,it=>it.Id)//+4重载 List menus = new(); var dbMenuList = await _db.Queryable().ToTreeAsync(dm => dm.Items, dm => dm.ParentId, 0); foreach (var item in dbMenuList) { menus.Add(item.CopyTo()); } return menus; } public async Task AddMenu(MenuBean menu) { return await _db.Insertable(menu.CopyTo()).ExecuteCommandAsync(); } public async Task AddDeviceMenu(Device device) { bool result = false; var deviceMainMenu = await _db.Queryable().FirstAsync(m => m.Name == "设备"); if (deviceMainMenu == null) throw new InvalidOperationException("没有找到设备菜单!!"); // 添加菜单项 MenuBean menu = new MenuBean() { Name = device.Name, Type = MenuType.DeviceMenu, DataId = device.Id, Icon = SegoeFluentIcons.Devices4.Glyph, }; menu.ParentId=deviceMainMenu.Id; var addDeviceMenuId = await _db.Insertable(menu.CopyTo()) .ExecuteReturnIdentityAsync(); if (addDeviceMenuId == 0) throw new InvalidOperationException($"{menu.Name},设备菜单添加失败!!"); var defVarTable= await _db.Queryable().FirstAsync(v=>v.DeviceId==device.Id && v.Name=="默认变量表"); if (defVarTable == null) throw new InvalidOperationException($"没有找到{device.Name}的默认变量表。"); var defVarTableMenu=new MenuBean() { Name = "默认变量表", Icon = SegoeFluentIcons.Tablet.Glyph, Type = MenuType.VariableTableMenu, ParentId = addDeviceMenuId, DataId = defVarTable.Id }; var addVarTable=new MenuBean() { Name = "添加变量表", Icon = SegoeFluentIcons.Add.Glyph, Type = MenuType.AddVariableTableMenu, ParentId = addDeviceMenuId, }; var defTableRes = await _db.Insertable(defVarTableMenu).ExecuteCommandAsync(); var addTableRes = await _db.Insertable(addVarTable).ExecuteCommandAsync(); if ((addTableRes+defTableRes) != 2) { // 如果出错删除原来添加的设备菜单 await _db.Deleteable().Where(m=>m.Id==addDeviceMenuId).ExecuteCommandAsync(); throw new InvalidOperationException("添加默认变量表时发生了错误!!"); } return true; } }