DMS/Services/DataServices.cs

using CommunityToolkit.Mvvm.ComponentModel;
using CommunityToolkit.Mvvm.Messaging;
using Microsoft.Extensions.DependencyInjection;
using Microsoft.Extensions.Logging;
using PMSWPF.Data.Repositories;
using PMSWPF.Enums;
using PMSWPF.Extensions;
using PMSWPF.Helper;
using PMSWPF.Message;
using PMSWPF.Models;
using PMSWPF.ViewModels;

namespace PMSWPF.Services;

public partial class DataServices : ObservableRecipient, IRecipient<LoadMessage>
{
    private readonly ILogger<DataServices> _logger;
    [ObservableProperty] private List<Device> _devices;
    [ObservableProperty] private List<VariableTable> _variableTables;
    [ObservableProperty] private List<MenuBean> menuBeans;
    private readonly DeviceRepository _deviceRepository;
    private readonly MenuRepository _menuRepository;

    public event Action<List<Device>> OnDeviceListChanged;
    public event Action<List<MenuBean>> OnMenuListChanged;


    partial void OnDevicesChanged(List<Device> devices)
    {
        OnDeviceListChanged?.Invoke(devices);
        if (menuBeans != null && Devices != null)
        {
            FillMenuData(MenuBeans, Devices);
        }
    }

    partial void OnMenuBeansChanged(List<MenuBean> menuBeans)
    {
        OnMenuListChanged?.Invoke(menuBeans);
        if (MenuBeans != null && Devices != null)
        {
            FillMenuData(MenuBeans, Devices);
        }
    }


    public DataServices(ILogger<DataServices> logger)
    {
        _logger = logger;
        IsActive = true;
        _deviceRepository = new DeviceRepository();
        _menuRepository = new MenuRepository();
    }


    /// <summary>
    /// 给Menu菜单的Data填充数据
    /// </summary>
    /// <param name="menuBeans"></param>
    private void FillMenuData(List<MenuBean> menuBeans, List<Device> devices)
    {
        if (menuBeans == null || menuBeans.Count == 0)
            return;

        foreach (MenuBean menuBean in menuBeans)
        {
            switch (menuBean.Type)
            {
                case MenuType.MainMenu:
                    menuBean.ViewModel = GetMainViewModel(menuBean.Name);
                    break;
                case MenuType.DeviceMenu:
                    menuBean.ViewModel = App.Current.Services.GetRequiredService<DeviceDetailViewModel>();
                    menuBean.Data = devices.FirstOrDefault(d => d.Id == menuBean.DataId);
                    break;
                case MenuType.VariableTableMenu:
                    var varTableVM = App.Current.Services.GetRequiredService<VariableTableViewModel>();
                    varTableVM.VariableTable = FindVarTableForDevice(menuBean.DataId);
                    menuBean.ViewModel = varTableVM;
                    menuBean.Data = varTableVM.VariableTable;
                    break;
                case MenuType.AddVariableTableMenu:
                    break;
            }

            if (menuBean.Items != null && menuBean.Items.Count > 0)
            {
                FillMenuData(menuBean.Items, devices);
            }
        }
    }

    private ViewModelBase GetMainViewModel(string name)
    {
        ViewModelBase navgateVM = App.Current.Services.GetRequiredService<HomeViewModel>();
        switch (name)
        {
            case "主页":
                navgateVM = App.Current.Services.GetRequiredService<HomeViewModel>();
                break;
            case "设备":
                navgateVM = App.Current.Services.GetRequiredService<DevicesViewModel>();
                break;
            case "数据转换":
                navgateVM = App.Current.Services.GetRequiredService<DataTransformViewModel>();
                break;
            case "设置":
                navgateVM = App.Current.Services.GetRequiredService<SettingViewModel>();
                break;
        }

        return navgateVM;
    }

    /// <summary>
    /// 从设备列表中找到变量表VarTable对象
    /// </summary>
    /// <param name="vtableId">VarTable的ID</param>
    /// <returns>如果找到择返回对象，否则返回null</returns>
    private VariableTable FindVarTableForDevice(int vtableId)
    {
        VariableTable varTable = null;
        foreach (var device in _devices)
        {
            varTable = device.VariableTables.FirstOrDefault(v => v.Id == vtableId);
            if (varTable != null)
                return varTable;
        }

        return varTable;
    }

    /// <summary>
    /// 接受加载消息，收到消息后从数据库加载对应的数据
    /// </summary>
    /// <param name="message">消息的类型，如加载菜单LoadMessage.Menu</param>
    /// <exception cref="ArgumentException"></exception>
    public async void Receive(LoadMessage message)
    {
        if (!(message.Value is LoadTypes))
            throw new ArgumentException($"接受到的加载类型错误：{message.Value}");
        try
        {
            switch ((LoadTypes)message.Value)
            {
                case LoadTypes.All:
                    Devices = await _deviceRepository.GetAll();
                    await LoadMenus();
                    break;
                case LoadTypes.Devices:
                    Devices = await _deviceRepository.GetAll();
                    break;
                case LoadTypes.Menu:
                    await LoadMenus();
                    break;
            }
        }
        catch (Exception e)
        {
            NotificationHelper.ShowMessage($"加载数据出现了错误：{e.Message}");
            _logger.LogError($"加载数据出现了错误：{e}");
        }
    }

    private async Task LoadMenus()
    {
        MenuBeans = await _menuRepository.GetMenu();
        foreach (MenuBean menu in MenuBeans)
        {
            MenuHelper.MenuAddParent(menu);
        }
    }
}