added test api for recently added items

yuval-ziv · Jul 21, 2024 · cc63cc5 · cc63cc5
1 parent 19937c0
commit cc63cc5
Show file tree

Hide file tree

Showing 3 changed files with 34 additions and 0 deletions.
diff --git a/Announcarr/Program.cs b/Announcarr/Program.cs
@@ -47,6 +47,10 @@
     async (ICalendarService calendarService, [FromQuery(Name = "start")] DateTimeOffset? start, [FromQuery(Name = "end")] DateTimeOffset? end) =>
     Results.Ok((object?)await calendarService.GetAllCalendarItemsAsync(start, end)));
 
+app.MapGet("/recentlyAdded",
+    async (ICalendarService calendarService, [FromQuery(Name = "start")] DateTimeOffset? start, [FromQuery(Name = "end")] DateTimeOffset? end) =>
+    Results.Ok((object?)await calendarService.GetAllRecentlyAddedItemsAsync(start, end)));
+
 app.MapGet("/testExporters", async (IEnumerable<IExporterService> exporterServices) => await Task.WhenAll(exporterServices.Select(exporterService => exporterService.TestExporterAsync())));
 
 app.Run();
diff --git a/Announcarr/Services/CalendarService.cs b/Announcarr/Services/CalendarService.cs
@@ -31,6 +31,21 @@ public async Task<CalendarResponse> GetAllCalendarItemsAsync(DateTimeOffset? sta
         return calendarResponse;
     }
 
+    public async Task<RecentlyAddedResponse> GetAllRecentlyAddedItemsAsync(DateTimeOffset? start, DateTimeOffset? end, CancellationToken cancellationToken = default)
+    {
+        start ??= DateTimeOffset.Now.AddDays(-7);
+        end ??= start.Value.AddDays(7);
+
+        RecentlyAddedResponse[] recentlyAddedResponses = await Task.WhenAll(_integrationServices.Where(integration => integration.IsEnabled())
+            .Select(async serviceIntegration => await GetRecentlyAddedItemsAsync(serviceIntegration, start.Value, end.Value, cancellationToken)));
+        RecentlyAddedResponse recentlyAddedResponse = recentlyAddedResponses.Length != 0 ? recentlyAddedResponses.Aggregate(RecentlyAddedResponse.Merge) : new RecentlyAddedResponse();
+
+        await Task.WhenAll(_exporterServices.Where(exporter => exporter.IsEnabled())
+            .Select(exporter => exporter.ExportRecentlyAddedAsync(recentlyAddedResponse, start.Value, end.Value, cancellationToken)));
+
+        return recentlyAddedResponse;
+    }
+
     private async Task<CalendarResponse> GetCalendarResponseAsync(IIntegrationService integrationService, DateTimeOffset start, DateTimeOffset end, CancellationToken cancellationToken = default)
     {
         try
@@ -43,4 +58,18 @@ private async Task<CalendarResponse> GetCalendarResponseAsync(IIntegrationServic
             return new CalendarResponse();
         }
     }
+
+    private async Task<RecentlyAddedResponse> GetRecentlyAddedItemsAsync(IIntegrationService integrationService, DateTimeOffset start, DateTimeOffset end,
+        CancellationToken cancellationToken = default)
+    {
+        try
+        {
+            return await integrationService.GetRecentlyAddedAsync(start, end, cancellationToken);
+        }
+        catch (Exception e)
+        {
+            _logger.LogError(e, "Unable to get recently added items from {IntegrationServiceName}", integrationService.GetName());
+            return new RecentlyAddedResponse();
+        }
+    }
 }
diff --git a/Announcarr/Services/ICalendarService.cs b/Announcarr/Services/ICalendarService.cs
@@ -5,4 +5,5 @@ namespace Announcarr.Services;
 public interface ICalendarService
 {
     Task<CalendarResponse> GetAllCalendarItemsAsync(DateTimeOffset? start, DateTimeOffset? end, CancellationToken cancellationToken = default);
+    Task<RecentlyAddedResponse> GetAllRecentlyAddedItemsAsync(DateTimeOffset? start, DateTimeOffset? end, CancellationToken cancellationToken = default);
 }