src/VisualStudio/Core/Def/ValueTracking/ValueTrackingTreeViewModel.cs - main - githubmirror/roslyn

src/VisualStudio/Core/Def/ValueTracking/ValueTrackingTreeViewModel.cs
125 строк4 KB
Cyrus Najmabadi
Switch to using field+auto-prop
24 янв 2025, 11:26
24 янв 2025, 11:26733debf
// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
// See the LICENSE file in the project root for more information.

using System.Collections.Generic;
using System.Collections.ObjectModel;
using System.ComponentModel;
using System.Runtime.CompilerServices;
using System.Windows.Media;
using Microsoft.CodeAnalysis.Editor.ReferenceHighlighting;
using Microsoft.CodeAnalysis.Editor.Shared.Utilities;
using Microsoft.VisualStudio.Text.Classification;

namespace Microsoft.VisualStudio.LanguageServices.ValueTracking;

internal sealed class ValueTrackingTreeViewModel : INotifyPropertyChanged
{
    public Brush? HighlightBrush
    {
        get;
        set => SetProperty(ref field, value);
    }

    public IClassificationFormatMap ClassificationFormatMap { get; }
    public ClassificationTypeMap ClassificationTypeMap { get; }
    public IEditorFormatMapService FormatMapService { get; }
    public ObservableCollection<TreeItemViewModel> Roots { get; } = [];
    public string AutomationName => ServicesVSResources.Value_Tracking;

    public TreeViewItemBase? SelectedItem
    {
        get;
        set => SetProperty(ref field, value);
    }
    public string SelectedItemFile
    {
        get;
        set => SetProperty(ref field, value);
    } = "";
    public int SelectedItemLine
    {
        get;
        set => SetProperty(ref field, value);
    }
    public bool IsLoading
    {
        get;
        private set => SetProperty(ref field, value);
    }
    public int LoadingCount
    {
        get;
        set => SetProperty(ref field, value);
    }

    public bool ShowDetails => SelectedItem is TreeItemViewModel;

    public event PropertyChangedEventHandler? PropertyChanged;

    public ValueTrackingTreeViewModel(IClassificationFormatMap classificationFormatMap, ClassificationTypeMap classificationTypeMap, IEditorFormatMapService formatMapService)
    {
        ClassificationFormatMap = classificationFormatMap;
        ClassificationTypeMap = classificationTypeMap;
        FormatMapService = formatMapService;

        var editorMap = FormatMapService.GetEditorFormatMap("text");
        SetHighlightBrush(editorMap);

        editorMap.FormatMappingChanged += (s, e) =>
        {
            SetHighlightBrush(editorMap);
        };

        PropertyChanged += Self_PropertyChanged;
    }

    private void SetHighlightBrush(IEditorFormatMap editorMap)
    {
        var properties = editorMap.GetProperties(ReferenceHighlightTag.TagId);
        HighlightBrush = properties["Background"] as Brush;
    }

    private void Self_PropertyChanged(object sender, PropertyChangedEventArgs e)
    {
        if (e.PropertyName == nameof(SelectedItem))
        {
            if (SelectedItem is not null)
            {
                SelectedItem.IsNodeSelected = true;

                if (SelectedItem is TreeItemViewModel itemWithInfo)
                {
                    SelectedItemFile = itemWithInfo?.FileName ?? "";
                    SelectedItemLine = itemWithInfo?.LineNumber ?? 0;
                }
                else
                {
                    SelectedItemFile = string.Empty;
                    SelectedItemLine = 0;
                }
            }

            NotifyPropertyChanged(nameof(ShowDetails));
        }

        if (e.PropertyName == nameof(LoadingCount))
        {
            IsLoading = LoadingCount > 0;
        }
    }

    private void SetProperty<T>(ref T field, T value, [CallerMemberName] string name = "")
    {
        if (EqualityComparer<T>.Default.Equals(field, value))
        {
            return;
        }

        field = value;
        NotifyPropertyChanged(name);
    }

    private void NotifyPropertyChanged(string name)
        => PropertyChanged?.Invoke(this, new PropertyChangedEventArgs(name));
}