Features/subscription details dto (#29)
* DTO was updated to reflect the new options * SubscriptionsController.cs added id's as part of the path not query * Refactored DiscordNotificationJob.cs to break apart the nesting * Added a test to make sure commits would not be sent based on model values
This commit is contained in:
parent
84b4137bdd
commit
70440aa3f5
@ -55,7 +55,7 @@ public class SubscriptionsController : ControllerBase
|
|||||||
_subscription.Delete(id);
|
_subscription.Delete(id);
|
||||||
}
|
}
|
||||||
|
|
||||||
[HttpGet("by/discordid")]
|
[HttpGet("by/discordid/{id}")]
|
||||||
public IEnumerable<SubscriptionDto> GetByDiscordId(Guid id)
|
public IEnumerable<SubscriptionDto> GetByDiscordId(Guid id)
|
||||||
{
|
{
|
||||||
var res = new List<SubscriptionDto>();
|
var res = new List<SubscriptionDto>();
|
||||||
@ -64,7 +64,7 @@ public class SubscriptionsController : ControllerBase
|
|||||||
return res;
|
return res;
|
||||||
}
|
}
|
||||||
|
|
||||||
[HttpGet("by/sourceId")]
|
[HttpGet("by/sourceId/{id}")]
|
||||||
public IEnumerable<SubscriptionDto> GetBySourceId(Guid id)
|
public IEnumerable<SubscriptionDto> GetBySourceId(Guid id)
|
||||||
{
|
{
|
||||||
var res = new List<SubscriptionDto>();
|
var res = new List<SubscriptionDto>();
|
||||||
|
@ -5,6 +5,8 @@ namespace Newsbot.Collector.Domain.Dto;
|
|||||||
public class SubscriptionDetailsDto
|
public class SubscriptionDetailsDto
|
||||||
{
|
{
|
||||||
public Guid Id { get; set; }
|
public Guid Id { get; set; }
|
||||||
|
public bool CodeAllowReleases { get; set; }
|
||||||
|
public bool CodeAllowCommits { get; set; }
|
||||||
public SourceDto? Source { get; set; }
|
public SourceDto? Source { get; set; }
|
||||||
public DiscordWebHookDto? DiscordWebHook { get; set; }
|
public DiscordWebHookDto? DiscordWebHook { get; set; }
|
||||||
|
|
||||||
@ -14,6 +16,8 @@ public class SubscriptionDetailsDto
|
|||||||
return new SubscriptionDetailsDto
|
return new SubscriptionDetailsDto
|
||||||
{
|
{
|
||||||
Id = subscription.Id,
|
Id = subscription.Id,
|
||||||
|
CodeAllowCommits = subscription.CodeAllowCommits,
|
||||||
|
CodeAllowReleases = subscription.CodeAllowReleases,
|
||||||
Source = SourceDto.Convert(source),
|
Source = SourceDto.Convert(source),
|
||||||
DiscordWebHook = DiscordWebHookDto.Convert(discord)
|
DiscordWebHook = DiscordWebHookDto.Convert(discord)
|
||||||
};
|
};
|
||||||
|
@ -6,6 +6,21 @@ using Serilog;
|
|||||||
|
|
||||||
namespace Newsbot.Collector.Services.Jobs;
|
namespace Newsbot.Collector.Services.Jobs;
|
||||||
|
|
||||||
|
public class MessageTypeNotRequestedException : Exception
|
||||||
|
{
|
||||||
|
public MessageTypeNotRequestedException()
|
||||||
|
{
|
||||||
|
}
|
||||||
|
|
||||||
|
public MessageTypeNotRequestedException(string message) : base(message)
|
||||||
|
{
|
||||||
|
}
|
||||||
|
|
||||||
|
public MessageTypeNotRequestedException(string message, Exception inner) : base(message, inner)
|
||||||
|
{
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
public class DiscordNotificationJobOptions
|
public class DiscordNotificationJobOptions
|
||||||
{
|
{
|
||||||
public string? ConnectionString { get; init; }
|
public string? ConnectionString { get; init; }
|
||||||
@ -63,7 +78,12 @@ public class DiscordNotificationJob
|
|||||||
var requests = _queue.List(100);
|
var requests = _queue.List(100);
|
||||||
_logger.Debug($"{JobName} - Collected {requests.Count} items to send");
|
_logger.Debug($"{JobName} - Collected {requests.Count} items to send");
|
||||||
|
|
||||||
foreach (var request in requests)
|
foreach (var request in requests) ProcessQueueItem(request);
|
||||||
|
|
||||||
|
_logger.Information($"{JobName} - Loop has been completed.");
|
||||||
|
}
|
||||||
|
|
||||||
|
public void ProcessQueueItem(DiscordQueueModel request)
|
||||||
{
|
{
|
||||||
_logger.Debug($"{JobName} - Processing {request.ID}");
|
_logger.Debug($"{JobName} - Processing {request.ID}");
|
||||||
// Get all details on the article in the queue
|
// Get all details on the article in the queue
|
||||||
@ -76,7 +96,7 @@ public class DiscordNotificationJob
|
|||||||
_logger.Error(
|
_logger.Error(
|
||||||
$"{JobName} - Article ({articleDetails.ID}) was linked to a empty Source ID. Removing from the queue.");
|
$"{JobName} - Article ({articleDetails.ID}) was linked to a empty Source ID. Removing from the queue.");
|
||||||
_queue.Delete(request.ID);
|
_queue.Delete(request.ID);
|
||||||
continue;
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
var sourceIcon = new IconModel();
|
var sourceIcon = new IconModel();
|
||||||
@ -86,24 +106,33 @@ public class DiscordNotificationJob
|
|||||||
}
|
}
|
||||||
catch
|
catch
|
||||||
{
|
{
|
||||||
_logger.Warning($"{JobName} - Source ID '{sourceDetails.ID}' is missing an icon.");
|
_logger.Warning("{JobName} - Source ID \'{SourceDetailsId}\' is missing an icon", JobName,
|
||||||
|
sourceDetails.ID);
|
||||||
}
|
}
|
||||||
|
|
||||||
// Find all the subscriptions for that source
|
// Find all the subscriptions for that source
|
||||||
var allSubscriptions = _subs.ListBySourceID(sourceDetails.ID);
|
var allSubscriptions = _subs.ListBySourceID(sourceDetails.ID);
|
||||||
|
|
||||||
foreach (var sub in allSubscriptions)
|
foreach (var sub in allSubscriptions)
|
||||||
|
SendSubscriptionNotification(request.ID, articleDetails, sourceDetails, sourceIcon, sub);
|
||||||
|
|
||||||
|
_logger.Debug("{JobName} - Removing {RequestId} from the queue", JobName, request.ID);
|
||||||
|
_queue.Delete(request.ID);
|
||||||
|
}
|
||||||
|
|
||||||
|
public void SendSubscriptionNotification(Guid requestId, ArticlesModel articleDetails, SourceModel sourceDetails,
|
||||||
|
IconModel sourceIcon, SubscriptionModel sub)
|
||||||
{
|
{
|
||||||
// Check if the subscription code flags
|
// Check if the subscription code flags
|
||||||
// If the article is a code commit and the subscription does not want them, skip.
|
// If the article is a code commit and the subscription does not want them, skip.
|
||||||
if (articleDetails.CodeIsCommit && sub.CodeAllowCommits == false) continue;
|
if (articleDetails.CodeIsCommit && !sub.CodeAllowCommits) throw new MessageTypeNotRequestedException("Message was a code commit and was not requested by the subscription.");
|
||||||
|
|
||||||
// same for releases
|
// same for releases
|
||||||
if (articleDetails.CodeIsRelease && sub.CodeAllowReleases == false) continue;
|
if (articleDetails.CodeIsRelease && !sub.CodeAllowReleases) throw new MessageTypeNotRequestedException("Message was a code release and was not requested by the subscription");
|
||||||
|
|
||||||
// find the discord webhooks we need to post to
|
// find the discord webhooks we need to post to
|
||||||
var discordDetails = _webhook.GetByID(sub.DiscordWebHookId);
|
var discordDetails = _webhook.GetByID(sub.DiscordWebHookId);
|
||||||
if (discordDetails.Enabled == false) continue;
|
if (discordDetails.Enabled == false) return;
|
||||||
|
|
||||||
var client = new DiscordWebhookClient(discordDetails.Url);
|
var client = new DiscordWebhookClient(discordDetails.Url);
|
||||||
try
|
try
|
||||||
@ -112,23 +141,16 @@ public class DiscordNotificationJob
|
|||||||
}
|
}
|
||||||
catch (Exception e)
|
catch (Exception e)
|
||||||
{
|
{
|
||||||
_logger.Error($"Failed to post message to Discord. {e}");
|
_logger.Error("Failed to post message to Discord. {ErrorMessage}", e.Message);
|
||||||
_logger.Debug($"Queue Record: {request.ID}");
|
_logger.Debug("Queue Record: {RequestId}", requestId);
|
||||||
_logger.Debug($"Article: {articleDetails.ID}");
|
_logger.Debug("Article: {ArticleDetailsId}", articleDetails.ID);
|
||||||
_logger.Debug($"Source: {sourceDetails.ID}");
|
_logger.Debug("Source: {SourceDetailsId}", sourceDetails.ID);
|
||||||
_logger.Debug($"Subscription: {sub.Id}");
|
_logger.Debug("Subscription: {SubId}", sub.Id);
|
||||||
}
|
}
|
||||||
|
|
||||||
Thread.Sleep(3000);
|
Thread.Sleep(3000);
|
||||||
}
|
}
|
||||||
|
|
||||||
_logger.Debug($"{JobName} - Removing {request.ID} from the queue.");
|
|
||||||
_queue.Delete(request.ID);
|
|
||||||
}
|
|
||||||
|
|
||||||
_logger.Information($"{JobName} - Loop has been completed.");
|
|
||||||
}
|
|
||||||
|
|
||||||
public DiscordMessage GenerateDiscordMessage(SourceModel source, ArticlesModel article, IconModel icon)
|
public DiscordMessage GenerateDiscordMessage(SourceModel source, ArticlesModel article, IconModel icon)
|
||||||
{
|
{
|
||||||
var embed = new DiscordMessageEmbed
|
var embed = new DiscordMessageEmbed
|
||||||
|
@ -6,7 +6,6 @@ namespace Newsbot.Collector.Tests.Jobs;
|
|||||||
|
|
||||||
public class DiscordNotificationJobTest
|
public class DiscordNotificationJobTest
|
||||||
{
|
{
|
||||||
|
|
||||||
[Fact]
|
[Fact]
|
||||||
public void PostTestMessage()
|
public void PostTestMessage()
|
||||||
{
|
{
|
||||||
@ -23,7 +22,7 @@ public class DiscordNotificationJobTest
|
|||||||
Value = "a",
|
Value = "a",
|
||||||
Enabled = true,
|
Enabled = true,
|
||||||
Url = "https://github.com",
|
Url = "https://github.com",
|
||||||
Tags = "Unit, Testing",
|
Tags = "Unit, Testing"
|
||||||
},
|
},
|
||||||
new ArticlesModel
|
new ArticlesModel
|
||||||
{
|
{
|
||||||
@ -42,4 +41,54 @@ public class DiscordNotificationJobTest
|
|||||||
});
|
});
|
||||||
webhookClient.SendMessage(msg);
|
webhookClient.SendMessage(msg);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
[Fact]
|
||||||
|
public void SkipsCodeCommitWhenSubscriptionDoesNotWantThem()
|
||||||
|
{
|
||||||
|
var client = new DiscordNotificationJob();
|
||||||
|
try
|
||||||
|
{
|
||||||
|
client.SendSubscriptionNotification(
|
||||||
|
new Guid(),
|
||||||
|
new ArticlesModel
|
||||||
|
{
|
||||||
|
Tags = "more,unit,testing",
|
||||||
|
Title = "Nope not real",
|
||||||
|
URL = "https://github.com/jtom38",
|
||||||
|
PubDate = DateTime.Now,
|
||||||
|
Thumbnail =
|
||||||
|
"https://cdn.arstechnica.net/wp-content/uploads/2023/03/GettyImages-944827400-800x534.jpg",
|
||||||
|
Description = "Please work",
|
||||||
|
AuthorName = "No one knows",
|
||||||
|
CodeIsCommit = true
|
||||||
|
},
|
||||||
|
new SourceModel
|
||||||
|
{
|
||||||
|
ID = Guid.NewGuid(),
|
||||||
|
Site = "Unit Test",
|
||||||
|
Source = "placeholder",
|
||||||
|
Type = "a",
|
||||||
|
Value = "a",
|
||||||
|
Enabled = true,
|
||||||
|
Url = "https://github.com",
|
||||||
|
Tags = "Unit, Testing"
|
||||||
|
},
|
||||||
|
new IconModel
|
||||||
|
{
|
||||||
|
Id = Guid.NewGuid(),
|
||||||
|
FileName = "https://www.redditstatic.com/desktop2x/img/favicon/android-icon-192x192.png"
|
||||||
|
},
|
||||||
|
new SubscriptionModel
|
||||||
|
{
|
||||||
|
CodeAllowCommits = false,
|
||||||
|
CodeAllowReleases = true
|
||||||
|
});
|
||||||
|
Assert.Fail("Expected a error to come back.");
|
||||||
|
}
|
||||||
|
catch (MessageTypeNotRequestedException)
|
||||||
|
{
|
||||||
|
Console.Write($"Message did not send as expected");
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
}
|
}
|
Loading…
Reference in New Issue
Block a user