osu-lazer/osu.Game/Graphics/Sprites/OsuLinkSpriteText.cs

// Copyright (c) 2007-2017 ppy Pty Ltd <contact@ppy.sh>.
// Licensed under the MIT Licence - https://raw.githubusercontent.com/ppy/osu/master/LICENCE

using OpenTK.Graphics;
using osu.Framework.Allocation;
using osu.Framework.Graphics;
using osu.Framework.Graphics.Containers;
using osu.Framework.Input;
using osu.Game.Beatmaps;
using osu.Game.Graphics.Containers;
using osu.Game.Overlays;
using System;
using System.Collections.Generic;
using System.Diagnostics;
using System.Linq;
using System.Text;
using System.Threading.Tasks;

namespace osu.Game.Graphics.Sprites
{
    public class OsuLinkSpriteText : OsuSpriteText
    {
        private ChatOverlay chat;

        private readonly OsuHoverContainer content;

        private BeatmapSetOverlay beatmapSetOverlay;

        public override bool HandleInput => content.Action != null;

        protected override Container<Drawable> Content => content ?? (Container<Drawable>)this;

        protected override IEnumerable<Drawable> FlowingChildren => Children;

        private string url;

        public string Url
        {
            get
            {
                return url;
            }
            set
            {
                if (value != null)
                {
                    url = value;
                    loadAction();
                }
            }
        }

        public OsuLinkSpriteText()
        {
            AddInternal(content = new OsuHoverContainer
            {
                AutoSizeAxes = Axes.Both,
            });
        }

        [BackgroundDependencyLoader]
        private void load(BeatmapSetOverlay beatmapSetOverlay, ChatOverlay chat)
        {
            this.beatmapSetOverlay = beatmapSetOverlay;
            this.chat = chat;
        }

        private void loadAction()
        {
            if (Url == null || String.IsNullOrEmpty(Url))
                return;

            var url = Url;

            // Client-internal stuff
            if (url.StartsWith("osu://"))
            {
                var firstPath = url.Substring(6, 5);
                url = url.Substring(11);

                if (firstPath == "chan/")
                {
                    var nextSlashIndex = url.IndexOf('/');
                    var channelName = nextSlashIndex != -1 ? url.Remove(nextSlashIndex) : url;

                    var foundChannel = chat.AvailableChannels.Find(channel => channel.Name == channelName);

                    if (foundChannel != null)
                        content.Action = () => chat.OpenChannel(foundChannel);
                }
                else if (firstPath == "edit/")
                {
                    // Open editor here, then goto specified time
                    // how to push new screen from here? we'll see
                    content.Action = () => new Framework.Screens.Screen().Push(new Screens.Edit.Editor());
                }
                else
                    throw new ArgumentException($"Unknown osu:// link at {nameof(OsuLinkSpriteText)} ({firstPath}).");
            }
            else if (url.StartsWith("http://") || url.StartsWith("https://"))
            {
                var osuUrlIndex = url.IndexOf("osu.ppy.sh/");
                if (osuUrlIndex == -1)
                {
                    content.Action = () => Process.Start(url);
                    return;
                }

                url = url.Substring(osuUrlIndex + 11);
                if (url.StartsWith("s/") || url.StartsWith("beatmapsets/") || url.StartsWith("d/"))
                    content.Action = () => beatmapSetOverlay.ShowBeatmapSet(getIdFromUrl(url));
                else if (url.StartsWith("b/") || url.StartsWith("beatmaps/"))
                    content.Action = () => beatmapSetOverlay.ShowBeatmap(getIdFromUrl(url));
            }
            else
                content.Action = () => Process.Start(url);
        }

        private int getIdFromUrl(string url)
        {
            var lastSlashIndex = url.LastIndexOf('/');
            // Remove possible trailing slash
            if (lastSlashIndex == url.Length)
            {
                url = url.Remove(url.Length - 1);
                lastSlashIndex = url.LastIndexOf('/');
            }

            var lastQuestionMarkIndex = url.LastIndexOf('?');
            // Filter out possible queries like mode specifications (e.g. /b/252238?m=0)
            if (lastQuestionMarkIndex > lastSlashIndex)
                url = url.Remove(lastQuestionMarkIndex);

            return int.Parse(url.Substring(lastSlashIndex + 1));
        }
    }
}