osu-lazer/osu.Game/Graphics/UserInterface/PageSelector/PageSelector.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 osu.Framework.Graphics.Containers;
using osu.Framework.Graphics;
using osu.Framework.Bindables;
using osu.Framework.Allocation;
using osu.Framework.Graphics.Shapes;
using osu.Framework.Graphics.Sprites;
using osu.Framework.Extensions.Color4Extensions;
using System;
using osuTK;
using osu.Game.Graphics.Containers;
using System.Collections.Generic;
using osu.Framework.Extensions.IEnumerableExtensions;
using osu.Game.Graphics.Sprites;
using osuTK.Graphics;

namespace osu.Game.Graphics.UserInterface.PageSelector
{
    public class PageSelector : CompositeDrawable
    {
        public const int HEIGHT = 20;

        public readonly BindableInt CurrentPage = new BindableInt(1);
        public readonly BindableInt MaxPages = new BindableInt(1);

        private readonly FillFlowContainer itemsFlow;

        private readonly Button previousPageButton;
        private readonly Button nextPageButton;

        public PageSelector()
        {
            AutoSizeAxes = Axes.Both;
            InternalChild = new FillFlowContainer
            {
                AutoSizeAxes = Axes.Both,
                Direction = FillDirection.Horizontal,
                Children = new Drawable[]
                {
                    previousPageButton = new Button(false, "prev")
                    {
                        Action = () => CurrentPage.Value -= 1,
                    },
                    itemsFlow = new FillFlowContainer
                    {
                        AutoSizeAxes = Axes.Both,
                        Direction = FillDirection.Horizontal,
                    },
                    nextPageButton = new Button(true, "next")
                    {
                        Action = () => CurrentPage.Value += 1
                    }
                }
            };
        }

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

            MaxPages.BindValueChanged(pagesAmount => onMaxPagesChanged(pagesAmount.NewValue), true);
            CurrentPage.BindValueChanged(page => onCurrentPageChanged(page.NewValue), true);
        }

        private void onMaxPagesChanged(int pagesAmount)
        {
            if (pagesAmount < 1)
            {
                MaxPages.Value = 1;
                return;
            }

            if (CurrentPage.Value > pagesAmount)
            {
                CurrentPage.Value = pagesAmount;
                return;
            }

            redraw();
        }

        private void onCurrentPageChanged(int newPage)
        {
            if (newPage > MaxPages.Value)
            {
                CurrentPage.Value = MaxPages.Value;
                return;
            }

            if (newPage < 1)
            {
                CurrentPage.Value = 1;
                return;
            }

            redraw();
        }

        private void redraw()
        {
            int newPage = CurrentPage.Value;
            int maxPages = MaxPages.Value;

            previousPageButton.Enabled.Value = newPage != 1;
            nextPageButton.Enabled.Value = newPage != maxPages;

            itemsFlow.Clear();

            if (newPage > 3)
                addDrawablePage(1);

            if (newPage > 4)
                addPlaceholder();

            for (int i = Math.Max(newPage - 2, 1); i <= Math.Min(newPage + 2, maxPages); i++)
            {
                if (i == newPage)
                    addDrawableCurrentPage();
                else
                    addDrawablePage(i);
            }

            if (newPage + 2 < maxPages - 1)
                addPlaceholder();

            if (newPage + 2 < maxPages)
                addDrawablePage(maxPages);
        }

        private void addDrawablePage(int page) => itemsFlow.Add(new DrawablePage(page.ToString())
        {
            Action = () => CurrentPage.Value = page,
        });

        private void addPlaceholder() => itemsFlow.Add(new Placeholder());

        private void addDrawableCurrentPage() => itemsFlow.Add(new SelectedPage(CurrentPage.Value.ToString()));

        private abstract class PageItem : OsuHoverContainer
        {
            private const int margin = 10;
            private const int height = 20;

            protected override Container<Drawable> Content => contentContainer;

            private readonly CircularContainer contentContainer;

            protected PageItem(string text)
            {
                AutoSizeAxes = Axes.X;
                Height = height;

                base.Content.Add(contentContainer = new CircularContainer
                {
                    AutoSizeAxes = Axes.X,
                    RelativeSizeAxes = Axes.Y,
                    Masking = true,
                });

                var background = CreateBackground();
                if (background != null)
                    Add(background);

                var drawableText = CreateText(text);

                if (drawableText != null)
                {
                    drawableText.Margin = new MarginPadding { Horizontal = margin };
                    Add(drawableText);
                }
            }

            [BackgroundDependencyLoader]
            private void load(OsuColour colours)
            {
                IdleColour = colours.Lime;
                HoverColour = colours.Lime.Lighten(20f);
            }

            protected abstract Drawable CreateText(string text);

            protected virtual Drawable CreateBackground() => null;
        }

        private class DrawablePage : PageItem
        {
            protected SpriteText SpriteText;

            protected override IEnumerable<Drawable> EffectTargets => new[] { SpriteText };

            public DrawablePage(string text)
                : base(text)
            {
            }

            protected override Drawable CreateText(string text) => SpriteText = new OsuSpriteText
            {
                Anchor = Anchor.Centre,
                Origin = Anchor.Centre,
                Text = text,
                Font = OsuFont.GetFont(size: 12),
            };
        }

        private class SelectedPage : DrawablePage
        {
            private Box background;

            protected override IEnumerable<Drawable> EffectTargets => new[] { background };

            public SelectedPage(string text)
                : base(text)
            {
            }

            protected override Drawable CreateBackground() => background = new Box
            {
                RelativeSizeAxes = Axes.Both,
            };

            [BackgroundDependencyLoader]
            private void load(OsuColour colours)
            {
                SpriteText.Colour = colours.GreySeafoamDark;
            }
        }

        private class Placeholder : DrawablePage
        {
            public Placeholder()
                : base("...")
            {
            }
        }

        private class Button : PageItem
        {
            private const int duration = 100;

            private Box background;
            private FillFlowContainer textContainer;
            private SpriteIcon icon;
            private readonly Box fadeBox;

            protected override IEnumerable<Drawable> EffectTargets => new[] { textContainer };

            public Button(bool rightAligned, string text)
                : base(text)
            {
                var alignment = rightAligned ? Anchor.x0 : Anchor.x2;

                textContainer.ForEach(drawable =>
                {
                    drawable.Anchor = Anchor.y1 | alignment;
                    drawable.Origin = Anchor.y1 | alignment;
                });

                icon.Icon = alignment == Anchor.x2 ? FontAwesome.Solid.ChevronLeft : FontAwesome.Solid.ChevronRight;

                Add(fadeBox = new Box
                {
                    RelativeSizeAxes = Axes.Both,
                    Colour = Color4.Black.Opacity(100)
                });
            }

            [BackgroundDependencyLoader]
            private void load(OsuColour colours)
            {
                background.Colour = colours.GreySeafoamDark;
            }

            protected override void LoadComplete()
            {
                base.LoadComplete();
                Enabled.BindValueChanged(onEnabledChanged, true);
            }

            private void onEnabledChanged(ValueChangedEvent<bool> enabled)
            {
                fadeBox.FadeTo(enabled.NewValue ? 0 : 1, duration);
            }

            protected override Drawable CreateBackground() => background = new Box
            {
                RelativeSizeAxes = Axes.Both,
            };

            protected override Drawable CreateText(string text) => textContainer = new FillFlowContainer
            {
                AutoSizeAxes = Axes.Both,
                Anchor = Anchor.Centre,
                Origin = Anchor.Centre,
                Direction = FillDirection.Horizontal,
                Children = new Drawable[]
                {
                    new OsuSpriteText
                    {
                        Text = text.ToUpper(),
                        Font = OsuFont.GetFont(size: 12),
                    },
                    icon = new SpriteIcon
                    {
                        Size = new Vector2(8),
                    },
                }
            };
        }
    }
}