osu-lazer/osu.Game/Screens/Select/FooterV2/BeatmapOptionsPopover.cs

// Copyright (c) ppy Pty Ltd <contact@ppy.sh>. Licensed under the MIT Licence.
// See the LICENCE file in the repository root for full licence text.

using System;
using System.Linq;
using osu.Framework.Allocation;
using osu.Framework.Bindables;
using osu.Framework.Graphics;
using osu.Framework.Graphics.Containers;
using osu.Framework.Graphics.Sprites;
using osu.Framework.Input.Events;
using osu.Framework.Localisation;
using osu.Game.Collections;
using osu.Game.Graphics;
using osu.Game.Graphics.Sprites;
using osu.Game.Graphics.UserInterface;
using osu.Game.Graphics.UserInterfaceV2;
using osu.Game.Overlays;
using osuTK;
using osuTK.Graphics;
using osuTK.Input;

namespace osu.Game.Screens.Select.FooterV2
{
    public partial class BeatmapOptionsPopover : OsuPopover
    {
        private FillFlowContainer<OptionButton> buttonFlow = null!;
        private readonly FooterButtonOptionsV2 footerButton;

        public BeatmapOptionsPopover(FooterButtonOptionsV2 footerButton)
        {
            this.footerButton = footerButton;
        }

        [BackgroundDependencyLoader]
        private void load(ManageCollectionsDialog? manageCollectionsDialog, SongSelect? songSelect, OsuColour colours)
        {
            Content.Padding = new MarginPadding(5);

            Child = buttonFlow = new FillFlowContainer<OptionButton>
            {
                AutoSizeAxes = Axes.Both,
                Direction = FillDirection.Vertical,
                Spacing = new Vector2(3),
            };

            addButton(@"Manage collections", FontAwesome.Solid.Book, () => manageCollectionsDialog?.Show());
            addButton(@"Delete all difficulties", FontAwesome.Solid.Trash, () => songSelect?.DeleteBeatmap(), colours.Red);
            addButton(@"Remove from unplayed", FontAwesome.Regular.TimesCircle, null);
            addButton(@"Clear local scores", FontAwesome.Solid.Eraser, () => songSelect?.ClearScores());

            if (songSelect != null && songSelect.AllowEditing)
                addButton(@"Edit beatmap", FontAwesome.Solid.PencilAlt, () => songSelect.Edit());
        }

        protected override void LoadComplete()
        {
            base.LoadComplete();

            ScheduleAfterChildren(() => GetContainingInputManager().ChangeFocus(this));
        }

        private void addButton(LocalisableString text, IconUsage icon, Action? action, Color4? colour = null)
        {
            var button = new OptionButton
            {
                Text = text,
                Icon = icon,
                TextColour = colour,
                Action = () =>
                {
                    Hide();
                    action?.Invoke();
                },
            };

            buttonFlow.Add(button);
        }

        private partial class OptionButton : OsuButton
        {
            public IconUsage Icon { get; init; }
            public Color4? TextColour { get; init; }

            public OptionButton()
            {
                Size = new Vector2(265, 50);
            }

            [BackgroundDependencyLoader]
            private void load(OverlayColourProvider colourProvider)
            {
                BackgroundColour = colourProvider.Background3;

                SpriteText.Colour = TextColour ?? Color4.White;
                Content.CornerRadius = 10;

                Add(new SpriteIcon
                {
                    Blending = BlendingParameters.Additive,
                    Anchor = Anchor.CentreLeft,
                    Origin = Anchor.CentreLeft,
                    Size = new Vector2(17),
                    X = 15,
                    Icon = Icon,
                    Colour = TextColour ?? Color4.White,
                });
            }

            protected override SpriteText CreateText() => new OsuSpriteText
            {
                Depth = -1,
                Origin = Anchor.CentreLeft,
                Anchor = Anchor.CentreLeft,
                X = 40
            };
        }

        protected override bool OnKeyDown(KeyDownEvent e)
        {
            // don't absorb control as ToolbarRulesetSelector uses control + number to navigate
            if (e.ControlPressed) return false;

            if (!e.Repeat && e.Key >= Key.Number1 && e.Key <= Key.Number9)
            {
                int requested = e.Key - Key.Number1;

                OptionButton? found = buttonFlow.Children.ElementAtOrDefault(requested);

                if (found != null)
                {
                    found.TriggerClick();
                    return true;
                }
            }

            return base.OnKeyDown(e);
        }

        protected override void UpdateState(ValueChangedEvent<Visibility> state)
        {
            base.UpdateState(state);

            if (state.NewValue == Visibility.Hidden)
                footerButton.IsActive.Value = false;
        }
    }
}