mirror of
https://github.com/ppy/osu.git
synced 2024-12-24 13:22:55 +08:00
163 lines
5.4 KiB
C#
163 lines
5.4 KiB
C#
// 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 System;
|
|
using System.Text;
|
|
using DiscordRPC;
|
|
using DiscordRPC.Message;
|
|
using osu.Framework.Allocation;
|
|
using osu.Framework.Bindables;
|
|
using osu.Framework.Graphics;
|
|
using osu.Framework.Logging;
|
|
using osu.Game.Configuration;
|
|
using osu.Game.Extensions;
|
|
using osu.Game.Online.API;
|
|
using osu.Game.Online.API.Requests.Responses;
|
|
using osu.Game.Rulesets;
|
|
using osu.Game.Users;
|
|
using LogLevel = osu.Framework.Logging.LogLevel;
|
|
|
|
namespace osu.Desktop
|
|
{
|
|
internal class DiscordRichPresence : Component
|
|
{
|
|
private const string client_id = "367827983903490050";
|
|
|
|
private DiscordRpcClient client;
|
|
|
|
[Resolved]
|
|
private IBindable<RulesetInfo> ruleset { get; set; }
|
|
|
|
private IBindable<APIUser> user;
|
|
|
|
private readonly IBindable<UserStatus> status = new Bindable<UserStatus>();
|
|
private readonly IBindable<UserActivity> activity = new Bindable<UserActivity>();
|
|
|
|
private readonly Bindable<DiscordRichPresenceMode> privacyMode = new Bindable<DiscordRichPresenceMode>();
|
|
|
|
private readonly RichPresence presence = new RichPresence
|
|
{
|
|
Assets = new Assets { LargeImageKey = "osu_logo_lazer", }
|
|
};
|
|
|
|
[BackgroundDependencyLoader]
|
|
private void load(IAPIProvider provider, OsuConfigManager config)
|
|
{
|
|
client = new DiscordRpcClient(client_id)
|
|
{
|
|
SkipIdenticalPresence = false // handles better on discord IPC loss, see updateStatus call in onReady.
|
|
};
|
|
|
|
client.OnReady += onReady;
|
|
|
|
// safety measure for now, until we performance test / improve backoff for failed connections.
|
|
client.OnConnectionFailed += (_, __) => client.Deinitialize();
|
|
|
|
client.OnError += (_, e) => Logger.Log($"An error occurred with Discord RPC Client: {e.Code} {e.Message}", LoggingTarget.Network);
|
|
|
|
config.BindWith(OsuSetting.DiscordRichPresence, privacyMode);
|
|
|
|
(user = provider.LocalUser.GetBoundCopy()).BindValueChanged(u =>
|
|
{
|
|
status.UnbindBindings();
|
|
status.BindTo(u.NewValue.Status);
|
|
|
|
activity.UnbindBindings();
|
|
activity.BindTo(u.NewValue.Activity);
|
|
}, true);
|
|
|
|
ruleset.BindValueChanged(_ => updateStatus());
|
|
status.BindValueChanged(_ => updateStatus());
|
|
activity.BindValueChanged(_ => updateStatus());
|
|
privacyMode.BindValueChanged(_ => updateStatus());
|
|
|
|
client.Initialize();
|
|
}
|
|
|
|
private void onReady(object _, ReadyMessage __)
|
|
{
|
|
Logger.Log("Discord RPC Client ready.", LoggingTarget.Network, LogLevel.Debug);
|
|
updateStatus();
|
|
}
|
|
|
|
private void updateStatus()
|
|
{
|
|
if (!client.IsInitialized)
|
|
return;
|
|
|
|
if (status.Value is UserStatusOffline || privacyMode.Value == DiscordRichPresenceMode.Off)
|
|
{
|
|
client.ClearPresence();
|
|
return;
|
|
}
|
|
|
|
if (status.Value is UserStatusOnline && activity.Value != null)
|
|
{
|
|
presence.State = truncate(activity.Value.Status);
|
|
presence.Details = truncate(getDetails(activity.Value));
|
|
}
|
|
else
|
|
{
|
|
presence.State = "Idle";
|
|
presence.Details = string.Empty;
|
|
}
|
|
|
|
// update user information
|
|
if (privacyMode.Value == DiscordRichPresenceMode.Limited)
|
|
presence.Assets.LargeImageText = string.Empty;
|
|
else
|
|
presence.Assets.LargeImageText = $"{user.Value.Username}" + (user.Value.Statistics?.GlobalRank > 0 ? $" (rank #{user.Value.Statistics.GlobalRank:N0})" : string.Empty);
|
|
|
|
// update ruleset
|
|
presence.Assets.SmallImageKey = ruleset.Value.IsLegacyRuleset() ? $"mode_{ruleset.Value.OnlineID}" : "mode_custom";
|
|
presence.Assets.SmallImageText = ruleset.Value.Name;
|
|
|
|
client.SetPresence(presence);
|
|
}
|
|
|
|
private static readonly int ellipsis_length = Encoding.UTF8.GetByteCount(new[] { '…' });
|
|
|
|
private string truncate(string str)
|
|
{
|
|
if (Encoding.UTF8.GetByteCount(str) <= 128)
|
|
return str;
|
|
|
|
ReadOnlyMemory<char> strMem = str.AsMemory();
|
|
|
|
do
|
|
{
|
|
strMem = strMem[..^1];
|
|
} while (Encoding.UTF8.GetByteCount(strMem.Span) + ellipsis_length > 128);
|
|
|
|
return string.Create(strMem.Length + 1, strMem, (span, mem) =>
|
|
{
|
|
mem.Span.CopyTo(span);
|
|
span[^1] = '…';
|
|
});
|
|
}
|
|
|
|
private string getDetails(UserActivity activity)
|
|
{
|
|
switch (activity)
|
|
{
|
|
case UserActivity.InGame game:
|
|
return game.BeatmapInfo.ToString();
|
|
|
|
case UserActivity.Editing edit:
|
|
return edit.BeatmapInfo.ToString();
|
|
|
|
case UserActivity.InLobby lobby:
|
|
return privacyMode.Value == DiscordRichPresenceMode.Limited ? string.Empty : lobby.Room.Name.Value;
|
|
}
|
|
|
|
return string.Empty;
|
|
}
|
|
|
|
protected override void Dispose(bool isDisposing)
|
|
{
|
|
client.Dispose();
|
|
base.Dispose(isDisposing);
|
|
}
|
|
}
|
|
}
|