PluralKit/PluralKit.Core/Dispatch/DispatchService.cs

using System;
using System.Linq;
using System.Collections.Generic;
using System.Net.Http;
using System.Threading.Tasks;

using Autofac;

using Serilog;

namespace PluralKit.Core
{
    public class DispatchService
    {
        private readonly ILogger _logger;
        private readonly ILifetimeScope _provider;
        private readonly HttpClient _client = new();
        public DispatchService(ILogger logger, ILifetimeScope provider, CoreConfig cfg)
        {
            _logger = logger;
            _provider = provider;
        }

        public async Task DoPostRequest(SystemId system, string webhookUrl, HttpContent content, bool isVerify = false)
        {
            if (!await DispatchExt.ValidateUri(webhookUrl))
            {
                _logger.Warning("Failed to dispatch webhook for system {SystemId}: URL is invalid or points to a private address", system);
                return;
            }

            try
            {
                await _client.PostAsync(webhookUrl, content);
            }
            catch (HttpRequestException e)
            {
                if (isVerify)
                    throw;
                else
                    _logger.Error("Could not dispatch webhook request!", e);
            }
        }

        public async Task Dispatch(SystemId systemId, UpdateDispatchData data)
        {
            if (data.EventData != null && data.EventData.Count == 0)
                return;

            var repo = _provider.Resolve<ModelRepository>();
            var system = await repo.GetSystem(systemId);
            if (system.WebhookUrl == null)
                return;

            data.SigningToken = system.WebhookToken;
            data.SystemId = system.Uuid.ToString();

            _logger.Debug("Dispatching webhook for system {SystemId}", systemId);
            await DoPostRequest(system.Id, system.WebhookUrl, data.GetPayloadBody());
        }

        public async Task Dispatch(MemberId memberId, UpdateDispatchData data)
        {
            if (data.EventData != null && data.EventData.Count == 0)
                return;

            var repo = _provider.Resolve<ModelRepository>();
            var member = await repo.GetMember(memberId);
            var system = await repo.GetSystem(member.System);
            if (system.WebhookUrl == null)
                return;

            data.SigningToken = system.WebhookToken;
            data.SystemId = system.Uuid.ToString();
            data.EntityId = member.Uuid.ToString();

            _logger.Debug("Dispatching webhook for member {MemberId} (system {SystemId})", memberId, system.Id);
            await DoPostRequest(system.Id, system.WebhookUrl, data.GetPayloadBody());
        }

        public async Task Dispatch(GroupId groupId, UpdateDispatchData data)
        {
            if (data.EventData != null && data.EventData.Count == 0)
                return;

            var repo = _provider.Resolve<ModelRepository>();
            var group = await repo.GetGroup(groupId);
            var system = await repo.GetSystem(group.System);
            if (system.WebhookUrl == null)
                return;

            data.SigningToken = system.WebhookToken;
            data.SystemId = system.Uuid.ToString();
            data.EntityId = group.Uuid.ToString();

            _logger.Debug("Dispatching webhook for group {GroupId} (system {SystemId})", groupId, system.Id);
            await DoPostRequest(system.Id, system.WebhookUrl, data.GetPayloadBody());
        }

        public async Task Dispatch(Dictionary<GroupId, MemberId> dict, DispatchEvent evt)
        {
            var repo = _provider.Resolve<ModelRepository>();
            var g = await repo.GetGroup(dict.Keys.FirstOrDefault());
            var system = await repo.GetSystem(g.System);
            if (system.WebhookUrl == null)
                return;

            var data = new UpdateDispatchData();
            data.Event = DispatchEvent.UPDATE_GROUP_MEMBERS;
            data.SystemId = system.Uuid.ToString();


            _logger.Debug("Dispatching webhook for group member update (system {SystemId})", system.Id);
            await DoPostRequest(system.Id, system.WebhookUrl, data.GetPayloadBody());
        }

        public async Task Dispatch(SwitchId swId, UpdateDispatchData data)
        {
            var repo = _provider.Resolve<ModelRepository>();
            var sw = await repo.GetSwitch(swId);
            var system = await repo.GetSystem(sw.System);
            if (system.WebhookUrl == null)
                return;

            data.SigningToken = system.WebhookToken;
            data.SystemId = system.Uuid.ToString();
            data.EntityId = sw.Uuid.ToString();

            _logger.Debug("Dispatching webhook for switch {SwitchId} (system {SystemId})", sw.Id, system.Id);
            await DoPostRequest(system.Id, system.WebhookUrl, data.GetPayloadBody());
        }

        public async Task Dispatch(SystemId systemId, PKMessage newMessage)
        {
            var repo = _provider.Resolve<ModelRepository>();
            var system = await repo.GetSystem(systemId);
            if (system.WebhookUrl == null)
                return;

            var member = await repo.GetMember(newMessage.Member);

            var data = new UpdateDispatchData();
            data.Event = DispatchEvent.CREATE_MESSAGE;
            data.SigningToken = system.WebhookToken;
            data.SystemId = system.Uuid.ToString();
            data.EventData = newMessage.ToDispatchJson(member.Uuid.ToString());

            _logger.Debug("Dispatching webhook for message create (system {SystemId})", system.Id);
            await DoPostRequest(system.Id, system.WebhookUrl, data.GetPayloadBody());
        }

        public async Task Dispatch(SystemId systemId, ulong guild_id, SystemGuildPatch patch)
        {
            var repo = _provider.Resolve<ModelRepository>();
            var system = await repo.GetSystem(systemId);
            if (system.WebhookUrl == null)
                return;

            string memberRef = null;
            if (patch.AutoproxyMember.Value != null)
            {
                var member = await repo.GetMember(patch.AutoproxyMember.Value.Value);
                memberRef = member.Uuid.ToString();
            }

            var data = new UpdateDispatchData();
            data.Event = DispatchEvent.UPDATE_SYSTEM_GUILD;
            data.SigningToken = system.WebhookToken;
            data.SystemId = system.Uuid.ToString();
            data.GuildId = guild_id;
            data.EventData = patch.ToJson(memberRef);

            _logger.Debug("Dispatching webhook for system {SystemId} in guild {GuildId}", system.Id, guild_id);
            await DoPostRequest(system.Id, system.WebhookUrl, data.GetPayloadBody());
        }

        public async Task Dispatch(MemberId memberId, ulong guild_id, MemberGuildPatch patch)
        {
            var repo = _provider.Resolve<ModelRepository>();
            var member = await repo.GetMember(memberId);
            var system = await repo.GetSystem(member.System);
            if (system.WebhookUrl == null)
                return;

            var data = new UpdateDispatchData();
            data.Event = DispatchEvent.UPDATE_MEMBER_GUILD;
            data.SigningToken = system.WebhookToken;
            data.SystemId = system.Uuid.ToString();
            data.EntityId = member.Uuid.ToString();
            data.GuildId = guild_id;
            data.EventData = patch.ToJson();

            _logger.Debug("Dispatching webhook for member {MemberId} (system {SystemId}) in guild {GuildId}", member.Id, system.Id, guild_id);
            await DoPostRequest(system.Id, system.WebhookUrl, data.GetPayloadBody());
        }

        public async Task Dispatch(ulong accountId, AccountPatch patch)
        {
            var repo = _provider.Resolve<ModelRepository>();
            var system = await repo.GetSystemByAccount(accountId);
            if (system.WebhookUrl == null)
                return;

            var data = new UpdateDispatchData();
            data.Event = DispatchEvent.UPDATE_MEMBER_GUILD;
            data.SigningToken = system.WebhookToken;
            data.EntityId = accountId.ToString();
            data.EventData = patch.ToJson();

            _logger.Debug("Dispatching webhook for account {AccountId} (system {SystemId})", accountId, system.Id);
            await DoPostRequest(system.Id, system.WebhookUrl, data.GetPayloadBody());
        }

        public async Task Dispatch(SystemId systemId, Guid uuid, DispatchEvent evt)
        {
            var repo = _provider.Resolve<ModelRepository>();
            var system = await repo.GetSystem(systemId);
            if (system.WebhookUrl == null)
                return;

            var data = new UpdateDispatchData();
            data.Event = evt;

            data.SigningToken = system.WebhookToken;
            data.SystemId = system.Uuid.ToString();
            data.EntityId = uuid.ToString();

            _logger.Debug("Dispatching webhook for entity delete (system {SystemId})", system.Id);
            await DoPostRequest(system.Id, system.WebhookUrl, data.GetPayloadBody());
        }
    }
}