Merge pull request #72 from InfinityGhost/feature/github-code-ref

Add support for file references in messages

Author: InfinityGhost

TabletBot.Common/GuildSettings.cs

@@ -31,6 +31,7 @@ public static IServiceCollection Build(Settings settings, State state, DiscordSo
                 .AddSingleton<CommandService>()
                 // Message watchers
                 .AddWatcher<CommandMessageWatcher>()
+                .AddWatcher<CodeMessageWatcher>()
                 .AddWatcher<IssueMessageWatcher>()
                 .AddWatcher<SpamMessageWatcher>()
                 .AddWatcher<ModMailMessageWatcher>()

TabletBot.Discord/BotServiceCollection.cs

@@ -1,4 +1,5 @@
 using System;
+using System.Text;
 using Discord;
 
 namespace TabletBot.Discord
@@ -18,5 +19,13 @@ public static class Formatting
         public static string CodeString(string text) => $"{CODE_AFFIX}{text}{CODE_AFFIX}";
         public static string CodeBlock(string text, string? lang = null) =>
             $"{CODE_BLOCK}{lang}{Environment.NewLine}{text}{Environment.NewLine}{CODE_BLOCK}";
+
+        public static void AppendCodeBlock(this StringBuilder stringBuilder, string[] lines, string? lang = null)
+        {
+            stringBuilder.AppendLine(CODE_BLOCK + lang);
+            foreach (var line in lines)
+                stringBuilder.AppendLine(line);
+            stringBuilder.AppendLine(CODE_BLOCK);
+        }
     }
 }

TabletBot.Discord/Formatting.cs

@@ -0,0 +1,60 @@
+using System;
+using System.IO;
+using System.Text;
+using System.Text.RegularExpressions;
+using System.Threading.Tasks;
+using Discord;
+using Octokit;
+using TabletBot.Discord.Commands;
+
+namespace TabletBot.Discord.Watchers.GitHub
+{
+    public class CodeMessageWatcher : IMessageWatcher
+    {
+        private readonly GitHubClient _client;
+
+        public CodeMessageWatcher(GitHubClient client)
+        {
+            _client = client;
+        }
+
+        private static readonly Regex CodeRefRegex = new Regex(
+            @"https:\/\/github.com\/(?<Owner>.+?)\/(?<Repo>.+?)\/blob\/(?<GitRef>.+?)\/(?<Path>.+?)#L(?<StartLine>.+?)(?:-L(?<EndLine>.+?))?($| |>)",
+            RegexOptions.Compiled
+        );
+
+        public async Task Receive(IMessage message)
+        {
+            var match = CodeRefRegex.Match(message.Content);
+            if (message is IUserMessage userMessage && match.Success)
+            {
+                using (message.Channel.EnterTypingState())
+                {
+                    var owner = match.Groups["Owner"].Value;
+                    var repo = match.Groups["Repo"].Value;
+                    var gitRef = match.Groups["GitRef"].Value;
+                    var path = match.Groups["Path"].Value;
+                    var startLine = int.Parse(match.Groups["StartLine"].Value) - 1;
+                    var endLine = match.Groups["EndLine"].Success ? int.Parse(match.Groups["EndLine"].Value) : startLine + 1;
+
+                    var url = $"https://raw.githubusercontent.com/{owner}/{repo}/{gitRef}/{path}";
+                    var rawContent = await _client.Connection.GetHtml(new Uri(url));
+
+                    var fileContent = rawContent.Body.Split(Environment.NewLine);
+                    var lines = fileContent[startLine..endLine];
+
+                    var extension = Path.GetExtension(path).Replace(".", string.Empty);
+
+                    var sb = new StringBuilder();
+                    sb.AppendLine(Formatting.CodeString(path));
+                    sb.AppendCodeBlock(lines, extension);
+
+                    await message.Channel.SendMessageAsync(sb.ToString(), messageReference: userMessage.ToReference());
+                }
+            }
+        }
+
+        public Task Deleted(Cacheable<IMessage, ulong> message, Cacheable<IMessageChannel, ulong> channel) =>
+            Task.CompletedTask;
+    }
+}

TabletBot.Discord/Watchers/GitHub/CodeMessageWatcher.cs

@@ -6,7 +6,6 @@
   </ItemGroup>
 
   <ItemGroup>
-    <PackageReference Include="System.CommandLine.DragonFruit" Version="0.3.0-alpha.19577.1" />
     <PackageReference Include="System.CommandLine.Experimental" Version="0.3.0-alpha.19577.1" />
     <PackageReference Include="System.CommandLine.Rendering" Version="0.3.0-alpha.21216.1" />
     <PackageReference Include="System.Reactive.Core" Version="5.0.0" />

TabletBot/TabletBot.csproj

@@ -18,7 +18,6 @@
   (fetchNuGet { pname = "Octokit"; version = "0.51.0"; sha256 = "1qmmw3vva026lz70xw68pwv6m18ry53a6z5didy2b4isd5bnrs09"; })
   (fetchNuGet { pname = "System.Collections.Immutable"; version = "5.0.0"; sha256 = "1kvcllagxz2q92g81zkz81djkn2lid25ayjfgjalncyc68i15p0r"; })
   (fetchNuGet { pname = "System.CommandLine"; version = "2.0.0-beta1.21216.1"; sha256 = "10p3i10sh9aarqfzac3bgmv9zabb8m5c2aylf3chwlh71kp9l63l"; })
-  (fetchNuGet { pname = "System.CommandLine.DragonFruit"; version = "0.3.0-alpha.19577.1"; sha256 = "09xv1ii0irr69snc9bjzkdg17ccasc3ib40ydakqqwf50lk5za0x"; })
   (fetchNuGet { pname = "System.CommandLine.Experimental"; version = "0.3.0-alpha.19577.1"; sha256 = "1qrj0pvz5zvw0a12sn8xc9i1bh58dsgkdq8gjinzj10l4aa38g1a"; })
   (fetchNuGet { pname = "System.CommandLine.Rendering"; version = "0.3.0-alpha.21216.1"; sha256 = "0kwrjm1i4xc9vgjid1l4wlxddk92fsyypjzg36zs8qxcw07kh9mw"; })
   (fetchNuGet { pname = "System.Interactive.Async"; version = "5.0.0"; sha256 = "00flf80ahpyhrsny2kfl9bsyh1mxmsdsigpq3rzhaps9wgdbmzjz"; })