From da5d6cb1d48c7622e8f4e365dbea1a840fd48e22 Mon Sep 17 00:00:00 2001
From: Lucas A <game4allyt@gmail.com>
Date: Fri, 12 Apr 2019 22:54:35 +0200
Subject: [PATCH] Add Beatmap fields to UserStatusSoloGame & UserStatusEditing
 so they can carry metadata about the played / edited beatmap

---
 osu.Game/Screens/Edit/Editor.cs           |  3 +++
 osu.Game/Screens/Play/Player.cs           |  3 +++
 osu.Game/Screens/Select/PlaySongSelect.cs |  3 +++
 osu.Game/Users/UserStatus.cs              | 27 +++++++++++++++++++++++
 4 files changed, 36 insertions(+)

diff --git a/osu.Game/Screens/Edit/Editor.cs b/osu.Game/Screens/Edit/Editor.cs
index 0ba1e74aca..bf00d23903 100644
--- a/osu.Game/Screens/Edit/Editor.cs
+++ b/osu.Game/Screens/Edit/Editor.cs
@@ -24,6 +24,7 @@ using osu.Game.Screens.Edit.Design;
 using osuTK.Input;
 using System.Collections.Generic;
 using osu.Framework;
+using osu.Game.Users;
 
 namespace osu.Game.Screens.Edit
 {
@@ -47,6 +48,8 @@ namespace osu.Game.Screens.Edit
         private DependencyContainer dependencies;
         private GameHost host;
 
+        protected override UserStatus ScreenStatus => new UserStatusEditing(Beatmap.Value.BeatmapInfo);
+
         protected override IReadOnlyDependencyContainer CreateChildDependencies(IReadOnlyDependencyContainer parent)
             => dependencies = new DependencyContainer(base.CreateChildDependencies(parent));
 
diff --git a/osu.Game/Screens/Play/Player.cs b/osu.Game/Screens/Play/Player.cs
index 0eebefec86..edb6f9f865 100644
--- a/osu.Game/Screens/Play/Player.cs
+++ b/osu.Game/Screens/Play/Player.cs
@@ -26,6 +26,7 @@ using osu.Game.Scoring;
 using osu.Game.Screens.Ranking;
 using osu.Game.Skinning;
 using osu.Game.Storyboards.Drawables;
+using osu.Game.Users;
 
 namespace osu.Game.Screens.Play
 {
@@ -33,6 +34,8 @@ namespace osu.Game.Screens.Play
     {
         protected override bool AllowBackButton => false; // handled by HoldForMenuButton
 
+        protected override UserStatus ScreenStatus => new UserStatusSoloGame(Beatmap.Value.BeatmapInfo);
+
         public override float BackgroundParallaxAmount => 0.1f;
 
         public override bool HideOverlaysOnEnter => true;
diff --git a/osu.Game/Screens/Select/PlaySongSelect.cs b/osu.Game/Screens/Select/PlaySongSelect.cs
index 340ceb6864..384064d2a8 100644
--- a/osu.Game/Screens/Select/PlaySongSelect.cs
+++ b/osu.Game/Screens/Select/PlaySongSelect.cs
@@ -7,6 +7,7 @@ using osu.Framework.Graphics.Sprites;
 using osu.Framework.Screens;
 using osu.Game.Graphics;
 using osu.Game.Screens.Play;
+using osu.Game.Users;
 using osuTK.Input;
 
 namespace osu.Game.Screens.Select
@@ -18,6 +19,8 @@ namespace osu.Game.Screens.Select
 
         public override bool AllowExternalScreenChange => true;
 
+        protected override UserStatus ScreenStatus => new UserStatusChoosingBeatmap();
+
         [BackgroundDependencyLoader]
         private void load(OsuColour colours)
         {
diff --git a/osu.Game/Users/UserStatus.cs b/osu.Game/Users/UserStatus.cs
index 14b4538a00..60e637d7e4 100644
--- a/osu.Game/Users/UserStatus.cs
+++ b/osu.Game/Users/UserStatus.cs
@@ -3,6 +3,7 @@
 
 using osuTK.Graphics;
 using osu.Game.Graphics;
+using osu.Game.Beatmaps;
 
 namespace osu.Game.Users
 {
@@ -41,7 +42,33 @@ namespace osu.Game.Users
 
     public class UserStatusSoloGame : UserStatusBusy
     {
+        public UserStatusSoloGame(BeatmapInfo info)
+        {
+            Beatmap = info;
+        }
+
         public override string Message => @"Solo Game";
+
+        public BeatmapInfo Beatmap { get; }
+    }
+
+    public class UserStatusEditing : UserStatusBusy
+    {
+        public UserStatusEditing(BeatmapInfo info)
+        {
+            Beatmap = info;
+        }
+
+        public override string Message => @"Editing a beatmap";
+
+        public override Color4 GetAppropriateColour(OsuColour colours) => colours.GreenDarker;
+
+        public BeatmapInfo Beatmap { get; }
+    }
+
+    public class UserStatusChoosingBeatmap : UserStatusOnline
+    {
+        public override string Message => @"Choosing a beatmap";
     }
 
     public class UserStatusMultiplayerGame : UserStatusBusy