using Discord; using Discord.Commands; using NadekoBot.Extensions; using NadekoBot.Core.Services; using Newtonsoft.Json.Linq; using System; using System.Globalization; using System.IO; using System.Net.Http; using System.Text.RegularExpressions; using System.Threading.Tasks; using NadekoBot.Common.Attributes; namespace NadekoBot.Modules.Searches { public partial class Searches { [Group] public class OsuCommands : NadekoSubmodule { private readonly IGoogleApiService _google; private readonly IBotCredentials _creds; public OsuCommands(IGoogleApiService google, IBotCredentials creds) { _google = google; _creds = creds; } [NadekoCommand, Usage, Description, Aliases] public async Task Osu(string usr, [Remainder] string mode = null) { if (string.IsNullOrWhiteSpace(usr)) return; using (var http = new HttpClient()) { try { var m = 0; if (!string.IsNullOrWhiteSpace(mode)) { m = ResolveGameMode(mode); } http.AddFakeHeaders(); var res = await http.GetStreamAsync(new Uri($"http://lemmmy.pw/osusig/sig.php?uname={ usr }&flagshadow&xpbar&xpbarhex&pp=2&mode={m}")).ConfigureAwait(false); var ms = new MemoryStream(); res.CopyTo(ms); ms.Position = 0; await Context.Channel.SendFileAsync(ms, $"{usr}.png", $"🎧 **{GetText("profile_link")}** \n`Image provided by https://lemmmy.pw/osusig`").ConfigureAwait(false); } catch (Exception ex) { await ReplyErrorLocalized("osu_failed").ConfigureAwait(false); _log.Warn(ex); } } } [NadekoCommand, Usage, Description, Aliases] public async Task Osub([Remainder] string map) { if (string.IsNullOrWhiteSpace(_creds.OsuApiKey)) { await ReplyErrorLocalized("osu_api_key").ConfigureAwait(false); return; } if (string.IsNullOrWhiteSpace(map)) return; try { using (var http = new HttpClient()) { var mapId = ResolveMap(map); var reqString = $"https://osu.ppy.sh/api/get_beatmaps?k={_creds.OsuApiKey}&{mapId}"; var obj = JArray.Parse(await http.GetStringAsync(reqString).ConfigureAwait(false))[0]; var sb = new System.Text.StringBuilder(); var starRating = Math.Round(double.Parse($"{obj["difficultyrating"]}", CultureInfo.InvariantCulture), 2); var time = TimeSpan.FromSeconds(double.Parse($"{obj["total_length"]}")).ToString(@"mm\:ss"); sb.AppendLine($"{obj["artist"]} - {obj["title"]}, mapped by {obj["creator"]}. https://osu.ppy.sh/s/{obj["beatmapset_id"]}"); sb.AppendLine($"{starRating} stars, {obj["bpm"]} BPM | AR{obj["diff_approach"]}, CS{obj["diff_size"]}, OD{obj["diff_overall"]} | Length: {time}"); await Context.Channel.SendMessageAsync(sb.ToString()).ConfigureAwait(false); } } catch (Exception ex) { await ReplyErrorLocalized("something_went_wrong").ConfigureAwait(false); _log.Warn(ex); } } [NadekoCommand, Usage, Description, Aliases] public async Task Osu5(string user, [Remainder] string mode = null) { var channel = (ITextChannel)Context.Channel; if (string.IsNullOrWhiteSpace(_creds.OsuApiKey)) { await channel.SendErrorAsync("An osu! API key is required.").ConfigureAwait(false); return; } if (string.IsNullOrWhiteSpace(user)) { await channel.SendErrorAsync("Please provide a username.").ConfigureAwait(false); return; } using (var http = new HttpClient()) { try { 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 obj = JArray.Parse(await http.GetStringAsync(reqString).ConfigureAwait(false)); var sb = new System.Text.StringBuilder($"`Top 5 plays for {user}:`\n```xl" + Environment.NewLine); foreach (var item in obj) { var mapReqString = $"https://osu.ppy.sh/api/get_beatmaps?k={_creds.OsuApiKey}&b={item["beatmap_id"]}"; var map = JArray.Parse(await http.GetStringAsync(mapReqString).ConfigureAwait(false))[0]; var pp = Math.Round(double.Parse($"{item["pp"]}", CultureInfo.InvariantCulture), 2); var acc = CalculateAcc(item, m); var mods = ResolveMods(int.Parse($"{item["enabled_mods"]}")); sb.AppendLine(mods != "+" ? $"{pp + "pp",-7} | {acc + "%",-7} | {map["artist"] + "-" + map["title"] + " (" + map["version"] + ")",-40} | **{mods,-10}** | /b/{item["beatmap_id"]}" : $"{pp + "pp",-7} | {acc + "%",-7} | {map["artist"] + "-" + map["title"] + " (" + map["version"] + ")",-40} | /b/{item["beatmap_id"]}"); } sb.Append("```"); await channel.SendMessageAsync(sb.ToString()).ConfigureAwait(false); } catch (Exception ex) { await ReplyErrorLocalized("something_went_wrong").ConfigureAwait(false); _log.Warn(ex); } } } //https://osu.ppy.sh/wiki/Accuracy private static double CalculateAcc(JToken play, int mode) { if (mode == 0) { var hitPoints = double.Parse($"{play["count50"]}") * 50 + double.Parse($"{play["count100"]}") * 100 + double.Parse($"{play["count300"]}") * 300; var totalHits = double.Parse($"{play["count50"]}") + double.Parse($"{play["count100"]}") + double.Parse($"{play["count300"]}") + double.Parse($"{play["countmiss"]}"); totalHits *= 300; return Math.Round(hitPoints / totalHits * 100, 2); } else if (mode == 1) { var hitPoints = double.Parse($"{play["countmiss"]}") * 0 + double.Parse($"{play["count100"]}") * 0.5 + double.Parse($"{play["count300"]}") * 1; var totalHits = double.Parse($"{play["countmiss"]}") + double.Parse($"{play["count100"]}") + double.Parse($"{play["count300"]}"); hitPoints *= 300; totalHits *= 300; return Math.Round(hitPoints / totalHits * 100, 2); } else if (mode == 2) { var fruitsCaught = double.Parse($"{play["count50"]}") + double.Parse($"{play["count100"]}") + double.Parse($"{play["count300"]}"); var totalFruits = double.Parse($"{play["countmiss"]}") + double.Parse($"{play["count50"]}") + double.Parse($"{play["count100"]}") + double.Parse($"{play["count300"]}") + double.Parse($"{play["countkatu"]}"); return Math.Round(fruitsCaught / totalFruits * 100, 2); } else { var hitPoints = double.Parse($"{play["count50"]}") * 50 + double.Parse($"{play["count100"]}") * 100 + double.Parse($"{play["countkatu"]}") * 200 + (double.Parse($"{play["count300"]}") + double.Parse($"{play["countgeki"]}")) * 300; var totalHits = double.Parse($"{play["countmiss"]}") + double.Parse($"{play["count50"]}") + double.Parse($"{play["count100"]}") + double.Parse($"{play["countkatu"]}") + double.Parse($"{play["count300"]}") + double.Parse($"{play["countgeki"]}"); totalHits *= 300; return Math.Round(hitPoints / totalHits * 100, 2); } } private static string ResolveMap(string mapLink) { var s = new Regex(@"osu.ppy.sh\/s\/", RegexOptions.IgnoreCase).Match(mapLink); var b = new Regex(@"osu.ppy.sh\/b\/", RegexOptions.IgnoreCase).Match(mapLink); var p = new Regex(@"osu.ppy.sh\/p\/", RegexOptions.IgnoreCase).Match(mapLink); var m = new Regex(@"&m=", RegexOptions.IgnoreCase).Match(mapLink); if (s.Success) { var mapId = mapLink.Substring(mapLink.IndexOf("/s/") + 3); return $"s={mapId}"; } else if (b.Success) { if (m.Success) return $"b={mapLink.Substring(mapLink.IndexOf("/b/") + 3, mapLink.IndexOf("&m") - (mapLink.IndexOf("/b/") + 3))}"; else return $"b={mapLink.Substring(mapLink.IndexOf("/b/") + 3)}"; } else if (p.Success) { if (m.Success) return $"b={mapLink.Substring(mapLink.IndexOf("?b=") + 3, mapLink.IndexOf("&m") - (mapLink.IndexOf("?b=") + 3))}"; else return $"b={mapLink.Substring(mapLink.IndexOf("?b=") + 3)}"; } else { return $"s={mapLink}"; //just a default incase an ID number was provided by itself (non-url)? } } private static int ResolveGameMode(string mode) { switch (mode.ToLower()) { 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; } } //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; } } }