wip

DMS.WPF/Services/MenuViewService.cs

@@ -0,0 +1,157 @@
+using AutoMapper;
+using DMS.Application.Interfaces;
+using DMS.Application.Interfaces.Management;
+using DMS.Application.Services.Management;
+using DMS.Core.Models;
+using DMS.WPF.Interfaces;
+using DMS.WPF.ItemViewModel;
+using System.Collections.ObjectModel;
+using System.Threading.Tasks;
+
+namespace DMS.WPF.Services;
+
+/// <summary>
+/// 菜单数据服务类，负责管理菜单相关的数据和操作。
+/// </summary>
+public class MenuViewService : IMenuViewService
+{
+    private readonly IMapper _mapper;
+    private readonly IWpfDataService _wpfDataService;
+    private readonly IAppStorageService _appStorageService;
+    private readonly IMenuManagementService _menuManagementService;
+
+
+
+    /// <summary>
+    /// MenuDataService类的构造函数。
+    /// </summary>
+    /// <param name="mapper">AutoMapper 实例。</param>
+    /// <param name="appStorageService">数据服务中心实例。</param>
+    public MenuViewService(IMapper mapper, IWpfDataService dataStorageService, IAppStorageService appStorageService, IMenuManagementService menuManagementService)
+    {
+        _mapper = mapper;
+        _wpfDataService = dataStorageService;
+        _appStorageService = appStorageService;
+        _menuManagementService = menuManagementService;
+    }
+
+    public void LoadAllMenus()
+    {
+        _wpfDataService.Menus = _mapper.Map<ObservableCollection<MenuItem>>(_appStorageService.Menus.Values);
+        BuildMenuTrees();
+    }
+
+    /// <summary>
+    /// 构建菜单树。
+    /// </summary>
+    public void BuildMenuTrees()
+    {
+        // 创建一个新的临时列表来存储根菜单
+        var newRootMenus = new List<MenuItem>();
+        
+        // 首先，确保所有菜单项的Children集合是干净的
+        foreach (var menu in _wpfDataService.Menus)
+        {
+            // 避免循环引用，清空Children集合
+            menu.Children.Clear();
+        }
+
+        // 遍历所有菜单项，构建树形结构
+        foreach (var menu in _wpfDataService.Menus)
+        {
+            var parentMenu = _wpfDataService.Menus.FirstOrDefault(m => m.Id == menu.ParentId);
+            // 检查是否有父ID，并且父ID不为0（通常0或null表示根节点）
+            if (parentMenu != null && menu.ParentId != 0)
+            {
+                // 将当前菜单添加到父菜单的Children列表中
+                if (!parentMenu.Children.Contains(menu))
+                {
+                    parentMenu.Children.Add(menu);
+                }
+            }
+            else
+            {
+                // 如果没有父ID，则这是一个根菜单
+                newRootMenus.Add(menu);
+            }
+        }
+        
+        // 在UI线程安全地更新MenuTrees集合
+        _wpfDataService.MenuTrees.Clear();
+        foreach (var rootMenu in newRootMenus)
+        {
+            _wpfDataService.MenuTrees.Add(rootMenu);
+        }
+    }
+
+    /// <summary>
+    /// 添加菜单项。
+    /// </summary>
+    public void AddMenuToView(MenuItem MenuItem)
+    {
+        if (MenuItem is null) return;
+
+        var parentMenu = _wpfDataService.Menus.FirstOrDefault(m => m.Id == MenuItem.ParentId);
+        if (parentMenu is not null)
+        {
+            // 首先添加到菜单列表
+            _wpfDataService.Menus.Add(MenuItem);
+            
+            // 然后将当前菜单添加到父菜单的Children列表中
+            if (!parentMenu.Children.Contains(MenuItem))
+            {
+                parentMenu.Children.Add(MenuItem);
+            }
+        }
+    }
+
+    /// <summary>
+    /// 更新菜单项。
+    /// </summary>
+    public async Task UpdateMenuItem(MenuItem MenuItem)
+    {
+        if (MenuItem is null) return;
+
+        var menu = _wpfDataService.Menus.FirstOrDefault(m => m.Id == MenuItem.Id);
+        if (menu is not null)
+        {
+
+            var res = await _menuManagementService.UpdateMenuAsync(_mapper.Map<MenuBean>(MenuItem));
+            if (res > 0)
+            {
+                menu.Header = MenuItem.Header;
+            }
+
+        }
+    }
+
+
+    /// <summary>
+    /// 删除菜单项。
+    /// </summary>
+    public async Task DeleteMenuItem(MenuItem? MenuItem)
+    {
+        if (MenuItem is null) return;
+
+        await _menuManagementService.DeleteMenuAsync(MenuItem.Id);
+
+        // 从扁平菜单列表中移除
+        _wpfDataService.Menus.Remove(MenuItem);
+
+        // 从树形结构中移除
+        if (MenuItem.ParentId.HasValue && MenuItem.ParentId.Value != 0)
+        {
+            // 如果有父菜单，从父菜单的Children中移除
+            var parentMenu = _wpfDataService.Menus.FirstOrDefault(m => m.Id == MenuItem.ParentId.Value);
+            parentMenu?.Children.Remove(MenuItem);
+        }
+        else
+        {
+            // 如果是根菜单，从MenuTrees中移除
+            _wpfDataService.MenuTrees.Remove(MenuItem);
+        }
+
+        // 重新构建整个菜单树以确保一致性
+        BuildMenuTrees();
+    }
+}