Initial commitDiscordMessageRecord

1 files changed, 29 insertions, 1 deletions

diff --git a/Handlers/DiscordHandler.cs b/Handlers/DiscordHandler.cs
index 461aac4..7701a3d 100644
--- a/Handlers/DiscordHandler.cs
+++ b/Handlers/DiscordHandler.cs
@@ -2,6 +2,7 @@
 using System.ComponentModel.DataAnnotations.Schema;
 using System.Text.RegularExpressions;
 using Discord;
+using Discord.Rest;
 using Discord.WebSocket;
 using Microsoft.EntityFrameworkCore;
 using PagerParser.Handlers;
@@ -10,6 +11,7 @@ namespace PagerParser {
     public partial class PagerContext {
         public DbSet<DiscordAlertChannel>      DiscordAlertChannels      { get; set; }
         public DbSet<DiscordAlertSubscription> DiscordAlertSubscriptions { get; set; }
+        public DbSet<DiscordAlertMessage>      DiscordAlertMessages      { get; set; }
     }
 }
 
@@ -243,14 +245,34 @@ namespace PagerParser.Handlers {
                     $"\nPaged Services: {pagedServicesText}";
 
                 // Actually post the message
+                RestUserMessage messageResult;
                 try {
-                    await guild
+                    messageResult = await guild
                         .GetTextChannel(alertChannel.ChannelId)
                         .SendMessageAsync(discordMessage);
                 } catch(Exception e) {
                     logger.LogError(
                         e,
                         "Error sending Discord message");
+                    return;
+                }
+
+                // Create a record linking the pager message in the database to the Discord
+                // message we've just posted so that the Discord message may be amended later
+                // if so required.
+                var messageRecord = new DiscordAlertMessage() {
+                    DiscordMessageId   = messageResult.Id,
+                    ParsedPagerMessage = pm
+                };
+
+                // Commit the record to the database
+                try {
+                    await db.AddAsync(messageRecord);
+                    await db.SaveChangesAsync();
+                } catch (Exception e) {
+                    logger.LogError(
+                        e,
+                        "Error committing Discord message result to the database");
                 }
             }
         }
@@ -638,6 +660,12 @@ namespace PagerParser.Handlers {
         public DiscordPrincipalType PrincipalType              { get; set; }
     }
 
+    public class DiscordAlertMessage {
+        [Key]
+        public ulong              DiscordMessageId   { get; set; }
+        public ParsedPagerMessage ParsedPagerMessage { get; set; }
+    }
+
     public enum DiscordPrincipalType {
         User,
         Role