discord-bot/CompatBot/EventHandlers/GithubLinksHandler.cs
2019-01-11 22:45:57 +05:00

95 lines
4.2 KiB
C#

using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Text.RegularExpressions;
using System.Threading.Tasks;
using CompatBot.Utils;
using DSharpPlus.EventArgs;
namespace CompatBot.EventHandlers
{
internal sealed class GithubLinksHandler
{
private const RegexOptions DefaultOptions = RegexOptions.Compiled | RegexOptions.IgnoreCase | RegexOptions.Singleline | RegexOptions.ExplicitCapture;
private static readonly Regex IssueMention = new Regex(@"(\b(issue|pr|pull[ \-]request|bug)\s*#?\s*(?<number>\d+)|#(?<also_number>\d{4}))\b", DefaultOptions);
private static readonly Regex IssueLink = new Regex(@"github.com/RPCS3/rpcs3/issues/(?<number>\d+)", DefaultOptions);
private static readonly GithubClient.Client Client = new GithubClient.Client();
public static async Task OnMessageCreated(MessageCreateEventArgs args)
{
if (args.Author.IsBot)
return;
if (string.IsNullOrEmpty(args.Message.Content) || args.Message.Content.StartsWith(Config.CommandPrefix))
return;
var lastBotMessages = await args.Channel.GetMessagesBeforeAsync(args.Message.Id, 20, DateTime.UtcNow.AddSeconds(-30)).ConfigureAwait(false);
foreach (var msg in lastBotMessages)
if (BotShutupHandler.NeedToSilence(msg).needToChill)
return;
lastBotMessages = await args.Channel.GetMessagesBeforeAsync(args.Message.Id, Config.ProductCodeLookupHistoryThrottle).ConfigureAwait(false);
StringBuilder previousRepliesBuilder = null;
foreach (var msg in lastBotMessages)
{
if (msg.Author.IsCurrent)
{
previousRepliesBuilder = previousRepliesBuilder ?? new StringBuilder();
previousRepliesBuilder.AppendLine(msg.Content);
var embeds = msg.Embeds;
if (embeds?.Count > 0)
foreach (var embed in embeds)
previousRepliesBuilder.AppendLine(embed.Title).AppendLine(embed.Description);
}
}
var previousReplies = previousRepliesBuilder?.ToString() ?? "";
var idsFromPreviousReplies = GetIssueIdsFromLinks(previousReplies);
var issuesToLookup = GetIssueIds(args.Message.Content)
.Where(lnk => !idsFromPreviousReplies.Contains(lnk))
.Take(args.Channel.IsPrivate ? 50 : 5)
.ToList();
if (issuesToLookup.Count == 0)
return;
await args.Message.ReactWithAsync(args.Client, Config.Reactions.PleaseWait).ConfigureAwait(false);
var suffix = issuesToLookup.Count == 1 ? "" : "s";
try
{
var result = new StringBuilder($"Link{suffix} to the mentioned issue{suffix}:");
foreach (var issueId in issuesToLookup)
result.AppendLine().Append("https://github.com/RPCS3/rpcs3/issues/" + issueId);
await args.Channel.SendAutosplitMessageAsync(result, blockStart: null, blockEnd: null).ConfigureAwait(false);
}
finally
{
await args.Message.RemoveReactionAsync(Config.Reactions.PleaseWait).ConfigureAwait(false);
}
}
public static List<int> GetIssueIds(string input)
{
return IssueMention.Matches(input)
.SelectMany(match => new[] {match.Groups["number"].Value, match.Groups["also_number"].Value})
.Distinct()
.Select(n => {
int.TryParse(n, out var i);
return i;
})
.Where(n => n > 0)
.ToList();
}
public static HashSet<int> GetIssueIdsFromLinks(string input)
{
return new HashSet<int>(
IssueLink.Matches(input)
.Select(match =>
{
int.TryParse(match.Groups["number"].Value, out var n);
return n;
})
);
}
}
}