sourcegit/src/Views/FilterModeSwitchButton.axaml.cs

using Avalonia;
using Avalonia.Controls;
using Avalonia.Interactivity;
using Avalonia.VisualTree;

namespace SourceGit.Views
{
    public partial class FilterModeSwitchButton : UserControl
    {
        public static readonly StyledProperty<Models.FilterMode> ModeProperty =
            AvaloniaProperty.Register<FilterModeSwitchButton, Models.FilterMode>(nameof(Mode));

        public Models.FilterMode Mode
        {
            get => GetValue(ModeProperty);
            set => SetValue(ModeProperty, value);
        }

        public static readonly StyledProperty<bool> IsNoneVisibleProperty =
            AvaloniaProperty.Register<FilterModeSwitchButton, bool>(nameof(IsNoneVisible));

        public bool IsNoneVisible
        {
            get => GetValue(IsNoneVisibleProperty);
            set => SetValue(IsNoneVisibleProperty, value);
        }

        public static readonly StyledProperty<bool> IsContextMenuOpeningProperty =
            AvaloniaProperty.Register<FilterModeSwitchButton, bool>(nameof(IsContextMenuOpening));

        public bool IsContextMenuOpening
        {
            get => GetValue(IsContextMenuOpeningProperty);
            set => SetValue(IsContextMenuOpeningProperty, value);
        }

        public FilterModeSwitchButton()
        {
            IsVisible = false;
            InitializeComponent();
        }

        protected override void OnPropertyChanged(AvaloniaPropertyChangedEventArgs change)
        {
            base.OnPropertyChanged(change);

            if (change.Property == ModeProperty ||
                change.Property == IsNoneVisibleProperty ||
                change.Property == IsContextMenuOpeningProperty)
            {
                var visible = (Mode != Models.FilterMode.None || IsNoneVisible || IsContextMenuOpening);
                SetCurrentValue(IsVisibleProperty, visible);
            }
        }

        private void OnChangeFilterModeButtonClicked(object sender, RoutedEventArgs e)
        {
            var repoView = this.FindAncestorOfType<Repository>();
            if (repoView == null)
                return;

            var repo = repoView.DataContext as ViewModels.Repository;
            if (repo == null)
                return;

            var button = sender as Button;
            if (button == null)
                return;

            var menu = new ContextMenu();
            var mode = Models.FilterMode.None;
            if (DataContext is Models.Tag tag)
            {
                mode = tag.FilterMode;

                if (mode != Models.FilterMode.None)
                {
                    var unset = new MenuItem();
                    unset.Header = App.Text("Repository.FilterCommits.Default");
                    unset.Click += (_, ev) =>
                    {
                        repo.SetTagFilterMode(tag, Models.FilterMode.None);
                        ev.Handled = true;
                    };

                    menu.Items.Add(unset);
                    menu.Items.Add(new MenuItem() { Header = "-" });
                }

                var include = new MenuItem();
                include.Icon = App.CreateMenuIcon("Icons.Filter");
                include.Header = App.Text("Repository.FilterCommits.Include");
                include.IsEnabled = mode != Models.FilterMode.Included;
                include.Click += (_, ev) =>
                {
                    repo.SetTagFilterMode(tag, Models.FilterMode.Included);
                    ev.Handled = true;
                };

                var exclude = new MenuItem();
                exclude.Icon = App.CreateMenuIcon("Icons.EyeClose");
                exclude.Header = App.Text("Repository.FilterCommits.Exclude");
                exclude.IsEnabled = mode != Models.FilterMode.Excluded;
                exclude.Click += (_, ev) =>
                {
                    repo.SetTagFilterMode(tag, Models.FilterMode.Excluded);
                    ev.Handled = true;
                };

                menu.Items.Add(include);
                menu.Items.Add(exclude);
            }
            else if (DataContext is ViewModels.BranchTreeNode node)
            {
                mode = node.FilterMode;

                if (mode != Models.FilterMode.None)
                {
                    var unset = new MenuItem();
                    unset.Header = App.Text("Repository.FilterCommits.Default");
                    unset.Click += (_, ev) =>
                    {
                        repo.SetBranchFilterMode(node, Models.FilterMode.None, false, true);
                        ev.Handled = true;
                    };

                    menu.Items.Add(unset);
                    menu.Items.Add(new MenuItem() { Header = "-" });
                }

                var include = new MenuItem();
                include.Icon = App.CreateMenuIcon("Icons.Filter");
                include.Header = App.Text("Repository.FilterCommits.Include");
                include.IsEnabled = mode != Models.FilterMode.Included;
                include.Click += (_, ev) =>
                {
                    repo.SetBranchFilterMode(node, Models.FilterMode.Included, false, true);
                    ev.Handled = true;
                };

                var exclude = new MenuItem();
                exclude.Icon = App.CreateMenuIcon("Icons.EyeClose");
                exclude.Header = App.Text("Repository.FilterCommits.Exclude");
                exclude.IsEnabled = mode != Models.FilterMode.Excluded;
                exclude.Click += (_, ev) =>
                {
                    repo.SetBranchFilterMode(node, Models.FilterMode.Excluded, false, true);
                    ev.Handled = true;
                };

                menu.Items.Add(include);
                menu.Items.Add(exclude);
            }

            if (mode == Models.FilterMode.None)
            {
                IsContextMenuOpening = true;
                menu.Closed += (_, _) => IsContextMenuOpening = false;
            }

            menu.Open(button);
            e.Handled = true;
        }
    }
}