101 lines
3.2 KiB
C#
101 lines
3.2 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using System.Globalization;
|
|
using System.Net.Http;
|
|
using System.Net.Mime;
|
|
using System.Text;
|
|
using System.Threading.Tasks;
|
|
using Jellyfin.Plugin.Webhook.Extensions;
|
|
using MediaBrowser.Common.Net;
|
|
using Microsoft.Extensions.Logging;
|
|
|
|
namespace Jellyfin.Plugin.Webhook.Destinations.Discord;
|
|
|
|
/// <summary>
|
|
/// Client for the <see cref="DiscordOption"/>.
|
|
/// </summary>
|
|
public class DiscordClient : BaseClient, IWebhookClient<DiscordOption>
|
|
{
|
|
private readonly ILogger<DiscordClient> _logger;
|
|
private readonly IHttpClientFactory _httpClientFactory;
|
|
|
|
/// <summary>
|
|
/// Initializes a new instance of the <see cref="DiscordClient"/> class.
|
|
/// </summary>
|
|
/// <param name="logger">Instance of the <see cref="ILogger{DiscordDestination}"/> interface.</param>
|
|
/// <param name="httpClientFactory">Instance of the<see cref="IHttpClientFactory"/> interface.</param>
|
|
public DiscordClient(ILogger<DiscordClient> logger, IHttpClientFactory httpClientFactory)
|
|
{
|
|
_logger = logger;
|
|
_httpClientFactory = httpClientFactory;
|
|
}
|
|
|
|
/// <inheritdoc />
|
|
public async Task SendAsync(DiscordOption option, Dictionary<string, object> data)
|
|
{
|
|
try
|
|
{
|
|
if (string.IsNullOrEmpty(option.WebhookUri))
|
|
{
|
|
throw new ArgumentException(nameof(option.WebhookUri));
|
|
}
|
|
|
|
if (!SendWebhook(_logger, option, data))
|
|
{
|
|
return;
|
|
}
|
|
|
|
// Add discord specific properties.
|
|
data["MentionType"] = GetMentionType(option.MentionType);
|
|
if (!string.IsNullOrEmpty(option.EmbedColor))
|
|
{
|
|
data["EmbedColor"] = FormatColorCode(option.EmbedColor);
|
|
}
|
|
|
|
if (!string.IsNullOrEmpty(option.AvatarUrl))
|
|
{
|
|
data["AvatarUrl"] = option.AvatarUrl;
|
|
}
|
|
|
|
if (!string.IsNullOrEmpty(option.Username))
|
|
{
|
|
data["Username"] = option.Username;
|
|
data["BotUsername"] = option.Username;
|
|
}
|
|
|
|
var body = option.GetMessageBody(data);
|
|
if (!SendMessageBody(_logger, option, body))
|
|
{
|
|
return;
|
|
}
|
|
|
|
_logger.LogDebug("SendAsync Body: {@Body}", body);
|
|
using var content = new StringContent(body, Encoding.UTF8, MediaTypeNames.Application.Json);
|
|
using var response = await _httpClientFactory
|
|
.CreateClient(NamedClient.Default)
|
|
.PostAsync(new Uri(option.WebhookUri), content)
|
|
.ConfigureAwait(false);
|
|
await response.LogIfFailedAsync(_logger).ConfigureAwait(false);
|
|
}
|
|
catch (HttpRequestException e)
|
|
{
|
|
_logger.LogWarning(e, "Error sending notification");
|
|
}
|
|
}
|
|
|
|
private static int FormatColorCode(string hexCode)
|
|
{
|
|
return int.Parse(hexCode[1..6], NumberStyles.HexNumber, CultureInfo.InvariantCulture);
|
|
}
|
|
|
|
private static string GetMentionType(DiscordMentionType mentionType)
|
|
{
|
|
return mentionType switch
|
|
{
|
|
DiscordMentionType.Everyone => "@everyone",
|
|
DiscordMentionType.Here => "@here",
|
|
_ => string.Empty
|
|
};
|
|
}
|
|
}
|