NetworkTool/windows/NetworkTool.Client/MainWindow.xaml.cs

using System.Collections.Generic;
using System.Windows;
using System.Windows.Controls;
using System.Windows.Input;
using System.Windows.Media;
using System.Windows.Media.Animation;
using System.Windows.Threading;
using NetworkTool.Client.Models;
using NetworkTool.Client.Services;

namespace NetworkTool.Client;

public partial class MainWindow : Window
{
    private readonly NetworkAdapterService _networkAdapterService = new();
    private readonly PasswordStoreService _passwordStoreService = new();
    private readonly DiscoveryService _discoveryService = new();
    private readonly ServerApiService _serverApiService = new();
    private IReadOnlyList<AdapterInfo> _adapters = [];
    private IReadOnlyList<DiscoveredDevice> _discoveredDevices = [];
    private bool _isBusy;
    private CancellationTokenSource? _statusMessageCts;

    public MainWindow()
    {
        InitializeComponent();
        Loaded += MainWindow_OnLoaded;
    }

    private void MainWindow_OnLoaded(object sender, RoutedEventArgs e)
    {
        LoadInitialState();
    }

    private void LoadInitialState()
    {
        _adapters = _networkAdapterService.GetEthernetAdapters();
        _adapters = _adapters
            .OrderByDescending(adapter => adapter.RecommendationScore)
            .ThenBy(adapter => adapter.Name)
            .ToList();
        AdapterComboBox.ItemsSource = _adapters;

        var recommendedAdapter = _networkAdapterService.GetRecommendedAdapter(_adapters);

        if (recommendedAdapter is not null)
        {
            AdapterComboBox.SelectedItem = recommendedAdapter;
        }
        else if (_adapters.Count > 0)
        {
            AdapterComboBox.SelectedIndex = 0;
        }

        UpdateButtonStates();
    }

    private void AdapterComboBox_OnSelectionChanged(object sender, SelectionChangedEventArgs e)
    {
        if (AdapterComboBox.SelectedItem is not AdapterInfo adapter)
        {
            ClearDiscoveredDevices();
            SetStatus("请选择一块有线网卡。", false);
            UpdateButtonStates();
            return;
        }

        if (!adapter.HasLink)
        {
            ClearDiscoveredDevices();
            SetStatus("当前网卡未检测到链路，请检查网线连接。", true);
            UpdateButtonStates();
            return;
        }

        UpdateButtonStates();
        _ = SearchDevicesAsync(adapter);
    }

    private async void RefreshAdaptersButton_OnClick(object sender, RoutedEventArgs e)
    {
        SetBusyState(true);
        try
        {
            RefreshAdapters();
            SetStatus("已刷新本机网卡。", true);
        }
        catch (Exception ex)
        {
            SetStatus($"刷新本机网卡失败：{ex.Message}", true);
            MessageBox.Show(this, ex.Message, "刷新失败", MessageBoxButton.OK, MessageBoxImage.Error);
        }
        finally
        {
            SetBusyState(false);
        }

        if (AdapterComboBox.SelectedItem is AdapterInfo adapter)
        {
            await SearchDevicesAsync(adapter);
        }
    }

    private async void SearchDevicesButton_OnClick(object sender, RoutedEventArgs e)
    {
        if (AdapterComboBox.SelectedItem is AdapterInfo adapter)
        {
            await SearchDevicesAsync(adapter);
        }
        else
        {
            SetStatus("请先选择一块网卡。", true);
        }
    }

    private async Task SearchDevicesAsync(AdapterInfo adapter)
    {
        if (_isBusy)
        {
            return;
        }

        if (string.IsNullOrWhiteSpace(adapter.IPv4Address))
        {
            ClearDiscoveredDevices();
            SetStatus("当前网卡没有可用 IPv4，无法搜索设备。", true);
            return;
        }

        if (!adapter.HasLink)
        {
            ClearDiscoveredDevices();
            SetStatus("当前网卡未检测到链路，请检查网线连接。", true);
            return;
        }

        SetBusyState(true);
        ClearDiscoveredDevices();
        SetStatus("正在通过当前网卡广播搜索设备。", true);
        await Dispatcher.InvokeAsync(() => { }, System.Windows.Threading.DispatcherPriority.Render);

        try
        {
            _discoveredDevices = await _discoveryService.DiscoverManyAsync(adapter.IPv4Address);
            DiscoveredDevicesListView.ItemsSource = _discoveredDevices;
            if (_discoveredDevices.Count == 0)
            {
                SetStatus("未发现 169.254 开头的设备 IP，请确认网卡、网线、远端服务和维护网段配置。", true);
                return;
            }

            SetStatus($"已发现 {_discoveredDevices.Count} 台设备，请双击 IP 连接。", true);
        }
        catch (Exception ex)
        {
            SetStatus($"搜索设备失败：{ex.Message}", true);
            MessageBox.Show(this, ex.Message, "搜索设备失败", MessageBoxButton.OK, MessageBoxImage.Error);
        }
        finally
        {
            SetBusyState(false);
        }
    }

    private void SavePasswordForDevice(DiscoveredDevice device, string password)
    {
        var deviceKey = GetDevicePasswordKey(device);
        if (!string.IsNullOrWhiteSpace(deviceKey) && !string.IsNullOrWhiteSpace(password))
        {
            _passwordStoreService.SavePassword(deviceKey, password);
        }
    }

    private void UpdateButtonStates()
    {
        var adapter = AdapterComboBox.SelectedItem as AdapterInfo;
        var hasAdapter = adapter is not null;
        RefreshAdaptersButton.IsEnabled = !_isBusy;
        SearchDevicesButton.IsEnabled = !_isBusy && hasAdapter && adapter!.HasLink;
    }

    private void RefreshAdapters(string? selectedAdapterId = null)
    {
        _adapters = _networkAdapterService.GetEthernetAdapters();
        _adapters = _adapters
            .OrderByDescending(adapter => adapter.RecommendationScore)
            .ThenBy(adapter => adapter.Name)
            .ToList();
        AdapterComboBox.ItemsSource = _adapters;

        var selected = selectedAdapterId is null
            ? _networkAdapterService.GetRecommendedAdapter(_adapters)
            : _adapters.FirstOrDefault(adapter => adapter.Id == selectedAdapterId) ?? _networkAdapterService.GetRecommendedAdapter(_adapters);

        if (selected is not null)
        {
            AdapterComboBox.SelectedItem = selected;
        }
    }

    private void ClearDiscoveredDevices()
    {
        _discoveredDevices = [];
        DiscoveredDevicesListView.ItemsSource = _discoveredDevices;
        DiscoveryStateTextBlock.Text = "选择网卡后会自动搜索设备。";
    }

    private async void DiscoveredDevicesListView_OnMouseDoubleClick(object sender, MouseButtonEventArgs e)
    {
        if (!_isBusy && DiscoveredDevicesListView.SelectedItem is DiscoveredDevice device)
        {
            await ConnectToDeviceAsync(device);
        }
    }

    private void DiscoveredDevicesListView_OnSizeChanged(object sender, SizeChangedEventArgs e)
    {
        var availableWidth = DiscoveredDevicesListView.ActualWidth - 36;
        if (availableWidth <= 0)
        {
            return;
        }

        DeviceIpColumn.Width = Math.Max(130, availableWidth * 0.26);
        DeviceHostnameColumn.Width = Math.Max(150, availableWidth * 0.30);
        DeviceMacColumn.Width = Math.Max(180, availableWidth - DeviceIpColumn.Width - DeviceHostnameColumn.Width);
    }

    private async Task ConnectToDeviceAsync(DiscoveredDevice device)
    {
        var deviceKey = GetDevicePasswordKey(device);
        var savedPassword = _passwordStoreService.LoadPassword(deviceKey);
        var password = string.Empty;
        if (!string.IsNullOrWhiteSpace(savedPassword))
        {
            password = savedPassword;
        }
        else if (!TryPromptForPassword(device, out savedPassword))
        {
            return;
        }
        else
        {
            password = savedPassword;
        }

        if (string.IsNullOrWhiteSpace(password))
        {
            SetStatus("请输入管理密码。", true);
            return;
        }

        var selectedAdapter = AdapterComboBox.SelectedItem as AdapterInfo;
        SetBusyState(true);
        SetStatus($"正在连接 {device.Lan2Ip}。", true);

        try
        {
            var baseAddress = $"http://{device.Lan2Ip}:48888";
            var result = await _serverApiService.CheckHealthAsync(baseAddress, password, selectedAdapter?.IPv4Address ?? string.Empty);
            if (result.Success)
            {
                SavePasswordForDevice(device, password);
                SetStatus("连接成功。", true);
                OpenDeviceDetailsWindow(baseAddress, selectedAdapter?.IPv4Address ?? string.Empty, password);
                return;
            }

            if (result.StatusCode == 401)
            {
                SetStatus("管理密码错误，请确认密码是否正确。", true);
                MessageBox.Show(this, "管理密码校验失败，请确认密码是否正确。", "密码错误", MessageBoxButton.OK, MessageBoxImage.Warning);
                return;
            }

            SetStatus($"设备已发现，但 HTTP 验证失败：{result.Message}", true);
        }
        catch (Exception ex)
        {
            SetStatus($"连接失败：{ex.Message}", true);
            MessageBox.Show(this, ex.Message, "连接失败", MessageBoxButton.OK, MessageBoxImage.Error);
        }
        finally
        {
            SetBusyState(false);
        }

    }

    private void OpenDeviceDetailsWindow(string baseAddress, string localIPv4, string password)
    {
        var window = new DeviceDetailsWindow(baseAddress, localIPv4, password)
        {
            Owner = this,
        };
        window.ShowDialog();
    }

    private bool TryPromptForPassword(DiscoveredDevice device, out string password)
    {
        var label = string.IsNullOrWhiteSpace(device.Mac) ? device.Lan2Ip : device.Mac;
        var window = new PasswordPromptWindow(label)
        {
            Owner = this,
        };

        if (window.ShowDialog() == true)
        {
            password = window.Password;
            return true;
        }

        password = string.Empty;
        return false;
    }

    private static string GetDevicePasswordKey(DiscoveredDevice device)
    {
        if (!string.IsNullOrWhiteSpace(device.Mac))
        {
            return device.Mac;
        }

        return device.DeviceId;
    }

    private void SetStatus(string message, bool addLog)
    {
        ApplyStatusMessageStyle(message);
        DiscoveryStateTextBlock.Text = message;
        StatusTextBlock.Text = message;
        StatusMessageBorder.Opacity = 0;
        StatusMessageBorder.Visibility = Visibility.Visible;
        StatusMessageBorder.BeginAnimation(OpacityProperty, new DoubleAnimation(1, TimeSpan.FromMilliseconds(160)));
        _statusMessageCts?.Cancel();
        _statusMessageCts = new CancellationTokenSource();
        _ = HideStatusMessageAsync(_statusMessageCts.Token);
        _ = addLog;
    }

    private async Task HideStatusMessageAsync(CancellationToken cancellationToken)
    {
        try
        {
            await Task.Delay(3000, cancellationToken);
            await Dispatcher.InvokeAsync(() =>
            {
                var animation = new DoubleAnimation(0, TimeSpan.FromMilliseconds(200));
                animation.Completed += (_, _) =>
                {
                    if (!cancellationToken.IsCancellationRequested)
                    {
                        StatusMessageBorder.Visibility = Visibility.Collapsed;
                    }
                };
                StatusMessageBorder.BeginAnimation(OpacityProperty, animation);
            });
        }
        catch (TaskCanceledException)
        {
        }
    }

    private void ApplyStatusMessageStyle(string message)
    {
        var (background, foreground) = GetStatusMessageBrushes(message);
        StatusMessageBorder.Background = background;
        StatusTextBlock.Foreground = foreground;
    }

    private static (Brush Background, Brush Foreground) GetStatusMessageBrushes(string message)
    {
        if (ContainsAny(message, "失败", "错误", "拒绝", "超时", "不能为空", "不正确", "无法"))
        {
            return (new SolidColorBrush((Color)ColorConverter.ConvertFromString("#B91C1C")), Brushes.White);
        }

        if (ContainsAny(message, "未发现", "请", "重试", "警告", "需要"))
        {
            return (new SolidColorBrush((Color)ColorConverter.ConvertFromString("#C2410C")), Brushes.White);
        }

        if (ContainsAny(message, "成功", "已切换", "已刷新", "已读取", "已加载", "已发现", "已提交", "已回填"))
        {
            return (new SolidColorBrush((Color)ColorConverter.ConvertFromString("#047857")), Brushes.White);
        }

        return (new SolidColorBrush((Color)ColorConverter.ConvertFromString("#111827")), Brushes.White);
    }

    private static bool ContainsAny(string message, params string[] markers)
    {
        return markers.Any(marker => message.Contains(marker, StringComparison.Ordinal));
    }

    private void SetBusyState(bool isBusy)
    {
        _isBusy = isBusy;
        AdapterComboBox.IsEnabled = !isBusy;
        RefreshAdaptersButton.IsEnabled = !isBusy;
        SearchDevicesButton.IsEnabled = !isBusy && AdapterComboBox.SelectedItem is AdapterInfo adapter && adapter.HasLink;
    }
}