feat: Remove stale tokens on a regular basis

2 files changed, 37 insertions, 0 deletions

diff --git a/src/server/Jobs/JobRegister.cs b/src/server/Jobs/JobRegister.cs
new file mode 100644
index 0000000..98eac97
--- /dev/null
+++ b/src/server/Jobs/JobRegister.cs
@@ -0,0 +1,18 @@
+using Quartz;
+
+namespace IOL.BookmarkThing.Server.Jobs;
+
+public static class JobRegister
+{
+	public static readonly JobKey TokenCleanupKey = new("TokenCleanupJob");
+
+	public static IServiceCollectionQuartzConfigurator RegisterJobs(this IServiceCollectionQuartzConfigurator configurator) {
+		configurator.AddJob<TokenCleanupJob>(TokenCleanupKey);
+		configurator.AddTrigger(options => {
+			options.ForJob(TokenCleanupKey)
+				   .WithIdentity(TokenCleanupKey.Name + "-trigger")
+				   .WithCronSchedule(CronScheduleBuilder.DailyAtHourAndMinute(1, 0));
+		});
+		return configurator;
+	}
+}
diff --git a/src/server/Jobs/TokenCleanupJob.cs b/src/server/Jobs/TokenCleanupJob.cs
new file mode 100644
index 0000000..99e7d0d
--- /dev/null
+++ b/src/server/Jobs/TokenCleanupJob.cs
@@ -0,0 +1,19 @@
+namespace IOL.BookmarkThing.Server.Jobs;
+
+public class TokenCleanupJob : IJob
+{
+	private readonly ILogger<TokenCleanupJob> _logger;
+	private readonly AppDbContext _context;
+
+	public TokenCleanupJob(ILogger<TokenCleanupJob> logger, AppDbContext context) {
+		_logger = logger;
+		_context = context;
+	}
+
+	public Task Execute(IJobExecutionContext context) {
+		var staleTokens = _context.AccessTokens.Where(c => c.HasExpired);
+		_logger.LogInformation("Removing {0} stale tokens", staleTokens.Count());
+		_context.AccessTokens.RemoveRange();
+		return Task.CompletedTask;
+	}
+}