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; } private async Task DoPostRequest(SystemId system, string webhookUrl, HttpContent content) { 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) { _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(); 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(); 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(); 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 dict, DispatchEvent evt) { var repo = _provider.Resolve(); 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(); 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(); 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(); 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(); 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(); 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(); 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()); } } }