From 494fc9d7d1f08e05e1ee196bd0746900343b51dd Mon Sep 17 00:00:00 2001 From: ivarlovlie Date: Mon, 16 Jan 2023 00:13:54 +0100 Subject: feat: Fully implemented text/file api, almost finished autodelete service --- src/WallE.cs | 74 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 74 insertions(+) create mode 100644 src/WallE.cs (limited to 'src/WallE.cs') diff --git a/src/WallE.cs b/src/WallE.cs new file mode 100644 index 0000000..ae78050 --- /dev/null +++ b/src/WallE.cs @@ -0,0 +1,74 @@ +using IOL.Helpers; + +namespace BlobBin; + +public class WallE : BackgroundService +{ + public IServiceProvider Services { get; } + private readonly ILogger _logger; + + public WallE(IServiceProvider services, + ILogger logger) { + Services = services; + _logger = logger; + } + + protected override async Task ExecuteAsync(CancellationToken stoppingToken) { + _logger.LogInformation("WallE is running."); + await DoWork(stoppingToken); + } + + private async Task DoWork(CancellationToken stoppingToken) { + _logger.LogInformation("WallE is working."); + using var scope = Services.CreateScope(); + var eva = scope.ServiceProvider.GetRequiredService(); + var pastes = eva.Pastes.Where(c => c.DeletedAt != default || c.AutoDeleteAfter != default) + .Select(c => new Paste() { + Id = c.Id, + DeletedAt = c.DeletedAt, + AutoDeleteAfter = c.AutoDeleteAfter, + CreatedAt = c.CreatedAt + }) + .ToList(); + var files = eva.Files + .Where(c => c.DeletedAt != default || c.AutoDeleteAfter != default) + .Select(c => new File() { + Id = c.Id, + DeletedAt = c.DeletedAt, + AutoDeleteAfter = c.AutoDeleteAfter, + CreatedAt = c.CreatedAt + }) + .ToList(); + var now = DateTime.UtcNow; + foreach (var file in files) { + var path = Path.Combine(Tools.GetFilesDirectoryPath(), file.Id.ToString()); + if (DateTime.Compare(now, file.DeletedAt?.AddDays(7) ?? DateTime.MinValue) > 0) { + System.IO.File.Delete(path); + eva.Files.Remove(file); + } else if (file.AutoDeleteAfter.HasValue()) { + var autoDeleteDateTime = file.CreatedAt.Add(Tools.ParseHumanTimeSpan(file.AutoDeleteAfter)); + if (DateTime.Compare(now, autoDeleteDateTime) > 0) { + System.IO.File.Delete(path); + eva.Files.Remove(file); + } + } + } + + // If the file has lived more than +7 days from when it should be automatically deleted, delete the file completely. + foreach (var paste in pastes) { + if (DateTime.Compare(now, paste.DeletedAt?.AddDays(7) ?? DateTime.MinValue) > 0) { + eva.Pastes.Remove(paste); + } else if (paste.AutoDeleteAfter.HasValue()) { + var autoDeleteDateTime = paste.CreatedAt.Add(Tools.ParseHumanTimeSpan(paste.AutoDeleteAfter)); + if (DateTime.Compare(now, autoDeleteDateTime) > 0) { + eva.Pastes.Remove(paste); + } + } + } + } + + public override async Task StopAsync(CancellationToken stoppingToken) { + _logger.LogInformation("WallE is stopping."); + await base.StopAsync(stoppingToken); + } +} \ No newline at end of file -- cgit v1.3