122 lines
4.1 KiB
C#
122 lines
4.1 KiB
C#
using AutoMapper;
|
||
using DMS.Application.DTOs;
|
||
using DMS.Application.Interfaces;
|
||
using DMS.Application.Interfaces.Management;
|
||
using DMS.Application.Services.Management;
|
||
using DMS.WPF.Interfaces;
|
||
using DMS.WPF.ViewModels.Items;
|
||
using System.Collections.ObjectModel;
|
||
using System.Threading.Tasks;
|
||
|
||
namespace DMS.WPF.Services;
|
||
|
||
/// <summary>
|
||
/// 菜单数据服务类，负责管理菜单相关的数据和操作。
|
||
/// </summary>
|
||
public class MenuDataService : IMenuDataService
|
||
{
|
||
private readonly IMapper _mapper;
|
||
private readonly IDataStorageService _dataStorageService;
|
||
private readonly IAppDataStorageService _appDataStorageService;
|
||
private readonly IMenuManagementService _menuManagementService;
|
||
|
||
|
||
|
||
/// <summary>
|
||
/// MenuDataService类的构造函数。
|
||
/// </summary>
|
||
/// <param name="mapper">AutoMapper 实例。</param>
|
||
/// <param name="appDataStorageService">数据服务中心实例。</param>
|
||
public MenuDataService(IMapper mapper,IDataStorageService dataStorageService, IAppDataStorageService appDataStorageService,IMenuManagementService menuManagementService)
|
||
{
|
||
_mapper = mapper;
|
||
_dataStorageService = dataStorageService;
|
||
_appDataStorageService = appDataStorageService;
|
||
_menuManagementService = menuManagementService;
|
||
}
|
||
|
||
public void LoadAllMenus()
|
||
{
|
||
_dataStorageService.Menus = _mapper.Map<ObservableCollection<MenuItemViewModel>>(_appDataStorageService.Menus.Values);
|
||
BuildMenuTrees();
|
||
}
|
||
|
||
/// <summary>
|
||
/// 构建菜单树。
|
||
/// </summary>
|
||
public void BuildMenuTrees()
|
||
{
|
||
_dataStorageService.MenuTrees.Clear();
|
||
// 遍历所有菜单项，构建树形结构
|
||
foreach (var menu in _dataStorageService.Menus)
|
||
{
|
||
var parentMenu = _dataStorageService.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，则这是一个根菜单
|
||
_dataStorageService.MenuTrees.Add(menu);
|
||
}
|
||
}
|
||
}
|
||
|
||
/// <summary>
|
||
/// 添加菜单项。
|
||
/// </summary>
|
||
public async Task AddMenuItem(MenuItemViewModel menuItemViewModel)
|
||
{
|
||
if (menuItemViewModel is null) return;
|
||
|
||
var deviceMenu = _dataStorageService.Menus.FirstOrDefault(m => m.Id == menuItemViewModel.ParentId);
|
||
if (deviceMenu is not null)
|
||
{
|
||
|
||
var menuId= await _menuManagementService.CreateMenuAsync(_mapper.Map<MenuBeanDto>(menuItemViewModel));
|
||
if (menuId>0)
|
||
{
|
||
menuItemViewModel.Id = menuId;
|
||
deviceMenu.Children.Add(menuItemViewModel);
|
||
_dataStorageService.Menus.Add(menuItemViewModel);
|
||
BuildMenuTrees();
|
||
}
|
||
|
||
}
|
||
}
|
||
|
||
|
||
/// <summary>
|
||
/// 删除菜单项。
|
||
/// </summary>
|
||
public async Task DeleteMenuItem(MenuItemViewModel? menuItemViewModel)
|
||
{
|
||
if (menuItemViewModel is null) return;
|
||
|
||
await _menuManagementService.DeleteMenuAsync(menuItemViewModel.Id);
|
||
|
||
// 从扁平菜单列表中移除
|
||
_dataStorageService.Menus.Remove(menuItemViewModel);
|
||
|
||
//// 从树形结构中移除
|
||
if (menuItemViewModel.ParentId.HasValue && menuItemViewModel.ParentId.Value != 0)
|
||
{
|
||
// 如果有父菜单，从父菜单的Children中移除
|
||
var parentMenu = _dataStorageService.Menus.FirstOrDefault(m => m.Id == menuItemViewModel.ParentId.Value);
|
||
parentMenu?.Children.Remove(menuItemViewModel);
|
||
}
|
||
else
|
||
{
|
||
// 如果是根菜单，从MenuTrees中移除
|
||
_dataStorageService.MenuTrees.Remove(menuItemViewModel);
|
||
}
|
||
|
||
//BuildMenuTrees();
|
||
}
|
||
} |