osu-lazer/osu.Game/Screens/Play/HUD/PlayerTeamFlag.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.Allocation;
using osu.Framework.Bindables;
using osu.Framework.Extensions;
using osu.Framework.Graphics;
using osu.Framework.Graphics.Containers;
using osu.Game.Database;
using osu.Game.Online.API;
using osu.Game.Online.API.Requests.Responses;
using osu.Game.Skinning;
using osu.Game.Users.Drawables;
using osuTK;

namespace osu.Game.Screens.Play.HUD
{
    public partial class PlayerTeamFlag : CompositeDrawable, ISerialisableDrawable
    {
        protected override bool ReceivePositionalInputAtSubTree(Vector2 screenSpacePos) => false;

        private readonly UpdateableTeamFlag flag;

        private const float default_size = 40f;

        [Resolved]
        private GameplayState? gameplayState { get; set; }

        [Resolved]
        private IAPIProvider api { get; set; } = null!;

        private IBindable<APIUser>? apiUser;

        public PlayerTeamFlag()
        {
            Size = new Vector2(default_size, default_size / 2f);

            InternalChild = flag = new UpdateableTeamFlag
            {
                RelativeSizeAxes = Axes.Both,
            };
        }

        [BackgroundDependencyLoader]
        private void load(UserLookupCache userLookupCache)
        {
            if (gameplayState != null)
            {
                if (gameplayState.Score.ScoreInfo.User.Team != null)
                    flag.Team = gameplayState.Score.ScoreInfo.User.Team;
                else
                {
                    // We only store very basic information about a user to realm, so there's a high chance we don't have the team information.
                    userLookupCache.GetUserAsync(gameplayState.Score.ScoreInfo.User.Id)
                                   .ContinueWith(task => Schedule(() => flag.Team = task.GetResultSafely()?.Team));
                }
            }
            else
            {
                apiUser = api.LocalUser.GetBoundCopy();
                apiUser.BindValueChanged(u => flag.Team = u.NewValue.Team, true);
            }
        }

        public bool UsesFixedAnchor { get; set; }
    }
}