feat(webhooks): init, add service/models, add JSON to patch objects

2021-11-02 06:08:17 -04:00
parent 08c5b78cc2
commit 71aec0d419
14 changed files with 551 additions and 2 deletions
--- a/PluralKit.Core/Dispatch/DispatchModels.cs
+++ b/PluralKit.Core/Dispatch/DispatchModels.cs
@@ -0,0 +1,109 @@
+using System;
+using System.Net;
+using System.Net.Http;
+using System.Threading.Tasks;
+
+using Newtonsoft.Json;
+using Newtonsoft.Json.Linq;
+
+using NodaTime;
+
+namespace PluralKit.Core
+{
+    public enum DispatchEvent
+    {
+        PING,
+        UPDATE_SYSTEM,
+        CREATE_MEMBER,
+        UPDATE_MEMBER,
+        DELETE_MEMBER,
+        CREATE_GROUP,
+        UPDATE_GROUP,
+        UPDATE_GROUP_MEMBERS,
+        DELETE_GROUP,
+        LINK_ACCOUNT,
+        UNLINK_ACCOUNT,
+        UPDATE_SYSTEM_GUILD,
+        UPDATE_MEMBER_GUILD,
+        CREATE_MESSAGE,
+        CREATE_SWITCH,
+        UPDATE_SWITCH,
+        UPDATE_SWITCH_MEMBERS,
+        DELETE_SWITCH,
+        DELETE_ALL_SWITCHES,
+    }
+
+    public struct UpdateDispatchData
+    {
+        public DispatchEvent Event;
+        public string SystemId;
+        public string? EntityId;
+        public ulong? GuildId;
+        public string SigningToken;
+        public JObject? EventData;
+    }
+
+    public static class DispatchExt
+    {
+        public static StringContent GetPayloadBody(this UpdateDispatchData data)
+        {
+            var o = new JObject();
+
+            o.Add("type", data.Event.ToString());
+            o.Add("signing_token", data.SigningToken);
+            o.Add("system_id", data.SystemId);
+            if (data.EntityId != null)
+                o.Add("id", data.EntityId);
+            if (data.GuildId != null)
+                o.Add("guild_id", data.GuildId);
+            if (data.EventData != null)
+                o.Add("data", data.EventData);
+
+            return new StringContent(JsonConvert.SerializeObject(o));
+        }
+
+        public static JObject ToDispatchJson(this PKMessage msg, string memberRef)
+        {
+            var o = new JObject();
+
+            o.Add("timestamp", Instant.FromUnixTimeMilliseconds((long)(msg.Mid >> 22) + 1420070400000).FormatExport());
+            o.Add("id", msg.Mid.ToString());
+            o.Add("original", msg.OriginalMid.ToString());
+            o.Add("sender", msg.Sender.ToString());
+            o.Add("channel", msg.Channel.ToString());
+            o.Add("member", memberRef);
+
+            return o;
+        }
+
+        public static async Task<bool> ValidateUri(string uri)
+        {
+            IPHostEntry host = null;
+
+            try
+            {
+                host = await Dns.GetHostEntryAsync(uri);
+            }
+            catch (Exception) { }
+
+            if (host == null || host.AddressList.Length == 0)
+                return false;
+
+#pragma warning disable CS0618
+
+            foreach (var address in host.AddressList)
+            {
+                if ((address.Address & 0x7f000000) == 0x7f000000) // 127.0/8
+                    return false;
+                if ((address.Address & 0x0a000000) == 0x0a000000) // 10.0/8
+                    return false;
+                if ((address.Address & 0xa9fe0000) == 0xa9fe0000) // 169.254/16
+                    return false;
+                if ((address.Address & 0xac100000) == 0xac100000) // 172.16/12
+                    return false;
+            }
+
+            return true;
+        }
+    }
+}