297 lines
No EOL
11 KiB
C#
297 lines
No EOL
11 KiB
C#
#nullable disable
|
|
using Ellie.Modules.Searches.Common;
|
|
using Newtonsoft.Json;
|
|
|
|
namespace Ellie.Modules.Searches;
|
|
|
|
public partial class Searches
|
|
{
|
|
[Group]
|
|
public partial class OsuCommands : EllieModule
|
|
{
|
|
private readonly IBotCredentials _creds;
|
|
private readonly IHttpClientFactory _httpFactory;
|
|
|
|
public OsuCommands(IBotCredentials creds, IHttpClientFactory factory)
|
|
{
|
|
_creds = creds;
|
|
_httpFactory = factory;
|
|
}
|
|
|
|
[Cmd]
|
|
public async Task Osu(string user, [Leftover] string mode = null)
|
|
{
|
|
if (string.IsNullOrWhiteSpace(user))
|
|
return;
|
|
|
|
using var http = _httpFactory.CreateClient();
|
|
var modeNumber = string.IsNullOrWhiteSpace(mode) ? 0 : ResolveGameMode(mode);
|
|
|
|
try
|
|
{
|
|
if (string.IsNullOrWhiteSpace(_creds.OsuApiKey))
|
|
{
|
|
await ReplyErrorLocalizedAsync(strs.osu_api_key);
|
|
return;
|
|
}
|
|
|
|
var smode = ResolveGameMode(modeNumber);
|
|
var userReq = $"https://osu.ppy.sh/api/get_user?k={_creds.OsuApiKey}&u={user}&m={modeNumber}";
|
|
var userResString = await http.GetStringAsync(userReq);
|
|
var objs = JsonConvert.DeserializeObject<List<OsuUserData>>(userResString);
|
|
|
|
if (objs.Count == 0)
|
|
{
|
|
await ReplyErrorLocalizedAsync(strs.osu_user_not_found);
|
|
return;
|
|
}
|
|
|
|
var obj = objs[0];
|
|
var userId = obj.UserId;
|
|
|
|
await ctx.Channel.EmbedAsync(_eb.Create()
|
|
.WithOkColor()
|
|
.WithTitle($"osu! {smode} profile for {user}")
|
|
.WithThumbnailUrl($"https://a.ppy.sh/{userId}")
|
|
.WithDescription($"https://osu.ppy.sh/u/{userId}")
|
|
.AddField("Official Rank", $"#{obj.PpRank}", true)
|
|
.AddField("Country Rank",
|
|
$"#{obj.PpCountryRank} :flag_{obj.Country.ToLower()}:",
|
|
true)
|
|
.AddField("Total PP", Math.Round(obj.PpRaw, 2), true)
|
|
.AddField("Accuracy", Math.Round(obj.Accuracy, 2) + "%", true)
|
|
.AddField("Playcount", obj.Playcount, true)
|
|
.AddField("Level", Math.Round(obj.Level), true));
|
|
}
|
|
catch (ArgumentOutOfRangeException)
|
|
{
|
|
await ReplyErrorLocalizedAsync(strs.osu_user_not_found);
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
await ReplyErrorLocalizedAsync(strs.osu_failed);
|
|
Log.Warning(ex, "Osu command failed");
|
|
}
|
|
}
|
|
|
|
[Cmd]
|
|
public async Task Gatari(string user, [Leftover] string mode = null)
|
|
{
|
|
using var http = _httpFactory.CreateClient();
|
|
var modeNumber = string.IsNullOrWhiteSpace(mode) ? 0 : ResolveGameMode(mode);
|
|
|
|
var modeStr = ResolveGameMode(modeNumber);
|
|
var resString = await http.GetStringAsync($"https://api.gatari.pw/user/stats?u={user}&mode={modeNumber}");
|
|
|
|
var statsResponse = JsonConvert.DeserializeObject<GatariUserStatsResponse>(resString);
|
|
if (statsResponse.Code != 200 || statsResponse.Stats.Id == 0)
|
|
{
|
|
await ReplyErrorLocalizedAsync(strs.osu_user_not_found);
|
|
return;
|
|
}
|
|
|
|
var usrResString = await http.GetStringAsync($"https://api.gatari.pw/users/get?u={user}");
|
|
|
|
var userData = JsonConvert.DeserializeObject<GatariUserResponse>(usrResString).Users[0];
|
|
var userStats = statsResponse.Stats;
|
|
|
|
var embed = _eb.Create()
|
|
.WithOkColor()
|
|
.WithTitle($"osu!Gatari {modeStr} profile for {user}")
|
|
.WithThumbnailUrl($"https://a.gatari.pw/{userStats.Id}")
|
|
.WithDescription($"https://osu.gatari.pw/u/{userStats.Id}")
|
|
.AddField("Official Rank", $"#{userStats.Rank}", true)
|
|
.AddField("Country Rank",
|
|
$"#{userStats.CountryRank} :flag_{userData.Country.ToLower()}:",
|
|
true)
|
|
.AddField("Total PP", userStats.Pp, true)
|
|
.AddField("Accuracy", $"{Math.Round(userStats.AvgAccuracy, 2)}%", true)
|
|
.AddField("Playcount", userStats.Playcount, true)
|
|
.AddField("Level", userStats.Level, true);
|
|
|
|
await ctx.Channel.EmbedAsync(embed);
|
|
}
|
|
|
|
[Cmd]
|
|
public async Task Osu5(string user, [Leftover] string mode = null)
|
|
{
|
|
if (string.IsNullOrWhiteSpace(_creds.OsuApiKey))
|
|
{
|
|
await SendErrorAsync("An osu! API key is required.");
|
|
return;
|
|
}
|
|
|
|
if (string.IsNullOrWhiteSpace(user))
|
|
{
|
|
await SendErrorAsync("Please provide a username.");
|
|
return;
|
|
}
|
|
|
|
using var http = _httpFactory.CreateClient();
|
|
var m = 0;
|
|
if (!string.IsNullOrWhiteSpace(mode))
|
|
m = ResolveGameMode(mode);
|
|
|
|
var reqString = "https://osu.ppy.sh/api/get_user_best"
|
|
+ $"?k={_creds.OsuApiKey}"
|
|
+ $"&u={Uri.EscapeDataString(user)}"
|
|
+ "&type=string"
|
|
+ "&limit=5"
|
|
+ $"&m={m}";
|
|
|
|
var resString = await http.GetStringAsync(reqString);
|
|
var obj = JsonConvert.DeserializeObject<List<OsuUserBests>>(resString);
|
|
|
|
var mapTasks = obj.Select(async item =>
|
|
{
|
|
var mapReqString = "https://osu.ppy.sh/api/get_beatmaps"
|
|
+ $"?k={_creds.OsuApiKey}"
|
|
+ $"&b={item.BeatmapId}";
|
|
|
|
var mapResString = await http.GetStringAsync(mapReqString);
|
|
var map = JsonConvert.DeserializeObject<List<OsuMapData>>(mapResString).FirstOrDefault();
|
|
if (map is null)
|
|
return default;
|
|
var pp = Math.Round(item.Pp, 2);
|
|
var acc = CalculateAcc(item, m);
|
|
var mods = ResolveMods(item.EnabledMods);
|
|
|
|
var title = $"{map.Artist}-{map.Title} ({map.Version})";
|
|
var desc = $@"[/b/{item.BeatmapId}](https://osu.ppy.sh/b/{item.BeatmapId})
|
|
{pp + "pp",-7} | {acc + "%",-7}
|
|
";
|
|
if (mods != "+")
|
|
desc += Format.Bold(mods);
|
|
|
|
return (title, desc);
|
|
});
|
|
|
|
var eb = _eb.Create().WithOkColor().WithTitle($"Top 5 plays for {user}");
|
|
|
|
var mapData = await mapTasks.WhenAll();
|
|
foreach (var (title, desc) in mapData.Where(x => x != default))
|
|
eb.AddField(title, desc);
|
|
|
|
await ctx.Channel.EmbedAsync(eb);
|
|
}
|
|
|
|
//https://osu.ppy.sh/wiki/Accuracy
|
|
private static double CalculateAcc(OsuUserBests play, int mode)
|
|
{
|
|
double hitPoints;
|
|
double totalHits;
|
|
if (mode == 0)
|
|
{
|
|
hitPoints = (play.Count50 * 50) + (play.Count100 * 100) + (play.Count300 * 300);
|
|
totalHits = play.Count50 + play.Count100 + play.Count300 + play.Countmiss;
|
|
totalHits *= 300;
|
|
}
|
|
else if (mode == 1)
|
|
{
|
|
hitPoints = (play.Countmiss * 0) + (play.Count100 * 0.5) + play.Count300;
|
|
totalHits = (play.Countmiss + play.Count100 + play.Count300) * 300;
|
|
hitPoints *= 300;
|
|
}
|
|
else if (mode == 2)
|
|
{
|
|
hitPoints = play.Count50 + play.Count100 + play.Count300;
|
|
totalHits = play.Countmiss + play.Count50 + play.Count100 + play.Count300 + play.Countkatu;
|
|
}
|
|
else
|
|
{
|
|
hitPoints = (play.Count50 * 50)
|
|
+ (play.Count100 * 100)
|
|
+ (play.Countkatu * 200)
|
|
+ ((play.Count300 + play.Countgeki) * 300);
|
|
|
|
totalHits = (play.Countmiss
|
|
+ play.Count50
|
|
+ play.Count100
|
|
+ play.Countkatu
|
|
+ play.Count300
|
|
+ play.Countgeki)
|
|
* 300;
|
|
}
|
|
|
|
|
|
return Math.Round(hitPoints / totalHits * 100, 2);
|
|
}
|
|
|
|
private static int ResolveGameMode(string mode)
|
|
{
|
|
switch (mode.ToUpperInvariant())
|
|
{
|
|
case "STD":
|
|
case "STANDARD":
|
|
return 0;
|
|
case "TAIKO":
|
|
return 1;
|
|
case "CTB":
|
|
case "CATCHTHEBEAT":
|
|
return 2;
|
|
case "MANIA":
|
|
case "OSU!MANIA":
|
|
return 3;
|
|
default:
|
|
return 0;
|
|
}
|
|
}
|
|
|
|
private static string ResolveGameMode(int mode)
|
|
{
|
|
switch (mode)
|
|
{
|
|
case 0:
|
|
return "Standard";
|
|
case 1:
|
|
return "Taiko";
|
|
case 2:
|
|
return "Catch";
|
|
case 3:
|
|
return "Mania";
|
|
default:
|
|
return "Standard";
|
|
}
|
|
}
|
|
|
|
//https://github.com/ppy/osu-api/wiki#mods
|
|
private static string ResolveMods(int mods)
|
|
{
|
|
var modString = "+";
|
|
|
|
if (IsBitSet(mods, 0))
|
|
modString += "NF";
|
|
if (IsBitSet(mods, 1))
|
|
modString += "EZ";
|
|
if (IsBitSet(mods, 8))
|
|
modString += "HT";
|
|
|
|
if (IsBitSet(mods, 3))
|
|
modString += "HD";
|
|
if (IsBitSet(mods, 4))
|
|
modString += "HR";
|
|
if (IsBitSet(mods, 6) && !IsBitSet(mods, 9))
|
|
modString += "DT";
|
|
if (IsBitSet(mods, 9))
|
|
modString += "NC";
|
|
if (IsBitSet(mods, 10))
|
|
modString += "FL";
|
|
|
|
if (IsBitSet(mods, 5))
|
|
modString += "SD";
|
|
if (IsBitSet(mods, 14))
|
|
modString += "PF";
|
|
|
|
if (IsBitSet(mods, 7))
|
|
modString += "RX";
|
|
if (IsBitSet(mods, 11))
|
|
modString += "AT";
|
|
if (IsBitSet(mods, 12))
|
|
modString += "SO";
|
|
return modString;
|
|
}
|
|
|
|
private static bool IsBitSet(int mods, int pos)
|
|
=> (mods & (1 << pos)) != 0;
|
|
}
|
|
} |