Data Erasure plugin

2025-12-17 07:34:24 +01:00 · 2023-03-22 14:45:42 +01:00
parent 75337f364e
commit 37d68f1b3b
10 changed files with 373 additions and 0 deletions
--- a/Plugins/BTCPayServer.Plugins.DataErasure/DataErasureService.cs
+++ b/Plugins/BTCPayServer.Plugins.DataErasure/DataErasureService.cs
@@ -0,0 +1,132 @@
+using System;
+using System.Linq;
+using System.Threading;
+using System.Threading.Tasks;
+using BTCPayServer.Abstractions.Contracts;
+using BTCPayServer.Services.Invoices;
+using Microsoft.Extensions.Hosting;
+using Microsoft.Extensions.Logging;
+
+namespace BTCPayServer.Plugins.DataErasure
+{
+    public class DataErasureService : IHostedService
+    {
+        private readonly IStoreRepository _storeRepository;
+        private readonly ILogger<DataErasureService> _logger;
+        private readonly InvoiceRepository _invoiceRepository;
+
+        public DataErasureService(IStoreRepository storeRepository, ILogger<DataErasureService> logger,
+            InvoiceRepository invoiceRepository)
+        {
+            _storeRepository = storeRepository;
+            _logger = logger;
+            _invoiceRepository = invoiceRepository;
+        }
+
+        public async Task<DataErasureSettings> Get(string storeId)
+        {
+            return await _storeRepository.GetSettingAsync<DataErasureSettings>(storeId,
+                nameof(DataErasureSettings));
+        }
+
+        public async Task Set(string storeId, DataErasureSettings settings)
+        {
+            await _runningLock.WaitAsync();
+            var existing = await Get(storeId);
+            settings.LastRunCutoff = existing?.LastRunCutoff;
+            await SetCore(storeId, settings);
+            _runningLock.Release();
+        }
+
+        private async Task SetCore(string storeId, DataErasureSettings settings)
+        {
+            await _storeRepository.UpdateSetting(storeId, nameof(DataErasureSettings), settings);
+        }
+
+        public bool IsRunning { get; private set; }
+        private readonly SemaphoreSlim _runningLock = new(1, 1);
+
+        private async Task Run(CancellationToken cancellationToken)
+        {
+            while (!cancellationToken.IsCancellationRequested)
+            {
+                await _runningLock.WaitAsync(cancellationToken);
+                IsRunning = true;
+
+
+                var settings =
+                    await _storeRepository.GetSettingsAsync<DataErasureSettings>(nameof(DataErasureSettings));
+                foreach (var setting in settings.Where(setting => setting.Value.Enabled))
+                {
+                    var skip = 0;
+                    var count = 0;
+                    var cutoffDate = DateTimeOffset.UtcNow.Subtract(TimeSpan.FromDays(setting.Value.DaysToKeep));
+                    while (true)
+                    {
+                        var invoices = await _invoiceRepository.GetInvoices(new InvoiceQuery()
+                        {
+                            StartDate = setting.Value.LastRunCutoff,
+                            EndDate = cutoffDate,
+                            StoreId = new[] {setting.Key},
+                            Skip = skip,
+                            Take = 100
+                        });
+                        foreach (var invoice in invoices)
+                        {
+                            //replace all buyer info with "erased"
+                            if (!string.IsNullOrEmpty(invoice.Metadata.BuyerAddress1))
+                                invoice.Metadata.BuyerAddress1 = "erased";
+                            if (!string.IsNullOrEmpty(invoice.Metadata.BuyerAddress2))
+                                invoice.Metadata.BuyerAddress2 = "erased";
+                            if (!string.IsNullOrEmpty(invoice.Metadata.BuyerCity))
+                                invoice.Metadata.BuyerCity = "erased";
+                            if (!string.IsNullOrEmpty(invoice.Metadata.BuyerCountry))
+                                invoice.Metadata.BuyerCountry = "erased";
+                            if (!string.IsNullOrEmpty(invoice.Metadata.BuyerEmail))
+                                invoice.Metadata.BuyerEmail = "erased";
+                            if (!string.IsNullOrEmpty(invoice.Metadata.BuyerName))
+                                invoice.Metadata.BuyerName = "erased";
+                            if (!string.IsNullOrEmpty(invoice.Metadata.BuyerPhone))
+                                invoice.Metadata.BuyerPhone = "erased";
+                            if (!string.IsNullOrEmpty(invoice.Metadata.BuyerState))
+                                invoice.Metadata.BuyerState = "erased";
+                            if (!string.IsNullOrEmpty(invoice.Metadata.BuyerZip))
+                                invoice.Metadata.BuyerZip = "erased";
+                            await _invoiceRepository.UpdateInvoiceMetadata(invoice.Id, invoice.StoreId,
+                                invoice.Metadata.ToJObject());
+                            count++;
+                        }
+
+                        if (invoices.Length < 100)
+                        {
+                            break;
+                        }
+
+                        skip += 100;
+                    }
+
+                    _logger.LogInformation($"Erased {count} invoice data for store {setting.Key}");
+                    setting.Value.LastRunCutoff = cutoffDate;
+                    await SetCore(setting.Key, setting.Value);
+                }
+
+                IsRunning = false;
+
+
+                _runningLock.Release();
+                await Task.Delay(TimeSpan.FromMinutes(1), cancellationToken);
+            }
+        }
+
+        public Task StartAsync(CancellationToken cancellationToken)
+        {
+            _ = Run(cancellationToken);
+            return Task.CompletedTask;
+        }
+
+        public Task StopAsync(CancellationToken cancellationToken)
+        {
+            return Task.CompletedTask;
+        }
+    }
+}