Begin shaping the spectator streaming component

2025-08-08 00:53:56 +09:00 · 2020-10-22 13:41:54 +09:00
parent db4dd3182b
commit 93db75bd41
3 changed files with 58 additions and 23 deletions
--- a/osu.Game/Online/Spectator/SpectatorStreamingClient.cs
+++ b/osu.Game/Online/Spectator/SpectatorStreamingClient.cs
@ -0,0 +1,115 @@
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using System.Threading.Tasks;
+using Microsoft.AspNetCore.SignalR.Client;
+using Microsoft.Extensions.DependencyInjection;
+using osu.Framework.Allocation;
+using osu.Framework.Bindables;
+using osu.Framework.Graphics;
+using osu.Game.Online.API;
+
+namespace osu.Game.Online.Spectator
+{
+    public class SpectatorStreamingClient : Component, ISpectatorClient
+    {
+        private HubConnection connection;
+
+        private readonly List<string> watchingUsers = new List<string>();
+
+        private readonly IBindable<APIState> apiState = new Bindable<APIState>();
+
+        [Resolved]
+        private APIAccess api { get; set; }
+
+        [BackgroundDependencyLoader]
+        private void load()
+        {
+            apiState.BindTo(api.State);
+            apiState.BindValueChanged(apiStateChanged, true);
+        }
+
+        private void apiStateChanged(ValueChangedEvent<APIState> state)
+        {
+            switch (state.NewValue)
+            {
+                case APIState.Failing:
+                case APIState.Offline:
+                    connection?.StopAsync();
+                    connection = null;
+                    break;
+
+                case APIState.Online:
+                    connect();
+                    break;
+            }
+        }
+
+#if DEBUG
+        private const string endpoint = "http://localhost:5009/spectator";
+#else
+        private const string endpoint = "https://spectator.ppy.sh/spectator";
+#endif
+
+        private void connect()
+        {
+            connection = new HubConnectionBuilder()
+                         .WithUrl(endpoint, options =>
+                         {
+                             options.Headers.Add("Authorization", $"Bearer {api.AccessToken}");
+                         })
+                         .AddMessagePackProtocol()
+                         .Build();
+
+            // until strong typed client support is added, each method must be manually bound (see https://github.com/dotnet/aspnetcore/issues/15198)
+            connection.On<string, int>(nameof(ISpectatorClient.UserBeganPlaying), ((ISpectatorClient)this).UserBeganPlaying);
+            connection.On<string, FrameDataBundle>(nameof(ISpectatorClient.UserSentFrames), ((ISpectatorClient)this).UserSentFrames);
+            connection.On<string, int>(nameof(ISpectatorClient.UserFinishedPlaying), ((ISpectatorClient)this).UserFinishedPlaying);
+
+            connection.StartAsync();
+        }
+
+        Task ISpectatorClient.UserBeganPlaying(string userId, int beatmapId)
+        {
+            if (connection.ConnectionId != userId)
+            {
+                if (watchingUsers.Contains(userId))
+                {
+                    Console.WriteLine($"{connection.ConnectionId} received began playing for already watched user {userId}");
+                }
+                else
+                {
+                    Console.WriteLine($"{connection.ConnectionId} requesting watch other user {userId}");
+                    WatchUser(userId);
+                    watchingUsers.Add(userId);
+                }
+            }
+            else
+            {
+                Console.WriteLine($"{connection.ConnectionId} Received user playing event for self {beatmapId}");
+            }
+
+            return Task.CompletedTask;
+        }
+
+        Task ISpectatorClient.UserFinishedPlaying(string userId, int beatmapId)
+        {
+            Console.WriteLine($"{connection.ConnectionId} Received user finished event {beatmapId}");
+            return Task.CompletedTask;
+        }
+
+        Task ISpectatorClient.UserSentFrames(string userId, FrameDataBundle data)
+        {
+            Console.WriteLine($"{connection.ConnectionId} Received frames from {userId}: {data.Frames.First().ToString()}");
+            return Task.CompletedTask;
+        }
+
+        public Task BeginPlaying(int beatmapId) => connection.SendAsync(nameof(ISpectatorServer.BeginPlaySession), beatmapId);
+
+        public Task SendFrames(FrameDataBundle data) => connection.SendAsync(nameof(ISpectatorServer.SendFrameData), data);
+
+        public Task EndPlaying(int beatmapId) => connection.SendAsync(nameof(ISpectatorServer.EndPlaySession), beatmapId);
+
+        private Task WatchUser(string userId) => connection.SendAsync(nameof(ISpectatorServer.StartWatchingUser), userId);
+    }
+}