Add cleanup background task

Author: hishamco

src/OrchardCoreContrib.Contents/Services/ISharedDraftLinkService.cs

@@ -7,4 +7,6 @@ public interface ISharedDraftLinkService
     Task<string> GenerateLinkAsync(ContentItem contentItem);
 
     Task<ContentItem> GetDraftContentAsync(string token);
+
+    Task<int> CleanupExpiredLinksAsync();
 }

src/OrchardCoreContrib.Contents/Services/SharedDraftCleanupTask.cs

@@ -0,0 +1,22 @@
+using Microsoft.Extensions.DependencyInjection;
+using Microsoft.Extensions.Logging;
+using OrchardCore.BackgroundTasks;
+
+namespace OrchardCoreContrib.Contents.Services;
+
+[BackgroundTask(Schedule = "0 0 * * *", Description = "Shared Draft Cleanup Background Service")]
+public class SharedDraftCleanupTask(ISharedDraftLinkService linkService) : IBackgroundTask
+{
+    public async Task DoWorkAsync(IServiceProvider serviceProvider, CancellationToken cancellationToken)
+    {
+        var count = await linkService.CleanupExpiredLinksAsync();
+
+        if (count > 0)
+        {
+            var logger = serviceProvider.GetRequiredService<ILogger<SharedDraftCleanupTask>>();
+
+            logger.LogInformation("Cleaned up {Count} expired draft links.", count);
+        }
+    }
+}
+

src/OrchardCoreContrib.Contents/Services/SharedDraftLinkService.cs

@@ -50,4 +50,19 @@ public async Task<ContentItem> GetDraftContentAsync(string token)
             ? null
             : await contentManager.GetAsync(link.ContentItemId, VersionOptions.Draft);
     }
+
+    public async Task<int> CleanupExpiredLinksAsync()
+    {
+        var expiredLinks = await session.Query<SharedDraftLink, SharedDraftLinkIndex>()
+            .Where(l => l.ExpirationUtc < DateTime.UtcNow)
+            .ListAsync();
+
+        foreach (var link in expiredLinks)
+        {
+            session.Delete(link);
+        }
+
+        return expiredLinks.Count();
+    }
+
 }

src/OrchardCoreContrib.Contents/Startup.cs

@@ -1,4 +1,5 @@
 using Microsoft.Extensions.DependencyInjection;
+using OrchardCore.BackgroundTasks;
 using OrchardCore.ContentManagement.Display.ContentDisplay;
 using OrchardCore.Data;
 using OrchardCore.Data.Migration;
@@ -15,13 +16,15 @@ public sealed class Startup : StartupBase
 {
     public override void ConfigureServices(IServiceCollection services)
     {
-        services.AddScoped<ISharedDraftLinkService, SharedDraftLinkService>();
+        services.AddSingleton<IBackgroundTask, SharedDraftCleanupTask>();
 
         services.AddDataMigration<Migrations>();
 
         services.AddIndexProvider<SharedDraftLinkIndexProvider>();
 
         services.AddScoped<IContentDisplayDriver, ShareDraftContentDriver>();
+        services.AddScoped<ISharedDraftLinkService, SharedDraftLinkService>();
+
         services.AddPermissionProvider<Permissions>();
     }
 }