support atom by category on api level

EdiWang · Sep 18, 2023 · d7483bb · d7483bb
1 parent 9e4214c
commit d7483bb
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 6 deletions.
diff --git a/src/Moonglade.Core/BlogCachePartition.cs b/src/Moonglade.Core/BlogCachePartition.cs
@@ -8,5 +8,6 @@ public enum BlogCachePartition
     PostCountCategory,
     PostCountTag,
     PostCountFeatured,
-    RssCategory
+    RssCategory,
+    AtomCategory
 }
diff --git a/src/Moonglade.Syndication/GetAtomStringQuery.cs b/src/Moonglade.Syndication/GetAtomStringQuery.cs
@@ -5,7 +5,7 @@
 
 namespace Moonglade.Syndication;
 
-public record GetAtomStringQuery : IRequest<string>;
+public record GetAtomStringQuery(string CategoryName = null) : IRequest<string>;
 
 public class GetAtomStringQueryHandler : IRequestHandler<GetAtomStringQuery, string>
 {
@@ -31,7 +31,10 @@ public GetAtomStringQueryHandler(IBlogConfig blogConfig, ISyndicationDataSource
 
     public async Task<string> Handle(GetAtomStringQuery request, CancellationToken ct)
     {
-        _feedGenerator.FeedItemCollection = await _sdds.GetFeedDataAsync();
+        var data = await _sdds.GetFeedDataAsync(request.CategoryName);
+        if (data is null) return null;
+
+        _feedGenerator.FeedItemCollection = data;
         var xml = await _feedGenerator.WriteAtomAsync();
         return xml;
     }

diff --git a/src/Moonglade.Web/Controllers/SubscriptionController.cs b/src/Moonglade.Web/Controllers/SubscriptionController.cs
@@ -65,17 +65,18 @@ public async Task<IActionResult> Rss([MaxLength(64)] string routeName = null)
             });
     }
 
-    [HttpGet("atom")]
+    [HttpGet("atom/{routeName?}")]
     public async Task<IActionResult> Atom([MaxLength(64)] string routeName = null)
     {
         bool hasRoute = !string.IsNullOrWhiteSpace(routeName);
         var route = hasRoute ? routeName.ToLower().Trim() : null;
 
-        return await _cache.GetOrCreateAsync(BlogCachePartition.General.ToString(), "atom", async entry =>
+        return await _cache.GetOrCreateAsync(
+            hasRoute ? BlogCachePartition.AtomCategory.ToString() : BlogCachePartition.General.ToString(), route ?? "atom", async entry =>
         {
             entry.SlidingExpiration = TimeSpan.FromHours(1);
 
-            var xml = await _mediator.Send(new GetAtomStringQuery());
+            var xml = await _mediator.Send(new GetAtomStringQuery(routeName));
             return Content(xml, "text/xml");
         });
     }